400 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			400 lines
		
	
	
		
			15 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 * as React from "react";
 | 
						|
// eslint-disable-next-line deprecate/import
 | 
						|
import { mount, ReactWrapper } from "enzyme";
 | 
						|
import { MatrixEvent, MsgType, RoomMember } from "matrix-js-sdk/src/matrix";
 | 
						|
import { THREAD_RELATION_TYPE } from "matrix-js-sdk/src/models/thread";
 | 
						|
 | 
						|
import { createTestClient, mkEvent, mkStubRoom, stubClient } from "../../../test-utils";
 | 
						|
import MessageComposer, {
 | 
						|
    MessageComposer as MessageComposerClass,
 | 
						|
} from "../../../../src/components/views/rooms/MessageComposer";
 | 
						|
import MatrixClientContext from "../../../../src/contexts/MatrixClientContext";
 | 
						|
import { MatrixClientPeg } from "../../../../src/MatrixClientPeg";
 | 
						|
import RoomContext from "../../../../src/contexts/RoomContext";
 | 
						|
import { IRoomState } from "../../../../src/components/structures/RoomView";
 | 
						|
import ResizeNotifier from "../../../../src/utils/ResizeNotifier";
 | 
						|
import { RoomPermalinkCreator } from "../../../../src/utils/permalinks/Permalinks";
 | 
						|
import { LocalRoom } from "../../../../src/models/LocalRoom";
 | 
						|
import MessageComposerButtons from "../../../../src/components/views/rooms/MessageComposerButtons";
 | 
						|
import { Features } from "../../../../src/settings/Settings";
 | 
						|
import SettingsStore from "../../../../src/settings/SettingsStore";
 | 
						|
import { SettingLevel } from "../../../../src/settings/SettingLevel";
 | 
						|
import dis from "../../../../src/dispatcher/dispatcher";
 | 
						|
import { Action } from "../../../../src/dispatcher/actions";
 | 
						|
import { SendMessageComposer } from "../../../../src/components/views/rooms/SendMessageComposer";
 | 
						|
import { E2EStatus } from "../../../../src/utils/ShieldUtils";
 | 
						|
import { addTextToComposerEnzyme } from "../../../test-utils/composer";
 | 
						|
import UIStore, { UI_EVENTS } from "../../../../src/stores/UIStore";
 | 
						|
import { SendWysiwygComposer } from "../../../../src/components/views/rooms/wysiwyg_composer";
 | 
						|
 | 
						|
describe("MessageComposer", () => {
 | 
						|
    stubClient();
 | 
						|
    const cli = createTestClient();
 | 
						|
 | 
						|
    describe("for a Room", () => {
 | 
						|
        const room = mkStubRoom("!roomId:server", "Room 1", cli);
 | 
						|
 | 
						|
        it("Renders a SendMessageComposer and MessageComposerButtons by default", () => {
 | 
						|
            const wrapper = wrapAndRender({ room });
 | 
						|
 | 
						|
            expect(wrapper.find("SendMessageComposer")).toHaveLength(1);
 | 
						|
            expect(wrapper.find("MessageComposerButtons")).toHaveLength(1);
 | 
						|
        });
 | 
						|
 | 
						|
        it("Does not render a SendMessageComposer or MessageComposerButtons when user has no permission", () => {
 | 
						|
            const wrapper = wrapAndRender({ room }, false);
 | 
						|
 | 
						|
            expect(wrapper.find("SendMessageComposer")).toHaveLength(0);
 | 
						|
            expect(wrapper.find("MessageComposerButtons")).toHaveLength(0);
 | 
						|
            expect(wrapper.find(".mx_MessageComposer_noperm_error")).toHaveLength(1);
 | 
						|
        });
 | 
						|
 | 
						|
        it("Does not render a SendMessageComposer or MessageComposerButtons when room is tombstoned", () => {
 | 
						|
            const wrapper = wrapAndRender(
 | 
						|
                { room },
 | 
						|
                true,
 | 
						|
                false,
 | 
						|
                mkEvent({
 | 
						|
                    event: true,
 | 
						|
                    type: "m.room.tombstone",
 | 
						|
                    room: room.roomId,
 | 
						|
                    user: "@user1:server",
 | 
						|
                    skey: "",
 | 
						|
                    content: {},
 | 
						|
                    ts: Date.now(),
 | 
						|
                }),
 | 
						|
            );
 | 
						|
 | 
						|
            expect(wrapper.find("SendMessageComposer")).toHaveLength(0);
 | 
						|
            expect(wrapper.find("MessageComposerButtons")).toHaveLength(0);
 | 
						|
            expect(wrapper.find(".mx_MessageComposer_roomReplaced_header")).toHaveLength(1);
 | 
						|
        });
 | 
						|
 | 
						|
        describe("when receiving a »reply_to_event«", () => {
 | 
						|
            let wrapper: ReactWrapper;
 | 
						|
            let resizeNotifier: ResizeNotifier;
 | 
						|
 | 
						|
            beforeEach(() => {
 | 
						|
                jest.useFakeTimers();
 | 
						|
                resizeNotifier = {
 | 
						|
                    notifyTimelineHeightChanged: jest.fn(),
 | 
						|
                } as unknown as ResizeNotifier;
 | 
						|
                wrapper = wrapAndRender({
 | 
						|
                    room,
 | 
						|
                    resizeNotifier,
 | 
						|
                });
 | 
						|
            });
 | 
						|
 | 
						|
            it("should call notifyTimelineHeightChanged() for the same context", () => {
 | 
						|
                dis.dispatch({
 | 
						|
                    action: "reply_to_event",
 | 
						|
                    context: (wrapper.instance as unknown as MessageComposerClass).context,
 | 
						|
                });
 | 
						|
                wrapper.update();
 | 
						|
 | 
						|
                jest.advanceTimersByTime(150);
 | 
						|
                expect(resizeNotifier.notifyTimelineHeightChanged).toHaveBeenCalled();
 | 
						|
            });
 | 
						|
 | 
						|
            it("should not call notifyTimelineHeightChanged() for a different context", () => {
 | 
						|
                dis.dispatch({
 | 
						|
                    action: "reply_to_event",
 | 
						|
                    context: "test",
 | 
						|
                });
 | 
						|
                wrapper.update();
 | 
						|
 | 
						|
                jest.advanceTimersByTime(150);
 | 
						|
                expect(resizeNotifier.notifyTimelineHeightChanged).not.toHaveBeenCalled();
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        // test button display depending on settings
 | 
						|
        [
 | 
						|
            {
 | 
						|
                setting: "MessageComposerInput.showStickersButton",
 | 
						|
                prop: "showStickersButton",
 | 
						|
            },
 | 
						|
            {
 | 
						|
                setting: "MessageComposerInput.showPollsButton",
 | 
						|
                prop: "showPollsButton",
 | 
						|
            },
 | 
						|
            {
 | 
						|
                setting: Features.VoiceBroadcast,
 | 
						|
                prop: "showVoiceBroadcastButton",
 | 
						|
            },
 | 
						|
        ].forEach(({ setting, prop }) => {
 | 
						|
            [true, false].forEach((value: boolean) => {
 | 
						|
                describe(`when ${setting} = ${value}`, () => {
 | 
						|
                    let wrapper: ReactWrapper;
 | 
						|
 | 
						|
                    beforeEach(() => {
 | 
						|
                        SettingsStore.setValue(setting, null, SettingLevel.DEVICE, value);
 | 
						|
                        wrapper = wrapAndRender({ room });
 | 
						|
                    });
 | 
						|
 | 
						|
                    it(`should pass the prop ${prop} = ${value}`, () => {
 | 
						|
                        expect(wrapper.find(MessageComposerButtons).props()[prop]).toBe(value);
 | 
						|
                    });
 | 
						|
 | 
						|
                    describe(`and setting ${setting} to ${!value}`, () => {
 | 
						|
                        beforeEach(async () => {
 | 
						|
                            // simulate settings update
 | 
						|
                            await SettingsStore.setValue(setting, null, SettingLevel.DEVICE, !value);
 | 
						|
                            dis.dispatch(
 | 
						|
                                {
 | 
						|
                                    action: Action.SettingUpdated,
 | 
						|
                                    settingName: setting,
 | 
						|
                                    newValue: !value,
 | 
						|
                                },
 | 
						|
                                true,
 | 
						|
                            );
 | 
						|
                            wrapper.update();
 | 
						|
                        });
 | 
						|
 | 
						|
                        it(`should pass the prop ${prop} = ${!value}`, () => {
 | 
						|
                            expect(wrapper.find(MessageComposerButtons).props()[prop]).toBe(!value);
 | 
						|
                        });
 | 
						|
                    });
 | 
						|
                });
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        it("should not render the send button", () => {
 | 
						|
            const wrapper = wrapAndRender({ room });
 | 
						|
            expect(wrapper.find("SendButton")).toHaveLength(0);
 | 
						|
        });
 | 
						|
 | 
						|
        describe("when a message has been entered", () => {
 | 
						|
            let wrapper: ReactWrapper;
 | 
						|
 | 
						|
            beforeEach(() => {
 | 
						|
                wrapper = wrapAndRender({ room });
 | 
						|
                addTextToComposerEnzyme(wrapper, "Hello");
 | 
						|
                wrapper.update();
 | 
						|
            });
 | 
						|
 | 
						|
            it("should render the send button", () => {
 | 
						|
                expect(wrapper.find("SendButton")).toHaveLength(1);
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        describe("UIStore interactions", () => {
 | 
						|
            let wrapper: ReactWrapper;
 | 
						|
            let resizeCallback: Function;
 | 
						|
 | 
						|
            beforeEach(() => {
 | 
						|
                jest.spyOn(UIStore.instance, "on").mockImplementation((_event: string, listener: Function): any => {
 | 
						|
                    resizeCallback = listener;
 | 
						|
                });
 | 
						|
            });
 | 
						|
 | 
						|
            describe("when a non-resize event occurred in UIStore", () => {
 | 
						|
                let stateBefore: any;
 | 
						|
 | 
						|
                beforeEach(() => {
 | 
						|
                    wrapper = wrapAndRender({ room }).children();
 | 
						|
                    stateBefore = { ...wrapper.instance().state };
 | 
						|
                    resizeCallback("test", {});
 | 
						|
                    wrapper.update();
 | 
						|
                });
 | 
						|
 | 
						|
                it("should not change the state", () => {
 | 
						|
                    expect(wrapper.instance().state).toEqual(stateBefore);
 | 
						|
                });
 | 
						|
            });
 | 
						|
 | 
						|
            describe("when a resize to narrow event occurred in UIStore", () => {
 | 
						|
                beforeEach(() => {
 | 
						|
                    wrapper = wrapAndRender({ room }, true, true).children();
 | 
						|
 | 
						|
                    wrapper.setState({
 | 
						|
                        isMenuOpen: true,
 | 
						|
                        isStickerPickerOpen: true,
 | 
						|
                    });
 | 
						|
                    resizeCallback(UI_EVENTS.Resize, {});
 | 
						|
                    wrapper.update();
 | 
						|
                });
 | 
						|
 | 
						|
                it("isMenuOpen should be true", () => {
 | 
						|
                    expect(wrapper.state("isMenuOpen")).toBe(true);
 | 
						|
                });
 | 
						|
 | 
						|
                it("isStickerPickerOpen should be false", () => {
 | 
						|
                    expect(wrapper.state("isStickerPickerOpen")).toBe(false);
 | 
						|
                });
 | 
						|
            });
 | 
						|
 | 
						|
            describe("when a resize to non-narrow event occurred in UIStore", () => {
 | 
						|
                beforeEach(() => {
 | 
						|
                    wrapper = wrapAndRender({ room }, true, false).children();
 | 
						|
                    wrapper.setState({
 | 
						|
                        isMenuOpen: true,
 | 
						|
                        isStickerPickerOpen: true,
 | 
						|
                    });
 | 
						|
                    resizeCallback(UI_EVENTS.Resize, {});
 | 
						|
                    wrapper.update();
 | 
						|
                });
 | 
						|
 | 
						|
                it("isMenuOpen should be false", () => {
 | 
						|
                    expect(wrapper.state("isMenuOpen")).toBe(false);
 | 
						|
                });
 | 
						|
 | 
						|
                it("isStickerPickerOpen should be false", () => {
 | 
						|
                    expect(wrapper.state("isStickerPickerOpen")).toBe(false);
 | 
						|
                });
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        describe("when not replying to an event", () => {
 | 
						|
            it("should pass the expected placeholder to SendMessageComposer", () => {
 | 
						|
                const wrapper = wrapAndRender({ room });
 | 
						|
                expect(wrapper.find(SendMessageComposer).props().placeholder).toBe("Send a message…");
 | 
						|
            });
 | 
						|
 | 
						|
            it("and an e2e status it should pass the expected placeholder to SendMessageComposer", () => {
 | 
						|
                const wrapper = wrapAndRender({
 | 
						|
                    room,
 | 
						|
                    e2eStatus: E2EStatus.Normal,
 | 
						|
                });
 | 
						|
                expect(wrapper.find(SendMessageComposer).props().placeholder).toBe("Send an encrypted message…");
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        describe("when replying to an event", () => {
 | 
						|
            let replyToEvent: MatrixEvent;
 | 
						|
            let props: Partial<React.ComponentProps<typeof MessageComposer>>;
 | 
						|
 | 
						|
            const checkPlaceholder = (expected: string) => {
 | 
						|
                it("should pass the expected placeholder to SendMessageComposer", () => {
 | 
						|
                    const wrapper = wrapAndRender(props);
 | 
						|
                    expect(wrapper.find(SendMessageComposer).props().placeholder).toBe(expected);
 | 
						|
                });
 | 
						|
            };
 | 
						|
 | 
						|
            const setEncrypted = () => {
 | 
						|
                beforeEach(() => {
 | 
						|
                    props.e2eStatus = E2EStatus.Normal;
 | 
						|
                });
 | 
						|
            };
 | 
						|
 | 
						|
            beforeEach(() => {
 | 
						|
                replyToEvent = mkEvent({
 | 
						|
                    event: true,
 | 
						|
                    type: MsgType.Text,
 | 
						|
                    user: cli.getUserId(),
 | 
						|
                    content: {},
 | 
						|
                });
 | 
						|
 | 
						|
                props = {
 | 
						|
                    room,
 | 
						|
                    replyToEvent,
 | 
						|
                };
 | 
						|
            });
 | 
						|
 | 
						|
            describe("without encryption", () => {
 | 
						|
                checkPlaceholder("Send a reply…");
 | 
						|
            });
 | 
						|
 | 
						|
            describe("with encryption", () => {
 | 
						|
                setEncrypted();
 | 
						|
                checkPlaceholder("Send an encrypted reply…");
 | 
						|
            });
 | 
						|
 | 
						|
            describe("with a non-thread relation", () => {
 | 
						|
                beforeEach(() => {
 | 
						|
                    props.relation = { rel_type: "test" };
 | 
						|
                });
 | 
						|
 | 
						|
                checkPlaceholder("Send a reply…");
 | 
						|
            });
 | 
						|
 | 
						|
            describe("that is a thread", () => {
 | 
						|
                beforeEach(() => {
 | 
						|
                    props.relation = { rel_type: THREAD_RELATION_TYPE.name };
 | 
						|
                });
 | 
						|
 | 
						|
                checkPlaceholder("Reply to thread…");
 | 
						|
 | 
						|
                describe("with encryption", () => {
 | 
						|
                    setEncrypted();
 | 
						|
                    checkPlaceholder("Reply to encrypted thread…");
 | 
						|
                });
 | 
						|
            });
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe("for a LocalRoom", () => {
 | 
						|
        const localRoom = new LocalRoom("!room:example.com", cli, cli.getUserId()!);
 | 
						|
 | 
						|
        it("should pass the sticker picker disabled prop", () => {
 | 
						|
            const wrapper = wrapAndRender({ room: localRoom });
 | 
						|
            expect(wrapper.find(MessageComposerButtons).props().showStickersButton).toBe(false);
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    it("should render SendWysiwygComposer", () => {
 | 
						|
        const room = mkStubRoom("!roomId:server", "Room 1", cli);
 | 
						|
 | 
						|
        SettingsStore.setValue("feature_wysiwyg_composer", null, SettingLevel.DEVICE, true);
 | 
						|
        const wrapper = wrapAndRender({ room });
 | 
						|
 | 
						|
        SettingsStore.setValue("feature_wysiwyg_composer", null, SettingLevel.DEVICE, false);
 | 
						|
        expect(wrapper.find(SendWysiwygComposer)).toBeTruthy();
 | 
						|
    });
 | 
						|
});
 | 
						|
 | 
						|
function wrapAndRender(
 | 
						|
    props: Partial<React.ComponentProps<typeof MessageComposer>> = {},
 | 
						|
    canSendMessages = true,
 | 
						|
    narrow = false,
 | 
						|
    tombstone?: MatrixEvent,
 | 
						|
): ReactWrapper {
 | 
						|
    const mockClient = MatrixClientPeg.get();
 | 
						|
    const roomId = "myroomid";
 | 
						|
    const room: any = props.room || {
 | 
						|
        currentState: undefined,
 | 
						|
        roomId,
 | 
						|
        client: mockClient,
 | 
						|
        getMember: function (userId: string): RoomMember {
 | 
						|
            return new RoomMember(roomId, userId);
 | 
						|
        },
 | 
						|
    };
 | 
						|
 | 
						|
    const roomState = {
 | 
						|
        room,
 | 
						|
        canSendMessages,
 | 
						|
        tombstone,
 | 
						|
        narrow,
 | 
						|
    } as unknown as IRoomState;
 | 
						|
 | 
						|
    const defaultProps = {
 | 
						|
        room,
 | 
						|
        resizeNotifier: new ResizeNotifier(),
 | 
						|
        permalinkCreator: new RoomPermalinkCreator(room),
 | 
						|
    };
 | 
						|
 | 
						|
    return mount(
 | 
						|
        <MatrixClientContext.Provider value={mockClient}>
 | 
						|
            <RoomContext.Provider value={roomState}>
 | 
						|
                <MessageComposer {...defaultProps} {...props} />
 | 
						|
            </RoomContext.Provider>
 | 
						|
        </MatrixClientContext.Provider>,
 | 
						|
    );
 | 
						|
}
 |