from pathlib import Path


PATCH_TEXT = Path("fix_streamlit_nested_expanders.py").read_text(encoding="utf-8")


def test_scrub_key_helpers_are_imported_by_streamlit_patch():
    assert "from scrub_key import build_scrub_key, scrub_key_to_json, validate_scrub_key" in PATCH_TEXT


def test_scrub_key_ui_block_is_present_near_download_section():
    assert "Scrub Key (JSON)" in PATCH_TEXT
    assert "Download Scrub Key (.json)" in PATCH_TEXT
    assert "solidprivacy_scrub_key.json" in PATCH_TEXT
    assert "application/json" in PATCH_TEXT


def test_scrub_key_export_uses_reviewed_replacements_and_timestamp_in_ui_layer():
    assert "scrub_key_rows = edited_replacements_df.copy()" in PATCH_TEXT
    assert "scrub_key_timestamp = datetime.now(timezone.utc)" in PATCH_TEXT
    assert "build_scrub_key(scrub_key_rows)" in PATCH_TEXT
    assert "scrub_key_to_json(scrub_key)" in PATCH_TEXT
    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
    assert "Deel deze sleutel niet met AI-diensten of derden" in PATCH_TEXT


def test_scrub_key_ui_does_not_add_ai_output_cloud_or_auto_rehydration_flow():
    assert "AI-output" not in PATCH_TEXT
    assert "openai" not in PATCH_TEXT.lower()
    assert "requests.post" not in PATCH_TEXT.lower()
    assert "rehydrat" not in PATCH_TEXT.lower()
    assert "restore_original_document" not in PATCH_TEXT
    assert "scrub_key_from_json" not in PATCH_TEXT


def test_scrub_key_patch_does_not_remove_existing_download_or_export_markers():
    assert 'st.subheader("4. Download opgeschoonde bestanden")' in PATCH_TEXT
    assert "Eindcontrole vóór download" in PATCH_TEXT
    assert "Extra exportcontrole" in PATCH_TEXT
    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