74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
 
|
"""Tests for AI-outfit config helpers in config_service (no GUI)."""
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import sys
|
|||
|
|
import tempfile
|
|||
|
|
import unittest
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
|
|||
|
|
|
|||
|
|
import services.config_service as cs
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TestOutfitConfigHelpers(unittest.TestCase):
|
|||
|
|
def setUp(self):
|
|||
|
|
self.tmp = Path(tempfile.mkdtemp())
|
|||
|
|
self._env = os.environ.get("CMBOT_DATA_DIR")
|
|||
|
|
os.environ["CMBOT_DATA_DIR"] = str(self.tmp)
|
|||
|
|
self.config_dir = self.tmp / "config"
|
|||
|
|
self.config_dir.mkdir(parents=True, exist_ok=True)
|
|||
|
|
|
|||
|
|
def tearDown(self):
|
|||
|
|
if self._env is None:
|
|||
|
|
os.environ.pop("CMBOT_DATA_DIR", None)
|
|||
|
|
else:
|
|||
|
|
os.environ["CMBOT_DATA_DIR"] = self._env
|
|||
|
|
shutil.rmtree(str(self.tmp), ignore_errors=True)
|
|||
|
|
|
|||
|
|
# -- ai_models.json -------------------------------------------------
|
|||
|
|
|
|||
|
|
def test_load_ai_models_missing_returns_empty(self):
|
|||
|
|
self.assertEqual(cs.load_ai_models(), [])
|
|||
|
|
|
|||
|
|
def test_load_ai_models_object_with_models_list(self):
|
|||
|
|
(self.config_dir / "ai_models.json").write_text(
|
|||
|
|
json.dumps({"models": [{"name": "m1", "url": "https://x"}]}),
|
|||
|
|
encoding="utf-8")
|
|||
|
|
models = cs.load_ai_models()
|
|||
|
|
self.assertEqual(len(models), 1)
|
|||
|
|
self.assertEqual(models[0]["name"], "m1")
|
|||
|
|
|
|||
|
|
def test_load_ai_models_bare_list(self):
|
|||
|
|
(self.config_dir / "ai_models.json").write_text(
|
|||
|
|
json.dumps([{"name": "a"}, {"name": "b"}]), encoding="utf-8")
|
|||
|
|
self.assertEqual(len(cs.load_ai_models()), 2)
|
|||
|
|
|
|||
|
|
def test_load_ai_models_corrupt_returns_empty(self):
|
|||
|
|
(self.config_dir / "ai_models.json").write_text("{ not json", encoding="utf-8")
|
|||
|
|
self.assertEqual(cs.load_ai_models(), [])
|
|||
|
|
|
|||
|
|
def test_load_ai_models_tolerates_bom(self):
|
|||
|
|
(self.config_dir / "ai_models.json").write_bytes(
|
|||
|
|
"".encode("utf-8") + json.dumps([{"name": "z"}]).encode("utf-8"))
|
|||
|
|
self.assertEqual(cs.load_ai_models()[0]["name"], "z")
|
|||
|
|
|
|||
|
|
# -- outfit_prompt.txt ----------------------------------------------
|
|||
|
|
|
|||
|
|
def test_prompt_default_when_missing(self):
|
|||
|
|
self.assertEqual(cs.load_outfit_prompt(), cs.DEFAULT_OUTFIT_PROMPT)
|
|||
|
|
|
|||
|
|
def test_prompt_save_then_load_roundtrip(self):
|
|||
|
|
cs.save_outfit_prompt("hello {title} {product_id}")
|
|||
|
|
self.assertEqual(cs.load_outfit_prompt(), "hello {title} {product_id}")
|
|||
|
|
|
|||
|
|
def test_prompt_saved_without_bom(self):
|
|||
|
|
cs.save_outfit_prompt("abc")
|
|||
|
|
raw = (self.config_dir / "outfit_prompt.txt").read_bytes()
|
|||
|
|
self.assertFalse(raw.startswith(b"\xef\xbb\xbf"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
unittest.main()
|