fix(tests): resolve flakiness in qwen oauth regression tests by patching base class directly

Author: Mateusz

tests/integration/test_qwen_oauth_static_routing_regression.py

@@ -75,8 +75,12 @@ async def test_qwen_oauth_static_routing_model_override_regression():
             mock_load_creds.return_value = True
 
             # Mock the parent OpenAIConnector.chat_completions method to capture the call
-            with patch(
-                "src.connectors.openai.OpenAIConnector.chat_completions",
+            # Use __bases__[0] to ensure we patch the exact base class object QwenOAuthConnector is using,
+            # which handles cases where modules might be reloaded during tests.
+            base_cls = QwenOAuthConnector.__bases__[0]
+            with patch.object(
+                base_cls,
+                "chat_completions",
                 new_callable=AsyncMock,
             ) as mock_parent_chat:
 
@@ -198,20 +202,24 @@ async def test_qwen_oauth_model_name_processing_with_static_routes():
             mock_load_creds.return_value = True
 
             # Mock the parent OpenAIConnector.chat_completions method
-            with patch(
-                "src.connectors.openai.OpenAIConnector.chat_completions",
+            # Use __bases__[0] to ensure we patch the exact base class object QwenOAuthConnector is using,
+            # which handles cases where modules might be reloaded during tests.
+            base_cls = QwenOAuthConnector.__bases__[0]
+            with patch.object(
+                base_cls,
+                "chat_completions",
                 new_callable=AsyncMock,
             ) as mock_parent_chat:
-
-                from src.core.domain.responses import ResponseEnvelope
-
-                mock_response_envelope = ResponseEnvelope(
-                    content="Test response", usage=MagicMock(total_tokens=10)
-                )
-                mock_parent_chat.return_value = mock_response_envelope
-
-                # Test various static routing scenarios
-                test_cases = [
+
+                from src.core.domain.responses import ResponseEnvelope
+
+                mock_response_envelope = ResponseEnvelope(
+                    content="Test response", usage=MagicMock(total_tokens=10)
+                )
+                mock_parent_chat.return_value = mock_response_envelope
+
+                # Test various static routing scenarios
+                test_cases = [
                     {
                         "original_model": "gemini-cli-oauth-personal:models/gemini-2.5-pro",
                         "static_override": "qwen-oauth:qwen3-coder-plus",
@@ -325,20 +333,24 @@ async def test_qwen_oauth_prevents_original_model_leakage():
             mock_load_creds.return_value = True
 
             # Mock the parent OpenAIConnector.chat_completions method
-            with patch(
-                "src.connectors.openai.OpenAIConnector.chat_completions",
+            # Use __bases__[0] to ensure we patch the exact base class object QwenOAuthConnector is using,
+            # which handles cases where modules might be reloaded during tests.
+            base_cls = QwenOAuthConnector.__bases__[0]
+            with patch.object(
+                base_cls,
+                "chat_completions",
                 new_callable=AsyncMock,
             ) as mock_parent_chat:
-
-                from src.core.domain.responses import ResponseEnvelope
-
-                mock_response_envelope = ResponseEnvelope(
-                    content="Test response", usage=MagicMock(total_tokens=10)
-                )
-                mock_parent_chat.return_value = mock_response_envelope
-
-                # Create a request with a completely different original model
-                original_model = (
+
+                from src.core.domain.responses import ResponseEnvelope
+
+                mock_response_envelope = ResponseEnvelope(
+                    content="Test response", usage=MagicMock(total_tokens=10)
+                )
+                mock_parent_chat.return_value = mock_response_envelope
+
+                # Create a request with a completely different original model
+                original_model = (
                     "some-other-backend:some-completely-different-model-name"
                 )
                 static_override_model = "qwen3-coder-plus"