"""中控 Bridge SSO / Token 冒烟."""
from __future__ import annotations

import unittest

from lib.hub.hub_bridge_auth import (
    make_sso_token,
    verify_sso_token,
    build_embed_open_url,
)


class HubBridgeAuthTests(unittest.TestCase):
    def test_sso_roundtrip(self):
        tok = make_sso_token("secret-token", ttl_sec=60)
        self.assertTrue(verify_sso_token(tok, "secret-token"))
        self.assertFalse(verify_sso_token(tok, "other"))
        self.assertFalse(verify_sso_token("bad", "secret-token"))

    def test_embed_open_url(self):
        url = build_embed_open_url(
            "https://okx.example.com",
            bridge_token="abc",
            embed_path="/embed",
        )
        self.assertTrue(url.startswith("https://okx.example.com/hub_sso?"))
        self.assertIn("token=", url)
        self.assertIn("next=%2Fembed", url)

    def test_embed_session_token(self):
        from lib.hub.hub_bridge_auth import make_embed_session_token

        tok = make_embed_session_token("secret-token", ttl_sec=120)
        self.assertTrue(verify_sso_token(tok, "secret-token"))


if __name__ == "__main__":
    unittest.main()