36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
with open("tests/test_visual_orchestration.py", "r", encoding="utf-8", newline="") as f:
|
|
content = f.read()
|
|
|
|
# Add more waits and verify provider was set
|
|
content = content.replace(
|
|
"""# Setup provider
|
|
client.set_value("current_provider", "gemini_cli")
|
|
time.sleep(0.5)
|
|
client.set_value(
|
|
"gcli_path",
|
|
f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"',
|
|
)
|
|
time.sleep(0.5)
|
|
client.set_value("manual_approve", True)
|
|
time.sleep(0.5)""",
|
|
"""# Setup provider - verify it took
|
|
client.set_value("current_provider", "gemini_cli")
|
|
time.sleep(1)
|
|
# Verify provider is set correctly
|
|
mma = client.get_mma_status()
|
|
print(f"[Test] After setting provider: {mma.get('mma_tier_usage', {})}")
|
|
client.set_value(
|
|
"gcli_path",
|
|
f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"',
|
|
)
|
|
time.sleep(1)
|
|
client.set_value("manual_approve", True)
|
|
time.sleep(1)
|
|
print(f"[Test] After all setup: {client.get_mma_status().get('mma_tier_usage', {})}")""",
|
|
)
|
|
|
|
with open("tests/test_visual_orchestration.py", "w", encoding="utf-8", newline="") as f:
|
|
f.write(content)
|
|
|
|
print("Added more debug to test")
|