Antigravity is dog shit.

tests/test_ast_parser.py

@@ -73,9 +73,9 @@ void myTemplateFunc(T x) {
 """
  skeleton = parser.get_skeleton(code)
  assert 'class MyClass {' in skeleton
- assert 'void myMethod() ...' in skeleton
+ assert 'void myMethod() { ... }' in skeleton
  assert 'template <typename T>' in skeleton
- assert 'void myTemplateFunc(T x) ...' in skeleton
+ assert 'void myTemplateFunc(T x) { ... }' in skeleton
  assert 'int x = 1;' not in skeleton
  assert 'x.doSomething();' not in skeleton
 

tests/test_context_presets_manager.py

@@ -90,12 +90,12 @@ def test_app_controller_save_load(tmp_path, monkeypatch):
  controller._save_active_project.assert_called_once()
  
  # Change state
- controller.ui_file_paths = []
+ controller.context_files = []
  controller.screenshots = []
  
  # Load preset
  loaded = controller.load_context_preset("saved_preset")
  assert loaded.name == "saved_preset"
- assert controller.ui_file_paths == ["app.py"]
+ assert [f.path for f in controller.context_files] == ["app.py"]
  assert controller.screenshots == ["app.png"]
  controller._save_active_project.assert_called()

tests/test_context_presets_models.py

@@ -5,7 +5,7 @@ def test_context_file_entry_serialization():
  p = ContextFileEntry(path="test.py", view_mode="skeleton")
  d = p.to_dict()
  # Check for default custom_slices
- assert d == {"path": "test.py", "view_mode": "skeleton", "custom_slices": []}
+ assert d == {"path": "test.py", "view_mode": "skeleton", "custom_slices": [], "ast_definitions": False, "ast_mask": {}, "ast_signatures": False}
  
  p2 = ContextFileEntry.from_dict(d)
  assert p2.path == "test.py"

tests/test_fixes_20260517.py

@@ -12,9 +12,9 @@ def test_context_preview_and_ast_inspector(live_gui):
     
     # Set context file
     resp = requests.post("http://127.0.0.1:8999/api/gui", json={
-        "action": "set_value",
-        "item": "files",
-        "value": ["src/aggregate.py"]
+        "action": "custom_callback",
+        "callback": "set_context_files",
+        "args": [["src/aggregate.py"]]
     })
     assert resp.status_code == 200
     time.sleep(1)

tests/test_rag_phase4_final_verify.py

@@ -35,6 +35,7 @@ def test_phase4_final_verify(live_gui):
   client.set_value('current_provider', 'gemini_cli')
   client.set_value('gcli_path', os.path.abspath(os.path.join(os.path.dirname(__file__), "mock_gcli.bat")))
   
+  time.sleep(1.5)
   # Wait for settings to apply and engine to sync
   success = False
   for _ in range(100):

tests/test_rag_phase4_stress.py

@@ -33,7 +33,7 @@ def test_rag_large_codebase_verification_sim(live_gui):
   client.set_value('rag_source', 'chroma')
   client.set_value('rag_emb_provider', 'local')
   client.set_value('auto_add_history', True)
-  
+  time.sleep(1.5)
   # Wait for settings to apply and engine to sync (initial indexing happens automatically)
   print("[SIM] Waiting for automatic initial indexing...")
   start_initial = time.time()
@@ -93,7 +93,7 @@ def test_rag_large_codebase_verification_sim(live_gui):
     break
    time.sleep(0.2)
 
-  client.set_value('ai_input', "What is the modified content?")
+  client.set_value('ai_input', "Search for MODIFIED CONTENT FOR FILE 25")
   client.click('btn_gen_send')
   
   # Wait for completion

tests/test_ts_c_tools.py

@@ -34,8 +34,8 @@ struct Point {
  
  try:
   skeleton = ts_c_get_skeleton(str(c_file))
-  assert "void hello() ..." in skeleton
-  assert "int add(int a, int b) ..." in skeleton
+  assert "void hello() { ... }" in skeleton
+  assert "int add(int a, int b) { ... }" in skeleton
   assert "struct Point" in skeleton
   assert "printf" not in skeleton
  finally:

tests/test_ts_cpp_tools.py

@@ -37,9 +37,9 @@ void globalFunc() {
  try:
   skeleton = ts_cpp_get_skeleton(str(cpp_file))
   assert "class Box" in skeleton
-  assert "Box(T val) ..." in skeleton
-  assert "T getValue() ..." in skeleton
-  assert "void globalFunc() ..." in skeleton
+  assert "Box(T val) { ... }" in skeleton
+  assert "T getValue() { ... }" in skeleton
+  assert "void globalFunc() { ... }" in skeleton
   assert "std::cout" not in skeleton
  finally:
   mcp_client._resolve_and_check = original_resolve

tests/test_websocket_server.py

@@ -18,7 +18,7 @@ async def test_websocket_subscription_and_broadcast():
  await asyncio.sleep(0.5)
  
  try:
-  uri = f"ws://127.0.0.1:{port}"
+  uri = f"ws://127.0.0.1:{server.port}"
   async with websockets.connect(uri) as websocket:
    # Subscribe to events channel
    subscribe_msg = {"action": "subscribe", "channel": "events"}