refactor(event_source): convert functional tests to unit tests (aws-powertools#2506)

Author: leandrodamascena

aws_lambda_powertools/utilities/data_classes/active_mq_event.py

@@ -1,4 +1,4 @@
-from typing import Any, Iterator, Optional
+from typing import Any, Dict, Iterator, Optional
 
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
 from aws_lambda_powertools.utilities.data_classes.shared_functions import base64_decode
@@ -112,6 +112,10 @@ class ActiveMQEvent(DictWrapper):
     - https://aws.amazon.com/blogs/compute/using-amazon-mq-as-an-event-source-for-aws-lambda/
     """
 
+    def __init__(self, data: Dict[str, Any]):
+        super().__init__(data)
+        self._messages: Optional[Iterator[ActiveMQMessage]] = None
+
     @property
     def event_source(self) -> str:
         return self["eventSource"]
@@ -128,4 +132,20 @@ def messages(self) -> Iterator[ActiveMQMessage]:
 
     @property
     def message(self) -> ActiveMQMessage:
-        return next(self.messages)
+        """
+        Returns the next ActiveMQ message using an iterator
+
+        Returns
+        -------
+        ActiveMQMessage
+            The next activemq message.
+
+        Raises
+        ------
+        StopIteration
+            If there are no more records available.
+
+        """
+        if self._messages is None:
+            self._messages = self.messages
+        return next(self._messages)

aws_lambda_powertools/utilities/data_classes/kafka_event.py

@@ -87,6 +87,10 @@ class KafkaEvent(DictWrapper):
     - https://docs.aws.amazon.com/lambda/latest/dg/with-msk.html
     """
 
+    def __init__(self, data: Dict[str, Any]):
+        super().__init__(data)
+        self._records: Optional[Iterator[KafkaEventRecord]] = None
+
     @property
     def event_source(self) -> str:
         """The AWS service from which the Kafka event record originated."""
@@ -116,5 +120,20 @@ def records(self) -> Iterator[KafkaEventRecord]:
 
     @property
     def record(self) -> KafkaEventRecord:
-        """The next Kafka record."""
-        return next(self.records)
+        """
+        Returns the next Kafka record using an iterator.
+
+        Returns
+        -------
+        KafkaEventRecord
+            The next Kafka record.
+
+        Raises
+        ------
+        StopIteration
+            If there are no more records available.
+
+        """
+        if self._records is None:
+            self._records = self.records
+        return next(self._records)