diff --git a/tests/claude/test_anthropic_telegram_inference_bot.py b/tests/claude/test_anthropic_telegram_inference_bot.py new file mode 100644 index 0000000..2e34635 --- /dev/null +++ b/tests/claude/test_anthropic_telegram_inference_bot.py @@ -0,0 +1,33 @@ +import unittest +from unittest.mock import patch, MagicMock +from anthropic_telegram_inference_bot import AnthropicTelegramInferenceBot + +class TestAnthropicTelegramInferenceBot(unittest.TestCase): + def setUp(self): + self.bot = AnthropicTelegramInferenceBot() + + @patch('anthropic_telegram_inference_bot.Anthropic') + def test_get_chat_response(self, MockAnthropic): + mock_anthropic = MockAnthropic.return_value + mock_anthropic.messages.create.return_value = MagicMock() + + messages = [{"role": "user", "content": "Hello"}] + response = self.bot.get_chat_response(messages) + + self.assertIsNotNone(response) + + @patch('anthropic_telegram_inference_bot.Anthropic') + def test_handle_message(self, MockAnthropic): + mock_anthropic = MockAnthropic.return_value + mock_anthropic.messages.create.return_value = MagicMock(content=[MagicMock(type="message", text="response content")]) + + user_id = "user123" + user_message = "Hello" + response = self.bot.handle_message(user_id, user_message) + + self.assertIsNotNone(response) + + # Additional testing for error cases and edge cases + +if __name__ == '__main__': + unittest.main()