File tree Expand file tree Collapse file tree 2 files changed +15
-16
lines changed
src/airflow/providers/microsoft/azure/triggers
tests/microsoft/azure/triggers Expand file tree Collapse file tree 2 files changed +15
-16
lines changed Original file line number Diff line number Diff 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 ()
Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments