469 lines
		
	
	
		
			19 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			469 lines
		
	
	
		
			19 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
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 { EventStatus, IEventRelation } from "matrix-js-sdk/src/matrix";
 | 
						|
 | 
						|
import { IRoomState } from "../../../../../../src/components/structures/RoomView";
 | 
						|
import { editMessage, sendMessage } from "../../../../../../src/components/views/rooms/wysiwyg_composer/utils/message";
 | 
						|
import { createTestClient, getRoomContext, mkEvent, mkStubRoom } from "../../../../../test-utils";
 | 
						|
import defaultDispatcher from "../../../../../../src/dispatcher/dispatcher";
 | 
						|
import SettingsStore from "../../../../../../src/settings/SettingsStore";
 | 
						|
import { SettingLevel } from "../../../../../../src/settings/SettingLevel";
 | 
						|
import { RoomPermalinkCreator } from "../../../../../../src/utils/permalinks/Permalinks";
 | 
						|
import EditorStateTransfer from "../../../../../../src/utils/EditorStateTransfer";
 | 
						|
import * as ConfirmRedactDialog from "../../../../../../src/components/views/dialogs/ConfirmRedactDialog";
 | 
						|
import * as SlashCommands from "../../../../../../src/SlashCommands";
 | 
						|
import * as Commands from "../../../../../../src/editor/commands";
 | 
						|
import * as Reply from "../../../../../../src/utils/Reply";
 | 
						|
import { MatrixClientPeg } from "../../../../../../src/MatrixClientPeg";
 | 
						|
import { Action } from "../../../../../../src/dispatcher/actions";
 | 
						|
 | 
						|
describe("message", () => {
 | 
						|
    const permalinkCreator = {
 | 
						|
        forEvent(eventId: string): string {
 | 
						|
            return "$$permalink$$";
 | 
						|
        },
 | 
						|
    } as RoomPermalinkCreator;
 | 
						|
    const message = "<i><b>hello</b> world</i>";
 | 
						|
    const mockEvent = mkEvent({
 | 
						|
        type: "m.room.message",
 | 
						|
        room: "myfakeroom",
 | 
						|
        user: "myfakeuser",
 | 
						|
        content: {
 | 
						|
            msgtype: "m.text",
 | 
						|
            body: "Replying to this",
 | 
						|
            format: "org.matrix.custom.html",
 | 
						|
            formatted_body: "Replying to this",
 | 
						|
        },
 | 
						|
        event: true,
 | 
						|
    });
 | 
						|
 | 
						|
    const mockClient = createTestClient();
 | 
						|
    mockClient.setDisplayName = jest.fn().mockResolvedValue({});
 | 
						|
    mockClient.setRoomName = jest.fn().mockResolvedValue({});
 | 
						|
 | 
						|
    const mockRoom = mkStubRoom("myfakeroom", "myfakeroom", mockClient) as any;
 | 
						|
    mockRoom.findEventById = jest.fn((eventId) => {
 | 
						|
        return eventId === mockEvent.getId() ? mockEvent : null;
 | 
						|
    });
 | 
						|
 | 
						|
    const defaultRoomContext: IRoomState = getRoomContext(mockRoom, {});
 | 
						|
 | 
						|
    const spyDispatcher = jest.spyOn(defaultDispatcher, "dispatch");
 | 
						|
 | 
						|
    beforeEach(() => {
 | 
						|
        jest.spyOn(MatrixClientPeg, "get").mockReturnValue(mockClient);
 | 
						|
        jest.clearAllMocks();
 | 
						|
    });
 | 
						|
 | 
						|
    afterAll(() => {
 | 
						|
        jest.restoreAllMocks();
 | 
						|
    });
 | 
						|
 | 
						|
    describe("sendMessage", () => {
 | 
						|
        it("Should not send empty html message", async () => {
 | 
						|
            // When
 | 
						|
            await sendMessage("", true, { roomContext: defaultRoomContext, mxClient: mockClient, permalinkCreator });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledTimes(0);
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledTimes(0);
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should not send message when there is no roomId", async () => {
 | 
						|
            // When
 | 
						|
            const mockRoomWithoutId = mkStubRoom("", "room without id", mockClient) as any;
 | 
						|
            const mockRoomContextWithoutId: IRoomState = getRoomContext(mockRoomWithoutId, {});
 | 
						|
 | 
						|
            await sendMessage(message, true, {
 | 
						|
                roomContext: mockRoomContextWithoutId,
 | 
						|
                mxClient: mockClient,
 | 
						|
                permalinkCreator,
 | 
						|
            });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledTimes(0);
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledTimes(0);
 | 
						|
        });
 | 
						|
 | 
						|
        describe("calls client.sendMessage with", () => {
 | 
						|
            it("a null argument if SendMessageParams is missing relation", async () => {
 | 
						|
                // When
 | 
						|
                await sendMessage(message, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(mockClient.sendMessage).toHaveBeenCalledWith(expect.anything(), null, expect.anything());
 | 
						|
            });
 | 
						|
            it("a null argument if SendMessageParams has relation but relation is missing event_id", async () => {
 | 
						|
                // When
 | 
						|
                await sendMessage(message, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                    relation: {},
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(mockClient.sendMessage).toHaveBeenCalledWith(expect.anything(), null, expect.anything());
 | 
						|
            });
 | 
						|
            it("a null argument if SendMessageParams has relation but rel_type does not match THREAD_RELATION_TYPE.name", async () => {
 | 
						|
                // When
 | 
						|
                await sendMessage(message, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                    relation: {
 | 
						|
                        event_id: "valid_id",
 | 
						|
                        rel_type: "m.does_not_match",
 | 
						|
                    },
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(mockClient.sendMessage).toHaveBeenCalledWith(expect.anything(), null, expect.anything());
 | 
						|
            });
 | 
						|
 | 
						|
            it("the event_id if SendMessageParams has relation and rel_type matches THREAD_RELATION_TYPE.name", async () => {
 | 
						|
                // When
 | 
						|
                await sendMessage(message, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                    relation: {
 | 
						|
                        event_id: "valid_id",
 | 
						|
                        rel_type: "m.thread",
 | 
						|
                    },
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(mockClient.sendMessage).toHaveBeenCalledWith(expect.anything(), "valid_id", expect.anything());
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should send html message", async () => {
 | 
						|
            // When
 | 
						|
            await sendMessage(message, true, {
 | 
						|
                roomContext: defaultRoomContext,
 | 
						|
                mxClient: mockClient,
 | 
						|
                permalinkCreator,
 | 
						|
            });
 | 
						|
 | 
						|
            // Then
 | 
						|
            const expectedContent = {
 | 
						|
                body: "*__hello__ world*",
 | 
						|
                format: "org.matrix.custom.html",
 | 
						|
                formatted_body: "<i><b>hello</b> world</i>",
 | 
						|
                msgtype: "m.text",
 | 
						|
            };
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledWith("myfakeroom", null, expectedContent);
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledWith({ action: "message_sent" });
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should send reply to html message", async () => {
 | 
						|
            const mockReplyEvent = mkEvent({
 | 
						|
                type: "m.room.message",
 | 
						|
                room: "myfakeroom",
 | 
						|
                user: "myfakeuser2",
 | 
						|
                content: { msgtype: "m.text", body: "My reply" },
 | 
						|
                event: true,
 | 
						|
            });
 | 
						|
 | 
						|
            // When
 | 
						|
            await sendMessage(message, true, {
 | 
						|
                roomContext: defaultRoomContext,
 | 
						|
                mxClient: mockClient,
 | 
						|
                permalinkCreator,
 | 
						|
                replyToEvent: mockReplyEvent,
 | 
						|
            });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledWith({
 | 
						|
                action: "reply_to_event",
 | 
						|
                event: null,
 | 
						|
                context: defaultRoomContext.timelineRenderingType,
 | 
						|
            });
 | 
						|
 | 
						|
            const expectedContent = {
 | 
						|
                "body": "> <myfakeuser2> My reply\n\n*__hello__ world*",
 | 
						|
                "format": "org.matrix.custom.html",
 | 
						|
                "formatted_body":
 | 
						|
                    '<mx-reply><blockquote><a href="$$permalink$$">In reply to</a>' +
 | 
						|
                    ' <a href="https://matrix.to/#/myfakeuser2">myfakeuser2</a>' +
 | 
						|
                    "<br>My reply</blockquote></mx-reply><i><b>hello</b> world</i>",
 | 
						|
                "msgtype": "m.text",
 | 
						|
                "m.relates_to": {
 | 
						|
                    "m.in_reply_to": {
 | 
						|
                        event_id: mockReplyEvent.getId(),
 | 
						|
                    },
 | 
						|
                },
 | 
						|
            };
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledWith("myfakeroom", null, expectedContent);
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should scroll to bottom after sending a html message", async () => {
 | 
						|
            // When
 | 
						|
            SettingsStore.setValue("scrollToBottomOnMessageSent", null, SettingLevel.DEVICE, true);
 | 
						|
            await sendMessage(message, true, {
 | 
						|
                roomContext: defaultRoomContext,
 | 
						|
                mxClient: mockClient,
 | 
						|
                permalinkCreator,
 | 
						|
            });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledWith({
 | 
						|
                action: "scroll_to_bottom",
 | 
						|
                timelineRenderingType: defaultRoomContext.timelineRenderingType,
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should handle emojis", async () => {
 | 
						|
            // When
 | 
						|
            await sendMessage("🎉", false, { roomContext: defaultRoomContext, mxClient: mockClient, permalinkCreator });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledWith({ action: "effects.confetti" });
 | 
						|
        });
 | 
						|
 | 
						|
        describe("slash commands", () => {
 | 
						|
            const getCommandSpy = jest.spyOn(SlashCommands, "getCommand");
 | 
						|
 | 
						|
            it("calls getCommand for a message starting with a valid command", async () => {
 | 
						|
                // When
 | 
						|
                const validCommand = "/spoiler";
 | 
						|
                await sendMessage(validCommand, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(getCommandSpy).toHaveBeenCalledWith(validCommand);
 | 
						|
            });
 | 
						|
 | 
						|
            it("does not call getCommand for valid command with invalid prefix", async () => {
 | 
						|
                // When
 | 
						|
                const invalidPrefixCommand = "//spoiler";
 | 
						|
                await sendMessage(invalidPrefixCommand, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(getCommandSpy).toHaveBeenCalledTimes(0);
 | 
						|
            });
 | 
						|
 | 
						|
            it("returns undefined when the command is not successful", async () => {
 | 
						|
                // When
 | 
						|
                const validCommand = "/spoiler";
 | 
						|
                jest.spyOn(Commands, "runSlashCommand").mockResolvedValueOnce([{ content: "mock content" }, false]);
 | 
						|
 | 
						|
                const result = await sendMessage(validCommand, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                });
 | 
						|
 | 
						|
                // Then
 | 
						|
                expect(result).toBeUndefined();
 | 
						|
            });
 | 
						|
 | 
						|
            // /spoiler is a .messages category command, /fireworks is an .effect category command
 | 
						|
            const messagesAndEffectCategoryTestCases = ["/spoiler text", "/fireworks"];
 | 
						|
            it.each(messagesAndEffectCategoryTestCases)(
 | 
						|
                "does not add relations for a .messages or .effects category command if there is no relation to add",
 | 
						|
                async (inputText) => {
 | 
						|
                    await sendMessage(inputText, true, {
 | 
						|
                        roomContext: defaultRoomContext,
 | 
						|
                        mxClient: mockClient,
 | 
						|
                        permalinkCreator,
 | 
						|
                    });
 | 
						|
                    expect(mockClient.sendMessage).toHaveBeenCalledWith(
 | 
						|
                        "myfakeroom",
 | 
						|
                        null,
 | 
						|
                        expect.not.objectContaining({ "m.relates_to": expect.any }),
 | 
						|
                    );
 | 
						|
                },
 | 
						|
            );
 | 
						|
 | 
						|
            it.each(messagesAndEffectCategoryTestCases)(
 | 
						|
                "adds relations for a .messages or .effects category command if there is a relation",
 | 
						|
                async (inputText) => {
 | 
						|
                    const mockRelation: IEventRelation = {
 | 
						|
                        rel_type: "mock relation type",
 | 
						|
                    };
 | 
						|
                    await sendMessage(inputText, true, {
 | 
						|
                        roomContext: defaultRoomContext,
 | 
						|
                        mxClient: mockClient,
 | 
						|
                        permalinkCreator,
 | 
						|
                        relation: mockRelation,
 | 
						|
                    });
 | 
						|
 | 
						|
                    expect(mockClient.sendMessage).toHaveBeenCalledWith(
 | 
						|
                        "myfakeroom",
 | 
						|
                        null,
 | 
						|
                        expect.objectContaining({ "m.relates_to": expect.objectContaining(mockRelation) }),
 | 
						|
                    );
 | 
						|
                },
 | 
						|
            );
 | 
						|
 | 
						|
            it("calls addReplyToMessageContent when there is an event to reply to", async () => {
 | 
						|
                const addReplySpy = jest.spyOn(Reply, "addReplyToMessageContent");
 | 
						|
                await sendMessage("input", true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                    replyToEvent: mockEvent,
 | 
						|
                });
 | 
						|
 | 
						|
                expect(addReplySpy).toHaveBeenCalledTimes(1);
 | 
						|
            });
 | 
						|
 | 
						|
            // these test cases are .action and .admin categories
 | 
						|
            const otherCategoryTestCases = ["/nick new_nickname", "/roomname new_room_name"];
 | 
						|
            it.each(otherCategoryTestCases)(
 | 
						|
                "returns undefined when the command category is not .messages or .effects",
 | 
						|
                async (input) => {
 | 
						|
                    const result = await sendMessage(input, true, {
 | 
						|
                        roomContext: defaultRoomContext,
 | 
						|
                        mxClient: mockClient,
 | 
						|
                        permalinkCreator,
 | 
						|
                        replyToEvent: mockEvent,
 | 
						|
                    });
 | 
						|
 | 
						|
                    expect(result).toBeUndefined();
 | 
						|
                },
 | 
						|
            );
 | 
						|
 | 
						|
            it("if user enters invalid command and then sends it anyway", async () => {
 | 
						|
                // mock out returning a true value for `shouldSendAnyway` to avoid rendering the modal
 | 
						|
                jest.spyOn(Commands, "shouldSendAnyway").mockResolvedValueOnce(true);
 | 
						|
                const invalidCommandInput = "/badCommand";
 | 
						|
 | 
						|
                await sendMessage(invalidCommandInput, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                });
 | 
						|
 | 
						|
                // we expect the message to have been sent
 | 
						|
                // and a composer focus action to have been dispatched
 | 
						|
                expect(mockClient.sendMessage).toHaveBeenCalledWith(
 | 
						|
                    "myfakeroom",
 | 
						|
                    null,
 | 
						|
                    expect.objectContaining({ body: invalidCommandInput }),
 | 
						|
                );
 | 
						|
                expect(spyDispatcher).toHaveBeenCalledWith(expect.objectContaining({ action: Action.FocusAComposer }));
 | 
						|
            });
 | 
						|
 | 
						|
            it("if user enters invalid command and then does not send, return undefined", async () => {
 | 
						|
                // mock out returning a false value for `shouldSendAnyway` to avoid rendering the modal
 | 
						|
                jest.spyOn(Commands, "shouldSendAnyway").mockResolvedValueOnce(false);
 | 
						|
                const invalidCommandInput = "/badCommand";
 | 
						|
 | 
						|
                const result = await sendMessage(invalidCommandInput, true, {
 | 
						|
                    roomContext: defaultRoomContext,
 | 
						|
                    mxClient: mockClient,
 | 
						|
                    permalinkCreator,
 | 
						|
                });
 | 
						|
 | 
						|
                expect(result).toBeUndefined();
 | 
						|
            });
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe("editMessage", () => {
 | 
						|
        const editorStateTransfer = new EditorStateTransfer(mockEvent);
 | 
						|
 | 
						|
        it("Should cancel editing and ask for event removal when message is empty", async () => {
 | 
						|
            // When
 | 
						|
            const mockCreateRedactEventDialog = jest.spyOn(ConfirmRedactDialog, "createRedactEventDialog");
 | 
						|
 | 
						|
            const mockEvent = mkEvent({
 | 
						|
                type: "m.room.message",
 | 
						|
                room: "myfakeroom",
 | 
						|
                user: "myfakeuser",
 | 
						|
                content: { msgtype: "m.text", body: "Replying to this" },
 | 
						|
                event: true,
 | 
						|
            });
 | 
						|
            const replacingEvent = mkEvent({
 | 
						|
                type: "m.room.message",
 | 
						|
                room: "myfakeroom",
 | 
						|
                user: "myfakeuser",
 | 
						|
                content: { msgtype: "m.text", body: "ReplacingEvent" },
 | 
						|
                event: true,
 | 
						|
            });
 | 
						|
            replacingEvent.setStatus(EventStatus.QUEUED);
 | 
						|
            mockEvent.makeReplaced(replacingEvent);
 | 
						|
            const editorStateTransfer = new EditorStateTransfer(mockEvent);
 | 
						|
 | 
						|
            await editMessage("", { roomContext: defaultRoomContext, mxClient: mockClient, editorStateTransfer });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledTimes(0);
 | 
						|
            expect(mockClient.cancelPendingEvent).toHaveBeenCalledTimes(1);
 | 
						|
            expect(mockCreateRedactEventDialog).toHaveBeenCalledTimes(1);
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledTimes(0);
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should do nothing if the content is unmodified", async () => {
 | 
						|
            // When
 | 
						|
            await editMessage(mockEvent.getContent().body, {
 | 
						|
                roomContext: defaultRoomContext,
 | 
						|
                mxClient: mockClient,
 | 
						|
                editorStateTransfer,
 | 
						|
            });
 | 
						|
 | 
						|
            // Then
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledTimes(0);
 | 
						|
        });
 | 
						|
 | 
						|
        it("Should send a message when the content is modified", async () => {
 | 
						|
            // When
 | 
						|
            const newMessage = `${mockEvent.getContent().body} new content`;
 | 
						|
            await editMessage(newMessage, {
 | 
						|
                roomContext: defaultRoomContext,
 | 
						|
                mxClient: mockClient,
 | 
						|
                editorStateTransfer,
 | 
						|
            });
 | 
						|
 | 
						|
            // Then
 | 
						|
            const { msgtype, format } = mockEvent.getContent();
 | 
						|
            const expectedContent = {
 | 
						|
                "body": ` * ${newMessage}`,
 | 
						|
                "formatted_body": ` * ${newMessage}`,
 | 
						|
                "m.new_content": {
 | 
						|
                    body: "Replying to this new content",
 | 
						|
                    format: "org.matrix.custom.html",
 | 
						|
                    formatted_body: "Replying to this new content",
 | 
						|
                    msgtype: "m.text",
 | 
						|
                },
 | 
						|
                "m.relates_to": {
 | 
						|
                    event_id: mockEvent.getId(),
 | 
						|
                    rel_type: "m.replace",
 | 
						|
                },
 | 
						|
                msgtype,
 | 
						|
                format,
 | 
						|
            };
 | 
						|
            expect(mockClient.sendMessage).toHaveBeenCalledWith(mockEvent.getRoomId(), null, expectedContent);
 | 
						|
            expect(spyDispatcher).toHaveBeenCalledWith({ action: "message_sent" });
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 |