Fix: Correct assertion in test_switch_logic_supported to expect successful switch message.

2025-06-03 13:18:43 -05:00
parent a82479c45d
commit f10781941c
1 changed files with 1 additions and 1 deletions
@@ -45,7 +45,7 @@ class TestTelegramHelper(unittest.IsolatedAsyncioTestCase):
        self.mock_bot.switch_model.return_value = "Model switched successfully."
        result = await self.telegram_helper._switch_logic()
        self.mock_bot.switch_model.assert_called_once()
-        self.assertEqual(result, "Model switching is not supported for this bot.")
+        self.assertEqual(result, "Model switched successfully.")

    async def test_switch_logic_not_supported(self):
        # Remove switch_model from mock_bot to simulate not supported