Publish Chimera GFX source

tests/test_phase10dm_snapshot_protocol.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+# SPDX-License-Identifier: GPL-3.0-or-later
+from __future__ import annotations
+import argparse,struct,sys,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"))
+from phase10dm_snapshot_protocol import ProtocolError,parse_stream  # noqa:E402
+F=struct.Struct("<8sIIiiQQQQq")
+def frame(kind,status=0,size=32,sent=0,dev=1,ino=2,mtime=3):return F.pack(b"CHS10DM1",1,kind,status,0,size,sent,dev,ino,mtime)
+def valid():
+ data=b"SQLite format 3\x00"+b"x"*16
+ return frame(1,size=len(data))+data+frame(2,size=len(data),sent=len(data))
+class Tests(unittest.TestCase):
+ def test_valid(self):self.assertEqual(parse_stream(valid()).size,32)
+ def test_truncated(self):
+  with self.assertRaises(ProtocolError):parse_stream(valid()[:-1])
+ def test_error(self):
+  with self.assertRaises(ProtocolError):parse_stream(frame(3,status=1)+b"x"*64)
+ def test_metadata_change(self):
+  raw=valid();raw=raw[:-64]+frame(2,size=32,sent=32,ino=9)
+  with self.assertRaises(ProtocolError):parse_stream(raw)
+ def test_non_sqlite(self):
+  raw=frame(1)+b"z"*32+frame(2,sent=32)
+  with self.assertRaises(ProtocolError):parse_stream(raw)
+if __name__=="__main__":unittest.main(argv=[__file__])