287 lines
13 KiB
Python
287 lines
13 KiB
Python
import fcntl, json, os, signal, subprocess, tempfile, time, unittest, uuid
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
ORCHESTRATOR = ROOT / "tools/sync_orchestrator.py"
|
|
|
|
|
|
def identity(prefix):
|
|
return prefix + str(uuid.uuid4())
|
|
|
|
|
|
class OrchestratorTest(unittest.TestCase):
|
|
def fixture(self, root, **overrides):
|
|
db = root / "trainlog.db"
|
|
db.touch()
|
|
state = Path(str(db) + ".sync-run.json")
|
|
run = identity("sy_")
|
|
request = identity("sy_")
|
|
state.write_text(
|
|
json.dumps(
|
|
{
|
|
"run_id": run,
|
|
"request_id": request,
|
|
"trigger": "web",
|
|
"phase": "requested",
|
|
"progress_revision": 1,
|
|
"result": "running",
|
|
}
|
|
)
|
|
)
|
|
config = {
|
|
"format": "trainlog-sync-orchestrator-config",
|
|
"version": 1,
|
|
"enabled": True,
|
|
"mode": "directory",
|
|
"expected_peer_id": identity("peer_"),
|
|
"transport_root": str(root / "transport"),
|
|
"owned_root": str(root / "owned"),
|
|
"timeout_seconds": 30,
|
|
} | overrides
|
|
(root / "transport").mkdir()
|
|
(root / "owned").mkdir()
|
|
path = root / "config.json"
|
|
path.write_text(json.dumps(config))
|
|
return db, state, path, run, request
|
|
|
|
def invoke(self, executable, values):
|
|
db, state, config, run, request = values
|
|
return subprocess.run(
|
|
[
|
|
"python3",
|
|
str(executable),
|
|
"--database",
|
|
str(db),
|
|
"--state",
|
|
str(state),
|
|
"--config",
|
|
str(config),
|
|
"--run-id",
|
|
run,
|
|
"--request-id",
|
|
request,
|
|
],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
def test_browser_configuration_cannot_select_command_or_boolean_timeout(self):
|
|
for override in (
|
|
{"peer_command": ["evil"]},
|
|
{"timeout_seconds": True},
|
|
{"mode": "unsupported"},
|
|
):
|
|
with self.subTest(override=override), tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
values = self.fixture(root, **override)
|
|
result = self.invoke(ORCHESTRATOR, values)
|
|
self.assertEqual(result.returncode, 2)
|
|
final = json.loads(values[1].read_text())
|
|
self.assertEqual(final["phase"], "failed")
|
|
self.assertEqual(final["error_code"], "internal_error")
|
|
|
|
def test_stdout_bound_is_enforced_before_complete_accumulation(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
sandbox = root / "program"
|
|
sandbox.mkdir()
|
|
(sandbox / "sync_orchestrator.py").write_bytes(ORCHESTRATOR.read_bytes())
|
|
(sandbox / "sync_peer_worker.py").write_text(
|
|
"import sys,time\nsys.stdout.write('x'*70000);sys.stdout.flush();time.sleep(30)\n"
|
|
)
|
|
values = self.fixture(root)
|
|
result = self.invoke(sandbox / "sync_orchestrator.py", values)
|
|
self.assertEqual(result.returncode, 2)
|
|
final = json.loads(values[1].read_text())
|
|
self.assertIn("exceeds 64 KiB", final["diagnostic"])
|
|
|
|
def test_shutdown_reaps_ignoring_helper_and_descendant(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
sandbox = root / "program"
|
|
sandbox.mkdir()
|
|
pidfile = root / "pids"
|
|
(sandbox / "sync_orchestrator.py").write_bytes(ORCHESTRATOR.read_bytes())
|
|
(sandbox / "sync_peer_worker.py").write_text(
|
|
"import os,signal,time\nsignal.signal(signal.SIGTERM,signal.SIG_IGN)\npid=os.fork()\nif pid==0:\n signal.signal(signal.SIGTERM,signal.SIG_IGN);open(%r,'a').write(str(os.getpid())+'\\n');time.sleep(60)\nelse:\n open(%r,'a').write(str(os.getpid())+'\\n');time.sleep(60)\n"
|
|
% (str(pidfile), str(pidfile))
|
|
)
|
|
values = self.fixture(root)
|
|
db, state, config, run, request = values
|
|
process = subprocess.Popen(
|
|
[
|
|
"python3",
|
|
str(sandbox / "sync_orchestrator.py"),
|
|
"--database",
|
|
str(db),
|
|
"--state",
|
|
str(state),
|
|
"--config",
|
|
str(config),
|
|
"--run-id",
|
|
run,
|
|
"--request-id",
|
|
request,
|
|
]
|
|
)
|
|
deadline = time.monotonic() + 5
|
|
while not pidfile.is_file() and time.monotonic() < deadline:
|
|
time.sleep(0.02)
|
|
process.send_signal(signal.SIGTERM)
|
|
self.assertEqual(process.wait(timeout=10), 2)
|
|
for pid in map(int, pidfile.read_text().split()):
|
|
with self.assertRaises(ProcessLookupError):
|
|
os.kill(pid, 0)
|
|
self.assertEqual(json.loads(state.read_text())["phase"], "interrupted")
|
|
|
|
def test_auto_transport_prefers_bluetooth_before_mtp(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
sandbox = root / "program"
|
|
sandbox.mkdir()
|
|
(sandbox / "sync_orchestrator.py").write_bytes(ORCHESTRATOR.read_bytes())
|
|
peer, desktop = identity("peer_"), identity("peer_")
|
|
inbound, outbound = identity("gen_"), identity("gen_")
|
|
report = {
|
|
"format": "trainlog-sync-worker-report", "version": 1,
|
|
"run_id": "RUN", "producer_peer_id": peer, "consumer_peer_id": desktop,
|
|
"inbound_generation_id": inbound, "outbound_generation_id": outbound,
|
|
"manifest_sha256": "a" * 64, "result": "completed",
|
|
"sessions_reconciled": 0, "domains": {}, "drafts": [],
|
|
"ai_midpoint": {"result": "not_configured"},
|
|
"ai_post_sync": {"result": "delivered_to_android"},
|
|
}
|
|
(sandbox / "sync_peer_worker.py").write_text(
|
|
"import json,sys\n"
|
|
"assert sys.argv[sys.argv.index('--mode')+1]=='bt'\n"
|
|
"assert '--bt-adapter' in sys.argv\n"
|
|
"v=" + repr(report) +
|
|
"\nv['run_id']=sys.argv[sys.argv.index('--run-id')+1]\nprint(json.dumps(v))\n"
|
|
)
|
|
bt_marker = root / "bt-probe"
|
|
bt_adapter = sandbox / "trainlog_generation_bt_adapter.py"
|
|
bt_adapter.write_text(f"#!/bin/sh\necho probe >> {bt_marker}\nexit 0\n")
|
|
bt_adapter.chmod(0o755)
|
|
mtp_marker = root / "mtp-probe"
|
|
mtp_adapter = root / "build/tui/trainlog-generation-mtp-adapter"
|
|
mtp_adapter.parent.mkdir(parents=True)
|
|
mtp_adapter.write_text(f"#!/bin/sh\necho probe >> {mtp_marker}\nexit 0\n")
|
|
mtp_adapter.chmod(0o755)
|
|
values = self.fixture(
|
|
root,
|
|
version=3,
|
|
mode="auto",
|
|
bluetooth_enabled=True,
|
|
bluetooth_device_address="AA:BB:CC:DD:EE:FF",
|
|
)
|
|
result = self.invoke(sandbox / "sync_orchestrator.py", values)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertTrue(bt_marker.is_file())
|
|
self.assertFalse(mtp_marker.exists())
|
|
final = json.loads(values[1].read_text())
|
|
self.assertEqual("completed", final["phase"])
|
|
self.assertEqual("bt", final["transport"])
|
|
self.assertEqual("success", final["bluetooth_state"])
|
|
self.assertEqual("unavailable", final["usb_state"])
|
|
|
|
def test_auto_transport_waits_for_mtp_lock_instead_of_falling_back_to_drive(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
sandbox = root / "program"
|
|
sandbox.mkdir()
|
|
(sandbox / "sync_orchestrator.py").write_bytes(ORCHESTRATOR.read_bytes())
|
|
peer, desktop = identity("peer_"), identity("peer_")
|
|
inbound, outbound = identity("gen_"), identity("gen_")
|
|
report = {
|
|
"format": "trainlog-sync-worker-report", "version": 1,
|
|
"run_id": "RUN", "producer_peer_id": peer, "consumer_peer_id": desktop,
|
|
"inbound_generation_id": inbound, "outbound_generation_id": outbound,
|
|
"manifest_sha256": "a" * 64, "result": "completed",
|
|
"sessions_reconciled": 0, "domains": {}, "drafts": [],
|
|
"ai_midpoint": {"result": "not_configured"},
|
|
"ai_post_sync": {"result": "not_configured"},
|
|
}
|
|
(sandbox / "sync_peer_worker.py").write_text(
|
|
"import json,sys\nv=" + repr(report) +
|
|
"\nv['run_id']=sys.argv[sys.argv.index('--run-id')+1]\nprint(json.dumps(v))\n"
|
|
)
|
|
adapter = root / "build/tui/trainlog-generation-mtp-adapter"
|
|
adapter.parent.mkdir(parents=True)
|
|
marker = root / "mtp-probe"
|
|
adapter.write_text(f"#!/bin/sh\necho probe >> {marker}\nexit 0\n")
|
|
adapter.chmod(0o755)
|
|
values = self.fixture(
|
|
root,
|
|
version=2,
|
|
mode="auto",
|
|
drive_enabled=True,
|
|
drive_remote="fake:Trainlog/Sync/v1",
|
|
)
|
|
db, state, config, run, request = values
|
|
mtp_lock = root / "mtp.lock"
|
|
lock_fd = os.open(mtp_lock, os.O_RDWR | os.O_CREAT, 0o600)
|
|
fcntl.flock(lock_fd, fcntl.LOCK_EX)
|
|
process = subprocess.Popen(
|
|
[
|
|
"python3", str(sandbox / "sync_orchestrator.py"),
|
|
"--database", str(db), "--state", str(state),
|
|
"--config", str(config), "--run-id", run, "--request-id", request,
|
|
],
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
)
|
|
try:
|
|
time.sleep(0.2)
|
|
self.assertIsNone(process.poll())
|
|
self.assertFalse(marker.exists())
|
|
finally:
|
|
fcntl.flock(lock_fd, fcntl.LOCK_UN)
|
|
os.close(lock_fd)
|
|
stdout, stderr = process.communicate(timeout=10)
|
|
self.assertEqual(process.returncode, 0, stderr or stdout)
|
|
self.assertTrue(marker.is_file())
|
|
final = json.loads(state.read_text())
|
|
self.assertEqual("completed", final["phase"])
|
|
self.assertEqual("mtp", final["transport"])
|
|
self.assertEqual("success", final["usb_state"])
|
|
|
|
def test_committed_primary_run_survives_drive_mirror_failure(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
sandbox = root / "program"
|
|
sandbox.mkdir()
|
|
(sandbox / "sync_orchestrator.py").write_bytes(ORCHESTRATOR.read_bytes())
|
|
(sandbox / "sync_drive_transport.py").write_text(
|
|
"import sys\nprint('injected Drive failure',file=sys.stderr)\nraise SystemExit(1)\n"
|
|
)
|
|
peer, desktop = identity("peer_"), identity("peer_")
|
|
inbound, outbound = identity("gen_"), identity("gen_")
|
|
report = {
|
|
"format": "trainlog-sync-worker-report", "version": 1,
|
|
"run_id": "RUN", "producer_peer_id": peer, "consumer_peer_id": desktop,
|
|
"inbound_generation_id": inbound, "outbound_generation_id": outbound,
|
|
"manifest_sha256": "a" * 64, "result": "completed",
|
|
"sessions_reconciled": 1, "domains": {"history-v4": True}, "drafts": [],
|
|
"ai_midpoint": {"result": "not_configured"},
|
|
"ai_post_sync": {"result": "not_configured"},
|
|
}
|
|
(sandbox / "sync_peer_worker.py").write_text(
|
|
"import json,sys\nv=" + repr(report) + "\nv['run_id']=sys.argv[sys.argv.index('--run-id')+1]\nprint(json.dumps(v))\n"
|
|
)
|
|
values = self.fixture(
|
|
root,
|
|
version=2,
|
|
drive_enabled=True,
|
|
drive_remote="fake:Trainlog/Sync/v1",
|
|
)
|
|
result = self.invoke(sandbox / "sync_orchestrator.py", values)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
final = json.loads(values[1].read_text())
|
|
self.assertEqual(final["phase"], "completed")
|
|
self.assertEqual(final["drive_state"], "failed")
|
|
self.assertIn("injected Drive failure", final["drive_diagnostic"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|