Skip to content

Commit c06cac9

Browse files
committed
fix: CI tests
1 parent 35e25f3 commit c06cac9

File tree

2 files changed

+15
-16
lines changed

2 files changed

+15
-16
lines changed

providers/src/airflow/providers/microsoft/azure/triggers/powerbi.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,8 +153,8 @@ async def fetch_refresh_status_and_error() -> tuple[str, str]:
153153
refresh_id=self.dataset_refresh_id,
154154
)
155155
return refresh_details["status"], refresh_details["error"]
156-
else:
157-
raise PowerBIDatasetRefreshException("Dataset refresh Id is missing.")
156+
157+
raise PowerBIDatasetRefreshException("Dataset refresh Id is missing.")
158158

159159
try:
160160
dataset_refresh_status, dataset_refresh_error = await fetch_refresh_status_and_error()

providers/tests/microsoft/azure/triggers/test_powerbi.py

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -220,20 +220,19 @@ async def test_powerbi_trigger_run_PowerBIDatasetRefreshException_during_refresh
220220
mock_get_refresh_details_by_refresh_id.side_effect = PowerBIDatasetRefreshException("Test exception")
221221
mock_trigger_dataset_refresh.return_value = DATASET_REFRESH_ID
222222

223-
with mock.patch("tenacity.wait.wait_exponential.__call__") as mock_retry:
224-
task = [i async for i in powerbi_trigger.run()]
225-
response = TriggerEvent(
226-
{
227-
"status": "error",
228-
"dataset_refresh_status": None,
229-
"message": "An error occurred: Test exception",
230-
"dataset_refresh_id": DATASET_REFRESH_ID,
231-
}
232-
)
233-
assert len(task) == 1
234-
assert response in task
235-
assert mock_cancel_dataset_refresh.call_count == 1
236-
assert mock_retry.call_count == 3
223+
task = [i async for i in powerbi_trigger.run()]
224+
response = TriggerEvent(
225+
{
226+
"status": "error",
227+
"dataset_refresh_status": None,
228+
"message": "An error occurred: Test exception",
229+
"dataset_refresh_id": DATASET_REFRESH_ID,
230+
}
231+
)
232+
assert mock_get_refresh_details_by_refresh_id.call_count == 3
233+
assert len(task) == 1
234+
assert response in task
235+
assert mock_cancel_dataset_refresh.call_count == 1
237236

238237
@pytest.mark.asyncio
239238
@mock.patch(f"{MODULE}.hooks.powerbi.PowerBIHook.cancel_dataset_refresh")

0 commit comments

Comments
 (0)