Progress on context composition

tests/test_ast_inspector_extended.py

@@ -26,6 +26,8 @@ def test_ast_inspector_line_range_parsing():
   mock_imgui.begin_child.return_value = True
   # radio_button returns (changed, active)
   mock_imgui.radio_button.return_value = (False, False)
+  mock_imgui.get_content_region_avail.return_value.y = 800.0
+  mock_imgui.get_frame_height_with_spacing.return_value = 24.0
   
   # Setup imscope mocks
   mock_imscope.window.return_value.__enter__.return_value = (True, True)

tests/test_fixes_20260517.py

@@ -0,0 +1,35 @@
+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": "set_value",
+        "item": "files",
+        "value": ["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.")