change: push powerbi_dataset_refresh_status before raising exception

Author: Ohashiro

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

@@ -153,11 +153,10 @@ def execute_complete(self, context: Context, event: dict[str, str]) -> Any:
         Relies on trigger to throw an exception, otherwise it assumes execution was successful.
         """
         if event:
-            if event["status"] == "error":
-                raise AirflowException(event["message"])
-
             self.xcom_push(
                 context=context,
                 key=f"{self.task_id}.powerbi_dataset_refresh_status",
                 value=event["dataset_refresh_status"],
             )
+            if event["status"] == "error":
+                raise AirflowException(event["message"])

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

@@ -144,7 +144,7 @@ def test_powerbi_operator_async_execute_complete_fail(self):
                     "dataset_refresh_id": "1234",
                 },
             )
-        assert context["ti"].xcom_push.call_count == 0
+        assert context["ti"].xcom_push.call_count == 1
         assert str(exc.value) == "error"
 
     def test_powerbi_operator_refresh_fail(self):
@@ -163,7 +163,7 @@ def test_powerbi_operator_refresh_fail(self):
                     "dataset_refresh_id": "1234",
                 },
             )
-        assert context["ti"].xcom_push.call_count == 0
+        assert context["ti"].xcom_push.call_count == 1
         assert str(exc.value) == "error message"
 
     def test_execute_complete_no_event(self):