From 74925b2c6d1f7de459bb6f35bb8cf472f476ccc3 Mon Sep 17 00:00:00 2001
From: Robin Townsend <robin@robin.town>
Date: Sat, 8 May 2021 19:51:51 -0400
Subject: [PATCH] Test ForwardDialog

Signed-off-by: Robin Townsend <robin@robin.town>
---
 .../views/dialogs/ForwardDialog-test.js       | 124 ++++++++++++++++++
 test/test-utils.js                            |   3 +-
 2 files changed, 126 insertions(+), 1 deletion(-)
 create mode 100644 test/components/views/dialogs/ForwardDialog-test.js

diff --git a/test/components/views/dialogs/ForwardDialog-test.js b/test/components/views/dialogs/ForwardDialog-test.js
new file mode 100644
index 0000000000..cabcbb0325
--- /dev/null
+++ b/test/components/views/dialogs/ForwardDialog-test.js
@@ -0,0 +1,124 @@
+/*
+Copyright 2021 The Matrix.org Foundation C.I.C.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+import "../../../skinned-sdk";
+
+import React from "react";
+import {configure, mount} from "enzyme";
+import Adapter from "enzyme-adapter-react-16";
+import {act} from "react-dom/test-utils";
+
+import * as TestUtils from "../../../test-utils";
+import {MatrixClientPeg} from "../../../../src/MatrixClientPeg";
+import DMRoomMap from "../../../../src/utils/DMRoomMap";
+import ForwardDialog from "../../../../src/components/views/dialogs/ForwardDialog";
+
+configure({ adapter: new Adapter() });
+
+describe("ForwardDialog", () => {
+    let client;
+    let wrapper;
+
+    const message = TestUtils.mkMessage({
+        room: "!111111111111111111:example.org",
+        user: "@alice:example.org",
+        msg: "Hello world!",
+        event: true,
+    });
+
+    beforeEach(async () => {
+        TestUtils.stubClient();
+        DMRoomMap.makeShared();
+        client = MatrixClientPeg.get();
+        client.getVisibleRooms = jest.fn().mockReturnValue(
+            ["a", "A", "b"].map(name => TestUtils.mkStubRoom(name, name)),
+        );
+        client.getUserId = jest.fn().mockReturnValue("@bob:example.org");
+
+        await act(async () => {
+            wrapper = mount(
+                <ForwardDialog
+                    cli={client}
+                    event={message}
+                    onFinished={jest.fn()}
+                />,
+            );
+            // Wait one tick for our profile data to load so the state update happens within act
+            await new Promise(resolve => setImmediate(resolve));
+        });
+    });
+
+    it("shows a preview with us as the sender", () => {
+        const previewBody = wrapper.find(".mx_EventTile_body");
+        expect(previewBody.text()).toBe("Hello world!");
+
+        // We would just test SenderProfile for the user ID, but it's stubbed
+        const previewAvatar = wrapper.find(".mx_EventTile_avatar .mx_BaseAvatar_image");
+        expect(previewAvatar.prop("title")).toBe("@bob:example.org");
+    });
+
+    it("filters the rooms", () => {
+        const roomsBefore = wrapper.find(".mx_ForwardList_entry");
+        expect(roomsBefore).toHaveLength(3);
+
+        const searchInput = wrapper.find(".mx_SearchBox input");
+        searchInput.instance().value = "a";
+        searchInput.simulate("change");
+
+        const roomsAfter = wrapper.find(".mx_ForwardList_entry");
+        expect(roomsAfter).toHaveLength(2);
+    });
+
+    it("tracks message sending progress across multiple rooms", async () => {
+        // Make sendEvent require manual resolution so we can see the sending state
+        let finishSend;
+        let cancelSend;
+        client.sendEvent = jest.fn(() => new Promise((resolve, reject) => {
+            finishSend = resolve;
+            cancelSend = reject;
+        }));
+
+        const firstRoom = wrapper.find(".mx_ForwardList_entry").first();
+        expect(firstRoom.find(".mx_AccessibleButton").text()).toBe("Send");
+
+        act(() => {
+            firstRoom.find(".mx_AccessibleButton").simulate("click");
+        });
+        expect(firstRoom.find(".mx_AccessibleButton").text()).toBe("Sending…");
+
+        await act(async () => {
+            cancelSend();
+            // Wait one tick for the button to realize the send failed
+            await new Promise(resolve => setImmediate(resolve));
+        });
+        expect(firstRoom.find(".mx_AccessibleButton").text()).toBe("Failed to send");
+
+        const secondRoom = wrapper.find(".mx_ForwardList_entry").at(1);
+        expect(secondRoom.find(".mx_AccessibleButton").text()).toBe("Send");
+
+        act(() => {
+            secondRoom.find(".mx_AccessibleButton").simulate("click");
+        });
+        expect(secondRoom.find(".mx_AccessibleButton").text()).toBe("Sending…");
+
+        await act(async () => {
+            finishSend();
+            // Wait one tick for the button to realize the send succeeded
+            await new Promise(resolve => setImmediate(resolve));
+        });
+        expect(secondRoom.find(".mx_AccessibleButton").text()).toBe("Sent");
+    });
+});
diff --git a/test/test-utils.js b/test/test-utils.js
index 6dc02463a5..985e9f804b 100644
--- a/test/test-utils.js
+++ b/test/test-utils.js
@@ -218,7 +218,7 @@ export function mkMessage(opts) {
     return mkEvent(opts);
 }
 
-export function mkStubRoom(roomId = null) {
+export function mkStubRoom(roomId = null, name) {
     const stubTimeline = { getEvents: () => [] };
     return {
         roomId,
@@ -254,6 +254,7 @@ export function mkStubRoom(roomId = null) {
         on: jest.fn(),
         removeListener: jest.fn(),
         getDMInviter: jest.fn(),
+        name,
         getAvatarUrl: () => 'mxc://avatar.url/room.png',
         getMxcAvatarUrl: () => 'mxc://avatar.url/room.png',
         isSpaceRoom: jest.fn(() => false),