From 8471b8efe57110485c231284a4a0fa1a2b44600e Mon Sep 17 00:00:00 2001
From: Zack <zack@zackbradshaw.com>
Date: Fri, 20 Oct 2023 17:21:14 -0500
Subject: [PATCH] feat: Add test for bingchat.py

---
 tests/models/bingchat.py | 58 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 tests/models/bingchat.py

diff --git a/tests/models/bingchat.py b/tests/models/bingchat.py
new file mode 100644
index 00000000..4e6a8271
--- /dev/null
+++ b/tests/models/bingchat.py
@@ -0,0 +1,58 @@
+import unittest
+import json
+import os
+
+# Assuming the BingChat class is in a file named "bing_chat.py"
+from bing_chat import BingChat, ConversationStyle
+
+class TestBingChat(unittest.TestCase):
+
+    def setUp(self):
+        # Path to a mock cookies file for testing
+        self.mock_cookies_path = "./mock_cookies.json"
+        with open(self.mock_cookies_path, 'w') as file:
+            json.dump({"mock_cookie": "mock_value"}, file)
+
+        self.chat = BingChat(cookies_path=self.mock_cookies_path)
+
+    def tearDown(self):
+        os.remove(self.mock_cookies_path)
+
+    def test_init(self):
+        self.assertIsInstance(self.chat, BingChat)
+        self.assertIsNotNone(self.chat.bot)
+
+    def test_call(self):
+        # Mocking the asynchronous behavior for the purpose of the test
+        self.chat.bot.ask = lambda *args, **kwargs: {"text": "Hello, Test!"}
+        response = self.chat("Test prompt")
+        self.assertEqual(response, "Hello, Test!")
+
+    def test_create_img(self):
+        # Mocking the ImageGen behavior for the purpose of the test
+        class MockImageGen:
+            def __init__(self, *args, **kwargs):
+                pass
+            
+            def get_images(self, *args, **kwargs):
+                return [{"path": "mock_image.png"}]
+            
+            @staticmethod
+            def save_images(*args, **kwargs):
+                pass
+
+        original_image_gen = BingChat.ImageGen
+        BingChat.ImageGen = MockImageGen
+
+        img_path = self.chat.create_img("Test prompt", auth_cookie="mock_auth_cookie")
+        self.assertEqual(img_path, "./output/mock_image.png")
+
+        BingChat.ImageGen = original_image_gen
+
+    def test_set_cookie_dir_path(self):
+        test_path = "./test_path"
+        BingChat.set_cookie_dir_path(test_path)
+        self.assertEqual(BingChat.Cookie.dir_path, test_path)
+
+if __name__ == "__main__":
+    unittest.main()