Add Antigravity v1internal gateway adapter

2026-10-07 01:47:47 +08:00 · 2026-05-25 06:58:15 +08:00
parent 505d9fd8bc
commit 28c3a5dbe4
15 changed files with 978 additions and 33 deletions
@@ -274,3 +274,86 @@ fn classifies_gemini_predict_long_running_as_video_route() {
    );
    assert!(decision.is_execution_runtime_candidate());
 }
+
+#[test]
+fn classifies_antigravity_v1internal_control_plane_routes() {
+    let headers = headers(&[
+        ("authorization", "Bearer ant-access-token"),
+        ("user-agent", "antigravity/cli/1.0.2 linux/arm64"),
+    ]);
+
+    for (path, route_kind) in [
+        ("/v1internal:loadCodeAssist", "load_code_assist"),
+        ("/v1internal:fetchAvailableModels", "fetch_available_models"),
+        ("/v1internal:fetchUserInfo", "fetch_user_info"),
+        ("/v1internal:fetchAdminControls", "fetch_admin_controls"),
+        ("/v1internal:setUserSettings", "set_user_settings"),
+        ("/v1internal:listExperiments", "list_experiments"),
+        (
+            "/v1internal:recordCodeAssistMetrics",
+            "record_code_assist_metrics",
+        ),
+    ] {
+        let uri: Uri = path.parse().expect("uri should parse");
+        let decision = classify_control_route(&http::Method::POST, &uri, &headers)
+            .expect("route should classify");
+
+        assert_eq!(decision.route_class.as_deref(), Some("ai_public"));
+        assert_eq!(decision.route_family.as_deref(), Some("antigravity"));
+        assert_eq!(decision.route_kind.as_deref(), Some(route_kind));
+        assert_eq!(
+            decision.request_auth_channel.as_deref(),
+            Some("bearer_like")
+        );
+        assert_eq!(
+            decision.auth_endpoint_signature.as_deref(),
+            Some("antigravity:v1internal")
+        );
+        assert!(
+            !decision.is_execution_runtime_candidate(),
+            "control-plane route {path} must be handled by local facade before execution runtime"
+        );
+    }
+}
+
+#[test]
+fn classifies_antigravity_stream_generate_content_as_execution_route() {
+    let headers = headers(&[
+        ("authorization", "Bearer ant-access-token"),
+        ("user-agent", "antigravity/cli/1.0.2 linux/arm64"),
+    ]);
+    let uri: Uri = "/v1internal:streamGenerateContent?alt=sse"
+        .parse()
+        .expect("uri should parse");
+    let decision =
+        classify_control_route(&http::Method::POST, &uri, &headers).expect("route should classify");
+
+    assert_eq!(decision.route_class.as_deref(), Some("ai_public"));
+    assert_eq!(decision.route_family.as_deref(), Some("antigravity"));
+    assert_eq!(
+        decision.route_kind.as_deref(),
+        Some("stream_generate_content")
+    );
+    assert_eq!(
+        decision.request_auth_channel.as_deref(),
+        Some("bearer_like")
+    );
+    assert_eq!(
+        decision.auth_endpoint_signature.as_deref(),
+        Some("antigravity:v1internal")
+    );
+    assert!(decision.is_execution_runtime_candidate());
+}
+
+#[test]
+fn rejects_unknown_antigravity_v1internal_route() {
+    let headers = headers(&[
+        ("authorization", "Bearer ant-access-token"),
+        ("user-agent", "antigravity/cli/1.0.2 linux/arm64"),
+    ]);
+    let uri: Uri = "/v1internal:deleteEverything"
+        .parse()
+        .expect("uri should parse");
+
+    assert!(classify_control_route(&http::Method::POST, &uri, &headers).is_none());
+}