From 73acf52329817c479891d48d4f2110032e746855 Mon Sep 17 00:00:00 2001
From: Roman <zajic@zajic.net>
Date: Fri, 7 Mar 2025 02:13:16 +0000
Subject: [PATCH] test: invalid rest API

---
 src/api_clients/invalid_rest.py | 67 +++++++++++++++++++++++++++++++++
 src/cli/nomos_cli.py            |  4 ++
 2 files changed, 71 insertions(+)
 create mode 100644 src/api_clients/invalid_rest.py

diff --git a/src/api_clients/invalid_rest.py b/src/api_clients/invalid_rest.py
new file mode 100644
index 0000000..4d026fd
--- /dev/null
+++ b/src/api_clients/invalid_rest.py
@@ -0,0 +1,67 @@
+import random
+import string
+
+from src.api_clients.rest import REST
+from src.libs.common import generate_random_bytes
+from src.libs.custom_logger import get_custom_logger
+import json
+
+logger = get_custom_logger(__name__)
+
+
+def alter_dispersal_data(data):
+
+    # Add random bytes to data and break padding
+    def alter_data_content():
+        random_n = random.randint(1, 31)
+        data["data"].extend(list(generate_random_bytes(random_n)))
+
+    # Change structure and content for metadata
+    def alter_metadata():
+        random_n = random.randint(7, 32)
+        data["metadata"] = list(generate_random_bytes(random_n))
+
+    # Add random property to the data object with random list content
+    def add_random_property():
+        random_k = random.randint(1, 16)
+        random_n = random.randint(7, 64)
+        random_str = "".join(random.choices(string.printable, k=random_k))
+        data[random_str] = list(generate_random_bytes(random_n))
+
+    choice = random.choice([alter_data_content, alter_metadata, add_random_property])
+    choice()
+
+    return data
+
+
+def alter_get_range_query(query):
+
+    # Swap range high with range low
+    def swap_range():
+        end = query["range"]["end"]
+        query["range"]["end"] = query["range"]["start"]
+        query["range"]["start"] = end
+
+    # Change app id
+    def alter_app_id():
+        random_n = random.randint(8, 33)
+        query["app_id"] = list(generate_random_bytes(random_n))
+
+    choice = random.choice([swap_range, alter_app_id])
+    choice()
+
+    return query
+
+
+class INVALID_REST(REST):
+    def __init__(self, port, host="127.0.0.1"):
+        super().__init__(port, host)
+
+    def send_dispersal_request(self, data):
+        data = alter_dispersal_data(data)
+        return self.rest_call("post", "disperse-data", json.dumps(data))
+
+    def send_get_range(self, query):
+        query = alter_get_range_query(query)
+        response = self.rest_call("post", "da/get-range", json.dumps(query))
+        return response.json()
diff --git a/src/cli/nomos_cli.py b/src/cli/nomos_cli.py
index 042fc65..af9a053 100644
--- a/src/cli/nomos_cli.py
+++ b/src/cli/nomos_cli.py
@@ -2,6 +2,7 @@ import json
 import os
 import re
 
+from src.api_clients.invalid_rest import INVALID_REST
 from src.api_clients.rest import REST
 from src.data_storage import DS
 from src.libs.common import generate_log_prefix, delay, remove_padding
@@ -112,6 +113,9 @@ class NomosCli:
     def set_rest_api(self, host, port):
         self._api = REST(port, host)
 
+    def set_invalid_rest_api(self, host, port):
+        self._api = INVALID_REST(port, host)
+
     @retry(stop=stop_after_delay(5), wait=wait_fixed(0.1), reraise=True)
     def stop(self):
         self._container = stop(self._container)