mirror of
https://github.com/logos-blockchain/logos-blockchain-e2e-tests.git
synced 2026-01-02 21:23:07 +00:00
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import pytest
|
|
|
|
from src.libs.common import delay, to_app_id, to_index
|
|
from src.libs.custom_logger import get_custom_logger
|
|
from src.steps.da import StepsDataAvailability
|
|
from src.test_data import DATA_TO_DISPERSE
|
|
|
|
logger = get_custom_logger(__name__)
|
|
|
|
|
|
class TestDosRobustness(StepsDataAvailability):
|
|
main_nodes = []
|
|
|
|
@pytest.mark.usefixtures("setup_2_node_cluster")
|
|
def test_spam_protection_valid_uploads(self):
|
|
num_samples = len(DATA_TO_DISPERSE)
|
|
missing_dispersals = num_samples
|
|
for i in range(num_samples):
|
|
try:
|
|
self.disperse_data(DATA_TO_DISPERSE[i], to_app_id(1), to_index(0), timeout_duration=0)
|
|
missing_dispersals -= 1
|
|
except Exception as ex:
|
|
logger.error(f"Dispersal #{i+1} was not successful with error {ex}")
|
|
break
|
|
|
|
delay(0.1)
|
|
|
|
assert missing_dispersals == 0, f"{missing_dispersals} dispersals were not successful"
|
|
|
|
@pytest.mark.usefixtures("setup_2_node_cluster")
|
|
def test_spam_protection_single_burst(self):
|
|
successful_dispersals = 0
|
|
for i in range(1000):
|
|
try:
|
|
self.disperse_data(DATA_TO_DISPERSE[0], to_app_id(1), to_index(0), timeout_duration=0)
|
|
successful_dispersals = i
|
|
except Exception as ex:
|
|
logger.debug(f"Dispersal #{i+1} was not successful with error {ex}")
|
|
break
|
|
|
|
assert successful_dispersals < 1000, "All 1000 consecutive dispersals were successful without any constraint"
|