Eliminate unnecessary lazy import

This commit is contained in:
Meng Xiangzhuo
2024-10-04 11:22:47 +08:00
parent 0fe772efa8
commit 903614140a
3 changed files with 20 additions and 9 deletions
+12 -4
View File
@@ -90,13 +90,17 @@ def test_ask_user_overwrite(mocker):
"""
Once https://github.com/tmbo/questionary/issues/35 is implemented, improve this test.
"""
prompt_mock = mocker.patch("questionary.prompt", return_value={"overwrite": False})
prompt_mock = mocker.patch(
"freqtrade.commands.build_config_commands.prompt", return_value={"overwrite": False}
)
assert not ask_user_overwrite(Path("test.json"))
assert prompt_mock.call_count == 1
prompt_mock.reset_mock()
prompt_mock = mocker.patch("questionary.prompt", return_value={"overwrite": True})
prompt_mock = mocker.patch(
"freqtrade.commands.build_config_commands.prompt", return_value={"overwrite": False}
)
assert ask_user_overwrite(Path("test.json"))
assert prompt_mock.call_count == 1
@@ -105,12 +109,16 @@ def test_ask_user_config(mocker):
"""
Once https://github.com/tmbo/questionary/issues/35 is implemented, improve this test.
"""
prompt_mock = mocker.patch("questionary.prompt", return_value={"overwrite": False})
prompt_mock = mocker.patch(
"freqtrade.commands.build_config_commands.prompt", return_value={"overwrite": False}
)
answers = ask_user_config()
assert isinstance(answers, dict)
assert prompt_mock.call_count == 1
prompt_mock = mocker.patch("questionary.prompt", return_value={})
prompt_mock = mocker.patch(
"freqtrade.commands.build_config_commands.prompt", return_value={"overwrite": False}
)
with pytest.raises(OperationalException, match=r"User interrupted interactive questions\."):
ask_user_config()