Add New Basic Tests

tests/test_timeout_session.py

@@ -0,0 +1,60 @@
+# flake8: noqa
+# mypy: ignore-errors
+"""Test timeout session system."""
+
+from unittest.mock import MagicMock, patch
+from requests import Session
+import timeout_session
+
+
+class TestTimeoutSession:
+
+    def test_adapter_injects_default_timeout(self):
+        """Verify the adapter sets the timeout if none is provided in the send call."""
+        adapter = timeout_session.TimeoutAdapter(timeout=42)
+
+        # We mock the parent HTTPAdapter.send method
+        with patch("requests.adapters.HTTPAdapter.send") as mock_send:
+            # Call send with timeout=None (which is what requests does by default)
+            adapter.send(MagicMock(), timeout=None)
+
+            # Check that the kwargs passed to super().send now contains our default
+            args, kwargs = mock_send.call_args
+            assert kwargs["timeout"] == 42
+
+    def test_adapter_respects_explicit_timeout(self):
+        """Verify the adapter doesn't override a timeout if the user explicitly provided one."""
+        adapter = timeout_session.TimeoutAdapter(timeout=42)
+
+        with patch("requests.adapters.HTTPAdapter.send") as mock_send:
+            # User explicitly wants 5 seconds
+            adapter.send(MagicMock(), timeout=5)
+
+            args, kwargs = mock_send.call_args
+            assert kwargs["timeout"] == 5
+
+    def test_new_session_configuration(self):
+        """Verify that new_session correctly mounts the adapter and sets headers."""
+        custom_timeout = 15
+        session = timeout_session.new_session(timeout=custom_timeout)
+
+        # Check User-Agent
+        assert "User-Agent" in session.headers
+
+        # Check mounting
+        # Prefix 'https://' should be mapped to our TimeoutAdapter
+        adapter = session.adapters.get("https://")
+        assert isinstance(adapter, timeout_session.TimeoutAdapter)
+        assert adapter.default_timeout == custom_timeout
+
+    def test_new_session_with_existing_object(self):
+        """Verify new_session can wrap an existing Session object."""
+        existing_session = Session()
+        existing_session.headers["X-Test"] = "Existing"
+
+        wrapped_session = timeout_session.new_session(session=existing_session)
+
+        assert wrapped_session.headers["X-Test"] == "Existing"
+        assert isinstance(
+            wrapped_session.adapters.get("http://"), timeout_session.TimeoutAdapter
+        )