chore: work on ci-cd integration

Author: lpm0073

azure_ai/tests/test_azureml_workspace.py

@@ -2,18 +2,24 @@
 Test script to verify Azure ML installation is working properly.
 """
 
+from logging import getLogger
+
 from azure_ai.ml_studio import AzureAIMLWorkspace
 
 from .test_base import AzureMLTestBase
 
 
+logger = getLogger(__name__)
+
+
 class AzureMLTestWorkspace(AzureMLTestBase):
     """Base class for all unit tests."""
 
     def test_azureml_workspace(self):
         """Test that we can create and use an Azure ML workspace."""
 
         if not self.is_testable:
+            logger.warning("Skipping test_azureml_workspace() as the environment is not testable.")
             return
 
         workspace = AzureAIMLWorkspace()

azure_ai/tests/test_base.py

@@ -5,6 +5,7 @@
 import os
 import unittest
 from logging import getLogger
+from unittest.mock import patch
 
 from dotenv import load_dotenv
 
@@ -24,4 +25,48 @@ def setUpClass(cls):
         """Set up class-level resources."""
         if not cls.is_testable:
             logger.warning("CI - skipping tests that require Azure ML workspace connection")
+            cls._setup_azure_mocks()
             return
+
+    @classmethod
+    def tearDownClass(cls):
+        """Clean up class-level resources."""
+        if not cls.is_testable:
+            # Stop all patches
+            patchers = [
+                "ml_client_patcher",
+                "credential_patcher",
+                "dataset_v2_patcher",
+                "workspace_patcher",
+                "dataset_v1_patcher",
+                "compute_patcher",
+                "experiment_patcher",
+            ]
+            for patcher_name in patchers:
+                if hasattr(cls, patcher_name):
+                    getattr(cls, patcher_name).stop()
+
+    @classmethod
+    def _setup_azure_mocks(cls):
+        """Set up mocks for Azure ML clients when not testable."""
+        logger.info("Setting up Azure ML mocks for non-testable CI-CD environment")
+
+        # Patch at the source before any imports happen
+        cls.ml_client_patcher = patch("azure.ai.ml.MLClient")
+        cls.mock_ml_client = cls.ml_client_patcher.start()
+
+        cls.credential_patcher = patch("azure.identity.DefaultAzureCredential")
+        cls.mock_credential = cls.credential_patcher.start()
+
+        # Also patch other Azure classes at their source
+        cls.workspace_patcher = patch("azureml.core.Workspace")
+        cls.mock_workspace = cls.workspace_patcher.start()
+
+        cls.dataset_v1_patcher = patch("azureml.core.Dataset")
+        cls.mock_dataset_v1 = cls.dataset_v1_patcher.start()
+
+        cls.compute_patcher = patch("azureml.core.compute.AmlCompute")
+        cls.mock_compute = cls.compute_patcher.start()
+
+        cls.experiment_patcher = patch("azureml.core.Experiment")
+        cls.mock_experiment = cls.experiment_patcher.start()

azure_ai/tests/test_cluster.py

@@ -3,13 +3,15 @@
 """
 
 import os
+from logging import getLogger
 
 from azure_ai.ml_studio import AzureAIMLStudioComputeCluster
 
 from .test_base import AzureMLTestBase
 
 
 HERE = os.path.abspath(os.path.dirname(__file__))
+logger = getLogger(__name__)
 
 
 class AzureMLTestCluster(AzureMLTestBase):
@@ -22,6 +24,7 @@ def test_existing_cluster(self):
         or create it if it does not.
         """
         if not self.is_testable:
+            logger.warning("Skipping test_existing_cluster() as the environment is not testable.")
             return
 
         cluster = AzureAIMLStudioComputeCluster(cluster_name="standard-cluster")

azure_ai/tests/test_config.py

@@ -78,6 +78,7 @@ def test_workspace_connection(self):
         """Test workspace connection using config.json."""
 
         if not self.is_testable:
+            logger.warning("Skipping test_workspace_connection() as the environment is not testable.")
             return
 
         try:

azure_ai/tests/test_dataset.py

@@ -3,13 +3,15 @@
 """
 
 import os
+from logging import getLogger
 
 from azure_ai.ml_studio import AzureAIMLAssetsDataset
 
 from .test_base import AzureMLTestBase
 
 
 HERE = os.path.abspath(os.path.dirname(__file__))
+logger = getLogger(__name__)
 
 
 class AzureMLTestDataset(AzureMLTestBase):
@@ -18,6 +20,7 @@ class AzureMLTestDataset(AzureMLTestBase):
     def test_existing_dataset(self):
         """Test that we can create and use an Azure ML workspace."""
         if not self.is_testable:
+            logger.warning("Skipping test_existing_dataset() as the environment is not testable.")
             return
 
         dataset = AzureAIMLAssetsDataset(
@@ -31,6 +34,7 @@ def test_kaggle_dataset(self):
         https://www.kaggle.com/datasets/whenamancodes/student-performance
         """
         if not self.is_testable:
+            logger.warning("Skipping test_kaggle_dataset() as the environment is not testable.")
             return
 
         dataset = AzureAIMLAssetsDataset(
@@ -50,6 +54,7 @@ def test_file_dataset(self):
         """
 
         if not self.is_testable:
+            logger.warning("Skipping test_file_dataset() as the environment is not testable.")
             return
 
         dataset = AzureAIMLAssetsDataset(