237 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			237 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2017 - 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 { Room } from 'matrix-js-sdk/src/matrix';
 | 
						|
 | 
						|
import { RoomViewStore } from '../../src/stores/RoomViewStore';
 | 
						|
import { Action } from '../../src/dispatcher/actions';
 | 
						|
import { getMockClientWithEventEmitter, untilDispatch, untilEmission } from '../test-utils';
 | 
						|
import SettingsStore from '../../src/settings/SettingsStore';
 | 
						|
import { SlidingSyncManager } from '../../src/SlidingSyncManager';
 | 
						|
import { TimelineRenderingType } from '../../src/contexts/RoomContext';
 | 
						|
import { MatrixDispatcher } from '../../src/dispatcher/dispatcher';
 | 
						|
import { UPDATE_EVENT } from '../../src/stores/AsyncStore';
 | 
						|
import { ActiveRoomChangedPayload } from '../../src/dispatcher/payloads/ActiveRoomChangedPayload';
 | 
						|
 | 
						|
jest.mock('../../src/utils/DMRoomMap', () => {
 | 
						|
    const mock = {
 | 
						|
        getUserIdForRoomId: jest.fn(),
 | 
						|
        getDMRoomsForUserId: jest.fn(),
 | 
						|
    };
 | 
						|
 | 
						|
    return {
 | 
						|
        shared: jest.fn().mockReturnValue(mock),
 | 
						|
        sharedInstance: mock,
 | 
						|
    };
 | 
						|
});
 | 
						|
 | 
						|
describe('RoomViewStore', function() {
 | 
						|
    const userId = '@alice:server';
 | 
						|
    const roomId = "!randomcharacters:aser.ver";
 | 
						|
    // we need to change the alias to ensure cache misses as the cache exists
 | 
						|
    // through all tests.
 | 
						|
    let alias = "#somealias2:aser.ver";
 | 
						|
    const mockClient = getMockClientWithEventEmitter({
 | 
						|
        joinRoom: jest.fn(),
 | 
						|
        getRoom: jest.fn(),
 | 
						|
        getRoomIdForAlias: jest.fn(),
 | 
						|
        isGuest: jest.fn(),
 | 
						|
    });
 | 
						|
    const room = new Room(roomId, mockClient, userId);
 | 
						|
    let dis: MatrixDispatcher;
 | 
						|
 | 
						|
    beforeEach(function() {
 | 
						|
        jest.clearAllMocks();
 | 
						|
        mockClient.credentials = { userId: userId };
 | 
						|
        mockClient.joinRoom.mockResolvedValue(room);
 | 
						|
        mockClient.getRoom.mockReturnValue(room);
 | 
						|
        mockClient.isGuest.mockReturnValue(false);
 | 
						|
 | 
						|
        // Reset the state of the store
 | 
						|
        dis = new MatrixDispatcher();
 | 
						|
        RoomViewStore.instance.reset();
 | 
						|
        RoomViewStore.instance.resetDispatcher(dis);
 | 
						|
    });
 | 
						|
 | 
						|
    it('can be used to view a room by ID and join', async () => {
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        dis.dispatch({ action: Action.JoinRoom });
 | 
						|
        await untilDispatch(Action.JoinRoomReady, dis);
 | 
						|
        expect(mockClient.joinRoom).toHaveBeenCalledWith(roomId, { viaServers: [] });
 | 
						|
        expect(RoomViewStore.instance.isJoining()).toBe(true);
 | 
						|
    });
 | 
						|
 | 
						|
    it('can auto-join a room', async () => {
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId, auto_join: true });
 | 
						|
        await untilDispatch(Action.JoinRoomReady, dis);
 | 
						|
        expect(mockClient.joinRoom).toHaveBeenCalledWith(roomId, { viaServers: [] });
 | 
						|
        expect(RoomViewStore.instance.isJoining()).toBe(true);
 | 
						|
    });
 | 
						|
 | 
						|
    it('emits ActiveRoomChanged when the viewed room changes', async () => {
 | 
						|
        const roomId2 = "!roomid:2";
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        let payload = await untilDispatch(Action.ActiveRoomChanged, dis) as ActiveRoomChangedPayload;
 | 
						|
        expect(payload.newRoomId).toEqual(roomId);
 | 
						|
        expect(payload.oldRoomId).toEqual(null);
 | 
						|
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId2 });
 | 
						|
        payload = await untilDispatch(Action.ActiveRoomChanged, dis) as ActiveRoomChangedPayload;
 | 
						|
        expect(payload.newRoomId).toEqual(roomId2);
 | 
						|
        expect(payload.oldRoomId).toEqual(roomId);
 | 
						|
    });
 | 
						|
 | 
						|
    it('invokes room activity listeners when the viewed room changes', async () => {
 | 
						|
        const roomId2 = "!roomid:2";
 | 
						|
        const callback = jest.fn();
 | 
						|
        RoomViewStore.instance.addRoomListener(roomId, callback);
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        await untilDispatch(Action.ActiveRoomChanged, dis) as ActiveRoomChangedPayload;
 | 
						|
        expect(callback).toHaveBeenCalledWith(true);
 | 
						|
        expect(callback).not.toHaveBeenCalledWith(false);
 | 
						|
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId2 });
 | 
						|
        await untilDispatch(Action.ActiveRoomChanged, dis) as ActiveRoomChangedPayload;
 | 
						|
        expect(callback).toHaveBeenCalledWith(false);
 | 
						|
    });
 | 
						|
 | 
						|
    it('can be used to view a room by alias and join', async () => {
 | 
						|
        mockClient.getRoomIdForAlias.mockResolvedValue({ room_id: roomId, servers: [] });
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_alias: alias });
 | 
						|
        await untilDispatch((p) => { // wait for the re-dispatch with the room ID
 | 
						|
            return p.action === Action.ViewRoom && p.room_id === roomId;
 | 
						|
        }, dis);
 | 
						|
 | 
						|
        // roomId is set to id of the room alias
 | 
						|
        expect(RoomViewStore.instance.getRoomId()).toBe(roomId);
 | 
						|
 | 
						|
        // join the room
 | 
						|
        dis.dispatch({ action: Action.JoinRoom }, true);
 | 
						|
 | 
						|
        await untilDispatch(Action.JoinRoomReady, dis);
 | 
						|
 | 
						|
        expect(RoomViewStore.instance.isJoining()).toBeTruthy();
 | 
						|
        expect(mockClient.joinRoom).toHaveBeenCalledWith(alias, { viaServers: [] });
 | 
						|
    });
 | 
						|
 | 
						|
    it('emits ViewRoomError if the alias lookup fails', async () => {
 | 
						|
        alias = "#something-different:to-ensure-cache-miss";
 | 
						|
        mockClient.getRoomIdForAlias.mockRejectedValue(new Error("network error or something"));
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_alias: alias });
 | 
						|
        const payload = await untilDispatch(Action.ViewRoomError, dis);
 | 
						|
        expect(payload.room_id).toBeNull();
 | 
						|
        expect(payload.room_alias).toEqual(alias);
 | 
						|
        expect(RoomViewStore.instance.getRoomAlias()).toEqual(alias);
 | 
						|
    });
 | 
						|
 | 
						|
    it('emits JoinRoomError if joining the room fails', async () => {
 | 
						|
        const joinErr = new Error("network error or something");
 | 
						|
        mockClient.joinRoom.mockRejectedValue(joinErr);
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        dis.dispatch({ action: Action.JoinRoom });
 | 
						|
        await untilDispatch(Action.JoinRoomError, dis);
 | 
						|
        expect(RoomViewStore.instance.isJoining()).toBe(false);
 | 
						|
        expect(RoomViewStore.instance.getJoinError()).toEqual(joinErr);
 | 
						|
    });
 | 
						|
 | 
						|
    it('remembers the event being replied to when swapping rooms', async () => {
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        await untilDispatch(Action.ActiveRoomChanged, dis);
 | 
						|
        const replyToEvent = {
 | 
						|
            getRoomId: () => roomId,
 | 
						|
        };
 | 
						|
        dis.dispatch({ action: 'reply_to_event', event: replyToEvent, context: TimelineRenderingType.Room });
 | 
						|
        await untilEmission(RoomViewStore.instance, UPDATE_EVENT);
 | 
						|
        expect(RoomViewStore.instance.getQuotingEvent()).toEqual(replyToEvent);
 | 
						|
        // view the same room, should remember the event.
 | 
						|
        // set the highlighed flag to make sure there is a state change so we get an update event
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId, highlighted: true });
 | 
						|
        await untilEmission(RoomViewStore.instance, UPDATE_EVENT);
 | 
						|
        expect(RoomViewStore.instance.getQuotingEvent()).toEqual(replyToEvent);
 | 
						|
    });
 | 
						|
 | 
						|
    it('swaps to the replied event room if it is not the current room', async () => {
 | 
						|
        const roomId2 = "!room2:bar";
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        await untilDispatch(Action.ActiveRoomChanged, dis);
 | 
						|
        const replyToEvent = {
 | 
						|
            getRoomId: () => roomId2,
 | 
						|
        };
 | 
						|
        dis.dispatch({ action: 'reply_to_event', event: replyToEvent, context: TimelineRenderingType.Room });
 | 
						|
        await untilDispatch(Action.ViewRoom, dis);
 | 
						|
        expect(RoomViewStore.instance.getQuotingEvent()).toEqual(replyToEvent);
 | 
						|
        expect(RoomViewStore.instance.getRoomId()).toEqual(roomId2);
 | 
						|
    });
 | 
						|
 | 
						|
    it('removes the roomId on ViewHomePage', async () => {
 | 
						|
        dis.dispatch({ action: Action.ViewRoom, room_id: roomId });
 | 
						|
        await untilDispatch(Action.ActiveRoomChanged, dis);
 | 
						|
        expect(RoomViewStore.instance.getRoomId()).toEqual(roomId);
 | 
						|
 | 
						|
        dis.dispatch({ action: Action.ViewHomePage });
 | 
						|
        await untilEmission(RoomViewStore.instance, UPDATE_EVENT);
 | 
						|
        expect(RoomViewStore.instance.getRoomId()).toBeNull();
 | 
						|
    });
 | 
						|
 | 
						|
    describe('Sliding Sync', function() {
 | 
						|
        beforeEach(() => {
 | 
						|
            jest.spyOn(SettingsStore, 'getValue').mockImplementation((settingName, roomId, value) => {
 | 
						|
                return settingName === "feature_sliding_sync"; // this is enabled, everything else is disabled.
 | 
						|
            });
 | 
						|
            RoomViewStore.instance.reset();
 | 
						|
        });
 | 
						|
 | 
						|
        it("subscribes to the room", async () => {
 | 
						|
            const setRoomVisible = jest.spyOn(SlidingSyncManager.instance, "setRoomVisible").mockReturnValue(
 | 
						|
                Promise.resolve(""),
 | 
						|
            );
 | 
						|
            const subscribedRoomId = "!sub1:localhost";
 | 
						|
            dis.dispatch({ action: Action.ViewRoom, room_id: subscribedRoomId });
 | 
						|
            await untilDispatch(Action.ActiveRoomChanged, dis);
 | 
						|
            expect(RoomViewStore.instance.getRoomId()).toBe(subscribedRoomId);
 | 
						|
            expect(setRoomVisible).toHaveBeenCalledWith(subscribedRoomId, true);
 | 
						|
        });
 | 
						|
 | 
						|
        // Regression test for an in-the-wild bug where rooms would rapidly switch forever in sliding sync mode
 | 
						|
        it("doesn't get stuck in a loop if you view rooms quickly", async () => {
 | 
						|
            const setRoomVisible = jest.spyOn(SlidingSyncManager.instance, "setRoomVisible").mockReturnValue(
 | 
						|
                Promise.resolve(""),
 | 
						|
            );
 | 
						|
            const subscribedRoomId = "!sub1:localhost";
 | 
						|
            const subscribedRoomId2 = "!sub2:localhost";
 | 
						|
            dis.dispatch({ action: Action.ViewRoom, room_id: subscribedRoomId }, true);
 | 
						|
            dis.dispatch({ action: Action.ViewRoom, room_id: subscribedRoomId2 }, true);
 | 
						|
            await untilDispatch(Action.ActiveRoomChanged, dis);
 | 
						|
            // sub(1) then unsub(1) sub(2), unsub(1)
 | 
						|
            const wantCalls = [
 | 
						|
                [subscribedRoomId, true],
 | 
						|
                [subscribedRoomId, false],
 | 
						|
                [subscribedRoomId2, true],
 | 
						|
                [subscribedRoomId, false],
 | 
						|
            ];
 | 
						|
            expect(setRoomVisible).toHaveBeenCalledTimes(wantCalls.length);
 | 
						|
            wantCalls.forEach((v, i) => {
 | 
						|
                try {
 | 
						|
                    expect(setRoomVisible.mock.calls[i][0]).toEqual(v[0]);
 | 
						|
                    expect(setRoomVisible.mock.calls[i][1]).toEqual(v[1]);
 | 
						|
                } catch (err) {
 | 
						|
                    throw new Error(`i=${i} got ${setRoomVisible.mock.calls[i]} want ${v}`);
 | 
						|
                }
 | 
						|
            });
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 |