Coder-Vippro/ChatGPT-Discord-Bot
1
0
Fork 0
Code Issues Pull Requests 3 Actions Packages Projects Releases 13 Wiki Activity

refactor: Update prepare_messages_for_api tests to reflect changes in message handling

Browse Source
This commit is contained in:
cauvang32
2025-08-28 15:31:45 +07:00
parent 7785a8f26e
commit 1cb47f1d25
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_bot.py
View File

@@ -216,31 +216,31 @@ class TestOpenAIUtils(unittest.TestCase):
self.assertEqual(untrimmed, short_text) self.assertEqual(untrimmed, short_text)
def test_prepare_messages_for_api(self): def test_prepare_messages_for_api(self):
# Test empty messages # Test empty messages - should return empty list (no system message added)
empty_result = prepare_messages_for_api([]) empty_result = prepare_messages_for_api([])
self.assertEqual(len(empty_result), 1) # Should have system message self.assertEqual(len(empty_result), 0) # Should be empty, no system message added
self.assertEqual(empty_result[0]["role"], "system") # Verify it's a system message
# Test regular messages # Test regular messages - should return messages as-is (no system message added)
messages = [ messages = [
{"role": "user", "content": "Hello"}, {"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi there!"}, {"role": "assistant", "content": "Hi there!"},
{"role": "user", "content": "How are you?"} {"role": "user", "content": "How are you?"}
] ]
result = prepare_messages_for_api(messages) result = prepare_messages_for_api(messages)
self.assertEqual(len(result), 4) # Should have system message + 3 original messages self.assertEqual(len(result), 3) # Should have 3 original messages only
self.assertEqual(result[0]["role"], "user")
self.assertEqual(result[0]["content"], "Hello")
# Test with null content # Test with null content - should filter out null content messages
messages_with_null = [ messages_with_null = [
{"role": "user", "content": None}, {"role": "user", "content": None},
{"role": "assistant", "content": "Response"} {"role": "assistant", "content": "Response"}
] ]
result_fixed = prepare_messages_for_api(messages_with_null) result_fixed = prepare_messages_for_api(messages_with_null)
self.assertEqual(len(result_fixed), 2) # Should have system message + 1 valid message self.assertEqual(len(result_fixed), 1) # Should have only 1 valid message (null filtered out)
# Verify the content is correct (system message + only the assistant message) # Verify the content is correct (only the assistant message)
self.assertEqual(result_fixed[0]["role"], "system") self.assertEqual(result_fixed[0]["role"], "assistant")
self.assertEqual(result_fixed[1]["role"], "assistant") self.assertEqual(result_fixed[0]["content"], "Response")
self.assertEqual(result_fixed[1]["content"], "Response")
class TestCodeUtils(unittest.TestCase): class TestCodeUtils(unittest.TestCase):
"""Test code utility functions""" """Test code utility functions"""