79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Real-clock scheduling test: set a slot 1 minute ahead, start the daemon,
|
||
|
|
and verify it fires the scheduled run on its own."""
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import tarfile
|
||
|
|
import time
|
||
|
|
|
||
|
|
ROOT = "/vol1/@appshare/fn-deepseek-harness/fnos-agent-checkin"
|
||
|
|
SIM = "/tmp/fnos-sim"
|
||
|
|
APPDEST = SIM + "/var/apps/AgentCheckin"
|
||
|
|
PKGVAR = SIM + "/appdata/AgentCheckin"
|
||
|
|
TARGET = APPDEST + "/target"
|
||
|
|
BACKEND = TARGET + "/backend"
|
||
|
|
|
||
|
|
shutil.rmtree(SIM, ignore_errors=True)
|
||
|
|
os.makedirs(APPDEST)
|
||
|
|
os.makedirs(PKGVAR)
|
||
|
|
os.makedirs(SIM + "/pkg")
|
||
|
|
subprocess.run(f"tar xzf {ROOT}/AgentCheckin/AgentCheckin.fpk -C {SIM}/pkg", shell=True, check=True)
|
||
|
|
with tarfile.open(SIM + "/pkg/app.tgz") as t:
|
||
|
|
t.extractall(TARGET)
|
||
|
|
for item in ("cmd", "config", "manifest", "wizard"):
|
||
|
|
src, dst = f"{SIM}/pkg/{item}", f"{APPDEST}/{item}"
|
||
|
|
shutil.copytree(src, dst) if os.path.isdir(src) else shutil.copy(src, dst)
|
||
|
|
subprocess.run(f"chmod +x {TARGET}/ui/index.cgi {APPDEST}/cmd/*", shell=True)
|
||
|
|
|
||
|
|
env = dict(os.environ, TRIM_APPDEST=APPDEST, TRIM_PKGVAR=PKGVAR)
|
||
|
|
subprocess.run(f"TRIM_PKGVAR={PKGVAR} bash {APPDEST}/cmd/install_callback",
|
||
|
|
shell=True, env=env, check=True)
|
||
|
|
|
||
|
|
add_account = (
|
||
|
|
"import sys; sys.path.insert(0, %r)\n"
|
||
|
|
"import engine\n"
|
||
|
|
"engine.add_trae('sched-test', 'eyJhbGciOiJSUzI1NiJ9.eyJ1c2VyUmVnaW9uIjp7InJlZ2lvbiI6IkNOIn19.sig')\n"
|
||
|
|
) % BACKEND
|
||
|
|
subprocess.run([sys.executable, "-c", add_account], env=env, check=True,
|
||
|
|
capture_output=True)
|
||
|
|
|
||
|
|
slot = time.strftime("%H:%M", time.localtime(time.time() + 60))
|
||
|
|
set_slot = (
|
||
|
|
"import sys; sys.path.insert(0, %r)\n"
|
||
|
|
"import common\n"
|
||
|
|
"s = common.load_settings(); s['schedule_times'] = [%r]\n"
|
||
|
|
"common.save_settings(s)\n"
|
||
|
|
"print('slot:', %r)\n"
|
||
|
|
) % (BACKEND, slot, slot)
|
||
|
|
r = subprocess.run([sys.executable, "-c", set_slot], env=env, check=True,
|
||
|
|
capture_output=True, text=True)
|
||
|
|
print(r.stdout.strip(), "| now:", time.strftime("%H:%M:%S"))
|
||
|
|
|
||
|
|
r = subprocess.run(f"bash {APPDEST}/cmd/main start", shell=True, env=env,
|
||
|
|
capture_output=True, text=True)
|
||
|
|
print("start:", r.stdout.strip() or r.stderr.strip())
|
||
|
|
|
||
|
|
deadline = time.time() + 170
|
||
|
|
fired = None
|
||
|
|
while time.time() < deadline:
|
||
|
|
try:
|
||
|
|
st = json.load(open(PKGVAR + "/state.json"))
|
||
|
|
except (OSError, ValueError):
|
||
|
|
time.sleep(3)
|
||
|
|
continue
|
||
|
|
reason = st.get("last_run_reason", "")
|
||
|
|
if reason.startswith("schedule:"):
|
||
|
|
fired = reason
|
||
|
|
break
|
||
|
|
time.sleep(3)
|
||
|
|
|
||
|
|
subprocess.run(f"bash {APPDEST}/cmd/main stop", shell=True, env=env,
|
||
|
|
capture_output=True)
|
||
|
|
print("SCHEDULED RUN:", fired)
|
||
|
|
ok = fired == "schedule:" + slot
|
||
|
|
print("RESULT:", "PASS" if ok else "FAIL")
|
||
|
|
sys.exit(0 if ok else 1)
|