status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Avoid modifying caller `phases` (`kw` is shallow copy)

This commit is contained in:
Etan Kissling 2022-12-11 23:41:08 +01:00
parent 0649e0662c
commit 0c3853e959
No known key found for this signature in database
GPG Key ID: B21DA824C5A3D03D
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/core/pyspec/eth2spec/test/context.py
View File

@ -604,20 +604,26 @@ def with_config_overrides(config_overrides, emitted_fork=None, emit=True):
"""
def decorator(fn):
def wrapper(*args, spec: Spec, **kw):
# Apply config overrides to spec
spec, output_config = spec_with_config_overrides(_get_copy_of_spec(spec), config_overrides)
if emit and emitted_fork is None:
yield 'config', 'cfg', output_config
# Apply config overrides to additional phases, if present
if 'phases' in kw:
phases = {}
for fork in kw['phases']:
if is_post_fork(fork, spec.fork):
kw['phases'][fork], output_config = \
spec_with_config_overrides(_get_copy_of_spec(kw['phases'][fork]), config_overrides)
if emit and emitted_fork == fork:
yield 'config', 'cfg', output_config
phases[fork], output = \
spec_with_config_overrides(_get_copy_of_spec(kw['phases'][fork]), config_overrides)
if emitted_fork == fork:
output_config = output
kw['phases'] = phases
# Run the function
out = fn(*args, spec=spec, **kw)
# Emit requested spec (with overrides)
if emit:
yield 'config', 'cfg', output_config
# If it's not returning None like a normal test function,
# it's generating things, and we need to complete it before setting back the config.
if out is not None: