Refactoring tests

Author: leandrodamascena

aws_lambda_powertools/utilities/kafka/deserializer/base.py

@@ -49,9 +49,4 @@ def deserialize(self, data: str) -> dict[str, Any] | str | object:
         raise NotImplementedError("Subclasses must implement the deserialize method")
 
     def _decode_input(self, data: bytes | str) -> bytes:
-        try:
-            return base64.b64decode(data)
-        except Exception as e:
-            raise TypeError(
-                f"Expected bytes or base64-encoded string, got {type(data).__name__}. Error: {str(e)}",
-            ) from e
+        return base64.b64decode(data)

tests/functional/kafka_consumer/required_dependencies/test_kafka_consumer.py

@@ -290,4 +290,4 @@ def handler(event: ConsumerRecords, context):
     assert result.original_value == kafka_event_with_json_data["records"]["my-topic-1"][0]["value"]
     assert result.original_key == kafka_event_with_json_data["records"]["my-topic-1"][0]["key"]
     assert result.original_headers == kafka_event_with_json_data["records"]["my-topic-1"][0]["headers"]
-    assert result.headers
+    assert result.headers == {"headerKey": b"headerValue"}