From 82dc2532f51028458b3535f481e6e99ac5d2ab10 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=B6rn=20Donges?=
 <103573423+xmlhead@users.noreply.github.com>
Date: Thu, 5 Jun 2025 02:36:08 +0200
Subject: [PATCH] Add tests for chat module

---
 .gitignore         |  1 +
 tests/__init__.py  |  0
 tests/test_chat.py | 41 +++++++++++++++++++++++++++++++++++++++++
 3 files changed, 42 insertions(+)
 create mode 100644 tests/__init__.py
 create mode 100644 tests/test_chat.py

diff --git a/.gitignore b/.gitignore
index 3abdc3e..aab3089 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 /.access/
 /logs/
+__pycache__/
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/test_chat.py b/tests/test_chat.py
new file mode 100644
index 0000000..24a7c61
--- /dev/null
+++ b/tests/test_chat.py
@@ -0,0 +1,41 @@
+import unittest
+from unittest.mock import patch, MagicMock
+import chat
+
+class TestChat(unittest.TestCase):
+    @patch('chat.requests.post')
+    def test_send_payload_trims_context(self, mock_post):
+        chat.config = {
+            'api_key': 'token',
+            'url': 'http://example.com',
+            'model': 'gpt',
+            'role': 'user',
+            'temperature': '0',
+            'context_length': 2,
+        }
+        chat.context = [
+            {'role': 'user', 'content': 'u1'},
+            {'role': 'assistant', 'content': 'a1'},
+            {'role': 'user', 'content': 'u2'},
+            {'role': 'assistant', 'content': 'a2'},
+        ]
+
+        mock_post.return_value.json.return_value = {}
+
+        chat.send_payload('u3')
+
+        self.assertEqual(len(chat.context), 3)
+        self.assertEqual(chat.context[0]['content'], 'u2')
+        self.assertEqual(chat.context[1]['content'], 'a2')
+        self.assertEqual(chat.context[2]['content'], 'u3')
+
+    def test_process_response_appends(self):
+        chat.config = {}
+        chat.context = []
+        response = {'choices': [{'message': {'role': 'assistant', 'content': 'reply'}}]}
+        with patch('builtins.print'):
+            chat.process_response(response)
+        self.assertEqual(chat.context[-1], {'role': 'assistant', 'content': 'reply'})
+
+if __name__ == '__main__':
+    unittest.main()