fix: update test_get_command_completions to handle matching_commands return value

This commit is contained in:
Paul Gauthier (aider) 2024-08-28 16:01:10 -07:00
parent 6687a24547
commit d2bc24cb18

View file

@ -169,6 +169,7 @@ class TestInputOutput(unittest.TestCase):
commands = MagicMock() commands = MagicMock()
commands.get_commands.return_value = ["model", "chat", "help"] commands.get_commands.return_value = ["model", "chat", "help"]
commands.get_completions.return_value = ["gpt-3.5-turbo", "gpt-4"] commands.get_completions.return_value = ["gpt-3.5-turbo", "gpt-4"]
commands.matching_commands.return_value = (["model"], None, None)
autocompleter = AutoCompleter(root, rel_fnames, addable_rel_fnames, commands, "utf-8") autocompleter = AutoCompleter(root, rel_fnames, addable_rel_fnames, commands, "utf-8")
@ -176,6 +177,7 @@ class TestInputOutput(unittest.TestCase):
result = autocompleter.get_command_completions("/model gpt", ["/model", "gpt"]) result = autocompleter.get_command_completions("/model gpt", ["/model", "gpt"])
self.assertEqual(result, ["gpt-3.5-turbo", "gpt-4"]) self.assertEqual(result, ["gpt-3.5-turbo", "gpt-4"])
commands.get_completions.assert_called_once_with("model") commands.get_completions.assert_called_once_with("model")
commands.matching_commands.assert_called_once_with("/model")
if __name__ == "__main__": if __name__ == "__main__":