mirror of
https://github.com/akadeepesh/contextzip.git
synced 2026-10-20 08:36:20 +02:00
83 lines
3.5 KiB
Python
83 lines
3.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import os
|
||
|
|
import time
|
||
|
|
import zipfile
|
||
|
|
|
||
|
|
from contextzip import snapshots
|
||
|
|
from contextzip.api import create_zip, get_files
|
||
|
|
from tests.helpers import TempProject
|
||
|
|
|
||
|
|
|
||
|
|
class SnapshotTests(TempProject):
|
||
|
|
def setUp(self):
|
||
|
|
super().setUp()
|
||
|
|
(self.root / ".git").mkdir()
|
||
|
|
self.write("src/app.py", "print(1)\n")
|
||
|
|
|
||
|
|
def package(self, **kw):
|
||
|
|
return create_zip(get_files(self.root, use_gitignore=False), **kw)
|
||
|
|
|
||
|
|
def test_every_package_is_a_numbered_snapshot(self):
|
||
|
|
a, b = self.package(), self.package()
|
||
|
|
self.assertEqual((a.snapshot_id, b.snapshot_id), (1, 2))
|
||
|
|
self.assertRegex(a.snapshot, r"^1-[a-z0-9]{6}$")
|
||
|
|
snaps = snapshots.list_snapshots(self.root)
|
||
|
|
self.assertEqual([s.id for s in snaps], [2, 1])
|
||
|
|
self.assertTrue(all(s.is_complete for s in snaps))
|
||
|
|
|
||
|
|
def test_zip_carries_the_guide_with_the_snapshot(self):
|
||
|
|
pkg = self.package()
|
||
|
|
with zipfile.ZipFile(pkg.zip_path) as zf:
|
||
|
|
text = zf.read("CONTEXTZIP.md").decode()
|
||
|
|
self.assertIn(f"Snapshot: {pkg.snapshot}", text)
|
||
|
|
self.assertNotIn("CONTEXTZIP.md", snapshots.find(self.root, pkg.snapshot).load()["files"])
|
||
|
|
|
||
|
|
def test_find(self):
|
||
|
|
pkg = self.package()
|
||
|
|
token = pkg.snapshot.split("-")[1]
|
||
|
|
for ref in ("1", "#1", pkg.snapshot, token):
|
||
|
|
self.assertEqual(snapshots.find(self.root, ref).id, 1, ref)
|
||
|
|
self.assertIsNone(snapshots.find(self.root, "1-zzzzzz")) # another clone's #1
|
||
|
|
|
||
|
|
def test_output_flag_still_snapshots(self):
|
||
|
|
pkg = self.package(output=self.tmp / "out" / "custom.zip")
|
||
|
|
self.assertEqual(pkg.snapshot_id, 1)
|
||
|
|
self.assertTrue(snapshots.find(self.root, 1).zip_path.is_file())
|
||
|
|
|
||
|
|
def test_pruning_keeps_recent_and_young(self):
|
||
|
|
for _ in range(4):
|
||
|
|
self.package()
|
||
|
|
self.assertEqual(snapshots.prune_candidates(self.root, keep=2, max_age_days=14), [])
|
||
|
|
old = time.time() - 30 * 86400
|
||
|
|
for snap in snapshots.list_snapshots(self.root):
|
||
|
|
os.utime(snap.manifest_path, (old, old))
|
||
|
|
pruned = sorted(p.name for p in snapshots.prune_candidates(self.root, keep=2, max_age_days=14))
|
||
|
|
self.assertEqual(pruned, ["0001", "0002"])
|
||
|
|
|
||
|
|
def test_pruning_never_removes_a_snapshot_the_inbox_waits_on(self):
|
||
|
|
first = self.package()
|
||
|
|
for _ in range(3):
|
||
|
|
self.package()
|
||
|
|
old = time.time() - 30 * 86400
|
||
|
|
for snap in snapshots.list_snapshots(self.root):
|
||
|
|
os.utime(snap.manifest_path, (old, old))
|
||
|
|
inbox = self.root / ".contextzip" / "inbox"
|
||
|
|
inbox.mkdir(parents=True, exist_ok=True)
|
||
|
|
with zipfile.ZipFile(inbox / "reply.zip", "w") as zf:
|
||
|
|
zf.writestr("CONTEXTZIP.md", f"Snapshot: {first.snapshot}\n")
|
||
|
|
zf.writestr("src/app.py", "print(2)\n")
|
||
|
|
protected = snapshots.protected_by_inbox(self.root)
|
||
|
|
pruned = {p.name for p in snapshots.prune_candidates(self.root, keep=1, max_age_days=1, protected=protected)}
|
||
|
|
self.assertNotIn("0001", pruned)
|
||
|
|
self.assertEqual(pruned, {"0002", "0003"})
|
||
|
|
|
||
|
|
def test_guide_can_be_turned_off(self):
|
||
|
|
from contextzip.packager import create_zip_silent
|
||
|
|
from contextzip.filters import ResolveResult
|
||
|
|
|
||
|
|
pkg = create_zip_silent(ResolveResult(included=[self.root / "src/app.py"]), self.root, None, guide=False)
|
||
|
|
with zipfile.ZipFile(pkg.zip_path) as zf:
|
||
|
|
self.assertNotIn("CONTEXTZIP.md", zf.namelist())
|
||
|
|
self.assertIsNotNone(pkg.snapshot) # the snapshot is still kept locally
|