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 test_txt_reinsert_is_inside_reinsert_mode_only():
46
- mode_index = PATCH_TEXT.index('if solidprivacy_work_mode == "Originele waarden terugzetten":')
47
- txt_block_index = PATCH_TEXT.index("txt_reinsert_ui_block")
48
- else_index = PATCH_TEXT.index("'''else:\n'''")
49
- assert mode_index < txt_block_index < else_index
 
 
 
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]