data-factory-testing-framework 0.2.4__tar.gz → 0.2.6__tar.gz

{data_factory_testing_framework-0.2.4 → data_factory_testing_framework-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: data-factory-testing-framework
-Version: 0.2.4
+Version: 0.2.6
 Summary: A stand-alone test framework that allows to write unit tests for Data Factory pipelines on Microsoft Fabric and Azure Data Factory.
 License: MIT
 Keywords: fabric,datafactory,unit-testing,functional-testing,azure

{data_factory_testing_framework-0.2.4 → data_factory_testing_framework-0.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "data-factory-testing-framework"
-version = "0.2.4"
+version = "0.2.6"
 description = "A stand-alone test framework that allows to write unit tests for Data Factory pipelines on Microsoft Fabric and Azure Data Factory."
 authors = ["Arjen Kroezen <arjenkroezen@microsoft.com>", "Yennifer Santos <ysantos@microsoft.com>", "Jaya Kumar <kumarjaya@microsoft.com>", "Leonard Herold <lherold@microsoft.com>"]
 readme = "README.md"

{data_factory_testing_framework-0.2.4 → data_factory_testing_framework-0.2.6}/src/data_factory_testing_framework/_deserializers/shared/_activity_deserializer.py

@@ -43,7 +43,11 @@ def _get_activity_from_activity_data(activity_data: dict) -> Activity:
         child_activities = _get_activity_from_activities_data(type_properties["activities"])
         return ForEachActivity(activities=child_activities, **activity_data)
     elif activity_data["type"] == "Switch":
-        default_activities = _get_activity_from_activities_data(type_properties["defaultActivities"])
+        default_activities = (
+            _get_activity_from_activities_data(type_properties["defaultActivities"])
+            if "defaultActivities" in type_properties
+            else []
+        )
         cases_activities = {}
         for case in type_properties["cases"]:
             case_value = case["value"]

{data_factory_testing_framework-0.2.4 → data_factory_testing_framework-0.2.6}/src/data_factory_testing_framework/_repositories/data_factory_repository.py

@@ -38,4 +38,4 @@ class DataFactoryRepository:
             if pipeline.name == name:
                 return pipeline
 
-        raise PipelineNotFoundError(f"Pipeline with name {name} not found")
+        raise PipelineNotFoundError(f"Pipeline with name: '{name}' not found")

{data_factory_testing_framework-0.2.4 → data_factory_testing_framework-0.2.6}/src/data_factory_testing_framework/_test_framework.py

@@ -54,21 +54,32 @@ class TestFramework:
             The repository attribute will be populated with the data factory entities if provided.
             should_evaluate_child_pipelines: optional boolean indicating whether child pipelines should be evaluated. Defaults to False.
         """
-        if framework_type == TestFrameworkType.Fabric:
+        self._framework_type = framework_type
+
+        if self._framework_type == TestFrameworkType.Fabric:
             if root_folder_path is not None:
                 self._repository = FabricRepositoryFactory().parse_from_folder(root_folder_path)
             else:
                 self._repository = DataFactoryRepository([])
-        elif framework_type == TestFrameworkType.DataFactory:
+        elif self._framework_type == TestFrameworkType.DataFactory:
             if root_folder_path is not None:
                 self._repository = DataFactoryRepositoryFactory().parse_from_folder(root_folder_path)
             else:
                 self._repository = DataFactoryRepository([])
-        elif framework_type == TestFrameworkType.Synapse:
+        elif self._framework_type == TestFrameworkType.Synapse:
             raise NotImplementedError("Synapse test framework is not implemented yet.")
 
         self._should_evaluate_child_pipelines = should_evaluate_child_pipelines
 
+    @property
+    def framework_type(self) -> TestFrameworkType:
+        """Indicates which test framework is used.
+
+        Returns:
+            A TestFrameworkType object.
+        """
+        return self._framework_type
+
     @property
     def should_evaluate_child_pipelines(self) -> bool:
         """Indicates whether child pipelines should be evaluated.
@@ -146,10 +157,16 @@ class TestFramework:
                     activities_iterator = []
                     if isinstance(activity, ExecutePipelineActivity) and self.should_evaluate_child_pipelines:
                         execute_pipeline_activity: ExecutePipelineActivity = activity
-                        pipeline = self.get_pipeline_by_id(
-                            # Note: in the future they will probably rename this to referenceId for Fabric
-                            execute_pipeline_activity.type_properties["pipeline"]["referenceName"],
-                        )
+                        pipeline: Pipeline = None
+                        if self.framework_type == TestFrameworkType.Fabric:
+                            pipeline = self.get_pipeline_by_id(
+                                # Note: in the future they will probably rename this to referenceId for Fabric
+                                execute_pipeline_activity.type_properties["pipeline"]["referenceName"],
+                            )
+                        else:
+                            pipeline = self.get_pipeline_by_name(
+                                execute_pipeline_activity.type_properties["pipeline"]["referenceName"],
+                            )
                         activities_iterator = execute_pipeline_activity.evaluate_pipeline(
                             pipeline,
                             activity.get_child_run_parameters(state),