You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.9 KiB
59 lines
1.9 KiB
1 year ago
|
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()
|