Class TestSerializeDeserialize

tests/unit/ast_cache_test.py:169–206

class TestSerializeDeserialize(unittest.TestCase):
    def _make_entry(self) -> FileCacheEntry:
        src = "def foo(): pass\n"
        tree = ast.parse(src)
        full_json, chunks = _build_ast_json_and_chunks(tree, src, {})
        return FileCacheEntry(
            file_path="/tmp/test_file.py",
            file_hash=hashlib.sha256(src.encode()).hexdigest(),
            mtime=1234567890.0,
            full_ast_json_z=zlib.compress(full_json.encode()),
            chunks=chunks,
        )

    def test_roundtrip(self):
        entry = self._make_entry()
        restored = _deserialize_entry(_serialize_entry(entry))
        self.assertEqual(restored.file_path, entry.file_path)
        self.assertEqual(restored.file_hash, entry.file_hash)
        self.assertEqual(restored.mtime, entry.mtime)
        self.assertEqual(restored.full_ast_json_z, entry.full_ast_json_z)
        self.assertEqual(restored.version, entry.version)
        self.assertEqual(set(restored.chunks.keys()), set(entry.chunks.keys()))

    def test_serialized_is_json_not_pickle(self):
        entry = self._make_entry()
        raw = _serialize_entry(entry)
        # Must be valid JSON
        d = json.loads(raw)
        self.assertIn("version", d)
        self.assertIn("file_hash", d)
        # Must NOT be a pickle stream (pickle starts with 0x80 or b'\x80')
        self.assertFalse(raw.encode()[0:1] == b'\x80')
        # Must start with '{' (JSON object)
        self.assertEqual(raw[0], '{')

    def test_deserialize_raises_on_garbage(self):
        with self.assertRaises(Exception):
            _deserialize_entry("not json at all }{")


# ── TestIncrementalAstCache ───────────────────────────────────────────────────