logos-blockchain/logos-blockchain-e2e-tests
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-e2e-tests.git synced 2026-01-09 08:33:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: invalid REST API

Browse Source
This commit is contained in:
Roman 2025-03-11 23:39:50 +00:00
parent e69c023010
commit b8c6764bf1
No known key found for this signature in database
GPG Key ID: BB3828275C58EFF1
4 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/api_clients/invalid_rest.py
View File

@ -31,6 +31,7 @@ def alter_dispersal_data(data):
choice = random.choice([alter_data_content, alter_metadata, add_random_property])
choice()
logger.debug(f"Data for dispersal request has been altered to: {data}")
return data
@ -50,16 +51,18 @@ def alter_get_range_query(query):
choice = random.choice([swap_range, alter_app_id])
choice()
logger.debug(f"Get-range query has been altered to: {query}")
return query
class INVALID_REST(REST):
class InvalidRest(REST):
def __init__(self, rest_port):
super().__init__(rest_port)
def send_dispersal_request(self, data):
data = alter_dispersal_data(data)
return self.rest_call("post", "disperse-data", json.dumps(data))
response = self.rest_call("post", "disperse-data", json.dumps(data))
return response
def send_get_range(self, query):
query = alter_get_range_query(query)

2
src/client/nomos_cli.py
View File

@ -2,8 +2,6 @@ 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
from src.libs.custom_logger import get_custom_logger

4
src/client/proxy_client.py
View File

@ -2,7 +2,7 @@ import json
import os
import re
from src.api_clients.invalid_rest import INVALID_REST
from src.api_clients.invalid_rest import InvalidRest
from src.api_clients.rest import REST
from src.data_storage import DS
from src.libs.common import generate_log_prefix, delay, remove_padding
@ -81,7 +81,7 @@ class ProxyClient:
self._api = REST(self._tcp_port)
def set_invalid_rest_api(self):
self._api = INVALID_REST(self._tcp_port)
self._api = InvalidRest(self._tcp_port)
@retry(stop=stop_after_delay(5), wait=wait_fixed(0.1), reraise=True)
def stop(self):

2
tests/dos_robustness/test_high_load_dos.py
View File

@ -170,7 +170,7 @@ class TestHighLoadDos(StepsDataAvailability):
break
dispersal_cl, download_cl = random.choice(self.client_nodes[1::2]), random.choice(self.client_nodes[::2])
invalid = random.choice([False])
invalid = random.choice([False, True])
delay(0.01)
try: