feat: complete pre-ai local completion v3

2026-09-21 16:30:27 -03:00
parent 6731eeca08
commit 84b08c2150
41 changed files with 4141 additions and 9 deletions
@@ -0,0 +1,25 @@
+from scripts.rollback_run import preview_records
+
+
+def test_rollback_preview_is_scoped_and_detects_drift():
+    records = [{
+        "run_id": "run-1", "db": "ACME", "product_code": "P1",
+        "fields": {"IPI": {"before": "0", "after": "5"}}, "outcome": "SUCCEEDED",
+    }]
+
+    preview = preview_records(records, run_id="run-1", database="ACME", field="IPI", product="P1",
+                              current={("ACME", "P1", "IPI"): "4"})
+
+    assert preview[0]["target"] == "0"
+    assert preview[0]["affected_rows"] == 1
+    assert preview[0]["status"] == "CONFLICT"
+    assert preview[0]["reason"] == "current != expected_after"
+
+
+def test_rollback_preview_does_not_include_other_runs_or_fields():
+    records = [
+        {"run_id": "run-1", "db": "ACME", "product_code": "P1", "fields": {"IPI": {"before": "0", "after": "5"}}, "outcome": "SUCCEEDED"},
+        {"run_id": "run-2", "db": "ACME", "product_code": "P1", "fields": {"CEST": {"before": "1", "after": "2"}}, "outcome": "SUCCEEDED"},
+    ]
+
+    assert len(preview_records(records, run_id="run-1", database="ACME", field="IPI")) == 1