36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import pytest
|
|
import time
|
|
import requests
|
|
import json
|
|
import os
|
|
|
|
def test_context_preview_and_ast_inspector(live_gui):
|
|
process, script_path = live_gui
|
|
|
|
# Give it a second to stabilize
|
|
time.sleep(1)
|
|
|
|
# Set context file
|
|
resp = requests.post("http://127.0.0.1:8999/api/gui", json={
|
|
"action": "custom_callback",
|
|
"callback": "set_context_files",
|
|
"args": [["src/aggregate.py"]]
|
|
})
|
|
assert resp.status_code == 200
|
|
time.sleep(1)
|
|
|
|
# Trigger Context Preview
|
|
resp = requests.post("http://127.0.0.1:8999/api/gui", json={
|
|
"action": "click",
|
|
"item": "btn_preview_ctx" # wait, there is no btn_preview_ctx registered in _clickable_actions
|
|
})
|
|
|
|
# Wait, the best way to verify if _do_generate works without crashing is
|
|
# checking /api/v1/context
|
|
resp = requests.get("http://127.0.0.1:8999/api/v1/context")
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert "## Files" in data["markdown"]
|
|
assert "aggregate.py" in data["markdown"]
|
|
print("SUCCESS: Context Generation works.")
|