Add tests

Author: pankajastro

tests/amazon/aws/sensors/test_s3_sensors.py

@@ -4,7 +4,7 @@
 from unittest import mock
 
 import pytest
-from airflow.exceptions import AirflowException, TaskDeferred
+from airflow.exceptions import AirflowException, AirflowSkipException, TaskDeferred
 from airflow.models import DAG, DagRun, TaskInstance
 from airflow.models.variable import Variable
 from airflow.utils import timezone
@@ -245,6 +245,16 @@ def test_s3_key_sensor_with_wildcard_async(self, mock_hook, mock_poke, context):
 
         assert isinstance(exc.value.trigger, S3KeyTrigger), "Trigger is not a S3KeyTrigger"
 
+    def test_soft_fail(self):
+        """Raise AirflowSkipException in case soft_fail is true"""
+        sensor = S3KeySensorAsync(
+            task_id="s3_key_sensor_async", bucket_key="key", bucket_name="bucket", soft_fail=True
+        )
+        with pytest.raises(AirflowSkipException):
+            sensor.execute_complete(
+                context={}, event={"status": "error", "message": "mocked error", "soft_fail": True}
+            )
+
 
 class TestS3KeysUnchangedSensorAsync:
     @mock.patch(f"{MODULE}.S3KeysUnchangedSensorAsync.defer")