test(unraid): add in-host validation runner

This commit is contained in:
Codex
2026-08-15 09:37:17 -07:00
parent d85088d6f8
commit fed8fc66dd
3 changed files with 68 additions and 0 deletions
@@ -0,0 +1,41 @@
import json
import pathlib
import time
import urllib.error
import urllib.request
from websockets.sync.client import connect
token = pathlib.Path("/run/secrets/api-token").read_text().strip()
headers = {"Authorization": "Bearer " + token}
def get(path):
req = urllib.request.Request("http://127.0.0.1:18120" + path, headers=headers)
with urllib.request.urlopen(req, timeout=10) as response:
return json.load(response)
health = get("/v1/health")
capabilities = get("/v1/capabilities")
with connect("ws://127.0.0.1:18120/v1/poses", additional_headers=headers, open_timeout=10) as ws:
messages = [json.loads(ws.recv(timeout=20)) for _ in range(10)]
frame_ids = [m["frame_id"] for m in messages]
summary = {
"tested_at_unix_ns": time.time_ns(),
"health": health,
"capabilities": capabilities,
"websocket": {
"messages": len(messages),
"frame_ids": frame_ids,
"strictly_increasing": all(b > a for a, b in zip(frame_ids, frame_ids[1:])),
"source_alive_all": all(m["source_alive"] for m in messages),
"people_counts": [len(m["people"]) for m in messages],
"coordinate_space_all": all(m["frame"]["coordinate_space"] == "ndi_buffer_unmodified" for m in messages),
"horizontal_flip_all_false": all(not m["frame"]["horizontal_flip_applied"] for m in messages),
"source_to_sent_ms": [round(m["timing"]["source_to_sent_ms"], 3) for m in messages],
},
}
print(json.dumps(summary, ensure_ascii=False, indent=2))
@@ -96,3 +96,28 @@ services:
retries: 6
start_period: 180s
validate:
image: tail2-pose-server:0.1.0
container_name: tail2-pose-validate
network_mode: host
gpus: all
restart: "no"
entrypoint: ["/bin/bash", "-lc"]
depends_on:
tail2-pose-server:
condition: service_healthy
volumes:
- /mnt/user/appdata/tail2-pose-server/app:/app:ro
- /mnt/user/appdata/tail2-pose-server/models:/models:ro
- /mnt/user/appdata/tail2-pose-server/secrets/api-token:/run/secrets/api-token:ro
- ./validation:/validation
command:
- |
set -euo pipefail
python -m pytest -q /opt/tail2-tests > /validation/unit-tests.txt 2>&1
python /app/tail2/validate_runtime.py > /validation/runtime.json
python -m pip freeze | sort > /validation/pip-freeze.txt
find / -xdev -type f -name 'libndi.so*' -print > /validation/ndi-runtime-files.txt 2>/dev/null || true
nvidia-smi --query-gpu=name,driver_version,memory.total,memory.used,temperature.gpu --format=csv,noheader > /validation/gpu.csv
sha256sum /models/*.pth > /validation/model-sha256.txt
chmod 0666 /validation/*.txt /validation/*.json /validation/*.csv
@@ -0,0 +1,2 @@
*
!.gitignore