configuration: the cache folder can now be specified in the configuration file
This commit is contained in:
@@ -57,6 +57,7 @@ class TestConfig(unittest.TestCase):
|
||||
|
||||
def test_from_dict_should_create_config_from_dict(self) -> None:
|
||||
source_dict = {
|
||||
'cache': '.',
|
||||
'db': './test_db/',
|
||||
'ais': {
|
||||
'myopenai': {
|
||||
@@ -73,6 +74,7 @@ class TestConfig(unittest.TestCase):
|
||||
}
|
||||
}
|
||||
config = Config.from_dict(source_dict)
|
||||
self.assertEqual(config.cache, '.')
|
||||
self.assertEqual(config.db, './test_db/')
|
||||
self.assertEqual(len(config.ais), 1)
|
||||
self.assertEqual(config.ais['myopenai'].name, 'openai')
|
||||
@@ -89,6 +91,7 @@ class TestConfig(unittest.TestCase):
|
||||
|
||||
def test_from_file_should_load_config_from_file(self) -> None:
|
||||
source_dict = {
|
||||
'cache': './test_cache/',
|
||||
'db': './test_db/',
|
||||
'ais': {
|
||||
'default': {
|
||||
@@ -108,6 +111,7 @@ class TestConfig(unittest.TestCase):
|
||||
yaml.dump(source_dict, f)
|
||||
config = Config.from_file(self.test_file.name)
|
||||
self.assertIsInstance(config, Config)
|
||||
self.assertEqual(config.cache, './test_cache/')
|
||||
self.assertEqual(config.db, './test_db/')
|
||||
self.assertEqual(len(config.ais), 1)
|
||||
self.assertIsInstance(config.ais['default'], AIConfig)
|
||||
@@ -115,6 +119,7 @@ class TestConfig(unittest.TestCase):
|
||||
|
||||
def test_to_file_should_save_config_to_file(self) -> None:
|
||||
config = Config(
|
||||
cache='./test_cache/',
|
||||
db='./test_db/',
|
||||
ais={
|
||||
'myopenai': OpenAIConfig(
|
||||
@@ -133,12 +138,14 @@ class TestConfig(unittest.TestCase):
|
||||
config.to_file(Path(self.test_file.name))
|
||||
with open(self.test_file.name, 'r') as f:
|
||||
saved_config = yaml.load(f, Loader=yaml.FullLoader)
|
||||
self.assertEqual(saved_config['cache'], './test_cache/')
|
||||
self.assertEqual(saved_config['db'], './test_db/')
|
||||
self.assertEqual(len(saved_config['ais']), 1)
|
||||
self.assertEqual(saved_config['ais']['myopenai']['system'], 'Custom system')
|
||||
|
||||
def test_from_file_error_unknown_ai(self) -> None:
|
||||
source_dict = {
|
||||
'cache': './test_cache/',
|
||||
'db': './test_db/',
|
||||
'ais': {
|
||||
'default': {
|
||||
|
||||
Reference in New Issue
Block a user