import pytest

from src.env_vars import CONSENSUS_SLOT_TIME
from src.libs.common import to_app_id, to_index, delay
from src.steps.da import StepsDataAvailability
from src.test_data import DATA_TO_DISPERSE


@pytest.mark.usefixtures("setup_2_node_mod_da_cluster")
@pytest.mark.mod_da_node
class TestDispersalResilience(StepsDataAvailability):
    main_nodes = []

    @pytest.mark.parametrize("setup_2_node_mod_da_cluster", [{"validator_version": "d8bbc46", "executor_version": "d19a1f3"}], indirect=True)
    def test_integrity_kzg_commitments(self):
        # Confirm validator node has rejected dispersal request from executor - there is a mismatch between
        # column data and proofs.
        self.disperse_data(DATA_TO_DISPERSE[3], to_app_id(1), to_index(0))
        delay(CONSENSUS_SLOT_TIME)
        try:
            rcv_data = self.get_data_range(self.node2, to_app_id(1), to_index(0), to_index(5))
        except AssertionError as ae:
            assert "Get data range response is empty" in str(ae), "Get data range response should be empty"

        if rcv_data:
            raise AssertionError("Get data range response should be empty")

    @pytest.mark.parametrize("setup_2_node_mod_da_cluster", [{"validator_version": "d8bbc46", "executor_version": "7f54114"}], indirect=True)
    def test_chunkification_robustness_different_chunk_size(self):
        # Confirm validator node has rejected dispersal request from executor with different data alignment
        try:
            self.disperse_data(DATA_TO_DISPERSE[4], to_app_id(1), to_index(0), timeout_duration=0)
        except Exception as e:
            assert "does not match destination slice length" in str(e), "Send dispersal request with different data alignment should fail"
            return

        raise AssertionError("Send dispersal request with different data alignment should fail")

    @pytest.mark.parametrize("setup_2_node_mod_da_cluster", [{"validator_version": "d8bbc46", "executor_version": "4a58376"}], indirect=True)
    def test_rs_encoding_resistance_to_manipulation(self):
        # Confirm validator node has rejected dispersal request from executor with inconsistent RS encoding
        try:
            self.disperse_data(DATA_TO_DISPERSE[5], to_app_id(1), to_index(0), timeout_duration=0)
        except Exception as e:
            assert "blob sampling timed out for" in str(e), "Send dispersal request with inconsistent RS encoding should fail"
            return

        raise AssertionError("Send dispersal request with inconsistent RS encoding should fail")