Skip to content

Commit ffaf334

Browse files
Refactoring tests
1 parent 01f72a2 commit ffaf334

File tree

1 file changed

+4
-0
lines changed
  • aws_lambda_powertools/utilities/kafka/serialization

1 file changed

+4
-0
lines changed

aws_lambda_powertools/utilities/kafka/serialization/dataclass.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from __future__ import annotations
22

3+
from dataclasses import is_dataclass
34
from typing import TYPE_CHECKING, Any, cast
45

56
from aws_lambda_powertools.utilities.kafka.serialization.base import OutputSerializerBase
@@ -11,4 +12,7 @@
1112

1213
class DataclassOutputSerializer(OutputSerializerBase):
1314
def serialize(self, data: dict[str, Any], output: type[T] | Callable | None = None) -> T | dict[str, Any]:
15+
if not is_dataclass(output):
16+
raise ValueError("Output class must be a dataclass")
17+
1418
return cast(T, output(**data))

0 commit comments

Comments
 (0)