from eth2spec.test.helpers.constants import PHASE0, ALTAIR, BELLATRIX, CAPELLA, DENEB, ELECTRA
from eth2spec.gen_helpers.gen_from_tests.gen import run_state_test_generators, combine_mods


if __name__ == "__main__":
    phase_0_mods = {key: 'eth2spec.test.phase0.sanity.test_' + key for key in [
        'blocks',
        'slots',
    ]}

    _new_altair_mods = {key: 'eth2spec.test.altair.sanity.test_' + key for key in [
        'blocks',
    ]}
    altair_mods = combine_mods(_new_altair_mods, phase_0_mods)

    _new_bellatrix_mods = {key: 'eth2spec.test.bellatrix.sanity.test_' + key for key in [
        'blocks',
    ]}
    bellatrix_mods = combine_mods(_new_bellatrix_mods, altair_mods)

    _new_capella_mods = {key: 'eth2spec.test.capella.sanity.test_' + key for key in [
        'blocks',
    ]}
    capella_mods = combine_mods(_new_capella_mods, bellatrix_mods)

    _new_deneb_mods = {key: 'eth2spec.test.deneb.sanity.test_' + key for key in [
        'blocks',
    ]}
    deneb_mods = combine_mods(_new_deneb_mods, capella_mods)

    # This is a "hack" which allows other test files (e.g., test_deposit_transition.py)
    # to reuse the sanity/block test format. If a new test file is added or removed,
    # do not forget to update sanity/block/__init__.py accordingly.
    _new_electra_mods = {key: 'eth2spec.test.electra.sanity.' + key for key in [
        'blocks',
    ]}
    electra_mods = combine_mods(_new_electra_mods, deneb_mods)

    all_mods = {
        PHASE0: phase_0_mods,
        ALTAIR: altair_mods,
        BELLATRIX: bellatrix_mods,
        CAPELLA: capella_mods,
        DENEB: deneb_mods,
        ELECTRA: electra_mods,
    }

    run_state_test_generators(runner_name="sanity", all_mods=all_mods)