feat(test): Add MOCK_MODE environment variable support to mock_gemini_cli.py

This commit is contained in:
2026-03-06 10:37:14 -05:00
parent 09261cf69b
commit 0e23d6afb7

View File

@@ -7,6 +7,18 @@ def main() -> None:
sys.stderr.write(f"DEBUG: GEMINI_CLI_HOOK_CONTEXT: {os.environ.get('GEMINI_CLI_HOOK_CONTEXT')}\n")
sys.stderr.flush()
mock_mode = os.environ.get("MOCK_MODE", "success")
if mock_mode == "malformed_json":
print("{broken_json: ", flush=True)
sys.exit(1)
elif mock_mode == "error_result":
print(json.dumps({"type": "result", "status": "error", "error": "Mock simulated error"}), flush=True)
sys.exit(1)
elif mock_mode == "timeout":
import time
time.sleep(120)
sys.exit(1)
# Read prompt from stdin
try:
prompt = sys.stdin.read()