46 lines
1.6 KiB
Python

import pytest
from src.env_vars import CONSENSUS_SLOT_TIME
from src.libs.common import delay, generate_text_data, to_app_id, to_index
from src.libs.custom_logger import get_custom_logger
from src.steps.da import StepsDataAvailability
logger = get_custom_logger(__name__)
class TestLargeVolume(StepsDataAvailability):
@pytest.mark.usefixtures("setup_4_node_cluster")
@pytest.mark.parametrize(
"setup_4_node_cluster,raw_data_size",
[
({"subnet_size": 4, "dispersal_factor": 1}, 50), # => ~~0.5kB
({"subnet_size": 64, "dispersal_factor": 16}, 800), # => ~~ 4kB
({"subnet_size": 2048, "dispersal_factor": 512}, 53 * 1024), # => ~~254kB, spec limit: 256kB
],
indirect=["setup_4_node_cluster"],
)
def test_large_volume_dispersal(self, raw_data_size):
data = generate_text_data(raw_data_size)
try:
response = self.disperse_data(data, to_app_id(1), to_index(0), timeout_duration=0)
except Exception as ex:
raise Exception(f"Dispersal was not successful with error {ex}")
assert response.status_code == 200
delay(CONSENSUS_SLOT_TIME)
self.get_data_range(self.node2, to_app_id(1), to_index(0), to_index(5), timeout_duration=20, interval=1)
@pytest.mark.usefixtures("setup_2_node_cluster")
@pytest.mark.parametrize(
"setup_2_node_cluster,raw_data_size",
[
({"subnet_size": 2, "dispersal_factor": 2}, 50),
],
indirect=["setup_2_node_cluster"],
)
def test_large_volume_dispersal_2node(self, raw_data_size):
self.test_large_volume_dispersal(raw_data_size)