Merge pull request #2688 from ethereum/fork-cli-fix

This commit is contained in:
Hsiao-Wei Wang 2021-10-21 23:30:45 +08:00 committed by GitHub
commit a89b55d7f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 4 deletions

View File

@ -49,6 +49,15 @@ def pytest_addoption(parser):
)
def _validate_fork_name(forks):
for fork in forks:
if fork not in ALL_PHASES:
raise ValueError(
f'The given --fork argument "{fork}" is not an available fork.'
f' The available forks: {ALL_PHASES}'
)
@fixture(autouse=True)
def preset(request):
context.DEFAULT_TEST_PRESET = request.config.getoption("--preset")
@ -56,10 +65,11 @@ def preset(request):
@fixture(autouse=True)
def run_phases(request):
phases = request.config.getoption("--fork")
if phases:
phases = [phase.lower() for phase in phases]
context.DEFAULT_PYTEST_FORKS = set(phases)
forks = request.config.getoption("--fork", default=None)
if forks:
forks = [fork.lower() for fork in forks]
_validate_fork_name(forks)
context.DEFAULT_PYTEST_FORKS = set(forks)
else:
context.DEFAULT_PYTEST_FORKS = ALL_PHASES