data-factory-testing-framework 1.0.2__tar.gz → 1.0.4__tar.gz

{data_factory_testing_framework-1.0.2/src/data_factory_testing_framework.egg-info → data_factory_testing_framework-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: data-factory-testing-framework
-Version: 1.0.2
+Version: 1.0.4
 Summary: A stand-alone test framework that allows to write unit tests for Data Factory pipelines on Microsoft Fabric and Azure Data Factory.
 Author-email: Data Factory Testing Framework <dftf@microsoft.com>
 License:     MIT License

data_factory_testing_framework-1.0.4/VERSION

@@ -0,0 +1 @@
+1.0.4

{data_factory_testing_framework-1.0.2 → data_factory_testing_framework-1.0.4}/src/data_factory_testing_framework/models/_pipeline.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, List
+from typing import TYPE_CHECKING, Any, List, Union
 
 from data_factory_testing_framework.exceptions import ActivityNotFoundError
 
@@ -79,10 +79,25 @@ class Pipeline:
 
         return parameters
 
+    def _get_default_value_for_variable(self, variable_value: dict[str, Any]) -> Union[str, int, bool, List[Any]]:
+        """Get the default value of a variable."""
+        if variable_value["type"] == "String":
+            return variable_value.get("defaultValue", "")
+        elif variable_value["type"] == "Integer":
+            return variable_value.get("defaultValue", 0)
+        elif variable_value["type"] == "Boolean":
+            return variable_value.get("defaultValue", False)
+        elif variable_value["type"] == "Array":
+            return variable_value.get("defaultValue", [])
+        else:
+            raise NotImplementedError(f"Type {variable_value['type']} not implemented")
+
     def get_run_variables(self) -> List[PipelineRunVariable]:
         """Get the run variables for the pipeline. This can be used to generate the instance variables for a pipeline run."""
         run_variables = []
         for variable_name, variable_value in self.variables.items():
-            run_variables.append(PipelineRunVariable(variable_name, variable_value.get("default_value", None)))
+            run_variables.append(
+                PipelineRunVariable(variable_name, self._get_default_value_for_variable(variable_value))
+            )
 
         return run_variables

{data_factory_testing_framework-1.0.2 → data_factory_testing_framework-1.0.4/src/data_factory_testing_framework.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: data-factory-testing-framework
-Version: 1.0.2
+Version: 1.0.4
 Summary: A stand-alone test framework that allows to write unit tests for Data Factory pipelines on Microsoft Fabric and Azure Data Factory.
 Author-email: Data Factory Testing Framework <dftf@microsoft.com>
 License:     MIT License

data_factory_testing_framework-1.0.2/VERSION

@@ -1 +0,0 @@
-1.0.2