fix unittestcase

Author: Pavan-Microsoft

src/tests/api/helpers/test_azure_credential_utils.py

@@ -7,93 +7,85 @@
 
 import helpers.azure_credential_utils as azure_credential_utils
 
+@pytest.fixture
+def mock_env_vars():
+    return {
+        "app_env": "dev"
+    }
+
 class TestAzureCredentialUtils:
-    @patch("helpers.azure_credential_utils.Config")
+    @patch.dict(os.environ, clear=True)
     @patch("helpers.azure_credential_utils.DefaultAzureCredential")
     @patch("helpers.azure_credential_utils.ManagedIdentityCredential")
-    def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_config):
+    def test_get_azure_credential_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_env_vars):
         """Test get_azure_credential in dev environment."""
         # Arrange
-        mock_config_instance = MagicMock()
-        mock_config_instance.app_env = "dev"
-        mock_config.return_value = mock_config_instance
-
+        os.environ.update(mock_env_vars)
         mock_default_credential = MagicMock()
         mock_default_azure_credential.return_value = mock_default_credential
 
         # Act
         credential = azure_credential_utils.get_azure_credential()
 
         # Assert
-        mock_config.assert_called_once()
         mock_default_azure_credential.assert_called_once()
         mock_managed_identity_credential.assert_not_called()
         assert credential == mock_default_credential
 
-    @patch("helpers.azure_credential_utils.Config")
+    @patch.dict(os.environ, clear=True)
     @patch("helpers.azure_credential_utils.DefaultAzureCredential")
     @patch("helpers.azure_credential_utils.ManagedIdentityCredential")
-    def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_config):
+    def test_get_azure_credential_non_dev_env(self, mock_managed_identity_credential, mock_default_azure_credential, mock_env_vars):
         """Test get_azure_credential in non-dev environment."""
         # Arrange
-        mock_config_instance = MagicMock()
-        mock_config_instance.app_env = "Prod"
-        mock_config.return_value = mock_config_instance
-
+        mock_env_vars["app_env"] = "prod"
+        os.environ.update(mock_env_vars)
         mock_managed_credential = MagicMock()
         mock_managed_identity_credential.return_value = mock_managed_credential
 
         # Act
         credential = azure_credential_utils.get_azure_credential(client_id="test-client-id")
 
         # Assert
-        mock_config.assert_called_once()
         mock_managed_identity_credential.assert_called_once_with(client_id="test-client-id")
         mock_default_azure_credential.assert_not_called()
         assert credential == mock_managed_credential
 
     @pytest.mark.asyncio
-    @patch("helpers.azure_credential_utils.Config")
+    @patch.dict(os.environ, clear=True)
     @patch("helpers.azure_credential_utils.AioDefaultAzureCredential")
     @patch("helpers.azure_credential_utils.AioManagedIdentityCredential")
-    async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_config):
+    async def test_get_azure_credential_async_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_env_vars):
         """Test get_azure_credential_async in dev environment."""
         # Arrange
-        mock_config_instance = MagicMock()
-        mock_config_instance.app_env = "dev"
-        mock_config.return_value = mock_config_instance
-
+        os.environ.update(mock_env_vars)
         mock_aio_default_credential = MagicMock()
         mock_aio_default_azure_credential.return_value = mock_aio_default_credential
 
         # Act
         credential = await azure_credential_utils.get_azure_credential_async()
 
         # Assert
-        mock_config.assert_called_once()
         mock_aio_default_azure_credential.assert_called_once()
         mock_aio_managed_identity_credential.assert_not_called()
         assert credential == mock_aio_default_credential
 
     @pytest.mark.asyncio
-    @patch("helpers.azure_credential_utils.Config")
+    @patch.dict(os.environ, clear=True)
     @patch("helpers.azure_credential_utils.AioDefaultAzureCredential")
     @patch("helpers.azure_credential_utils.AioManagedIdentityCredential")
-    async def test_get_azure_credential_async_non_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_config):
+    async def test_get_azure_credential_async_non_dev_env(self, mock_aio_managed_identity_credential, mock_aio_default_azure_credential, mock_env_vars):
         """Test get_azure_credential_async in non-dev environment."""
         # Arrange
-        mock_config_instance = MagicMock()
-        mock_config_instance.app_env = "Prod"
-        mock_config.return_value = mock_config_instance
-
+        mock_env_vars["app_env"] = "prod"
+        os.environ.update(mock_env_vars)
         mock_aio_managed_credential = MagicMock()
         mock_aio_managed_identity_credential.return_value = mock_aio_managed_credential
 
         # Act
         credential = await azure_credential_utils.get_azure_credential_async(client_id="test-client-id")
 
         # Assert
-        mock_config.assert_called_once()
         mock_aio_managed_identity_credential.assert_called_once_with(client_id="test-client-id")
         mock_aio_default_azure_credential.assert_not_called()
         assert credential == mock_aio_managed_credential