Chore: store v3 edge test cases

src/node/store_response.py

@@ -59,6 +59,16 @@ def message_hash(self, index):
         else:
             return None
 
+    def message_content(self, index):
+        try:
+            if self.messages is not None:
+                payload = self.messages[index]["message"]["contentTopic"]
+                return payload
+            else:
+                return None
+        except IndexError:
+            return None
+
     def message_payload(self, index):
         try:
             if self.messages is not None:

tests/store/test_api_flags.py

@@ -1,8 +1,14 @@
 import pytest
+from time import time
+from src.libs.custom_logger import get_custom_logger
 from src.libs.common import to_base64
 from src.node.waku_message import WakuMessage
+from src.node.store_response import StoreResponse
 from src.steps.store import StepsStore
 from src.test_data import SAMPLE_INPUTS
+from src.test_data import PUBSUB_TOPICS_STORE
+
+logger = get_custom_logger(__name__)
 
 
 @pytest.mark.usefixtures("node_setup")
@@ -11,6 +17,36 @@ def test_store_with_peerAddr(self):
         self.publish_message()
         self.check_published_message_is_stored(store_node=self.store_node1, peer_addr=self.multiaddr_list[0])
 
+    def test_store_with_wrongPeerAddr(self):
+        self.publish_message()
+        wrong_peer_addr = self.multiaddr_list[0][1:]
+        logger.debug(f"Running test with wrong_peer_addr: {wrong_peer_addr}")
+        try:
+            self.check_published_message_is_stored(store_node=self.store_node1, peer_addr=wrong_peer_addr)
+            raise Exception("Message stored with wrong peer address")
+        except Exception as ex:
+            logger.debug(f" Response with wrong peer address is { ex.args[0]}")
+            assert ex.args[0].find("Invalid MultiAddress") != -1
+        # try to send wrong peer id
+        wrong_peer_id = self.multiaddr_list[0][:-1]
+        logger.debug(f"Running test with wrong_peer_addr {wrong_peer_id}")
+        try:
+            self.check_published_message_is_stored(store_node=self.store_node1, peer_addr=wrong_peer_id)
+            raise Exception("Message restored with wrong peer id")
+        except Exception as ex:
+            logger.debug(f" Response with wrong peer id is { ex.args[0]}")
+            assert ex.args[0].find("Failed parsing remote peer info") != -1
+
+        # send address without /tcp number
+        wrong_peer_addr = self.multiaddr_list[0].replace("/tcp", "")
+        logger.debug(f"logger is {wrong_peer_addr}")
+        try:
+            self.check_published_message_is_stored(store_node=self.store_node1, peer_addr=wrong_peer_addr)
+            raise Exception("Message stored with wrong peer address")
+        except Exception as ex:
+            logger.debug(f" Response with wrong peer address is { ex.args[0]}")
+            assert ex.args[0].find("Unsupported protocol") != -1
+
     def test_store_include_data(self):
         message_list = []
         for payload in SAMPLE_INPUTS:
@@ -25,3 +61,24 @@ def test_store_include_data(self):
                 assert store_response.message_pubsub_topic(index) == self.test_pubsub_topic
                 waku_message = WakuMessage([store_response.message_at(index)])
                 waku_message.assert_received_message(message_list[index])
+
+    def test_store_not_include_data(self):
+        self.publish_message(message=self.create_message())
+        store_response = self.get_messages_from_store(self.store_node1, include_data="false")
+        logger.debug(f" Message restored with hash only is {store_response.messages} ")
+        assert "message" not in store_response.messages
+
+    def test_get_store_messages_with_wrong_typo_content_topic(self):
+        # positive scenario
+        content_topic = "/myapp/1/latest/protoo"
+        logger.debug(f"Trying to publish msg with content topic {content_topic}")
+        msg = self.publish_message(message=self.create_message(contentTopic=content_topic))
+        store_response = self.get_messages_from_store(self.store_node1, include_data="true", content_topics=content_topic)
+        try:
+            if store_response.messages is not None:
+                stored_contentTopic = store_response.message_content(0)
+                logger.debug(f"stored content topic is {stored_contentTopic}")
+                assert stored_contentTopic == content_topic, "content topics don't match"
+
+        except Exception as e:
+            raise Exception(f"can't get message with content topic {content_topic}")

tests/store/test_topics.py

@@ -2,6 +2,9 @@
 from src.env_vars import NODE_2
 from src.steps.store import StepsStore
 from src.test_data import CONTENT_TOPICS_DIFFERENT_SHARDS
+from src.libs.custom_logger import get_custom_logger
+
+logger = get_custom_logger(__name__)
 
 
 @pytest.mark.xfail("go-waku" in NODE_2, reason="Bug reported: https://github.com/waku-org/go-waku/issues/1108")
@@ -79,3 +82,29 @@ def test_store_without_pubsub_topic_and_content_topic(self):
         for node in self.store_nodes:
             store_response = node.get_store_messages(page_size=20, ascending="true")
             assert len(store_response["messages"]) == len(CONTENT_TOPICS_DIFFERENT_SHARDS), "Message count mismatch"
+
+    def test_store_with_not_valid_content_topic(self):
+        empty_content_topic = ""
+        for node in self.store_nodes:
+            store_response = node.get_store_messages(page_size=20, include_data="true", ascending="true", content_topics=empty_content_topic)
+            assert len(store_response["messages"]) == len(CONTENT_TOPICS_DIFFERENT_SHARDS), "Message count mismatch"
+        # test with space string content topic
+        space_content_topic = " "
+        try:
+            store_response = self.store_nodes[0].get_store_messages(
+                page_size=20, include_data="true", ascending="true", content_topics=space_content_topic
+            )
+            logger.debug(f" response for empty content_topic {store_response}")
+            assert store_response["messages"] == [], "message stored with wrong topic "
+        except Exception as e:
+            raise Exception("couldn't get stored message")
+        # test with wrong url
+        wrong_content_topic = "myapp/1/latest/proto"
+        try:
+            store_response = self.store_nodes[0].get_store_messages(
+                page_size=20, include_data="true", ascending="true", content_topics=wrong_content_topic
+            )
+            logger.debug(f" response for wrong url content topic is {store_response}")
+            assert store_response["messages"] == [], "message stored with wrong topic "
+        except Exception as e:
+            raise Exception("couldn't get stored message")