227 lines
		
	
	
		
			8.8 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			227 lines
		
	
	
		
			8.8 KiB
		
	
	
	
		
			TypeScript
		
	
	
| /*
 | |
| Copyright 2024 New Vector Ltd.
 | |
| Copyright 2022, 2023 The Matrix.org Foundation C.I.C.
 | |
| 
 | |
| SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only OR LicenseRef-Element-Commercial
 | |
| Please see LICENSE files in the repository root for full details.
 | |
| */
 | |
| 
 | |
| import type { Page } from "@playwright/test";
 | |
| import { test, expect } from "../../element-web-test";
 | |
| import { openIntegrationManager } from "./utils";
 | |
| import type { UserWidget } from "../../../src/utils/WidgetUtils-types.ts";
 | |
| 
 | |
| const ROOM_NAME = "Integration Manager Test";
 | |
| 
 | |
| const INTEGRATION_MANAGER_TOKEN = "DefinitelySecret_DoNotUseThisForReal";
 | |
| const INTEGRATION_MANAGER_HTML = `
 | |
|     <html lang="en">
 | |
|         <head>
 | |
|             <title>Fake Integration Manager</title>
 | |
|         </head>
 | |
|         <body>
 | |
|             <input type="text" id="target-room-id"/>
 | |
|             <input type="text" id="event-type"/>
 | |
|             <input type="text" id="state-key"/>
 | |
|             <button name="Send" id="send-action">Press to send action</button>
 | |
|             <button name="Close" id="close">Press to close</button>
 | |
|             <p id="message-response">No response</p>
 | |
|             <script>
 | |
|                 document.getElementById("send-action").onclick = () => {
 | |
|                     window.parent.postMessage(
 | |
|                         {
 | |
|                             action: "read_events",
 | |
|                             room_id: document.getElementById("target-room-id").value,
 | |
|                             type: document.getElementById("event-type").value,
 | |
|                             state_key: JSON.parse(document.getElementById("state-key").value),
 | |
|                         },
 | |
|                         '*',
 | |
|                     );
 | |
|                 };
 | |
|                 document.getElementById("close").onclick = () => {
 | |
|                     window.parent.postMessage(
 | |
|                         {
 | |
|                             action: "close_scalar",
 | |
|                         },
 | |
|                         '*',
 | |
|                     );
 | |
|                 };
 | |
|                 // Listen for a postmessage response
 | |
|                 window.addEventListener("message", (event) => {
 | |
|                     document.getElementById("message-response").innerText = JSON.stringify(event.data);
 | |
|                 });
 | |
|             </script>
 | |
|         </body>
 | |
|     </html>
 | |
| `;
 | |
| 
 | |
| async function sendActionFromIntegrationManager(
 | |
|     page: Page,
 | |
|     integrationManagerUrl: string,
 | |
|     targetRoomId: string,
 | |
|     eventType: string,
 | |
|     stateKey: string | boolean,
 | |
| ) {
 | |
|     const iframe = page.frameLocator(`iframe[src*="${integrationManagerUrl}"]`);
 | |
|     await iframe.locator("#target-room-id").fill(targetRoomId);
 | |
|     await iframe.locator("#event-type").fill(eventType);
 | |
|     await iframe.locator("#state-key").fill(JSON.stringify(stateKey));
 | |
|     await iframe.locator("#send-action").click();
 | |
| }
 | |
| 
 | |
| test.describe("Integration Manager: Read Events", () => {
 | |
|     test.use({
 | |
|         displayName: "Alice",
 | |
|         room: async ({ user, app }, use) => {
 | |
|             const roomId = await app.client.createRoom({
 | |
|                 name: ROOM_NAME,
 | |
|             });
 | |
|             await use({ roomId });
 | |
|         },
 | |
|     });
 | |
| 
 | |
|     let integrationManagerUrl: string;
 | |
|     test.beforeEach(async ({ page, webserver }) => {
 | |
|         integrationManagerUrl = webserver.start(INTEGRATION_MANAGER_HTML);
 | |
| 
 | |
|         await page.addInitScript(
 | |
|             ({ token, integrationManagerUrl }) => {
 | |
|                 window.localStorage.setItem("mx_scalar_token", token);
 | |
|                 window.localStorage.setItem(`mx_scalar_token_at_${integrationManagerUrl}`, token);
 | |
|             },
 | |
|             {
 | |
|                 token: INTEGRATION_MANAGER_TOKEN,
 | |
|                 integrationManagerUrl,
 | |
|             },
 | |
|         );
 | |
|     });
 | |
| 
 | |
|     test.beforeEach(async ({ page, user, app, room }) => {
 | |
|         await app.client.setAccountData("m.widgets", {
 | |
|             "m.integration_manager": {
 | |
|                 content: {
 | |
|                     type: "m.integration_manager",
 | |
|                     name: "Integration Manager",
 | |
|                     url: integrationManagerUrl,
 | |
|                     data: {
 | |
|                         api_url: integrationManagerUrl,
 | |
|                     },
 | |
|                 },
 | |
|                 id: "integration-manager",
 | |
|             } as unknown as UserWidget,
 | |
|         });
 | |
| 
 | |
|         // Succeed when checking the token is valid
 | |
|         await page.route(
 | |
|             `${integrationManagerUrl}/account?scalar_token=${INTEGRATION_MANAGER_TOKEN}*`,
 | |
|             async (route) => {
 | |
|                 await route.fulfill({
 | |
|                     json: {
 | |
|                         user_id: user.userId,
 | |
|                     },
 | |
|                 });
 | |
|             },
 | |
|         );
 | |
| 
 | |
|         await app.viewRoomByName(ROOM_NAME);
 | |
|     });
 | |
| 
 | |
|     test("should read a state event by state key", async ({ page, app, room }) => {
 | |
|         const eventType = "io.element.integrations.installations";
 | |
|         const eventContent = {
 | |
|             foo: "bar",
 | |
|         };
 | |
|         const stateKey = "state-key-123";
 | |
| 
 | |
|         // Send a state event
 | |
|         const sendEventResponse = await app.client.sendStateEvent(room.roomId, eventType, eventContent, stateKey);
 | |
|         await openIntegrationManager(app);
 | |
| 
 | |
|         // Read state events
 | |
|         await sendActionFromIntegrationManager(page, integrationManagerUrl, room.roomId, eventType, stateKey);
 | |
| 
 | |
|         // Check the response
 | |
|         const iframe = page.frameLocator(`iframe[src*="${integrationManagerUrl}"]`);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(sendEventResponse.event_id);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(`"content":${JSON.stringify(eventContent)}`);
 | |
|     });
 | |
| 
 | |
|     test("should read a state event with empty state key", async ({ page, app, room }) => {
 | |
|         const eventType = "io.element.integrations.installations";
 | |
|         const eventContent = {
 | |
|             foo: "bar",
 | |
|         };
 | |
|         const stateKey = "";
 | |
| 
 | |
|         // Send a state event
 | |
|         const sendEventResponse = await app.client.sendStateEvent(room.roomId, eventType, eventContent, stateKey);
 | |
|         await openIntegrationManager(app);
 | |
| 
 | |
|         // Read state events
 | |
|         await sendActionFromIntegrationManager(page, integrationManagerUrl, room.roomId, eventType, stateKey);
 | |
| 
 | |
|         // Check the response
 | |
|         const iframe = page.frameLocator(`iframe[src*="${integrationManagerUrl}"]`);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(sendEventResponse.event_id);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(`"content":${JSON.stringify(eventContent)}`);
 | |
|     });
 | |
| 
 | |
|     test("should read state events with any state key", async ({ page, app, room }) => {
 | |
|         const eventType = "io.element.integrations.installations";
 | |
| 
 | |
|         const stateKey1 = "state-key-123";
 | |
|         const eventContent1 = {
 | |
|             foo1: "bar1",
 | |
|         };
 | |
|         const stateKey2 = "state-key-456";
 | |
|         const eventContent2 = {
 | |
|             foo2: "bar2",
 | |
|         };
 | |
|         const stateKey3 = "state-key-789";
 | |
|         const eventContent3 = {
 | |
|             foo3: "bar3",
 | |
|         };
 | |
| 
 | |
|         // Send state events
 | |
|         const sendEventResponses = await Promise.all([
 | |
|             app.client.sendStateEvent(room.roomId, eventType, eventContent1, stateKey1),
 | |
|             app.client.sendStateEvent(room.roomId, eventType, eventContent2, stateKey2),
 | |
|             app.client.sendStateEvent(room.roomId, eventType, eventContent3, stateKey3),
 | |
|         ]);
 | |
| 
 | |
|         await openIntegrationManager(app);
 | |
| 
 | |
|         // Read state events
 | |
|         await sendActionFromIntegrationManager(
 | |
|             page,
 | |
|             integrationManagerUrl,
 | |
|             room.roomId,
 | |
|             eventType,
 | |
|             true, // Any state key
 | |
|         );
 | |
| 
 | |
|         // Check the response
 | |
|         const iframe = page.frameLocator(`iframe[src*="${integrationManagerUrl}"]`);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(sendEventResponses[0].event_id);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(`"content":${JSON.stringify(eventContent1)}`);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(sendEventResponses[1].event_id);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(`"content":${JSON.stringify(eventContent2)}`);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(sendEventResponses[2].event_id);
 | |
|         await expect(iframe.locator("#message-response")).toContainText(`"content":${JSON.stringify(eventContent3)}`);
 | |
|     });
 | |
| 
 | |
|     test("should fail to read an event type which is not allowed", async ({ page, app, room }) => {
 | |
|         const eventType = "com.example.event";
 | |
|         const stateKey = "";
 | |
| 
 | |
|         await openIntegrationManager(app);
 | |
| 
 | |
|         // Read state events
 | |
|         await sendActionFromIntegrationManager(page, integrationManagerUrl, room.roomId, eventType, stateKey);
 | |
| 
 | |
|         // Check the response
 | |
|         const iframe = page.frameLocator(`iframe[src*="${integrationManagerUrl}"]`);
 | |
|         await expect(iframe.locator("#message-response")).toContainText("Failed to read events");
 | |
|     });
 | |
| });
 |