29 lines
1.9 KiB
Python
29 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
from __future__ import annotations
|
|
import argparse,json,sys,tempfile,unittest
|
|
from pathlib import Path
|
|
P=argparse.ArgumentParser();P.add_argument("--root",type=Path,required=True);R=P.parse_args().root;sys.path.insert(0,str(R/"tools"))
|
|
import phase10do_one_shot_snapshot_runner as runner # noqa:E402
|
|
|
|
def record(base:Path):
|
|
return {"active":True,"run_id":"DM_TEST","target":"192.0.2.1","port":9021,"artifact_size":runner.ARTIFACT_SIZE,"artifact_sha256":runner.ARTIFACT_SHA256,"snapshot_path":str((base/"snapshot.db").resolve()),"receipt_path":str((base/"consumed.json").resolve()),"not_before":99.0,"not_after":101.0,"one_connection":True,"one_transfer":True,"one_execution":True,"result_receive":True,"target_file_read":True,"device_write":False,"installation":False,"autoload":False,"retry":False,"reconnect":False}
|
|
class Tests(unittest.TestCase):
|
|
def test_inactive_tracked_manifest_rejected(self):
|
|
manifest=json.loads((R/"manifests/retroarch/phase-1.0do-inactive-snapshot-runner.json").read_text())
|
|
with self.assertRaises(runner.RunnerError):runner.validate_records(manifest,manifest,100.0)
|
|
def test_exact_active_pair_validates(self):
|
|
with tempfile.TemporaryDirectory() as d:
|
|
value=record(Path(d));self.assertEqual(runner.validate_records(value,dict(value),100.0)["run_id"],"DM_TEST")
|
|
def test_mismatch_and_forbidden_authority_rejected(self):
|
|
with tempfile.TemporaryDirectory() as d:
|
|
value=record(Path(d));other=dict(value);other["run_id"]="OTHER"
|
|
with self.assertRaises(runner.RunnerError):runner.validate_records(value,other,100.0)
|
|
value["device_write"]=True
|
|
with self.assertRaises(runner.RunnerError):runner.validate_records(value,value,100.0)
|
|
def test_receipt_is_exclusive(self):
|
|
with tempfile.TemporaryDirectory() as d:
|
|
value=record(Path(d));runner._consume(value)
|
|
with self.assertRaises(FileExistsError):runner._consume(value)
|
|
if __name__=="__main__":unittest.main(argv=[__file__])
|