Initial commit: grok-free-register-oss
Open-source Grok free registration CLI, xai_enroller auth pipeline, local auth service, tests and docs.
This commit is contained in:
commit
d10009d639
72 files changed
+18752
No files matched your search
@@ -0,0 +1,64 @@
|
||||
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",
|
||||
]
|
||||
Reference in new issue
Block a user