From 407dec7493c3314385324ce5dcc6b32505048cc7 Mon Sep 17 00:00:00 2001 From: Florin Barbu Date: Fri, 24 May 2024 12:52:47 +0300 Subject: [PATCH] fixes --- src/node/waku_message.py | 6 +++--- src/steps/store.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/node/waku_message.py b/src/node/waku_message.py index 7b39a677..5b7e2311 100644 --- a/src/node/waku_message.py +++ b/src/node/waku_message.py @@ -13,7 +13,7 @@ class MessageRpcResponse: timestamp: Optional[int] ephemeral: Optional[bool] meta: Optional[str] - rateLimitProof: str = field(default_factory=dict) + rateLimitProof: Optional[str] = field(default_factory=dict) rate_limit_proof: Optional[dict] = field(default_factory=dict) @@ -24,8 +24,8 @@ class WakuMessage: self.message_rpc_response_schema = class_schema(self.schema)() @allure.step - def assert_received_message(self, sent_message): - message = self.message_rpc_response_schema.load(self.received_messages) + def assert_received_message(self, sent_message, index=0): + message = self.message_rpc_response_schema.load(self.received_messages[index]) def assert_fail_message(field_name): return f"Incorrect field: {field_name}. Published: {sent_message[field_name]} Received: {getattr(message, field_name)}" diff --git a/src/steps/store.py b/src/steps/store.py index ae1f779b..fde10409 100644 --- a/src/steps/store.py +++ b/src/steps/store.py @@ -221,7 +221,7 @@ class StepsStore(StepsCommon): assert self.store_response.messages, f"Peer {node.image} couldn't find any messages. Actual response: {self.store_response}" assert len(self.store_response.messages) >= 1, "Expected at least 1 message but got none" store_message_index = -1 # we are looking for the last and most recent message in the store - waku_message = WakuMessage(self.store_response.messages[store_message_index:]) + waku_message = WakuMessage([self.store_response.messages[store_message_index:]]) if store_v == "v1": waku_message.assert_received_message(message_to_check) else: