From 03da2019dc76a8ebc120d129947998e7f22d0a26 Mon Sep 17 00:00:00 2001
From: seonghobae <8172694+seonghobae@users.noreply.github.com>
Date: Tue, 16 Jun 2026 05:10:14 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20Add=20tests=20for=20cmd=5Freview?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_vibesec.py | 65 ++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 64 insertions(+), 1 deletion(-)

diff --git a/tests/test_vibesec.py b/tests/test_vibesec.py
index a5a41f9..b96bea7 100644
--- a/tests/test_vibesec.py
+++ b/tests/test_vibesec.py
@@ -5,7 +5,7 @@
 
 import pytest
 
-from scanner.cli.vibesec import _collect_files, _print_scan_results, _scan_file, cmd_init, cmd_scan
+from scanner.cli.vibesec import _collect_files, _print_scan_results, _scan_file, cmd_init, cmd_scan, cmd_review, REVIEW_PROMPT_BASE, REVIEW_PROMPT_NEXTJS, REVIEW_PROMPT_SUPABASE, REVIEW_PROMPT_FIREBASE, REVIEW_PROMPT_STRIPE, REVIEW_PROMPT_FOOTER
 
 MOCK_RULES = [
     {
@@ -416,3 +416,66 @@ def test_sanitize_terminal_output():
 
     # Test non-strings
     assert _sanitize_terminal_output(None) is None
+
+# ---------------------------------------------------------------------------
+# cmd_review tests
+# ---------------------------------------------------------------------------
+
+from argparse import Namespace
+
+def test_cmd_review_base_prompt(capsys):
+    args = Namespace(stack=None, db=None, payments=None)
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_BASE in captured.out
+    assert REVIEW_PROMPT_FOOTER in captured.out
+    assert REVIEW_PROMPT_NEXTJS not in captured.out
+    assert REVIEW_PROMPT_SUPABASE not in captured.out
+    assert REVIEW_PROMPT_FIREBASE not in captured.out
+    assert REVIEW_PROMPT_STRIPE not in captured.out
+
+def test_cmd_review_nextjs(capsys):
+    args = Namespace(stack=["nextjs"], db=None, payments=None)
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_NEXTJS in captured.out
+
+def test_cmd_review_supabase(capsys):
+    args = Namespace(stack=None, db="supabase", payments=None)
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_SUPABASE in captured.out
+
+def test_cmd_review_supabase_via_stack(capsys):
+    args = Namespace(stack=["supabase"], db=None, payments=None)
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_SUPABASE in captured.out
+
+def test_cmd_review_firebase(capsys):
+    args = Namespace(stack=None, db="firebase", payments=None)
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_FIREBASE in captured.out
+
+def test_cmd_review_firebase_via_stack(capsys):
+    args = Namespace(stack=["firebase"], db=None, payments=None)
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_FIREBASE in captured.out
+
+def test_cmd_review_stripe(capsys):
+    args = Namespace(stack=None, db=None, payments="stripe")
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_STRIPE in captured.out
+
+def test_cmd_review_all_options(capsys):
+    args = Namespace(stack=["nextjs"], db="supabase", payments="stripe")
+    cmd_review(args)
+    captured = capsys.readouterr()
+    assert REVIEW_PROMPT_BASE in captured.out
+    assert REVIEW_PROMPT_NEXTJS in captured.out
+    assert REVIEW_PROMPT_SUPABASE in captured.out
+    assert REVIEW_PROMPT_STRIPE in captured.out
+    assert REVIEW_PROMPT_FOOTER in captured.out