diff --git a/src/node/store_response.py b/src/node/store_response.py index 53052060..602ad007 100644 --- a/src/node/store_response.py +++ b/src/node/store_response.py @@ -87,3 +87,7 @@ class StoreResponse: return self.messages[index]["pubsub_topic"] else: return None + + @property + def resp_json(self): + return self.response diff --git a/src/steps/store.py b/src/steps/store.py index fde10409..9a1cb099 100644 --- a/src/steps/store.py +++ b/src/steps/store.py @@ -218,7 +218,7 @@ class StepsStore(StepsCommon): **kwargs, ) - assert self.store_response.messages, f"Peer {node.image} couldn't find any messages. Actual response: {self.store_response}" + assert self.store_response.messages, f"Peer {node.image} couldn't find any messages. Actual response: {self.store_response.resp_json}" 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:]]) diff --git a/tests/store/test_api_flags.py b/tests/store/test_api_flags.py index 471d524b..8c0cba1b 100644 --- a/tests/store/test_api_flags.py +++ b/tests/store/test_api_flags.py @@ -23,5 +23,5 @@ class TestApiFlags(StepsStore): for index in range(len(store_response.messages)): assert store_response.message_payload(index) == message_list[index]["payload"] assert store_response.message_pubsub_topic(index) == self.test_pubsub_topic - waku_message = WakuMessage(store_response.message_at(index)) + waku_message = WakuMessage([store_response.message_at(index)]) waku_message.assert_received_message(message_list[index])