Fix local test test_get_remote (#2706)

Signed-off-by: Mecoli1219 <michaellai901026@gmail.com>
flyteorg · Aug 23, 2024 · 0f42872 · 0f42872
1 parent 94148df
commit 0f42872
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
@@ -48,8 +48,10 @@ def reset_flytectl_config_env_var() -> pytest.fixture():
     return os.environ[FLYTECTL_CONFIG_ENV_VAR]
 
 
+@mock.patch("flytekit.configuration.plugin.get_config_file")
 @mock.patch("flytekit.configuration.plugin.FlyteRemote")
-def test_get_remote(mock_remote, reset_flytectl_config_env_var):
+def test_get_remote(mock_remote, mock_config_file, reset_flytectl_config_env_var):
+    mock_config_file.return_value = None
     r = FlytekitPlugin.get_remote(None, "p", "d")
     assert r is not None
     mock_remote.assert_called_once_with(