fix: use getters for optional etag and size; default to sane defaults when possible

Author: heitorlessa

aws_lambda_powertools/utilities/data_classes/s3_event.py

@@ -32,14 +32,14 @@ def key(self) -> str:
         return unquote_plus(self["key"])
 
     @property
-    def size(self) -> str:
-        """Object size"""
-        return self["size"]
+    def size(self) -> Optional[int]:
+        """Object size. Object deletion event doesn't contain size."""
+        return self.get("size")
 
     @property
-    def etag(self) -> Optional[str]:
-        """Object etag"""
-        return self.get("etag")
+    def etag(self) -> str:
+        """Object etag. Object deletion event doesn't contain etag; we default to empty string"""
+        return self.get("etag", "")
 
     @property
     def version_id(self) -> str:
@@ -178,8 +178,8 @@ def size(self) -> int:
 
     @property
     def etag(self) -> str:
-        """object eTag"""
-        return self["s3"]["object"]["eTag"]
+        """Object eTag. Object deletion event doesn't contain eTag; we default to empty string"""
+        return self["s3"]["object"].get("eTag", "")
 
     @property
     def version_id(self) -> Optional[str]:

tests/unit/data_classes/test_s3_eventbridge_notification.py

@@ -26,10 +26,10 @@ def test_s3_eventbridge_notification_detail_parsed(raw_event: Dict):
     assert parsed_event.detail.deletion_type == raw_event["detail"].get("deletion-type")
     assert parsed_event.detail.destination_access_tier == raw_event["detail"].get("destination-access-tier")
     assert parsed_event.detail.destination_storage_class == raw_event["detail"].get("destination-storage-class")
-    assert parsed_event.detail.object.etag == raw_event["detail"]["object"]["etag"]
+    assert parsed_event.detail.object.etag == raw_event["detail"]["object"].get("etag", "")
     assert parsed_event.detail.object.key == raw_event["detail"]["object"]["key"]
     assert parsed_event.detail.object.sequencer == raw_event["detail"]["object"]["sequencer"]
-    assert parsed_event.detail.object.size == raw_event["detail"]["object"]["size"]
+    assert parsed_event.detail.object.size == raw_event["detail"]["object"].get("size")
     assert parsed_event.detail.reason == raw_event["detail"].get("reason")
     assert parsed_event.detail.version == raw_event["detail"].get("version")
     assert parsed_event.detail.request_id == raw_event["detail"]["request-id"]

tests/unit/parser/test_s3_notification.py

@@ -52,8 +52,8 @@ def test_s3_eventbridge_notification_object_deleted_event():
     assert model.detail.version == raw_event["detail"]["version"]
     assert model.detail.bucket.name == raw_event["detail"]["bucket"]["name"]
     assert model.detail.object.key == raw_event["detail"]["object"]["key"]
-    assert model.detail.object.size == raw_event["detail"]["object"]["size"]
-    assert model.detail.object.etag == raw_event["detail"]["object"]["etag"]
+    assert model.detail.object.size == raw_event["detail"]["object"].get("size")
+    assert model.detail.object.etag == raw_event["detail"]["object"].get("etag")
     assert model.detail.object.sequencer == raw_event["detail"]["object"]["sequencer"]
     assert model.detail.request_id == raw_event["detail"]["request-id"]
     assert model.detail.requester == raw_event["detail"]["requester"]