test(azure): fix missing argument issue

astronomer · Oct 18, 2023 · bdef5a1 · bdef5a1
1 parent 2009f93
commit bdef5a1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/microsoft/azure/sensors/test_data_factory.py b/tests/microsoft/azure/sensors/test_data_factory.py
@@ -19,6 +19,8 @@ class TestAzureDataFactoryPipelineRunStatusSensorAsync:
     SENSOR = AzureDataFactoryPipelineRunStatusSensorAsync(
         task_id="pipeline_run_sensor_async",
         run_id=RUN_ID,
+        factory_name="factory_name",
+        resource_group_name="resource_group_name",
     )
 
     @mock.patch(f"{MODULE}.AzureDataFactoryPipelineRunStatusSensorAsync.defer")
@@ -61,5 +63,9 @@ def test_poll_interval_deprecation_warning(self):
         # TODO: Remove once deprecated
         with pytest.warns(expected_warning=DeprecationWarning):
             AzureDataFactoryPipelineRunStatusSensorAsync(
-                task_id="pipeline_run_sensor_async", run_id=self.RUN_ID, poll_interval=5.0
+                task_id="pipeline_run_sensor_async",
+                run_id=self.RUN_ID,
+                poll_interval=5.0,
+                factory_name="factory_name",
+                resource_group_name="resource_group_name",
             )