fix: CI tests

apache · Jan 10, 2025 · c06cac9 · c06cac9
1 parent 35e25f3
commit c06cac9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 16 deletions.
diff --git a/providers/src/airflow/providers/microsoft/azure/triggers/powerbi.py b/providers/src/airflow/providers/microsoft/azure/triggers/powerbi.py
@@ -153,8 +153,8 @@ async def fetch_refresh_status_and_error() -> tuple[str, str]:
                     refresh_id=self.dataset_refresh_id,
                 )
                 return refresh_details["status"], refresh_details["error"]
-            else:
-                raise PowerBIDatasetRefreshException("Dataset refresh Id is missing.")
+
+            raise PowerBIDatasetRefreshException("Dataset refresh Id is missing.")
 
         try:
             dataset_refresh_status, dataset_refresh_error = await fetch_refresh_status_and_error()

diff --git a/providers/tests/microsoft/azure/triggers/test_powerbi.py b/providers/tests/microsoft/azure/triggers/test_powerbi.py
@@ -220,20 +220,19 @@ async def test_powerbi_trigger_run_PowerBIDatasetRefreshException_during_refresh
         mock_get_refresh_details_by_refresh_id.side_effect = PowerBIDatasetRefreshException("Test exception")
         mock_trigger_dataset_refresh.return_value = DATASET_REFRESH_ID
 
-        with mock.patch("tenacity.wait.wait_exponential.__call__") as mock_retry:
-            task = [i async for i in powerbi_trigger.run()]
-            response = TriggerEvent(
-                {
-                    "status": "error",
-                    "dataset_refresh_status": None,
-                    "message": "An error occurred: Test exception",
-                    "dataset_refresh_id": DATASET_REFRESH_ID,
-                }
-            )
-            assert len(task) == 1
-            assert response in task
-            assert mock_cancel_dataset_refresh.call_count == 1
-            assert mock_retry.call_count == 3
+        task = [i async for i in powerbi_trigger.run()]
+        response = TriggerEvent(
+            {
+                "status": "error",
+                "dataset_refresh_status": None,
+                "message": "An error occurred: Test exception",
+                "dataset_refresh_id": DATASET_REFRESH_ID,
+            }
+        )
+        assert mock_get_refresh_details_by_refresh_id.call_count == 3
+        assert len(task) == 1
+        assert response in task
+        assert mock_cancel_dataset_refresh.call_count == 1
 
     @pytest.mark.asyncio
     @mock.patch(f"{MODULE}.hooks.powerbi.PowerBIHook.cancel_dataset_refresh")