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

[GEMINI] Refactor test suite structure for new helpers and config modules

Browse Source
This commit is contained in:
venus
2026-07-19 00:53:00 -05:00
parent 8b82058cc6
commit 4bddd4a49b
6 changed files with 59 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_commands.py
View File

@@ -27,7 +27,7 @@ def test_basic_set_flag_format_cli():
"""
Verifies that 'basic set-flag-format' CLI command writes the pattern to config.toml.
"""
from ctf.utils import load_config
from ctf.config import load_config
runner = CliRunner()
config_file = Path("/home/venus/code/ctf/config.toml")
@@ -45,14 +45,14 @@ def test_basic_set_flag_format_cli():
assert updated_config["Competition"]["flag_format"] == "TEST_FLAG{[a-z]+}"
finally:
# Restore original config
from ctf.utils import write_config
from ctf.config import write_config
write_config(original_config, str(config_file))
def test_basic_set_flag_format_original_option_cli():
"""
Verifies that 'basic set-flag-format' with --original/-o option prompts for selection and writes it.
"""
from ctf.utils import load_config
from ctf.config import load_config
runner = CliRunner()
config_file = Path("/home/venus/code/ctf/config.toml")
@@ -78,7 +78,7 @@ def test_basic_set_flag_format_original_option_cli():
assert len(selected_pattern) > 0
finally:
# Restore original config
from ctf.utils import write_config
from ctf.config import write_config
write_config(original_config, str(config_file))
def test_basic_set_flag_format_validation_cli():
@@ -109,7 +109,7 @@ def test_basic_set_competition_cli():
"""
Verifies that 'basic set-competition' updates the competition name in config.toml.
"""
from ctf.utils import load_config
from ctf.config import load_config
runner = CliRunner()
config_file = Path("/home/venus/code/ctf/config.toml")
@@ -123,7 +123,7 @@ def test_basic_set_competition_cli():
updated_config = load_config(str(config_file))
assert updated_config["Competition"]["competition"] == "CyberCTF2026"
finally:
from ctf.utils import write_config
from ctf.config import write_config
write_config(original_config, str(config_file))
def test_basic_set_flag_format_warning_cli():
@@ -131,7 +131,7 @@ def test_basic_set_flag_format_warning_cli():
Verifies that if competition name is set and not present in the example flag,
set-flag-format outputs a warning.
"""
from ctf.utils import load_config, write_config
from ctf.config import load_config, write_config
runner = CliRunner()
config_file = Path("/home/venus/code/ctf/config.toml")