test(providers/microsoft): fix test_execute_failure_in_deferrable_mode_with_soft_fail expected exception

Author: Lee-W

tests/providers/amazon/aws/sensors/test_batch.py

@@ -20,7 +20,7 @@
 
 import pytest
 
-from airflow.exceptions import AirflowException, TaskDeferred
+from airflow.exceptions import AirflowException, AirflowSkipException, TaskDeferred
 from airflow.providers.amazon.aws.hooks.batch_client import BatchClientHook
 from airflow.providers.amazon.aws.sensors.batch import (
     BatchComputeEnvironmentSensor,
@@ -103,7 +103,7 @@ def test_execute_failure_in_deferrable_mode(self, deferrable_batch_sensor: Batch
     def test_execute_failure_in_deferrable_mode_with_soft_fail(self, deferrable_batch_sensor: BatchSensor):
         """Tests that an AirflowSkipException is raised in case of error event and soft_fail is set to True"""
         deferrable_batch_sensor.soft_fail = True
-        with pytest.raises(AirflowException):
+        with pytest.raises(AirflowSkipException):
             deferrable_batch_sensor.execute_complete(context={}, event={"status": "failure"})