55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import unittest
|
|
import json
|
|
from unittest.mock import Mock, PropertyMock, MagicMock, AsyncMock, patch, mock_open
|
|
from fjerkroa_bot import FjerkroaBot
|
|
from discord import User, Message, TextChannel
|
|
|
|
|
|
class TestBotBase(unittest.IsolatedAsyncioTestCase):
|
|
|
|
async def asyncSetUp(self):
|
|
self.mock_response = Mock()
|
|
self.mock_response.choices = [
|
|
Mock(text="Nice day today!")
|
|
]
|
|
self.config_data = {
|
|
"openai_key": "OPENAIKEY",
|
|
"engine": "gpt-4",
|
|
"max_tokens": 1024,
|
|
"n": 1,
|
|
"temperature": 0.9,
|
|
}
|
|
self.history_data = []
|
|
with patch.object(FjerkroaBot, 'load_config', new=lambda s, c: self.config_data), \
|
|
patch.object(FjerkroaBot, 'user', new_callable=PropertyMock) as mock_user:
|
|
mock_user.return_value = MagicMock(spec=User)
|
|
mock_user.return_value.id = 12
|
|
self.bot = FjerkroaBot('config.json')
|
|
|
|
def create_message(self, message: str) -> Message:
|
|
message = MagicMock(spec=Message)
|
|
message.content = "Hello, how are you?"
|
|
message.author = AsyncMock(spec=User)
|
|
message.author.id = 123
|
|
message.author.bot = False
|
|
message.channel = AsyncMock(spec=TextChannel)
|
|
message.channel.send = AsyncMock()
|
|
return message
|
|
|
|
|
|
class TestFunctionality(TestBotBase):
|
|
|
|
def test_load_config(self) -> None:
|
|
with patch('builtins.open', mock_open(read_data=json.dumps(self.config_data))):
|
|
result = FjerkroaBot.load_config('config.json')
|
|
self.assertEqual(result, self.config_data)
|
|
|
|
async def test_on_message_event(self) -> None:
|
|
message = self.create_message("Hello there! How are you?")
|
|
await self.bot.on_message(message)
|
|
message.channel.send.assert_called_once_with("Hello!")
|
|
|
|
|
|
if __name__ == "__mait__":
|
|
unittest.main()
|