chendelian
abe02f26cf
All checks were successful
Publish Python Package to Gitea / publish (push) Successful in 17s
126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""e2e_helpers 轻量单元测试（不依赖匠厂桌面 / IPC）。"""
|
||
from __future__ import annotations
|
||
|
||
import tempfile
|
||
import unittest
|
||
from pathlib import Path
|
||
|
||
from jiangchang_desktop_sdk.e2e_helpers import (
|
||
SkillDataDir,
|
||
_iter_text_and_newlines,
|
||
_is_placeholder_value,
|
||
_parse_env_file,
|
||
assert_skill_env_file,
|
||
)
|
||
|
||
|
||
def _segments(prompt: str) -> list[str | None]:
|
||
out: list[str | None] = []
|
||
for seg in _iter_text_and_newlines(prompt):
|
||
if seg is not None and not isinstance(seg, str):
|
||
out.append(None)
|
||
else:
|
||
out.append(seg) # type: ignore[arg-type]
|
||
return out
|
||
|
||
|
||
class TestIterTextAndNewlines(unittest.TestCase):
|
||
def test_lf(self) -> None:
|
||
self.assertEqual(_segments("a\nb"), ["a", None, "b"])
|
||
|
||
def test_crlf(self) -> None:
|
||
self.assertEqual(_segments("a\r\nb"), ["a", None, "b"])
|
||
|
||
def test_cr(self) -> None:
|
||
self.assertEqual(_segments("a\rb"), ["a", None, "b"])
|
||
|
||
def test_multiple_newlines(self) -> None:
|
||
self.assertEqual(_segments("x\n\ny"), ["x", None, None, "y"])
|
||
|
||
|
||
class TestParseEnvFile(unittest.TestCase):
|
||
def _write(self, path: Path, content: str) -> None:
|
||
path.write_text(content, encoding="utf-8")
|
||
|
||
def test_basic_and_comments(self) -> None:
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
env_path = Path(tmp) / ".env"
|
||
self._write(
|
||
env_path,
|
||
"# comment\n\nAPI_KEY=real-secret\n# KEY=ignored\n",
|
||
)
|
||
parsed = _parse_env_file(env_path)
|
||
self.assertEqual(parsed["API_KEY"], "real-secret")
|
||
self.assertNotIn("KEY", parsed)
|
||
|
||
def test_quoted_values(self) -> None:
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
env_path = Path(tmp) / ".env"
|
||
self._write(env_path, 'A="double"\nB=\'single\'\n')
|
||
parsed = _parse_env_file(env_path)
|
||
self.assertEqual(parsed["A"], "double")
|
||
self.assertEqual(parsed["B"], "single")
|
||
|
||
def test_placeholder_detection(self) -> None:
|
||
self.assertTrue(_is_placeholder_value("TODO", None))
|
||
self.assertTrue(_is_placeholder_value("sk-xxx", None))
|
||
self.assertTrue(_is_placeholder_value(" Changeme ", None))
|
||
self.assertFalse(_is_placeholder_value("sk-live-abc", None))
|
||
|
||
|
||
class TestAssertSkillEnvFile(unittest.TestCase):
|
||
def test_missing_file(self) -> None:
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
ctx = SkillDataDir(
|
||
root=Path(tmp),
|
||
path=Path(tmp) / "user" / "demo-skill",
|
||
user_id="12428",
|
||
skill_slug="demo-skill",
|
||
)
|
||
with self.assertRaises(AssertionError) as cm:
|
||
assert_skill_env_file(ctx, ["API_KEY"])
|
||
self.assertIn("config/.env", str(cm.exception))
|
||
|
||
def test_missing_and_placeholder_keys(self) -> None:
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
skill_dir = Path(tmp) / "12428" / "demo-skill"
|
||
env_dir = skill_dir / "config"
|
||
env_dir.mkdir(parents=True)
|
||
(env_dir / ".env").write_text(
|
||
"API_KEY=your-key\nOTHER=real\n",
|
||
encoding="utf-8",
|
||
)
|
||
ctx = SkillDataDir(
|
||
root=Path(tmp),
|
||
path=skill_dir,
|
||
user_id="12428",
|
||
skill_slug="demo-skill",
|
||
)
|
||
with self.assertRaises(AssertionError) as cm:
|
||
assert_skill_env_file(ctx, ["API_KEY", "MISSING", "OTHER"])
|
||
msg = str(cm.exception)
|
||
self.assertIn("API_KEY", msg)
|
||
self.assertIn("MISSING", msg)
|
||
self.assertNotIn("OTHER", msg)
|
||
|
||
def test_pass_returns_path(self) -> None:
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
skill_dir = Path(tmp) / "1" / "demo-skill"
|
||
env_dir = skill_dir / "config"
|
||
env_dir.mkdir(parents=True)
|
||
env_file = env_dir / ".env"
|
||
env_file.write_text("API_KEY=sk-live-ok\n", encoding="utf-8")
|
||
ctx = SkillDataDir(
|
||
root=Path(tmp),
|
||
path=skill_dir,
|
||
user_id="1",
|
||
skill_slug="demo-skill",
|
||
)
|
||
got = assert_skill_env_file(ctx, ["API_KEY"])
|
||
self.assertEqual(got, env_file)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|