Merge pull request #808 from microsoft/code-quality-fix

refactor: Code quality fix

Author: Roopan-Microsoft

src/tests/backend/auth/test_auth_utils.py

@@ -5,10 +5,8 @@
 import pytest
 import base64
 import json
-import logging
 import sys
 import os
-import importlib.util
 from unittest.mock import patch, MagicMock
 
 # Add the source root directory to the Python path for imports

src/tests/backend/common/config/test_app_config.py

@@ -13,9 +13,6 @@
 import os
 import logging
 from unittest.mock import patch, MagicMock, AsyncMock
-from azure.identity import DefaultAzureCredential, ManagedIdentityCredential
-from azure.cosmos import CosmosClient
-from azure.ai.projects.aio import AIProjectClient
 
 # Add the source root directory to the Python path for imports
 import sys

src/tests/backend/common/database/test_cosmosdb.py

@@ -4,10 +4,8 @@
 import logging
 import sys
 import os
-from typing import Any, Dict, List, Optional
-from unittest.mock import AsyncMock, MagicMock, Mock, patch
+from unittest.mock import AsyncMock, Mock, patch
 import pytest
-import uuid
 
 # Add the backend directory to the Python path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..', '..', 'backend'))

src/tests/backend/common/database/test_database_base.py

@@ -2,9 +2,9 @@
 
 import sys
 import os
-from abc import ABC, abstractmethod
+from abc import ABC
 from typing import Any, Dict, List, Optional, Type
-from unittest.mock import AsyncMock, Mock, patch
+from unittest.mock import Mock
 import pytest
 
 # Add the backend directory to the Python path

src/tests/backend/common/database/test_database_factory.py

@@ -3,8 +3,7 @@
 import logging
 import sys
 import os
-from typing import Optional
-from unittest.mock import AsyncMock, Mock, patch, MagicMock
+from unittest.mock import AsyncMock, Mock, patch
 import pytest
 
 # Add the backend directory to the Python path

src/tests/backend/common/utils/test_event_utils.py

@@ -3,7 +3,7 @@
 import logging
 import sys
 import os
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import Mock, patch
 import pytest
 
 # Mock external dependencies at module level

src/tests/backend/common/utils/test_otlp_tracing.py

@@ -2,7 +2,7 @@
 
 import sys
 import os
-from unittest.mock import Mock, patch, MagicMock, call
+from unittest.mock import Mock, patch, call
 import pytest
 
 # Mock external dependencies at module level

src/tests/backend/common/utils/test_utils_af.py

@@ -1,10 +1,8 @@
 """Unit tests for utils_af module."""
 
-import logging
 import sys
 import os
-import uuid
-from unittest.mock import Mock, patch, AsyncMock, MagicMock
+from unittest.mock import Mock, patch, AsyncMock
 import pytest
 
 # Add the backend directory to the Python path

src/tests/backend/common/utils/test_utils_agents.py

@@ -4,7 +4,6 @@
 This module tests the utility functions for agent ID generation and database operations.
 """
 
-import logging
 import string
 import sys
 import unittest
@@ -33,10 +32,8 @@
 sys.modules['common.models'] = Mock()
 sys.modules['common.models.messages_af'] = Mock()
 
-import pytest
-
 from backend.common.database.database_base import DatabaseBase
-from backend.common.models.messages_af import CurrentTeamAgent, DataType, TeamConfiguration
+from backend.common.models.messages_af import CurrentTeamAgent, TeamConfiguration
 from backend.common.utils.utils_agents import (
     generate_assistant_id,
     get_database_team_agent_id,

src/tests/backend/common/utils/test_utils_date.py

@@ -7,16 +7,12 @@
 
 import json
 import locale
-import logging
 import unittest
 import sys
 import os
 from datetime import datetime
-from typing import Optional
 from unittest.mock import Mock, patch
 
-import pytest
-
 # Add the backend directory to the Python path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..', '..', 'backend'))
 
@@ -90,9 +86,6 @@ def mock_parse(date_str):
 import re as real_re
 utils_date_module.re = real_re
 
-# Import dateutil.parser after mocking to avoid import errors
-from dateutil import parser
-
 
 class TestFormatDateForUser(unittest.TestCase):
     """Test cases for format_date_for_user function."""