[GEMINI] Reorganize tests for FlagDetectorStream and cli_helpers commands

tests/test_cli_helpers.py

@@ -0,0 +1,72 @@
+# tests/test_cli_helpers.py
+import sys
+import pytest
+from pathlib import Path
+from click.testing import CliRunner
+from unittest.mock import patch
+from ctf.config import load_config, write_config
+
+# {{{ test_flag_cmd_cli
+def test_flag_cmd_cli():
+    """
+    Verifies the ctf flag command outputs last_flag with and without --plain.
+    """
+    from ctf.main import cli
+    
+    config_file = Path("/home/venus/code/ctf/config.toml")
+    original_config = load_config(str(config_file))
+    
+    try:
+        # Pre-set last_flag in config
+        temp_config = load_config(str(config_file))
+        if "Competition" not in temp_config:
+            temp_config["Competition"] = {}
+        temp_config["Competition"]["last_flag"] = "flag{test_cli_flag}"
+        write_config(temp_config, str(config_file))
+        
+        runner = CliRunner()
+        
+        # Test standard flag command output
+        result_std = runner.invoke(cli, ["flag"])
+        assert result_std.exit_code == 0
+        assert "Last detected flag: flag{test_cli_flag}" in result_std.output
+        
+        # Test plain flag command output
+        result_plain = runner.invoke(cli, ["flag", "--plain"])
+        assert result_plain.exit_code == 0
+        assert result_plain.output.strip() == "flag{test_cli_flag}"
+    finally:
+        write_config(original_config, str(config_file))
+# }}}
+
+# {{{ test_flag_cmd_exemption
+def test_flag_cmd_exemption(capsys):
+    """
+    Verifies that running ctf flag is exempted from triggering the stdout flag warning box.
+    """
+    from ctf.main import main
+    
+    config_file = Path("/home/venus/code/ctf/config.toml")
+    original_config = load_config(str(config_file))
+    
+    try:
+        temp_config = load_config(str(config_file))
+        if "Competition" not in temp_config:
+            temp_config["Competition"] = {}
+        temp_config["Competition"]["flag_format"] = r"flag\{[a-z_]+\}"
+        temp_config["Competition"]["last_flag"] = "flag{test_exempt_flag}"
+        write_config(temp_config, str(config_file))
+        
+        with patch.object(sys, "argv", ["ctf", "flag"]):
+            with pytest.raises(SystemExit) as exc_info:
+                main()
+            assert exc_info.value.code == 0
+            
+        captured = capsys.readouterr()
+        # Verify that the warning box is NOT in stdout
+        assert "Potential flag(s) detected in command output" not in captured.out
+        # Verify that the actual flag info IS printed
+        assert "Last detected flag: flag{test_exempt_flag}" in captured.out
+    finally:
+        write_config(original_config, str(config_file))
+# }}}