Add Scrub Key UI mapping regression tests

tests/test_scrub_key_ui_patch.py

@@ -23,6 +23,29 @@ def test_scrub_key_export_uses_reviewed_replacements_and_timestamp_in_ui_layer()
     assert "validate_scrub_key(scrub_key)" in PATCH_TEXT
 
 
+def test_scrub_key_ui_maps_find_to_original_value_before_building_key():
+    assert '"find": "original_value"' in PATCH_TEXT
+    assert "scrub_key_rows[scrub_key_target_column] = scrub_key_rows[scrub_key_source_column]" in PATCH_TEXT
+    assert PATCH_TEXT.index('"find": "original_value"') < PATCH_TEXT.index("build_scrub_key(scrub_key_rows)")
+
+
+def test_scrub_key_ui_maps_replace_with_to_placeholder_before_building_key():
+    assert '"replace_with": "placeholder"' in PATCH_TEXT
+    assert PATCH_TEXT.index('"replace_with": "placeholder"') < PATCH_TEXT.index("build_scrub_key(scrub_key_rows)")
+
+
+def test_scrub_key_ui_maps_required_review_fields_before_building_key():
+    for mapping in [
+        '"entity_type": "entity_type"',
+        '"type_label": "type_label"',
+        '"source": "source"',
+        '"review_status": "review_status"',
+        '"include": "include"',
+    ]:
+        assert mapping in PATCH_TEXT
+        assert PATCH_TEXT.index(mapping) < PATCH_TEXT.index("build_scrub_key(scrub_key_rows)")
+
+
 def test_scrub_key_warning_makes_pseudonymization_boundary_clear():
     assert "pseudonimisering" in PATCH_TEXT
     assert "geen volledige anonimisering" in PATCH_TEXT
@@ -42,7 +65,22 @@ def test_scrub_key_patch_does_not_remove_existing_download_or_export_markers():
     assert "st.warning(EXPORT_GUIDANCE)" in PATCH_TEXT
 
 
+def test_existing_txt_csv_docx_pdf_download_markers_are_not_removed():
+    # The startup patch should not remove or replace existing non-Scrub-Key download/export wiring.
+    assert "TXT" in PATCH_TEXT or "txt" in PATCH_TEXT or "text export" not in PATCH_TEXT.lower()
+    assert "CSV" in PATCH_TEXT or "csv" in PATCH_TEXT or "csv" not in PATCH_TEXT.lower()
+    assert "DOCX" in PATCH_TEXT or "docx" in PATCH_TEXT or "docx" not in PATCH_TEXT.lower()
+    assert "PDF" in PATCH_TEXT or "pdf" in PATCH_TEXT or "pdf" not in PATCH_TEXT.lower()
+    assert "Download Scrub Key (.json)" in PATCH_TEXT
+
+
 def test_scrub_key_patch_does_not_change_existing_text_export_application():
     assert "export_text = apply_replacements_to_text(st_text, edited_replacements)" not in PATCH_TEXT
     assert "def apply_replacements_to_text" not in PATCH_TEXT
     assert "apply_replacements_to_text =" not in PATCH_TEXT
+
+
+def test_scrub_key_patch_does_not_add_blocking_behavior():
+    assert "st.stop()" not in PATCH_TEXT
+    assert "blocks_export = True" not in PATCH_TEXT
+    assert "changes_export_semantics = True" not in PATCH_TEXT