Spaces:
Paused
Paused
File size: 2,173 Bytes
99a7ebb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | import json
import tempfile
import unittest
from pathlib import Path
ROOT_DIR = Path(__file__).resolve().parents[1]
ROOT_CONFIG_FILE = ROOT_DIR / "config.json"
class ConfigLoadingTests(unittest.TestCase):
@classmethod
def setUpClass(cls) -> None:
cls._created_root_config = False
if not ROOT_CONFIG_FILE.exists():
ROOT_CONFIG_FILE.write_text(json.dumps({"auth-key": "test-auth"}), encoding="utf-8")
cls._created_root_config = True
from services import config as config_module
cls.config_module = config_module
@classmethod
def tearDownClass(cls) -> None:
if cls._created_root_config and ROOT_CONFIG_FILE.exists():
ROOT_CONFIG_FILE.unlink()
def test_load_settings_ignores_directory_config_path(self) -> None:
with tempfile.TemporaryDirectory() as tmp_dir:
base_dir = Path(tmp_dir)
data_dir = base_dir / "data"
config_dir = base_dir / "config.json"
os_auth_key = "env-auth"
config_dir.mkdir()
module = self.config_module
old_base_dir = module.BASE_DIR
old_data_dir = module.DATA_DIR
old_config_file = module.CONFIG_FILE
old_env_auth_key = module.os.environ.get("CHATGPT2API_AUTH_KEY")
try:
module.BASE_DIR = base_dir
module.DATA_DIR = data_dir
module.CONFIG_FILE = config_dir
module.os.environ["CHATGPT2API_AUTH_KEY"] = os_auth_key
settings = module._load_settings()
self.assertEqual(settings.auth_key, os_auth_key)
self.assertEqual(settings.refresh_account_interval_minute, 5)
finally:
module.BASE_DIR = old_base_dir
module.DATA_DIR = old_data_dir
module.CONFIG_FILE = old_config_file
if old_env_auth_key is None:
module.os.environ.pop("CHATGPT2API_AUTH_KEY", None)
else:
module.os.environ["CHATGPT2API_AUTH_KEY"] = old_env_auth_key
if __name__ == "__main__":
unittest.main()
|