From 2d41e39f897a061516f1fca7942763df39fdaa0d Mon Sep 17 00:00:00 2001 From: CadeYu Date: Sat, 21 Mar 2026 13:38:42 +0800 Subject: [PATCH] feat(cli): allow custom OpenRouter model IDs --- tests/test_openrouter_model_selection.py | 25 ++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/test_openrouter_model_selection.py b/tests/test_openrouter_model_selection.py index 5b832e81..dba5b131 100644 --- a/tests/test_openrouter_model_selection.py +++ b/tests/test_openrouter_model_selection.py @@ -19,6 +19,31 @@ class OpenRouterModelSelectionTests(unittest.TestCase): ) self.assertEqual(chosen, "minimax/minimax-m2.1") + def test_exit_on_no_choice(self): + with self.assertRaises(SystemExit) as cm: + resolve_model_choice("openrouter", None, "Quick-Thinking") + self.assertEqual(cm.exception.code, 1) + + def test_exit_on_empty_custom_model_input(self): + with self.assertRaises(SystemExit) as cm: + resolve_model_choice( + "openrouter", + CUSTOM_OPENROUTER_MODEL, + "Deep-Thinking", + prompt_fn=lambda _: " ", + ) + self.assertEqual(cm.exception.code, 1) + + def test_exit_on_none_custom_model_input(self): + with self.assertRaises(SystemExit) as cm: + resolve_model_choice( + "openrouter", + CUSTOM_OPENROUTER_MODEL, + "Deep-Thinking", + prompt_fn=lambda _: None, + ) + self.assertEqual(cm.exception.code, 1) + if __name__ == "__main__": unittest.main()