Fix local test test_saving_remote (#2707)

Signed-off-by: Mecoli1219 <[email protected]>
flyteorg · Aug 23, 2024 · a50eb4b · a50eb4b
1 parent 0f42872
commit a50eb4b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/flytekit/unit/cli/pyflyte/test_register.py b/tests/flytekit/unit/cli/pyflyte/test_register.py
@@ -59,8 +59,10 @@ def test_get_remote(mock_remote, mock_config_file, reset_flytectl_config_env_var
     )
 
 
+@mock.patch("flytekit.configuration.plugin.get_config_file")
 @mock.patch("flytekit.configuration.plugin.FlyteRemote")
-def test_saving_remote(mock_remote):
+def test_saving_remote(mock_remote, mock_config_file):
+    mock_config_file.return_value = None
     mock_context = mock.MagicMock
     mock_context.obj = {}
     get_and_save_remote_with_click_context(mock_context, "p", "d")