import unittest
from unittest.mock import MagicMock, patch

import urllib3

from domainCheck.detect import register


class RegisterTimeoutConfigTests(unittest.TestCase):
    def test_proxy_timeout_uses_bounded_total(self):
        with patch.dict("os.environ", {}, clear=False):
            timeout = register._resolve_register_timeout({"http": "http://127.0.0.1:8080"})
        self.assertIsInstance(timeout, urllib3.Timeout)
        self.assertEqual(timeout.connect_timeout, 1.0)
        self.assertEqual(timeout.read_timeout, 1.5)
        self.assertEqual(timeout.total, 2.2)

    def test_direct_timeout_respects_override(self):
        with patch.dict(
            "os.environ",
            {
                "DOMAINCHECK_REGISTER_DIRECT_CONNECT_TIMEOUT": "0.9",
                "DOMAINCHECK_REGISTER_DIRECT_READ_TIMEOUT": "1.4",
                "DOMAINCHECK_REGISTER_DIRECT_TOTAL_TIMEOUT": "1.7",
            },
            clear=False,
        ):
            timeout = register._resolve_register_timeout(None)
        self.assertEqual(timeout.connect_timeout, 0.9)
        self.assertEqual(timeout.read_timeout, 1.4)
        self.assertEqual(timeout.total, 1.7)

    def test_budget_clamps_proxy_timeout_into_remaining_window(self):
        with patch.dict("os.environ", {}, clear=False):
            timeout = register._resolve_register_timeout(
                {"http": "http://127.0.0.1:8080"},
                budget_seconds=0.8,
            )
        self.assertEqual(timeout.total, 0.8)
        self.assertLessEqual(timeout.connect_timeout, 0.35)
        self.assertLessEqual(timeout.read_timeout, 0.61)

    def test_request_register_converts_urllib3_timeout_for_requests(self):
        session = MagicMock()
        response = MagicMock()
        session.get.return_value = response
        timeout = urllib3.Timeout(connect=1.1, read=1.7, total=2.4)
        with patch("domainCheck.detect.register._get_http_manager", return_value=session):
            result = register._request_register(
                "https://rdap.verisign.com/com/v1/domain/example",
                proxies={"http": "http://127.0.0.1:8080", "https": "http://127.0.0.1:8080"},
                timeout=timeout,
            )
        self.assertIs(result, response)
        session.get.assert_called_once_with(
            "https://rdap.verisign.com/com/v1/domain/example",
            timeout=(1.1, 1.7),
            proxies={"http": "http://127.0.0.1:8080", "https": "http://127.0.0.1:8080"},
            allow_redirects=True,
        )


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