Switch to TOML for config file, fix bugs.
This commit is contained in:
+3
-2
@@ -3,6 +3,7 @@ import unittest
|
||||
import pytest
|
||||
import aiohttp
|
||||
import json
|
||||
import toml
|
||||
import openai
|
||||
import logging
|
||||
from unittest.mock import Mock, PropertyMock, MagicMock, AsyncMock, patch, mock_open, ANY
|
||||
@@ -18,7 +19,7 @@ class TestBotBase(unittest.IsolatedAsyncioTestCase):
|
||||
Mock(text="Nice day today!")
|
||||
]
|
||||
self.config_data = {
|
||||
"openai_token": os.environ['OPENAI_TOKEN'],
|
||||
"openai-token": os.environ['OPENAI_TOKEN'],
|
||||
"model": "gpt-4",
|
||||
"max_tokens": 1024,
|
||||
"temperature": 0.9,
|
||||
@@ -55,7 +56,7 @@ class TestBotBase(unittest.IsolatedAsyncioTestCase):
|
||||
class TestFunctionality(TestBotBase):
|
||||
|
||||
def test_load_config(self) -> None:
|
||||
with patch('builtins.open', mock_open(read_data=json.dumps(self.config_data))):
|
||||
with patch('builtins.open', mock_open(read_data=toml.dumps(self.config_data))):
|
||||
result = FjerkroaBot.load_config('config.json')
|
||||
self.assertEqual(result, self.config_data)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user