From 3e067ebe8414eca57e23a7d2ee92b367fc25719b Mon Sep 17 00:00:00 2001 From: boffman <686610+boffman@users.noreply.github.com> Date: Tue, 19 Mar 2024 16:02:39 +0100 Subject: [PATCH] Execute python script runs at module level (#308) --- grizzly/steps/scenario/setup.py | 2 +- tests/unit/test_grizzly/steps/test_setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/grizzly/steps/scenario/setup.py b/grizzly/steps/scenario/setup.py index cf44f859..2719dd85 100644 --- a/grizzly/steps/scenario/setup.py +++ b/grizzly/steps/scenario/setup.py @@ -264,7 +264,7 @@ def _execute_python_script(context: Context, source: str) -> None: if on_worker(context): return - exec(source) # noqa: S102 + exec(source, globals(), globals()) # noqa: S102 @then('execute python script "{script_path}"') def step_setup_execute_python_script(context: Context, script_path: str) -> None: diff --git a/tests/unit/test_grizzly/steps/test_setup.py b/tests/unit/test_grizzly/steps/test_setup.py index 3f50bf5d..e5b99f7c 100644 --- a/tests/unit/test_grizzly/steps/test_setup.py +++ b/tests/unit/test_grizzly/steps/test_setup.py @@ -278,5 +278,5 @@ def test__execute_python_script(behave_fixture: BehaveFixture, mocker: MockerFix _execute_python_script(context, "print('foobar')") on_worker_mock.assert_called_once_with(context) - exec_mock.assert_called_once_with("print('foobar')") + exec_mock.assert_called_once_with("print('foobar')", ANY(dict), ANY(dict))