93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
import asyncio
|
|
import uuid
|
|
import time
|
|
from typing import Dict, Optional
|
|
|
|
_tasks: Dict[str, dict] = {}
|
|
|
|
|
|
def create_task() -> str:
|
|
task_id = str(uuid.uuid4())
|
|
_tasks[task_id] = {
|
|
"id": task_id,
|
|
"status": "pending",
|
|
"logs": [],
|
|
"created": time.time(),
|
|
}
|
|
return task_id
|
|
|
|
|
|
def get_task(task_id: str) -> Optional[dict]:
|
|
return _tasks.get(task_id)
|
|
|
|
|
|
async def run_test(task_id: str) -> None:
|
|
task = _tasks[task_id]
|
|
task["status"] = "running"
|
|
|
|
def log(msg: str, type: str = "info") -> None:
|
|
task["logs"].append({"msg": msg, "type": type, "ts": time.strftime("%H:%M:%S")})
|
|
|
|
log("▸ Checking UERANSIM Docker image…", "run")
|
|
check = await asyncio.create_subprocess_exec(
|
|
"docker", "images", "-q", "ueransim",
|
|
stdout=asyncio.subprocess.PIPE,
|
|
stderr=asyncio.subprocess.PIPE,
|
|
)
|
|
out, _ = await check.communicate()
|
|
if not out.strip():
|
|
log("✗ UERANSIM image not found.", "err")
|
|
log(" SSH to host and run: bash /opt/p5g-marvis/build-ueransim.sh", "err")
|
|
task["status"] = "error"
|
|
return
|
|
|
|
log(" UERANSIM image ready", "ok")
|
|
log("▸ Starting test container — allow up to 60s…", "run")
|
|
|
|
env_file = "/opt/p5g-marvis/config/ueransim.env"
|
|
|
|
try:
|
|
proc = await asyncio.create_subprocess_exec(
|
|
"docker", "run", "--rm",
|
|
"--network=host",
|
|
"--privileged",
|
|
"--env-file", env_file,
|
|
"ueransim",
|
|
stdout=asyncio.subprocess.PIPE,
|
|
stderr=asyncio.subprocess.STDOUT,
|
|
)
|
|
try:
|
|
stdout, _ = await asyncio.wait_for(proc.communicate(), timeout=90)
|
|
except asyncio.TimeoutError:
|
|
proc.kill()
|
|
log("✗ Test timed out after 90s — container killed", "err")
|
|
task["status"] = "error"
|
|
return
|
|
|
|
for line in stdout.decode(errors="replace").splitlines():
|
|
line = line.strip()
|
|
if not line:
|
|
continue
|
|
if "ERROR" in line:
|
|
log(line, "err")
|
|
elif "PASSED" in line or "established" in line or "successful" in line:
|
|
log(line, "ok")
|
|
elif "WARNING" in line:
|
|
log(line, "warn")
|
|
else:
|
|
log(line, "info")
|
|
|
|
if proc.returncode == 0:
|
|
log("✓ Emulated data session completed successfully", "ok")
|
|
task["status"] = "done"
|
|
elif proc.returncode == 2:
|
|
log("⚠ Credentials not configured — edit /opt/p5g-marvis/config/ueransim.env", "warn")
|
|
task["status"] = "error"
|
|
else:
|
|
log(f"✗ Test exited with code {proc.returncode}", "err")
|
|
task["status"] = "error"
|
|
|
|
except Exception as exc:
|
|
log(f"✗ Unexpected error: {exc}", "err")
|
|
task["status"] = "error"
|