417 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			417 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2024 New Vector Ltd.
 | 
						|
Copyright 2022 The Matrix.org Foundation C.I.C.
 | 
						|
 | 
						|
SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only OR LicenseRef-Element-Commercial
 | 
						|
Please see LICENSE files in the repository root for full details.
 | 
						|
*/
 | 
						|
 | 
						|
import React, { ComponentProps } from "react";
 | 
						|
import { act, fireEvent, render } from "jest-matrix-react";
 | 
						|
import * as maplibregl from "maplibre-gl";
 | 
						|
import {
 | 
						|
    BeaconEvent,
 | 
						|
    getBeaconInfoIdentifier,
 | 
						|
    RelationType,
 | 
						|
    MatrixEvent,
 | 
						|
    EventType,
 | 
						|
    Relations,
 | 
						|
    M_BEACON,
 | 
						|
    Room,
 | 
						|
} from "matrix-js-sdk/src/matrix";
 | 
						|
 | 
						|
import MBeaconBody from "../../../../../src/components/views/messages/MBeaconBody";
 | 
						|
import {
 | 
						|
    getMockClientWithEventEmitter,
 | 
						|
    makeBeaconEvent,
 | 
						|
    makeBeaconInfoEvent,
 | 
						|
    makeRoomWithBeacons,
 | 
						|
    makeRoomWithStateEvents,
 | 
						|
} from "../../../../test-utils";
 | 
						|
import { RoomPermalinkCreator } from "../../../../../src/utils/permalinks/Permalinks";
 | 
						|
import { MediaEventHelper } from "../../../../../src/utils/MediaEventHelper";
 | 
						|
import MatrixClientContext from "../../../../../src/contexts/MatrixClientContext";
 | 
						|
import Modal from "../../../../../src/Modal";
 | 
						|
import { TILE_SERVER_WK_KEY } from "../../../../../src/utils/WellKnownUtils";
 | 
						|
import * as mapUtilHooks from "../../../../../src/utils/location/useMap";
 | 
						|
import { LocationShareError } from "../../../../../src/utils/location";
 | 
						|
 | 
						|
describe("<MBeaconBody />", () => {
 | 
						|
    // 14.03.2022 16:15
 | 
						|
    const now = 1647270879403;
 | 
						|
    // stable date for snapshots
 | 
						|
    jest.spyOn(global.Date, "now").mockReturnValue(now);
 | 
						|
    const roomId = "!room:server";
 | 
						|
    const aliceId = "@alice:server";
 | 
						|
 | 
						|
    const mapOptions = { container: {} as unknown as HTMLElement, style: "" };
 | 
						|
    const mockMap = new maplibregl.Map(mapOptions);
 | 
						|
    const mockMarker = new maplibregl.Marker();
 | 
						|
 | 
						|
    const mockClient = getMockClientWithEventEmitter({
 | 
						|
        getClientWellKnown: jest.fn().mockReturnValue({
 | 
						|
            [TILE_SERVER_WK_KEY.name]: { map_style_url: "maps.com" },
 | 
						|
        }),
 | 
						|
        getUserId: jest.fn().mockReturnValue(aliceId),
 | 
						|
        getRoom: jest.fn(),
 | 
						|
        redactEvent: jest.fn(),
 | 
						|
    });
 | 
						|
 | 
						|
    const defaultEvent = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-1");
 | 
						|
 | 
						|
    const defaultProps: ComponentProps<typeof MBeaconBody> = {
 | 
						|
        mxEvent: defaultEvent,
 | 
						|
        highlights: [],
 | 
						|
        highlightLink: "",
 | 
						|
        onHeightChanged: jest.fn(),
 | 
						|
        onMessageAllowed: jest.fn(),
 | 
						|
        // we dont use these and they pollute the snapshots
 | 
						|
        permalinkCreator: {} as unknown as RoomPermalinkCreator,
 | 
						|
        mediaEventHelper: {} as unknown as MediaEventHelper,
 | 
						|
    };
 | 
						|
 | 
						|
    const getComponent = (props = {}) =>
 | 
						|
        render(
 | 
						|
            <MatrixClientContext.Provider value={mockClient}>
 | 
						|
                <MBeaconBody {...defaultProps} {...props} />
 | 
						|
            </MatrixClientContext.Provider>,
 | 
						|
        );
 | 
						|
 | 
						|
    const modalSpy = jest.spyOn(Modal, "createDialog").mockReturnValue({
 | 
						|
        finished: Promise.resolve([true]),
 | 
						|
        close: () => {},
 | 
						|
    });
 | 
						|
 | 
						|
    beforeEach(() => {
 | 
						|
        jest.clearAllMocks();
 | 
						|
    });
 | 
						|
 | 
						|
    const testBeaconStatuses = () => {
 | 
						|
        it("renders stopped beacon UI for an explicitly stopped beacon", () => {
 | 
						|
            const beaconInfoEvent = makeBeaconInfoEvent(aliceId, roomId, { isLive: false }, "$alice-room1-1");
 | 
						|
            makeRoomWithStateEvents([beaconInfoEvent], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
            expect(component.container).toHaveTextContent("Live location ended");
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders stopped beacon UI for an expired beacon", () => {
 | 
						|
            const beaconInfoEvent = makeBeaconInfoEvent(
 | 
						|
                aliceId,
 | 
						|
                roomId,
 | 
						|
                // puts this beacons live period in the past
 | 
						|
                { isLive: true, timestamp: now - 600000, timeout: 500 },
 | 
						|
                "$alice-room1-1",
 | 
						|
            );
 | 
						|
            makeRoomWithStateEvents([beaconInfoEvent], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
            expect(component.container).toHaveTextContent("Live location ended");
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders loading beacon UI for a beacon that has not started yet", () => {
 | 
						|
            const beaconInfoEvent = makeBeaconInfoEvent(
 | 
						|
                aliceId,
 | 
						|
                roomId,
 | 
						|
                // puts this beacons start timestamp in the future
 | 
						|
                { isLive: true, timestamp: now + 60000, timeout: 500 },
 | 
						|
                "$alice-room1-1",
 | 
						|
            );
 | 
						|
            makeRoomWithStateEvents([beaconInfoEvent], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
            expect(component.container).toHaveTextContent("Loading live location…");
 | 
						|
        });
 | 
						|
 | 
						|
        it("does not open maximised map when on click when beacon is stopped", () => {
 | 
						|
            const beaconInfoEvent = makeBeaconInfoEvent(
 | 
						|
                aliceId,
 | 
						|
                roomId,
 | 
						|
                // puts this beacons live period in the past
 | 
						|
                { isLive: true, timestamp: now - 600000, timeout: 500 },
 | 
						|
                "$alice-room1-1",
 | 
						|
            );
 | 
						|
            makeRoomWithStateEvents([beaconInfoEvent], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
            fireEvent.click(component.container.querySelector(".mx_MBeaconBody_map")!);
 | 
						|
 | 
						|
            expect(modalSpy).not.toHaveBeenCalled();
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders stopped UI when a beacon event is not the latest beacon for a user", () => {
 | 
						|
            const aliceBeaconInfo1 = makeBeaconInfoEvent(
 | 
						|
                aliceId,
 | 
						|
                roomId,
 | 
						|
                // this one is a little older
 | 
						|
                { isLive: true, timestamp: now - 500 },
 | 
						|
                "$alice-room1-1",
 | 
						|
            );
 | 
						|
            aliceBeaconInfo1.event.origin_server_ts = now - 500;
 | 
						|
            const aliceBeaconInfo2 = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-2");
 | 
						|
 | 
						|
            makeRoomWithStateEvents([aliceBeaconInfo1, aliceBeaconInfo2], { roomId, mockClient });
 | 
						|
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo1 });
 | 
						|
            // beacon1 has been superceded by beacon2
 | 
						|
            expect(component.container).toHaveTextContent("Live location ended");
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders stopped UI when a beacon event is replaced", () => {
 | 
						|
            const aliceBeaconInfo1 = makeBeaconInfoEvent(
 | 
						|
                aliceId,
 | 
						|
                roomId,
 | 
						|
                // this one is a little older
 | 
						|
                { isLive: true, timestamp: now - 500 },
 | 
						|
                "$alice-room1-1",
 | 
						|
            );
 | 
						|
            aliceBeaconInfo1.event.origin_server_ts = now - 500;
 | 
						|
            const aliceBeaconInfo2 = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-2");
 | 
						|
 | 
						|
            const room = makeRoomWithStateEvents([aliceBeaconInfo1], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo1 });
 | 
						|
 | 
						|
            const beaconInstance = room.currentState.beacons.get(getBeaconInfoIdentifier(aliceBeaconInfo1))!;
 | 
						|
            // update alice's beacon with a new edition
 | 
						|
            // beacon instance emits
 | 
						|
            act(() => {
 | 
						|
                beaconInstance.update(aliceBeaconInfo2);
 | 
						|
            });
 | 
						|
 | 
						|
            // beacon1 has been superceded by beacon2
 | 
						|
            expect(component.container).toHaveTextContent("Live location ended");
 | 
						|
        });
 | 
						|
    };
 | 
						|
 | 
						|
    testBeaconStatuses();
 | 
						|
 | 
						|
    describe("on liveness change", () => {
 | 
						|
        it("renders stopped UI when a beacon stops being live", () => {
 | 
						|
            const aliceBeaconInfo = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-1");
 | 
						|
 | 
						|
            const room = makeRoomWithStateEvents([aliceBeaconInfo], { roomId, mockClient });
 | 
						|
            const beaconInstance = room.currentState.beacons.get(getBeaconInfoIdentifier(aliceBeaconInfo))!;
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo });
 | 
						|
 | 
						|
            act(() => {
 | 
						|
                // @ts-ignore cheat to force beacon to not live
 | 
						|
                beaconInstance._isLive = false;
 | 
						|
                beaconInstance.emit(BeaconEvent.LivenessChange, false, beaconInstance);
 | 
						|
            });
 | 
						|
 | 
						|
            // stopped UI
 | 
						|
            expect(component.container).toHaveTextContent("Live location ended");
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe("latestLocationState", () => {
 | 
						|
        const aliceBeaconInfo = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-1");
 | 
						|
 | 
						|
        const location1 = makeBeaconEvent(aliceId, {
 | 
						|
            beaconInfoId: aliceBeaconInfo.getId(),
 | 
						|
            geoUri: "geo:51,41",
 | 
						|
            timestamp: now + 1,
 | 
						|
        });
 | 
						|
        const location2 = makeBeaconEvent(aliceId, {
 | 
						|
            beaconInfoId: aliceBeaconInfo.getId(),
 | 
						|
            geoUri: "geo:52,42",
 | 
						|
            timestamp: now + 10000,
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders a live beacon without a location correctly", () => {
 | 
						|
            makeRoomWithStateEvents([aliceBeaconInfo], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo });
 | 
						|
 | 
						|
            expect(component.container).toHaveTextContent("Loading live location…");
 | 
						|
        });
 | 
						|
 | 
						|
        it("does nothing on click when a beacon has no location", () => {
 | 
						|
            makeRoomWithStateEvents([aliceBeaconInfo], { roomId, mockClient });
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo });
 | 
						|
 | 
						|
            fireEvent.click(component.container.querySelector(".mx_MBeaconBody_map")!);
 | 
						|
 | 
						|
            expect(modalSpy).not.toHaveBeenCalled();
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders a live beacon with a location correctly", () => {
 | 
						|
            const room = makeRoomWithStateEvents([aliceBeaconInfo], { roomId, mockClient });
 | 
						|
            const beaconInstance = room.currentState.beacons.get(getBeaconInfoIdentifier(aliceBeaconInfo))!;
 | 
						|
            beaconInstance.addLocations([location1]);
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo });
 | 
						|
 | 
						|
            expect(component.container.querySelector(".maplibregl-canvas-container")).toBeDefined();
 | 
						|
        });
 | 
						|
 | 
						|
        it("opens maximised map view on click when beacon has a live location", () => {
 | 
						|
            const room = makeRoomWithStateEvents([aliceBeaconInfo], { roomId, mockClient });
 | 
						|
            const beaconInstance = room.currentState.beacons.get(getBeaconInfoIdentifier(aliceBeaconInfo))!;
 | 
						|
            beaconInstance.addLocations([location1]);
 | 
						|
            const component = getComponent({ mxEvent: aliceBeaconInfo });
 | 
						|
 | 
						|
            fireEvent.click(component.container.querySelector(".mx_Map")!);
 | 
						|
 | 
						|
            // opens modal
 | 
						|
            expect(modalSpy).toHaveBeenCalled();
 | 
						|
        });
 | 
						|
 | 
						|
        it("updates latest location", () => {
 | 
						|
            const room = makeRoomWithStateEvents([aliceBeaconInfo], { roomId, mockClient });
 | 
						|
            getComponent({ mxEvent: aliceBeaconInfo });
 | 
						|
 | 
						|
            const beaconInstance = room.currentState.beacons.get(getBeaconInfoIdentifier(aliceBeaconInfo))!;
 | 
						|
            act(() => {
 | 
						|
                beaconInstance.addLocations([location1]);
 | 
						|
            });
 | 
						|
 | 
						|
            expect(mockMap.setCenter).toHaveBeenCalledWith({ lat: 51, lon: 41 });
 | 
						|
            expect(mockMarker.setLngLat).toHaveBeenCalledWith({ lat: 51, lon: 41 });
 | 
						|
 | 
						|
            act(() => {
 | 
						|
                beaconInstance.addLocations([location2]);
 | 
						|
            });
 | 
						|
 | 
						|
            expect(mockMap.setCenter).toHaveBeenCalledWith({ lat: 52, lon: 42 });
 | 
						|
            expect(mockMarker.setLngLat).toHaveBeenCalledWith({ lat: 52, lon: 42 });
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe("redaction", () => {
 | 
						|
        const makeEvents = (): {
 | 
						|
            beaconInfoEvent: MatrixEvent;
 | 
						|
            location1: MatrixEvent;
 | 
						|
            location2: MatrixEvent;
 | 
						|
        } => {
 | 
						|
            const beaconInfoEvent = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-1");
 | 
						|
 | 
						|
            const location1 = makeBeaconEvent(
 | 
						|
                aliceId,
 | 
						|
                { beaconInfoId: beaconInfoEvent.getId(), geoUri: "geo:51,41", timestamp: now + 1 },
 | 
						|
                roomId,
 | 
						|
            );
 | 
						|
            location1.event.event_id = "1";
 | 
						|
            const location2 = makeBeaconEvent(
 | 
						|
                aliceId,
 | 
						|
                { beaconInfoId: beaconInfoEvent.getId(), geoUri: "geo:52,42", timestamp: now + 10000 },
 | 
						|
                roomId,
 | 
						|
            );
 | 
						|
            location2.event.event_id = "2";
 | 
						|
            return { beaconInfoEvent, location1, location2 };
 | 
						|
        };
 | 
						|
 | 
						|
        const redactionEvent = new MatrixEvent({ type: EventType.RoomRedaction, content: { reason: "test reason" } });
 | 
						|
 | 
						|
        const setupRoomWithBeacon = (beaconInfoEvent: MatrixEvent, locationEvents: MatrixEvent[] = []): Room => {
 | 
						|
            const room = makeRoomWithStateEvents([beaconInfoEvent], { roomId, mockClient });
 | 
						|
            const beaconInstance = room.currentState.beacons.get(getBeaconInfoIdentifier(beaconInfoEvent))!;
 | 
						|
            beaconInstance.addLocations(locationEvents);
 | 
						|
            return room;
 | 
						|
        };
 | 
						|
        const mockGetRelationsForEvent = (locationEvents: MatrixEvent[] = []) => {
 | 
						|
            const relations = new Relations(RelationType.Reference, M_BEACON.name, mockClient);
 | 
						|
            jest.spyOn(relations, "getRelations").mockReturnValue(locationEvents);
 | 
						|
 | 
						|
            const getRelationsForEvent = jest.fn().mockReturnValue(relations);
 | 
						|
 | 
						|
            return getRelationsForEvent;
 | 
						|
        };
 | 
						|
 | 
						|
        it("does nothing when getRelationsForEvent is falsy", () => {
 | 
						|
            const { beaconInfoEvent, location1, location2 } = makeEvents();
 | 
						|
            const room = setupRoomWithBeacon(beaconInfoEvent, [location1, location2]);
 | 
						|
 | 
						|
            getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
 | 
						|
            act(() => {
 | 
						|
                beaconInfoEvent.makeRedacted(redactionEvent, room);
 | 
						|
            });
 | 
						|
 | 
						|
            // no error, no redactions
 | 
						|
            expect(mockClient.redactEvent).not.toHaveBeenCalled();
 | 
						|
        });
 | 
						|
 | 
						|
        it("cleans up redaction listener on unmount", () => {
 | 
						|
            const { beaconInfoEvent, location1, location2 } = makeEvents();
 | 
						|
            setupRoomWithBeacon(beaconInfoEvent, [location1, location2]);
 | 
						|
            const removeListenerSpy = jest.spyOn(beaconInfoEvent, "removeListener");
 | 
						|
 | 
						|
            const component = getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
 | 
						|
            act(() => {
 | 
						|
                component.unmount();
 | 
						|
            });
 | 
						|
 | 
						|
            expect(removeListenerSpy).toHaveBeenCalled();
 | 
						|
        });
 | 
						|
 | 
						|
        it("does nothing when beacon has no related locations", async () => {
 | 
						|
            const { beaconInfoEvent } = makeEvents();
 | 
						|
            // no locations
 | 
						|
            const room = setupRoomWithBeacon(beaconInfoEvent, []);
 | 
						|
            const getRelationsForEvent = await mockGetRelationsForEvent();
 | 
						|
 | 
						|
            getComponent({ mxEvent: beaconInfoEvent, getRelationsForEvent });
 | 
						|
 | 
						|
            act(() => {
 | 
						|
                beaconInfoEvent.makeRedacted(redactionEvent, room);
 | 
						|
            });
 | 
						|
 | 
						|
            expect(getRelationsForEvent).toHaveBeenCalledWith(
 | 
						|
                beaconInfoEvent.getId(),
 | 
						|
                RelationType.Reference,
 | 
						|
                M_BEACON.name,
 | 
						|
            );
 | 
						|
            expect(mockClient.redactEvent).not.toHaveBeenCalled();
 | 
						|
        });
 | 
						|
 | 
						|
        it("redacts related locations on beacon redaction", async () => {
 | 
						|
            const { beaconInfoEvent, location1, location2 } = makeEvents();
 | 
						|
            const room = setupRoomWithBeacon(beaconInfoEvent, [location1, location2]);
 | 
						|
 | 
						|
            const getRelationsForEvent = await mockGetRelationsForEvent([location1, location2]);
 | 
						|
 | 
						|
            getComponent({ mxEvent: beaconInfoEvent, getRelationsForEvent });
 | 
						|
 | 
						|
            act(() => {
 | 
						|
                beaconInfoEvent.makeRedacted(redactionEvent, room);
 | 
						|
            });
 | 
						|
 | 
						|
            expect(getRelationsForEvent).toHaveBeenCalledWith(
 | 
						|
                beaconInfoEvent.getId(),
 | 
						|
                RelationType.Reference,
 | 
						|
                M_BEACON.name,
 | 
						|
            );
 | 
						|
            expect(mockClient.redactEvent).toHaveBeenCalledTimes(2);
 | 
						|
            expect(mockClient.redactEvent).toHaveBeenCalledWith(roomId, location1.getId(), undefined, {
 | 
						|
                reason: "test reason",
 | 
						|
            });
 | 
						|
            expect(mockClient.redactEvent).toHaveBeenCalledWith(roomId, location2.getId(), undefined, {
 | 
						|
                reason: "test reason",
 | 
						|
            });
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    describe("when map display is not configured", () => {
 | 
						|
        beforeEach(() => {
 | 
						|
            // mock map utils to raise MapStyleUrlNotConfigured error
 | 
						|
            jest.spyOn(mapUtilHooks, "useMap").mockImplementation(({ onError }) => {
 | 
						|
                onError?.(new Error(LocationShareError.MapStyleUrlNotConfigured));
 | 
						|
                return mockMap;
 | 
						|
            });
 | 
						|
        });
 | 
						|
 | 
						|
        it("renders maps unavailable error for a live beacon with location", () => {
 | 
						|
            const beaconInfoEvent = makeBeaconInfoEvent(aliceId, roomId, { isLive: true }, "$alice-room1-1");
 | 
						|
            const location1 = makeBeaconEvent(aliceId, {
 | 
						|
                beaconInfoId: beaconInfoEvent.getId(),
 | 
						|
                geoUri: "geo:51,41",
 | 
						|
                timestamp: now + 1,
 | 
						|
            });
 | 
						|
 | 
						|
            makeRoomWithBeacons(roomId, mockClient, [beaconInfoEvent], [location1]);
 | 
						|
 | 
						|
            const component = getComponent({ mxEvent: beaconInfoEvent });
 | 
						|
            expect(component.getByTestId("map-rendering-error")).toMatchSnapshot();
 | 
						|
        });
 | 
						|
 | 
						|
        // test that statuses display as expected with a map display error
 | 
						|
        testBeaconStatuses();
 | 
						|
    });
 | 
						|
});
 |