fix: add marker for tests requiring node with modifications

This commit is contained in:
Roman 2025-06-20 16:44:51 +08:00
parent 0e78516607
commit ddb5d22d8d
No known key found for this signature in database
GPG Key ID: 583BDF43C238B83E
2 changed files with 16 additions and 0 deletions

View File

@ -17,6 +17,21 @@ from src.data_storage import DS
logger = get_custom_logger(__name__)
def pytest_addoption(parser):
parser.addoption("--run-with-mod-da-node", action="store_true", default=False, help="Run tests requiring nodes with modified da layer")
def pytest_configure(config):
config.addinivalue_line("markers", "mod_da_node: Mark test as requiring --run-with-mod-da-node")
def pytest_collection_modifyitems(config, items):
run_mod_da_node = config.getoption("--run-with-mod-da-node")
for item in items:
if "mod_da_node" in item.keywords and not run_mod_da_node:
item.add_marker(pytest.mark.skip(reason="Requires --run-with-mod-da-node option"))
# See https://docs.pytest.org/en/latest/example/simple.html#making-test-result-information-available-in-fixtures
@pytest.hookimpl(hookwrapper=True, tryfirst=True)
def pytest_runtest_makereport(item):

View File

@ -6,6 +6,7 @@ from src.steps.da import StepsDataAvailability
from src.test_data import DATA_TO_DISPERSE
@pytest.mark.mod_da_node
class TestDispersalResilience(StepsDataAvailability):
main_nodes = []