v0.33.8

scripts/tests/upgrade_performance_test.py

@@ -0,0 +1,80 @@
+#!/usr/bin/env python3
+"""Run source-upgrade preservation/refusal regressions in disposable directories."""
+from pathlib import Path
+import shutil
+import sqlite3
+import sys
+import tempfile
+
+ROOT = Path(__file__).resolve().parents[2]
+sys.path.insert(0, str(ROOT))
+from scripts.deploy.upgrade import upgrade
+
+
+def fixture(parent):
+    target = parent / "installed"
+    target.mkdir()
+    (target / "Cargo.toml").write_text('[package]\nname = "rust-proxy"\nversion = "0.33.7"\n')
+    (target / "src").mkdir()
+    (target / "src" / "obsolete.rs").write_text("// old managed source\n")
+    (target / "data").mkdir()
+    with sqlite3.connect(target / "data" / "fixture.sqlite") as db:
+        db.execute("CREATE TABLE preserved(value TEXT)")
+        db.execute("INSERT INTO preserved VALUES ('configuration and preferences')")
+    (target / ".env").write_text("LOCAL_FIXTURE=1\n")
+    (target / "langs").mkdir()
+    (target / "langs" / "custom.json").write_text('{"custom": true}\n')
+    return target
+
+
+def main():
+    with tempfile.TemporaryDirectory(prefix="rustproxy-upgrade-test-", dir=ROOT.parent) as temporary:
+        parent = Path(temporary)
+        target = fixture(parent)
+        try:
+            upgrade(ROOT, target, stopped=False)
+        except ValueError:
+            pass
+        else:
+            raise AssertionError("upgrade must require stopped confirmation")
+        assert (target / "src" / "obsolete.rs").is_file()
+        print("PASS upgrade refuses missing --stopped without modifying source")
+
+        if sys.platform != "win32":
+            import fcntl
+            with (target / "data" / "process.lock").open("w+") as lock:
+                fcntl.flock(lock, fcntl.LOCK_EX | fcntl.LOCK_NB)
+                try:
+                    upgrade(ROOT, target, stopped=True)
+                except ValueError:
+                    pass
+                else:
+                    raise AssertionError("upgrade must refuse a live process lock")
+            print("PASS upgrade refuses a data directory locked by a live process")
+
+        backup = upgrade(ROOT, target, stopped=True)
+        assert (target / ".env").read_text() == "LOCAL_FIXTURE=1\n"
+        assert (target / "langs" / "custom.json").read_text() == '{"custom": true}\n'
+        with sqlite3.connect(target / "data" / "fixture.sqlite") as db:
+            assert db.execute("SELECT value FROM preserved").fetchone() == ("configuration and preferences",)
+        assert not (target / "src" / "obsolete.rs").exists()
+        assert (backup / "src" / "obsolete.rs").is_file()
+        assert (target / "Cargo.toml").read_bytes() == (ROOT / "Cargo.toml").read_bytes()
+        assert (target / "src" / "proxy" / "cache_plan.rs").is_file()
+        print("PASS upgrade preserves SQLite, environment and custom language; backs up old source and replaces managed tree")
+
+        if sys.platform != "win32":
+            shutil.rmtree(target / "src")
+            (target / "src").symlink_to(backup / "src", target_is_directory=True)
+            try:
+                upgrade(ROOT, target, stopped=True)
+            except ValueError:
+                pass
+            else:
+                raise AssertionError("upgrade must refuse a managed symlink")
+            assert (backup / "src" / "obsolete.rs").is_file()
+            print("PASS upgrade refuses managed symlinks without changing their targets")
+
+
+if __name__ == "__main__":
+    main()