This commit is contained in:
Florin Barbu 2024-05-24 12:52:47 +03:00
parent 3df20332fa
commit 407dec7493
No known key found for this signature in database
GPG Key ID: 593D6DBC6D9E5095
2 changed files with 4 additions and 4 deletions

View File

@ -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)}"

View File

@ -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: