diff --git a/codeflash/main.py b/codeflash/main.py index da0d83db6..37b18954a 100644 --- a/codeflash/main.py +++ b/codeflash/main.py @@ -33,6 +33,8 @@ def main() -> None: from codeflash.telemetry import posthog_cf from codeflash.telemetry.sentry import init_sentry + if "--help" in sys.argv[1:] or "-h" in sys.argv[1:]: + print_codeflash_banner() args = parse_args() if args.command != "auth": print_codeflash_banner() diff --git a/tests/test_help_banner.py b/tests/test_help_banner.py new file mode 100644 index 000000000..c5d801b23 --- /dev/null +++ b/tests/test_help_banner.py @@ -0,0 +1,18 @@ +import subprocess +import sys + + +def test_help_displays_logo() -> None: + result = subprocess.run( + [sys.executable, "-c", "from codeflash.main import main; main()", "--help"], capture_output=True, text=True + ) + assert result.returncode == 0 + assert "codeflash.ai" in result.stdout + + +def test_help_short_flag_displays_logo() -> None: + result = subprocess.run( + [sys.executable, "-c", "from codeflash.main import main; main()", "-h"], capture_output=True, text=True + ) + assert result.returncode == 0 + assert "codeflash.ai" in result.stdout