From 8b82058cc60147b3edd41788229e9ea6378a254b Mon Sep 17 00:00:00 2001 From: venus Date: Sun, 19 Jul 2026 00:40:06 -0500 Subject: [PATCH] [GEMINI] Add test case for check_for_flag function --- tests/test_decoding.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tests/test_decoding.py b/tests/test_decoding.py index 8e1409a..2931f91 100644 --- a/tests/test_decoding.py +++ b/tests/test_decoding.py @@ -133,3 +133,20 @@ def test_try_decode_metadata_rot13_unique_only(): assert "rot13" not in res_duplicate # }}} +# {{{ test_check_for_flag +def test_check_for_flag(): + """ + Verifies that check_for_flag extracts multiple matching flags or returns empty list. + """ + from ctf.decoding import check_for_flag + # Default fallback - matches multiple + res = check_for_flag("flag{one} and flag{two}") + assert set(res) == {"flag{one}", "flag{two}"} + # Custom format + res_custom = check_for_flag("custom{first} custom{second}", r"custom\{[a-z]+\}") + assert set(res_custom) == {"custom{first}", "custom{second}"} + # No matches + assert check_for_flag("no flag matches here") == [] +# }}} + +