Add integration tests for MCP feature

src/test/java/io/stargate/sgv2/jsonapi/api/v1/mcp/McpFeatureDisabledIntegrationTest.java

@@ -0,0 +1,97 @@
+package io.stargate.sgv2.jsonapi.api.v1.mcp;
+
+import static io.stargate.sgv2.jsonapi.api.v1.util.DataApiCommandSenders.assertGeneralCommand;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assertions.*;
+
+import io.quarkiverse.mcp.server.MetaKey;
+import io.quarkiverse.mcp.server.test.McpAssured;
+import io.quarkus.test.common.QuarkusTestResource;
+import io.quarkus.test.junit.QuarkusIntegrationTest;
+import io.stargate.sgv2.jsonapi.api.model.command.CommandName;
+import io.stargate.sgv2.jsonapi.config.feature.ApiFeature;
+import io.stargate.sgv2.jsonapi.exception.SchemaException;
+import io.stargate.sgv2.jsonapi.testresource.DseTestResource;
+import io.vertx.core.json.JsonObject;
+import java.net.URI;
+import java.util.Map;
+import org.junit.jupiter.api.Test;
+
+/**
+ * Integration test to verify MCP feature flag enforcement. When MCP feature is disabled, MCP tool
+ * invocations should fail. But with a per-request header override, they should succeed.
+ */
+@QuarkusIntegrationTest
+@QuarkusTestResource(
+    value = McpFeatureDisabledIntegrationTest.TestResource.class,
+    restrictToAnnotatedClass = true)
+public class McpFeatureDisabledIntegrationTest extends McpIntegrationTestBase {
+
+  /** Custom test resource that disables the MCP feature flag. */
+  public static class TestResource extends DseTestResource {
+    public TestResource() {}
+
+    @Override
+    public String getFeatureFlagMcp() {
+      // return BLANK String to leave feature "undefined" (disabled unless per-request header
+      // override)
+      // ("false" would be "disabled" for all tests, regardless of headers)
+      return " ";
+    }
+  }
+
+  @Test
+  public void restCommandsStillWorkWhenMcpDisabled() {
+    // REST API commands should still function even when MCP feature flag is disabled
+    assertGeneralCommand().postFindEmbeddingProviders().wasSuccessful();
+  }
+
+  @Test
+  public void failToolCallWithMCPDisabled() {
+    callToolAndAssert(
+        CommandName.Names.FIND_KEYSPACES,
+        Map.of(),
+        response -> {
+          // Assert: should be an error with MCP_FEATURE_NOT_ENABLED in structuredContent, no meta_
+          // and content
+          assertThat(response.content()).isEmpty();
+          assertThat(response._meta()).isEmpty();
+          assertTrue(response.isError());
+          assertThat(response.structuredContent()).isNotNull();
+
+          assertThat(response.structuredContent()).isNotNull();
+          var errorContent = (JsonObject) response.structuredContent();
+          assertTrue(errorContent.containsKey("errors"));
+          assertEquals(
+              errorContent.getJsonArray("errors").getJsonObject(0).getString("errorCode"),
+              SchemaException.Code.MCP_FEATURE_NOT_ENABLED.name());
+        });
+  }
+
+  @Test
+  public void okToolCallWithHeaderEnableMCP() {
+    var mcpClientWithHeaderEnable =
+        McpAssured.newStreamableClient()
+            .setBaseUri(URI.create(MCP_HOSTNAME + getTestPort()))
+            .setMcpPath(MCP_PATH)
+            .setAdditionalHeaders(msg -> authHeaders().add(ApiFeature.MCP.httpHeaderName(), "true"))
+            .build()
+            .connect();
+
+    mcpClientWithHeaderEnable
+        .when()
+        .toolsCall(
+            CommandName.Names.FIND_KEYSPACES,
+            Map.of(),
+            response -> {
+              // Assert: should be no error, and have status in meta_
+              assertFalse(response.isError());
+              assertNotNull(response._meta());
+              assertNull(response.structuredContent());
+
+              var status = (JsonObject) response._meta().get(MetaKey.of("status"));
+              assertNotNull(status, "Status should not be null");
+            })
+        .thenAssertResults();
+  }
+}

src/test/java/io/stargate/sgv2/jsonapi/api/v1/mcp/McpIntegrationTestBase.java

@@ -25,10 +25,10 @@
 @TestInstance(TestInstance.Lifecycle.PER_CLASS)
 public abstract class McpIntegrationTestBase {
 
-  private static final String MCP_PATH = "/v1/mcp";
+  protected static final String MCP_PATH = "/v1/mcp";
 
   /** MCP Assured cannot automatically resolve the URI (like Rest Assured) */
-  private static final String MCP_HOSTNAME = "http://localhost:";
+  protected static final String MCP_HOSTNAME = "http://localhost:";
 
   /** Test keyspace name, with a random suffix for isolation. */
   protected final String keyspaceName =