d10009d639
Open-source Grok free registration CLI, xai_enroller auth pipeline, local auth service, tests and docs.
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def _entrypoint_workspace(tmp_path):
|
|
for name in ("start.sh", "auth-service.sh", "setup.sh"):
|
|
shutil.copy2(ROOT / name, tmp_path / name)
|
|
scripts = tmp_path / "scripts"
|
|
scripts.mkdir()
|
|
shutil.copy2(ROOT / "scripts" / "ensure_runtime.sh", scripts / "ensure_runtime.sh")
|
|
python = tmp_path / ".venv" / "bin" / "python"
|
|
python.parent.mkdir(parents=True)
|
|
python.write_text(
|
|
"#!/bin/sh\nprintf '%s\\n' \"$@\" > \"$ENTRY_CAPTURE\"\n",
|
|
encoding="utf-8",
|
|
)
|
|
python.chmod(0o755)
|
|
return tmp_path
|
|
|
|
|
|
def _run_entry(workspace, script, *args):
|
|
capture = workspace / "capture.txt"
|
|
environment = {**os.environ, "ENTRY_CAPTURE": str(capture)}
|
|
completed = subprocess.run(
|
|
["bash", script, *args],
|
|
cwd=workspace,
|
|
env=environment,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
assert completed.returncode == 0, completed.stderr
|
|
return capture.read_text(encoding="utf-8").splitlines()
|
|
|
|
|
|
def test_start_dispatches_registration_and_email_as_independent_modules(tmp_path):
|
|
workspace = _entrypoint_workspace(tmp_path)
|
|
(workspace / ".env").write_text("EMAIL_MODE=tempmail\n", encoding="utf-8")
|
|
|
|
assert _run_entry(workspace, "start.sh", "--target", "3") == [
|
|
"-m",
|
|
"grok_register.register",
|
|
"--target",
|
|
"3",
|
|
]
|
|
assert _run_entry(workspace, "start.sh", "--email-service", "--port", "9090") == [
|
|
"-m",
|
|
"grok_register.email_server",
|
|
"--port",
|
|
"9090",
|
|
]
|
|
|
|
|
|
def test_auth_service_keeps_the_supported_python_module_internal(tmp_path):
|
|
workspace = _entrypoint_workspace(tmp_path)
|
|
assert _run_entry(workspace, "auth-service.sh", "--debug") == [
|
|
"-m",
|
|
"xai_enroller.service",
|
|
"--debug",
|
|
]
|