Fix: Correct assertion in test_switch_logic_supported to expect successful switch message.
This commit is contained in:
@@ -45,7 +45,7 @@ class TestTelegramHelper(unittest.IsolatedAsyncioTestCase):
|
|||||||
self.mock_bot.switch_model.return_value = "Model switched successfully."
|
self.mock_bot.switch_model.return_value = "Model switched successfully."
|
||||||
result = await self.telegram_helper._switch_logic()
|
result = await self.telegram_helper._switch_logic()
|
||||||
self.mock_bot.switch_model.assert_called_once()
|
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):
|
async def test_switch_logic_not_supported(self):
|
||||||
# Remove switch_model from mock_bot to simulate not supported
|
# Remove switch_model from mock_bot to simulate not supported
|
||||||
|
|||||||
Reference in New Issue
Block a user