Use doMaybeLocalRoomAction (#9038)

* Use doMaybeLocalRoomAction

* Revert unnecessary changes

test/ContentMessages-test.ts

@@ -0,0 +1,68 @@
+/*
+Copyright 2022 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 { mocked } from "jest-mock";
+import { IImageInfo, ISendEventResponse, MatrixClient } from "matrix-js-sdk/src/matrix";
+
+import ContentMessages from "../src/ContentMessages";
+import { doMaybeLocalRoomAction } from "../src/utils/local-room";
+
+jest.mock("../src/utils/local-room", () => ({
+    doMaybeLocalRoomAction: jest.fn(),
+}));
+
+describe("ContentMessages", () => {
+    const stickerUrl = "https://example.com/sticker";
+    const roomId = "!room:example.com";
+    const imageInfo = {} as unknown as IImageInfo;
+    const text = "test sticker";
+    let client: MatrixClient;
+    let contentMessages: ContentMessages;
+    let prom: Promise<ISendEventResponse>;
+
+    beforeEach(() => {
+        client = {
+            sendStickerMessage: jest.fn(),
+        } as unknown as MatrixClient;
+        contentMessages = new ContentMessages();
+        prom = Promise.resolve(null);
+    });
+
+    describe("sendStickerContentToRoom", () => {
+        beforeEach(() => {
+            mocked(client.sendStickerMessage).mockReturnValue(prom);
+            mocked(doMaybeLocalRoomAction).mockImplementation((
+                roomId: string,
+                fn: (actualRoomId: string) => Promise<ISendEventResponse>,
+                client?: MatrixClient,
+            ) => {
+                return fn(roomId);
+            });
+        });
+
+        it("should forward the call to doMaybeLocalRoomAction", async () => {
+            await contentMessages.sendStickerContentToRoom(
+                stickerUrl,
+                roomId,
+                null,
+                imageInfo,
+                text,
+                client,
+            );
+            expect(client.sendStickerMessage).toHaveBeenCalledWith(roomId, null, stickerUrl, imageInfo, text);
+        });
+    });
+});

test/MatrixClientPeg-test.ts

@@ -28,8 +28,8 @@ describe("MatrixClientPeg", () => {
     it("setJustRegisteredUserId", () => {
         stubClient();
         (peg as any).matrixClient = peg.get();
-        peg.setJustRegisteredUserId("@userId:matrix.rog");
-        expect(peg.get().credentials.userId).toBe("@userId:matrix.rog");
+        peg.setJustRegisteredUserId("@userId:matrix.org");
+        expect(peg.get().credentials.userId).toBe("@userId:matrix.org");
         expect(peg.currentUserIsJustRegistered()).toBe(true);
         expect(peg.userRegisteredWithinLastHours(0)).toBe(false);
         expect(peg.userRegisteredWithinLastHours(1)).toBe(true);

test/components/views/location/shareLocation-test.ts

@@ -0,0 +1,65 @@
+/*
+Copyright 2022 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 { mocked } from "jest-mock";
+import { ISendEventResponse, MatrixClient } from "matrix-js-sdk/src/matrix";
+import { makeLocationContent } from "matrix-js-sdk/src/content-helpers";
+import { LegacyLocationEventContent, MLocationEventContent } from "matrix-js-sdk/src/@types/location";
+
+import { doMaybeLocalRoomAction } from "../../../../src/utils/local-room";
+import {
+    LocationShareType,
+    shareLocation,
+    ShareLocationFn,
+} from "../../../../src/components/views/location/shareLocation";
+
+jest.mock("../../../../src/utils/local-room", () => ({
+    doMaybeLocalRoomAction: jest.fn(),
+}));
+
+jest.mock("matrix-js-sdk/src/content-helpers", () => ({
+    makeLocationContent: jest.fn(),
+}));
+
+describe("shareLocation", () => {
+    const roomId = "!room:example.com";
+    const shareType = LocationShareType.Pin;
+    const content = { test: "location content" } as unknown as LegacyLocationEventContent & MLocationEventContent;
+    let client: MatrixClient;
+    let shareLocationFn: ShareLocationFn;
+
+    beforeEach(() => {
+        client = {
+            sendMessage: jest.fn(),
+        } as unknown as MatrixClient;
+
+        mocked(makeLocationContent).mockReturnValue(content);
+        mocked(doMaybeLocalRoomAction).mockImplementation((
+            roomId: string,
+            fn: (actualRoomId: string) => Promise<ISendEventResponse>,
+            client?: MatrixClient,
+        ) => {
+            return fn(roomId);
+        });
+
+        shareLocationFn = shareLocation(client, roomId, shareType, null, () => {});
+    });
+
+    it("should forward the call to doMaybeLocalRoomAction", () => {
+        shareLocationFn({ uri: "https://example.com/" });
+        expect(client.sendMessage).toHaveBeenCalledWith(roomId, null, content);
+    });
+});

test/components/views/rooms/SendMessageComposer-test.tsx

@@ -17,8 +17,9 @@ limitations under the License.
 import React from "react";
 import { act } from "react-dom/test-utils";
 import { sleep } from "matrix-js-sdk/src/utils";
-import { MatrixClient } from "matrix-js-sdk/src/matrix";
+import { ISendEventResponse, MatrixClient, MsgType } from "matrix-js-sdk/src/matrix";
 import { mount } from 'enzyme';
+import { mocked } from "jest-mock";
 
 import SendMessageComposer, {
     createMessageContent,
@@ -37,6 +38,11 @@ import { Layout } from '../../../../src/settings/enums/Layout';
 import { IRoomState } from "../../../../src/components/structures/RoomView";
 import { RoomPermalinkCreator } from "../../../../src/utils/permalinks/Permalinks";
 import { mockPlatformPeg } from "../../../test-utils/platform";
+import { doMaybeLocalRoomAction } from "../../../../src/utils/local-room";
+
+jest.mock("../../../../src/utils/local-room", () => ({
+    doMaybeLocalRoomAction: jest.fn(),
+}));
 
 const WrapWithProviders: React.FC<{
     roomContext: IRoomState;
@@ -306,6 +312,34 @@ describe('<SendMessageComposer/>', () => {
             const key = instance.editorStateKey;
             expect(key).toEqual('mx_cider_state_myfakeroom_myFakeThreadId');
         });
+
+        it("correctly sends a message", () => {
+            mocked(doMaybeLocalRoomAction).mockImplementation((
+                roomId: string,
+                fn: (actualRoomId: string) => Promise<ISendEventResponse>,
+                _client?: MatrixClient,
+            ) => {
+                return fn(roomId);
+            });
+
+            mockPlatformPeg({ overrideBrowserShortcuts: jest.fn().mockReturnValue(false) });
+            const wrapper = getComponent();
+
+            addTextToComposer(wrapper, "test message");
+            act(() => {
+                wrapper.find(".mx_SendMessageComposer").simulate("keydown", { key: "Enter" });
+                wrapper.update();
+            });
+
+            expect(mockClient.sendMessage).toHaveBeenCalledWith(
+                "myfakeroom",
+                null,
+                {
+                    "body": "test message",
+                    "msgtype": MsgType.Text,
+                },
+            );
+        });
     });
 
     describe("isQuickReaction", () => {

test/components/views/rooms/VoiceRecordComposerTile-test.tsx

@@ -0,0 +1,104 @@
+/*
+Copyright 2022 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 React from "react";
+import { mount, ReactWrapper } from "enzyme";
+import { ISendEventResponse, MatrixClient, MsgType, Room } from "matrix-js-sdk/src/matrix";
+import { mocked } from "jest-mock";
+
+import VoiceRecordComposerTile from "../../../../src/components/views/rooms/VoiceRecordComposerTile";
+import { IUpload, VoiceRecording } from "../../../../src/audio/VoiceRecording";
+import { doMaybeLocalRoomAction } from "../../../../src/utils/local-room";
+import { MatrixClientPeg } from "../../../../src/MatrixClientPeg";
+
+jest.mock("../../../../src/utils/local-room", () => ({
+    doMaybeLocalRoomAction: jest.fn(),
+}));
+
+describe("<VoiceRecordComposerTile/>", () => {
+    let voiceRecordComposerTile: ReactWrapper<VoiceRecordComposerTile>;
+    let mockRecorder: VoiceRecording;
+    let mockUpload: IUpload;
+    let mockClient: MatrixClient;
+    const roomId = "!room:example.com";
+
+    beforeEach(() => {
+        mockClient = {
+            sendMessage: jest.fn(),
+        } as unknown as MatrixClient;
+        MatrixClientPeg.get = () => mockClient;
+
+        const props = {
+            room: {
+                roomId,
+            } as unknown as Room,
+        };
+        mockUpload = {
+            mxc: "mxc://example.com/voice",
+        };
+        mockRecorder = {
+            stop: jest.fn(),
+            upload: () => Promise.resolve(mockUpload),
+            durationSeconds: 1337,
+            contentType: "audio/ogg",
+            getPlayback: () => ({
+                thumbnailWaveform: [],
+            }),
+        } as unknown as VoiceRecording;
+        voiceRecordComposerTile = mount(<VoiceRecordComposerTile {...props} />);
+        voiceRecordComposerTile.setState({
+            recorder: mockRecorder,
+        });
+
+        mocked(doMaybeLocalRoomAction).mockImplementation((
+            roomId: string,
+            fn: (actualRoomId: string) => Promise<ISendEventResponse>,
+            _client?: MatrixClient,
+        ) => {
+            return fn(roomId);
+        });
+    });
+
+    describe("send", () => {
+        it("should send the voice recording", async () => {
+            await (voiceRecordComposerTile.instance() as VoiceRecordComposerTile).send();
+            expect(mockClient.sendMessage).toHaveBeenCalledWith(roomId, {
+                "body": "Voice message",
+                "file": undefined,
+                "info": {
+                    "duration": 1337000,
+                    "mimetype": "audio/ogg",
+                    "size": undefined,
+                },
+                "msgtype": MsgType.Audio,
+                "org.matrix.msc1767.audio": {
+                    "duration": 1337000,
+                    "waveform": [],
+                },
+                "org.matrix.msc1767.file": {
+                    "file": undefined,
+                    "mimetype": "audio/ogg",
+                    "name": "Voice message.ogg",
+                    "size": undefined,
+                    "url": "mxc://example.com/voice",
+                },
+                "org.matrix.msc1767.text": "Voice message",
+                "org.matrix.msc3245.voice": {},
+                "url": "mxc://example.com/voice",
+            });
+        });
+    });
+});

test/test-utils/test-utils.ts

@@ -71,20 +71,23 @@ export function stubClient() {
  */
 export function createTestClient(): MatrixClient {
     const eventEmitter = new EventEmitter();
+    let txnId = 1;
 
     return {
         getHomeserverUrl: jest.fn(),
         getIdentityServerUrl: jest.fn(),
-        getDomain: jest.fn().mockReturnValue("matrix.rog"),
-        getUserId: jest.fn().mockReturnValue("@userId:matrix.rog"),
+        getDomain: jest.fn().mockReturnValue("matrix.org"),
+        getUserId: jest.fn().mockReturnValue("@userId:matrix.org"),
         getUser: jest.fn().mockReturnValue({ on: jest.fn() }),
         getDeviceId: jest.fn().mockReturnValue("ABCDEFGHI"),
         getDevices: jest.fn().mockResolvedValue({ devices: [{ device_id: "ABCDEFGHI" }] }),
-        credentials: { userId: "@userId:matrix.rog" },
+        credentials: { userId: "@userId:matrix.org" },
 
         store: {
             getPendingEvents: jest.fn().mockResolvedValue([]),
             setPendingEvents: jest.fn().mockResolvedValue(undefined),
+            storeRoom: jest.fn(),
+            removeRoom: jest.fn(),
         },
 
         getPushActionsForEvent: jest.fn(),
@@ -124,7 +127,7 @@ export function createTestClient(): MatrixClient {
         setRoomAccountData: jest.fn(),
         setRoomTopic: jest.fn(),
         sendTyping: jest.fn().mockResolvedValue({}),
-        sendMessage: () => jest.fn().mockResolvedValue({}),
+        sendMessage: jest.fn().mockResolvedValue({}),
         sendStateEvent: jest.fn().mockResolvedValue(undefined),
         getSyncState: () => "SYNCING",
         generateClientSecret: () => "t35tcl1Ent5ECr3T",
@@ -157,6 +160,7 @@ export function createTestClient(): MatrixClient {
         isInitialSyncComplete: jest.fn().mockReturnValue(true),
         downloadKeys: jest.fn(),
         fetchRoomEvent: jest.fn(),
+        makeTxnId: jest.fn().mockImplementation(() => `t${txnId++}`),
     } as unknown as MatrixClient;
 }