venus/CTF-tool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[GEMINI] Fix test_flag_detector_stream_persistence test case

Browse Source
This commit is contained in:
venus
2026-07-19 03:07:27 -05:00
parent 7bd450abed
commit eb06206e32
1 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_helpers.py
View File

@@ -86,21 +86,15 @@ def test_flag_detector_stream_persistence():
""" """
from io import StringIO from io import StringIO
from ctf.helpers import FlagDetectorStream from ctf.helpers import FlagDetectorStream
from ctf.config import load_config, write_config from ctf.config import Config
config_file = Path("/home/venus/code/ctf/config.toml") out = StringIO()
original_config = load_config(str(config_file)) stream = FlagDetectorStream(out, r"flag\{[a-z_]+\}")
stream.write("found flag{persisted_flag} in stdout")
try: # Reload config and check last_flag
out = StringIO() cfg = Config()
stream = FlagDetectorStream(out, r"flag\{[a-z_]+\}") assert cfg.data.get("Competition", {}).get("last_flag") == "flag{persisted_flag}"
stream.write("found flag{persisted_flag} in stdout")
# Reload config and check last_flag
updated_config = load_config(str(config_file))
assert updated_config.get("Competition", {}).get("last_flag") == "flag{persisted_flag}"
finally:
write_config(original_config, str(config_file))
# }}} # }}}