ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-03-06 10:37:14 -05:00
parent 09261cf69b
commit 0e23d6afb7
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/mock_gemini_cli.py
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.write(f"DEBUG: GEMINI_CLI_HOOK_CONTEXT: {os.environ.get('GEMINI_CLI_HOOK_CONTEXT')}\n")
sys.stderr.flush() 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 # Read prompt from stdin
try: try:
prompt = sys.stdin.read() prompt = sys.stdin.read()