Merge pull request #130 from bucolucas/enhance-unit-tests-issue-118-next-steps

Complete next steps for enhancing unit tests

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()

tests/claude/test_chatgpt_telegram_inference_bot.py

@@ -0,0 +1,38 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from chatgpt_telegram_inference_bot import ChatGPTTelegramInferenceBot
+
+class TestChatGPTTelegramInferenceBot(unittest.TestCase):
+    def setUp(self):
+        self.bot = ChatGPTTelegramInferenceBot()
+    
+    @patch('chatgpt_telegram_inference_bot.OpenAI')
+    def test_get_chat_response(self, MockOpenAI):
+        mock_ai = MockOpenAI.return_value
+        mock_ai.chat.completions.create.return_value = MagicMock()
+        
+        messages = [{"role": "user", "content": "Hello"}]
+        response = self.bot.get_chat_response(messages)
+        
+        self.assertIsNotNone(response)
+    
+    @patch('chatgpt_telegram_inference_bot.OpenAI')
+    def test_handle_message(self, MockOpenAI):
+        mock_ai = MockOpenAI.return_value
+        mock_ai.chat.completions.create.return_value = MagicMock(choices=[MagicMock(message={"content": "response content"}, finish_reason='stop')])
+        
+        user_id = "user123"
+        user_message = "Hello"
+        response = self.bot.handle_message(user_id, user_message)
+        
+        self.assertIsNotNone(response)
+
+    def test_switch_model(self):
+        initial_model = self.bot.model
+        self.bot.switch_model()
+        self.assertNotEqual(initial_model, self.bot.model)
+    
+    # Additional testing for error cases and edge cases
+
+if __name__ == '__main__':
+    unittest.main()