50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
import os
|
|
import plistlib
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
ROOT = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")
|
|
sys.path.insert(0, os.path.join(ROOT, "files"))
|
|
import claude_agents # noqa: E402
|
|
import launchagent # noqa: E402
|
|
|
|
SCRIPT = os.path.join(ROOT, "files", "launchagent.py")
|
|
|
|
|
|
class Plist(unittest.TestCase):
|
|
def test_label_matches_server(self):
|
|
self.assertEqual(launchagent.LABEL, "com.andrey.claude-agents")
|
|
self.assertEqual(launchagent.LABEL, claude_agents.SERVICE_LABEL)
|
|
|
|
def test_plist_shape(self):
|
|
self.assertEqual(
|
|
launchagent.plist("/opt/homebrew/bin/python3", "/x/claude_agents.py", "/l/panel.log"),
|
|
{"Label": "com.andrey.claude-agents",
|
|
"ProgramArguments": ["/opt/homebrew/bin/python3", "/x/claude_agents.py", "--headless"],
|
|
"RunAtLoad": True, "KeepAlive": True,
|
|
"StandardOutPath": "/l/panel.log", "StandardErrorPath": "/l/panel.log"})
|
|
|
|
|
|
class Cli(unittest.TestCase):
|
|
def test_cli_writes_plist_with_spaces(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
target = f"{tmp}/LaunchAgents/a.plist"
|
|
args = ("/usr/bin/python3", f"{tmp}/Application Support/claude_agents.py",
|
|
f"{tmp}/Logs/claude-agents/panel.log")
|
|
r = subprocess.run([sys.executable, SCRIPT, "write", target, *args],
|
|
capture_output=True, text=True)
|
|
self.assertEqual(r.returncode, 0, r.stderr)
|
|
with open(target, "rb") as f:
|
|
self.assertEqual(plistlib.load(f), launchagent.plist(*args))
|
|
self.assertTrue(os.path.isdir(f"{tmp}/Logs/claude-agents"))
|
|
|
|
def test_cli_bad_args(self):
|
|
r = subprocess.run([sys.executable, SCRIPT, "write"], capture_output=True, text=True)
|
|
self.assertEqual(r.returncode, 2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|