From d69dd5cee577fd78812306713ac80a619577bf58 Mon Sep 17 00:00:00 2001 From: Artem Petrov <58334441+wckdman@users.noreply.github.com> Date: Thu, 2 Nov 2023 11:33:18 +0100 Subject: [PATCH] fix test Signed-off-by: Artem Petrov <58334441+wckdman@users.noreply.github.com> --- tests/flytekit/unit/cli/pyflyte/test_register.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/flytekit/unit/cli/pyflyte/test_register.py b/tests/flytekit/unit/cli/pyflyte/test_register.py index eca2be3711..7a6bc68688 100644 --- a/tests/flytekit/unit/cli/pyflyte/test_register.py +++ b/tests/flytekit/unit/cli/pyflyte/test_register.py @@ -175,7 +175,9 @@ def test_non_fast_register_require_version(mock_client, mock_remote): @mock.patch("flytekit.clients.friendly.SynchronousFlyteClient", spec=SynchronousFlyteClient) @pytest.mark.parametrize("format", ["json", "yaml"]) def test_register_result_output(mock_client, mock_remote, format, expected_result_file_content): + ctx = FlyteContextManager.current_context() mock_remote._client = mock_client + mock_remote.return_value.context = ctx mock_remote.return_value._version_from_hash.return_value = "dummy_version_from_hash" mock_remote.return_value.fast_package.return_value = "dummy_md5_bytes", "dummy_native_url" runner = CliRunner()