Add planner previews and streaming

tests/test_endpoints.py

@@ -0,0 +1,32 @@
+from fastapi.testclient import TestClient
+
+from server.main import app, manager
+
+test_client = TestClient(app)
+
+
+def test_plan_endpoint(default_grid_request):
+    init_response = test_client.post("/grid/init", json=default_grid_request)
+    grid_id = init_response.json()["grid_id"]
+
+    plan_response = test_client.post(
+        f"/grid/{grid_id}/plan",
+        json={"preferred_label": None, "action": "click", "text": "hello"},
+    )
+    assert plan_response.status_code == 200
+    payload = plan_response.json()
+    assert payload["plan"]["grid_id"] == grid_id
+    assert payload["result"]["success"]
+
+
+def test_refresh_endpoint(default_grid_request):
+    init_response = test_client.post("/grid/init", json=default_grid_request)
+    grid_id = init_response.json()["grid_id"]
+
+    refresh_response = test_client.post(
+        f"/grid/{grid_id}/refresh", json={"screenshot_base64": "AAA", "memo": "updated"}
+    )
+    assert refresh_response.status_code == 200
+    grid = manager.get_grid(grid_id)
+    assert grid.screenshot == "AAA"
+    assert grid.memo == "updated"