OLS-2654 - adding MCP apps tests

cypress/support/commands.ts

@@ -42,6 +42,19 @@ declare global {
         query: string,
         errorMessage: string,
       ): Chainable<Element>;
+      interceptMCPQuery(
+        alias: string,
+        query: string,
+        toolName: string,
+        uiResourceUri: string,
+        conversationId?: string | null,
+      ): Chainable<Element>;
+      interceptMCPResources(
+        alias: string,
+        htmlContent: string,
+        serverName?: string,
+        uiResourceUri?: string,
+      ): Chainable<Element>;
     }
   }
 }
@@ -138,7 +151,7 @@ Cypress.Commands.add(
     cy.intercept('POST', getApiUrl('/v1/streaming_query'), (request) => {
       expect(request.body.media_type).to.equal('application/json');
       expect(request.body.conversation_id).to.equal(conversationId);
-      expect(request.body.query).to.include(query);
+      expect(request.body.query).to.equal(query);
 
       expect(request.body.attachments).to.have.lengthOf(attachments.length);
       attachments.forEach((a, i) => {
@@ -168,7 +181,7 @@ Cypress.Commands.add(
   'interceptQueryWithError',
   (alias: string, query: string, errorMessage: string) => {
     cy.intercept('POST', getApiUrl('/v1/streaming_query'), (request) => {
-      expect(request.body.query).to.include(query);
+      expect(request.body.query).to.equal(query);
       const responseBody = MOCK_STREAMED_RESPONSE_WITH_ERROR_BODY.replace(
         'MOCK_ERROR_MESSAGE',
         errorMessage,
@@ -204,3 +217,72 @@ Cypress.Commands.add(
     }).as(alias);
   },
 );
+
+const MOCK_MCP_STREAMED_RESPONSE_BODY_TEMPLATE = `data: {"event": "start", "data": {"conversation_id": "CONVERSATION_ID"}}
+
+data: {"event": "token", "data": {"id": 0, "token": "Here"}}
+
+data: {"event": "token", "data": {"id": 1, "token": " is"}}
+
+data: {"event": "token", "data": {"id": 2, "token": " your"}}
+
+data: {"event": "token", "data": {"id": 3, "token": " MCP"}}
+
+data: {"event": "token", "data": {"id": 4, "token": " dashboard"}}
+
+data: {"event": "tool_call", "data": {"id": 1, "name": "TOOL_NAME", "server_name": "test-server", "args": {}}}
+
+data: {"event": "tool_result", "data": {"id": 1, "content": "Dashboard loaded", "status": "success", "server_name": "test-server", "tool_meta": {"ui": {"resourceUri": "UI_RESOURCE_URI"}}}}
+
+data: {"event": "end", "data": {"referenced_documents": [], "truncated": false}}
+`;
+
+Cypress.Commands.add(
+  'interceptMCPQuery',
+  (
+    alias: string,
+    query: string,
+    toolName: string,
+    uiResourceUri: string,
+    conversationId: string | null = null,
+  ) => {
+    cy.intercept('POST', getApiUrl('/v1/streaming_query'), (request) => {
+      expect(request.body.media_type).to.equal('application/json');
+      expect(request.body.conversation_id).to.equal(conversationId);
+      expect(request.body.query).to.equal(query);
+
+      const responseBody = MOCK_MCP_STREAMED_RESPONSE_BODY_TEMPLATE.replace(
+        'CONVERSATION_ID',
+        conversationId || '5f424596-a4f9-4a3a-932b-46a768de3e7c',
+      )
+        .replace('TOOL_NAME', toolName)
+        .replace('UI_RESOURCE_URI', uiResourceUri);
+
+      request.reply({ body: responseBody, delay: 1000 });
+    }).as(alias);
+  },
+);
+
+Cypress.Commands.add(
+  'interceptMCPResources',
+  (
+    alias: string,
+    htmlContent: string,
+    serverName: string = 'test-server',
+    uiResourceUri: string = 'mcp://test-server/resources/dashboard',
+  ) => {
+    cy.intercept('POST', getApiUrl('/v1/mcp-apps/resources'), (request) => {
+      /* eslint-disable camelcase */
+      Cypress.log({
+        name: 'MCP Resources Request',
+        message: `server_name: ${request.body.server_name}, resource_uri: ${request.body.resource_uri}`,
+      });
+
+      expect(request.body.server_name).to.equal(serverName);
+      expect(request.body.resource_uri).to.equal(uiResourceUri);
+      /* eslint-enable camelcase */
+
+      request.reply({ body: { content: htmlContent }, delay: 500 });
+    }).as(alias);
+  },
+);