Spaces:
Running
Running
solidprivacy-nl commited on
Commit ·
e442d28
1
Parent(s): ef83cf5
Fix TXT reinsert UI mode placement test
Browse files
tests/test_txt_reinsert_ui_patch.py
CHANGED
|
@@ -42,11 +42,14 @@ def test_txt_reinsert_accepts_txt_only_and_requires_key():
|
|
| 42 |
assert 'st.session_state.get("active_scrub_key", {})' in txt_block
|
| 43 |
|
| 44 |
|
| 45 |
-
def
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
|
|
|
|
|
|
|
|
|
| 50 |
review_summary_start = PATCH_TEXT.index("review_summary_block =")
|
| 51 |
review_summary_end = PATCH_TEXT.index("text = replace_once(\n text,\n ''' st.subheader", review_summary_start)
|
| 52 |
review_summary_block = PATCH_TEXT[review_summary_start:review_summary_end]
|
|
|
|
| 42 |
assert 'st.session_state.get("active_scrub_key", {})' in txt_block
|
| 43 |
|
| 44 |
|
| 45 |
+
def test_txt_reinsert_is_injected_inside_reinsert_mode_only():
|
| 46 |
+
mode_start = PATCH_TEXT.index("two_mode_selection_block = '''")
|
| 47 |
+
mode_end = PATCH_TEXT.index("mode_marker =", mode_start)
|
| 48 |
+
mode_block = PATCH_TEXT[mode_start:mode_end]
|
| 49 |
+
mode_index = mode_block.index('if solidprivacy_work_mode == "Originele waarden terugzetten":')
|
| 50 |
+
txt_injection_index = mode_block.index("scrub_key_import_ui_block + reinsert_ui_block + txt_reinsert_ui_block")
|
| 51 |
+
else_index = mode_block.index("'''else:\n'''")
|
| 52 |
+
assert mode_index < txt_injection_index < else_index
|
| 53 |
review_summary_start = PATCH_TEXT.index("review_summary_block =")
|
| 54 |
review_summary_end = PATCH_TEXT.index("text = replace_once(\n text,\n ''' st.subheader", review_summary_start)
|
| 55 |
review_summary_block = PATCH_TEXT[review_summary_start:review_summary_end]
|