Skip to content

Commit

Permalink
[Event] Add topic property (#61)
Browse files Browse the repository at this point in the history
  • Loading branch information
TomerShor authored Jan 15, 2024
1 parent a03e92d commit c7661e0
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 2 deletions.
4 changes: 4 additions & 0 deletions nuclio_sdk/event.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@ def _from_parsed_data(parsed_data, body):
type_version=parsed_data["type_version"],
version=parsed_data["version"],
offset=parsed_data.get("offset", 0),
topic=parsed_data.get("topic"),
)

@staticmethod
Expand All @@ -86,6 +87,7 @@ def _from_parsed_data_bytes(parsed_data, body):
type_version=parsed_data[b"type_version"],
version=parsed_data[b"version"],
offset=parsed_data.get(b"offset", 0),
topic=parsed_data.get(b"topic"),
)


Expand Down Expand Up @@ -162,6 +164,7 @@ def __init__(
version=None,
last_in_batch=None,
offset=None,
topic=None,
):
self.body = body
self.content_type = content_type
Expand All @@ -181,6 +184,7 @@ def __init__(
self.version = version
self.last_in_batch = last_in_batch or False
self.offset = offset or 0
self.topic = topic

def to_json(self):
obj = {}
Expand Down
6 changes: 4 additions & 2 deletions nuclio_sdk/test/test_event.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ def test_event_to_json_bytes_body(self):
serialized_event.trigger.__dict__,
{"kind": "http", "name": "my-http-trigger"},
)
self.assertEqual(serialized_event.last_in_batch, False)
self.assertFalse(serialized_event.last_in_batch)
self.assertEqual(serialized_event.offset, 0)

def test_event_to_json_bytes_non_utf8able_body(self):
Expand All @@ -45,9 +45,11 @@ def test_event_to_json_bytes_non_utf8able_body(self):

def test_event_to_json_string_body(self):
request_body = "str-body"
event = nuclio_sdk.Event(body=request_body)
topic = "my-topic"
event = nuclio_sdk.Event(body=request_body, topic=topic)
serialized_event = self._deserialize_event(event)
self.assertEqual(request_body, serialized_event.body)
self.assertEqual(topic, serialized_event.topic)

def test_print_event(self):
"""
Expand Down

0 comments on commit c7661e0

Please sign in to comment.