|
8 | 8 | import os |
9 | 9 | import sys |
10 | 10 | import unittest |
11 | | -from unittest import IsolatedAsyncioTestCase |
12 | 11 | from unittest.mock import AsyncMock, Mock, patch |
13 | 12 |
|
14 | 13 | # Add the backend directory to the Python path |
@@ -162,7 +161,7 @@ def test_ad_token_provider(self, mock_config): |
162 | 161 | self.assertEqual(token, "test-token-123") |
163 | 162 | mock_credential.get_token.assert_called_once_with(mock_config.AZURE_COGNITIVE_SERVICES) |
164 | 163 |
|
165 | | -class TestAzureConfigAsync(IsolatedAsyncioTestCase): |
| 164 | +class TestAzureConfigAsync(unittest.IsolatedAsyncioTestCase): |
166 | 165 | """Async test cases for AzureConfig class.""" |
167 | 166 |
|
168 | 167 | @patch('backend.v4.config.settings.AzureOpenAIChatClient') |
@@ -284,7 +283,7 @@ def test_overwrite_existing_team(self): |
284 | 283 | self.assertEqual(config.get_current_team(user_id), team_config2) |
285 | 284 |
|
286 | 285 |
|
287 | | -class TestOrchestrationConfig(IsolatedAsyncioTestCase): |
| 286 | +class TestOrchestrationConfig(unittest.IsolatedAsyncioTestCase): |
288 | 287 | """Test cases for OrchestrationConfig class.""" |
289 | 288 |
|
290 | 289 | def test_orchestration_config_creation(self): |
@@ -490,7 +489,7 @@ def test_cleanup_clarification(self): |
490 | 489 | self.assertNotIn(request_id, config._clarification_events) |
491 | 490 |
|
492 | 491 |
|
493 | | -class TestConnectionConfig(IsolatedAsyncioTestCase): |
| 492 | +class TestConnectionConfig(unittest.IsolatedAsyncioTestCase): |
494 | 493 | """Test cases for ConnectionConfig class.""" |
495 | 494 |
|
496 | 495 | def test_connection_config_creation(self): |
|
0 commit comments