503 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			503 lines
		
	
	
		
			18 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 React from "react";
 | |
| // eslint-disable-next-line deprecate/import
 | |
| import { mount, ReactWrapper } from "enzyme";
 | |
| import { mocked } from "jest-mock";
 | |
| import { RoomMember } from "matrix-js-sdk/src/models/room-member";
 | |
| import { MatrixClient } from "matrix-js-sdk/src/client";
 | |
| import { RelationType } from "matrix-js-sdk/src/matrix";
 | |
| import { logger } from "matrix-js-sdk/src/logger";
 | |
| import { M_ASSET, LocationAssetType } from "matrix-js-sdk/src/@types/location";
 | |
| import { act } from "react-dom/test-utils";
 | |
| 
 | |
| import LocationShareMenu from "../../../../src/components/views/location/LocationShareMenu";
 | |
| import MatrixClientContext from "../../../../src/contexts/MatrixClientContext";
 | |
| import { ChevronFace } from "../../../../src/components/structures/ContextMenu";
 | |
| import SettingsStore from "../../../../src/settings/SettingsStore";
 | |
| import { MatrixClientPeg } from "../../../../src/MatrixClientPeg";
 | |
| import { LocationShareType } from "../../../../src/components/views/location/shareLocation";
 | |
| import {
 | |
|     findByTagAndTestId,
 | |
|     findByTestId,
 | |
|     flushPromisesWithFakeTimers,
 | |
|     getMockClientWithEventEmitter,
 | |
|     setupAsyncStoreWithClient,
 | |
| } from "../../../test-utils";
 | |
| import Modal from "../../../../src/Modal";
 | |
| import { DEFAULT_DURATION_MS } from "../../../../src/components/views/location/LiveDurationDropdown";
 | |
| import { OwnBeaconStore } from "../../../../src/stores/OwnBeaconStore";
 | |
| import { SettingLevel } from "../../../../src/settings/SettingLevel";
 | |
| import QuestionDialog from "../../../../src/components/views/dialogs/QuestionDialog";
 | |
| 
 | |
| jest.useFakeTimers();
 | |
| 
 | |
| jest.mock("../../../../src/utils/location/findMapStyleUrl", () => ({
 | |
|     findMapStyleUrl: jest.fn().mockReturnValue("test"),
 | |
| }));
 | |
| 
 | |
| jest.mock("../../../../src/settings/SettingsStore", () => ({
 | |
|     getValue: jest.fn(),
 | |
|     setValue: jest.fn(),
 | |
|     monitorSetting: jest.fn(),
 | |
|     watchSetting: jest.fn(),
 | |
|     unwatchSetting: jest.fn(),
 | |
| }));
 | |
| 
 | |
| jest.mock("../../../../src/stores/OwnProfileStore", () => ({
 | |
|     OwnProfileStore: {
 | |
|         instance: {
 | |
|             displayName: "Ernie",
 | |
|             getHttpAvatarUrl: jest.fn().mockReturnValue("image.com/img"),
 | |
|         },
 | |
|     },
 | |
| }));
 | |
| 
 | |
| jest.mock("../../../../src/Modal", () => ({
 | |
|     createDialog: jest.fn(),
 | |
|     on: jest.fn(),
 | |
|     off: jest.fn(),
 | |
|     ModalManagerEvent: { Opened: "opened" },
 | |
| }));
 | |
| 
 | |
| describe("<LocationShareMenu />", () => {
 | |
|     const userId = "@ernie:server.org";
 | |
|     const mockClient = getMockClientWithEventEmitter({
 | |
|         getUserId: jest.fn().mockReturnValue(userId),
 | |
|         getClientWellKnown: jest.fn().mockResolvedValue({
 | |
|             map_style_url: "maps.com",
 | |
|         }),
 | |
|         sendMessage: jest.fn(),
 | |
|         unstable_createLiveBeacon: jest.fn().mockResolvedValue({ event_id: "1" }),
 | |
|         unstable_setLiveBeacon: jest.fn().mockResolvedValue({ event_id: "1" }),
 | |
|         getVisibleRooms: jest.fn().mockReturnValue([]),
 | |
|     });
 | |
| 
 | |
|     const defaultProps = {
 | |
|         menuPosition: {
 | |
|             top: 1,
 | |
|             left: 1,
 | |
|             chevronFace: ChevronFace.Bottom,
 | |
|         },
 | |
|         onFinished: jest.fn(),
 | |
|         openMenu: jest.fn(),
 | |
|         roomId: "!room:server.org",
 | |
|         sender: new RoomMember("!room:server.org", userId),
 | |
|     };
 | |
| 
 | |
|     const position = {
 | |
|         coords: {
 | |
|             latitude: -36.24484561954707,
 | |
|             longitude: 175.46884959563613,
 | |
|             accuracy: 10,
 | |
|         },
 | |
|         timestamp: 1646305006802,
 | |
|         type: "geolocate",
 | |
|     };
 | |
| 
 | |
|     const makeOwnBeaconStore = async () => {
 | |
|         const store = OwnBeaconStore.instance;
 | |
| 
 | |
|         await setupAsyncStoreWithClient(store, mockClient);
 | |
|         return store;
 | |
|     };
 | |
| 
 | |
|     const getComponent = (props = {}) =>
 | |
|         mount(<LocationShareMenu {...defaultProps} {...props} />, {
 | |
|             wrappingComponent: MatrixClientContext.Provider,
 | |
|             wrappingComponentProps: { value: mockClient },
 | |
|         });
 | |
| 
 | |
|     beforeEach(async () => {
 | |
|         jest.spyOn(logger, "error").mockRestore();
 | |
|         mocked(SettingsStore).getValue.mockReturnValue(false);
 | |
|         mockClient.sendMessage.mockClear();
 | |
|         mockClient.unstable_createLiveBeacon.mockClear().mockResolvedValue({ event_id: "1" });
 | |
|         jest.spyOn(MatrixClientPeg, "get").mockReturnValue(mockClient as unknown as MatrixClient);
 | |
|         mocked(Modal).createDialog.mockClear();
 | |
| 
 | |
|         jest.clearAllMocks();
 | |
| 
 | |
|         await makeOwnBeaconStore();
 | |
|     });
 | |
| 
 | |
|     const getShareTypeOption = (component: ReactWrapper, shareType: LocationShareType) =>
 | |
|         findByTagAndTestId(component, `share-location-option-${shareType}`, "button");
 | |
| 
 | |
|     const getBackButton = (component: ReactWrapper) =>
 | |
|         findByTagAndTestId(component, "share-dialog-buttons-back", "button");
 | |
| 
 | |
|     const getCancelButton = (component: ReactWrapper) =>
 | |
|         findByTagAndTestId(component, "share-dialog-buttons-cancel", "button");
 | |
| 
 | |
|     const getSubmitButton = (component: ReactWrapper) =>
 | |
|         findByTagAndTestId(component, "location-picker-submit-button", "button");
 | |
| 
 | |
|     const setLocation = (component: ReactWrapper) => {
 | |
|         // set the location
 | |
|         const locationPickerInstance = component.find("LocationPicker").instance();
 | |
|         act(() => {
 | |
|             // @ts-ignore
 | |
|             locationPickerInstance.onGeolocate(position);
 | |
|             // make sure button gets enabled
 | |
|             component.setProps({});
 | |
|         });
 | |
|     };
 | |
| 
 | |
|     const setShareType = (component: ReactWrapper, shareType: LocationShareType) =>
 | |
|         act(() => {
 | |
|             getShareTypeOption(component, shareType).at(0).simulate("click");
 | |
|             component.setProps({});
 | |
|         });
 | |
| 
 | |
|     describe("when only Own share type is enabled", () => {
 | |
|         beforeEach(() => enableSettings([]));
 | |
| 
 | |
|         it("renders own and live location options", () => {
 | |
|             const component = getComponent();
 | |
|             expect(getShareTypeOption(component, LocationShareType.Own).length).toBe(1);
 | |
|             expect(getShareTypeOption(component, LocationShareType.Live).length).toBe(1);
 | |
|         });
 | |
| 
 | |
|         it("renders back button from location picker screen", () => {
 | |
|             const component = getComponent();
 | |
|             setShareType(component, LocationShareType.Own);
 | |
| 
 | |
|             expect(getBackButton(component).length).toBe(1);
 | |
|         });
 | |
| 
 | |
|         it("clicking cancel button from location picker closes dialog", () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             act(() => {
 | |
|                 getCancelButton(component).at(0).simulate("click");
 | |
|             });
 | |
| 
 | |
|             expect(onFinished).toHaveBeenCalled();
 | |
|         });
 | |
| 
 | |
|         it("creates static own location share event on submission", () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             setShareType(component, LocationShareType.Own);
 | |
| 
 | |
|             setLocation(component);
 | |
| 
 | |
|             act(() => {
 | |
|                 getSubmitButton(component).at(0).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             expect(onFinished).toHaveBeenCalled();
 | |
|             const [messageRoomId, relation, messageBody] = mockClient.sendMessage.mock.calls[0];
 | |
|             expect(messageRoomId).toEqual(defaultProps.roomId);
 | |
|             expect(relation).toEqual(null);
 | |
|             expect(messageBody).toEqual(
 | |
|                 expect.objectContaining({
 | |
|                     [M_ASSET.name]: {
 | |
|                         type: LocationAssetType.Self,
 | |
|                     },
 | |
|                 }),
 | |
|             );
 | |
|         });
 | |
|     });
 | |
| 
 | |
|     describe("with pin drop share type enabled", () => {
 | |
|         it("renders share type switch with own and pin drop options", () => {
 | |
|             const component = getComponent();
 | |
|             expect(component.find("LocationPicker").length).toBe(0);
 | |
| 
 | |
|             expect(getShareTypeOption(component, LocationShareType.Own).length).toBe(1);
 | |
|             expect(getShareTypeOption(component, LocationShareType.Pin).length).toBe(1);
 | |
|         });
 | |
| 
 | |
|         it("does not render back button on share type screen", () => {
 | |
|             const component = getComponent();
 | |
|             expect(getBackButton(component).length).toBe(0);
 | |
|         });
 | |
| 
 | |
|         it("clicking cancel button from share type screen closes dialog", () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             act(() => {
 | |
|                 getCancelButton(component).at(0).simulate("click");
 | |
|             });
 | |
| 
 | |
|             expect(onFinished).toHaveBeenCalled();
 | |
|         });
 | |
| 
 | |
|         it("selecting own location share type advances to location picker", () => {
 | |
|             const component = getComponent();
 | |
| 
 | |
|             setShareType(component, LocationShareType.Own);
 | |
| 
 | |
|             expect(component.find("LocationPicker").length).toBe(1);
 | |
|         });
 | |
| 
 | |
|         it("clicking back button from location picker screen goes back to share screen", () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             // advance to location picker
 | |
|             setShareType(component, LocationShareType.Own);
 | |
| 
 | |
|             expect(component.find("LocationPicker").length).toBe(1);
 | |
| 
 | |
|             act(() => {
 | |
|                 getBackButton(component).at(0).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             // back to share type
 | |
|             expect(component.find("ShareType").length).toBe(1);
 | |
|         });
 | |
| 
 | |
|         it("creates pin drop location share event on submission", () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             // advance to location picker
 | |
|             setShareType(component, LocationShareType.Pin);
 | |
| 
 | |
|             setLocation(component);
 | |
| 
 | |
|             act(() => {
 | |
|                 getSubmitButton(component).at(0).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             expect(onFinished).toHaveBeenCalled();
 | |
|             const [messageRoomId, relation, messageBody] = mockClient.sendMessage.mock.calls[0];
 | |
|             expect(messageRoomId).toEqual(defaultProps.roomId);
 | |
|             expect(relation).toEqual(null);
 | |
|             expect(messageBody).toEqual(
 | |
|                 expect.objectContaining({
 | |
|                     [M_ASSET.name]: {
 | |
|                         type: LocationAssetType.Pin,
 | |
|                     },
 | |
|                 }),
 | |
|             );
 | |
|         });
 | |
|     });
 | |
| 
 | |
|     describe("with live location disabled", () => {
 | |
|         beforeEach(() => enableSettings([]));
 | |
| 
 | |
|         const getToggle = (component: ReactWrapper) =>
 | |
|             findByTestId(component, "enable-live-share-toggle").find('[role="switch"]').at(0);
 | |
|         const getSubmitEnableButton = (component: ReactWrapper) =>
 | |
|             findByTestId(component, "enable-live-share-submit").at(0);
 | |
| 
 | |
|         it("goes to labs flag screen after live options is clicked", () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             setShareType(component, LocationShareType.Live);
 | |
| 
 | |
|             expect(findByTestId(component, "location-picker-enable-live-share")).toMatchSnapshot();
 | |
|         });
 | |
| 
 | |
|         it("disables OK button when labs flag is not enabled", () => {
 | |
|             const component = getComponent();
 | |
| 
 | |
|             setShareType(component, LocationShareType.Live);
 | |
| 
 | |
|             expect(getSubmitEnableButton(component).props()["disabled"]).toBeTruthy();
 | |
|         });
 | |
| 
 | |
|         it("enables OK button when labs flag is toggled to enabled", () => {
 | |
|             const component = getComponent();
 | |
| 
 | |
|             setShareType(component, LocationShareType.Live);
 | |
| 
 | |
|             act(() => {
 | |
|                 getToggle(component).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
|             expect(getSubmitEnableButton(component).props()["disabled"]).toBeFalsy();
 | |
|         });
 | |
| 
 | |
|         it("enables live share setting on ok button submit", () => {
 | |
|             const component = getComponent();
 | |
| 
 | |
|             setShareType(component, LocationShareType.Live);
 | |
| 
 | |
|             act(() => {
 | |
|                 getToggle(component).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             act(() => {
 | |
|                 getSubmitEnableButton(component).simulate("click");
 | |
|             });
 | |
| 
 | |
|             expect(SettingsStore.setValue).toHaveBeenCalledWith(
 | |
|                 "feature_location_share_live",
 | |
|                 undefined,
 | |
|                 SettingLevel.DEVICE,
 | |
|                 true,
 | |
|             );
 | |
|         });
 | |
| 
 | |
|         it("navigates to location picker when live share is enabled in settings store", () => {
 | |
|             // @ts-ignore
 | |
|             mocked(SettingsStore.watchSetting).mockImplementation((featureName, roomId, callback) => {
 | |
|                 callback(featureName, roomId, SettingLevel.DEVICE, "", "");
 | |
|                 window.setTimeout(() => {
 | |
|                     callback(featureName, roomId, SettingLevel.DEVICE, "", "");
 | |
|                 }, 1000);
 | |
|             });
 | |
|             mocked(SettingsStore.getValue).mockReturnValue(false);
 | |
|             const component = getComponent();
 | |
| 
 | |
|             setShareType(component, LocationShareType.Live);
 | |
| 
 | |
|             // we're on enable live share screen
 | |
|             expect(findByTestId(component, "location-picker-enable-live-share").length).toBeTruthy();
 | |
| 
 | |
|             act(() => {
 | |
|                 mocked(SettingsStore.getValue).mockReturnValue(true);
 | |
|                 // advance so watchSetting will update the value
 | |
|                 jest.runAllTimers();
 | |
|             });
 | |
| 
 | |
|             component.setProps({});
 | |
| 
 | |
|             // advanced to location picker
 | |
|             expect(component.find("LocationPicker").length).toBeTruthy();
 | |
|         });
 | |
|     });
 | |
| 
 | |
|     describe("Live location share", () => {
 | |
|         beforeEach(() => enableSettings(["feature_location_share_live"]));
 | |
| 
 | |
|         it("does not display live location share option when composer has a relation", () => {
 | |
|             const relation = {
 | |
|                 rel_type: RelationType.Thread,
 | |
|                 event_id: "12345",
 | |
|             };
 | |
|             const component = getComponent({ relation });
 | |
| 
 | |
|             expect(getShareTypeOption(component, LocationShareType.Live).length).toBeFalsy();
 | |
|         });
 | |
| 
 | |
|         it("creates beacon info event on submission", async () => {
 | |
|             const onFinished = jest.fn();
 | |
|             const component = getComponent({ onFinished });
 | |
| 
 | |
|             // advance to location picker
 | |
|             setShareType(component, LocationShareType.Live);
 | |
|             setLocation(component);
 | |
| 
 | |
|             act(() => {
 | |
|                 getSubmitButton(component).at(0).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             // flush stopping existing beacons promises
 | |
|             await flushPromisesWithFakeTimers();
 | |
| 
 | |
|             expect(onFinished).toHaveBeenCalled();
 | |
|             const [eventRoomId, eventContent] = mockClient.unstable_createLiveBeacon.mock.calls[0];
 | |
|             expect(eventRoomId).toEqual(defaultProps.roomId);
 | |
|             expect(eventContent).toEqual(
 | |
|                 expect.objectContaining({
 | |
|                     // default timeout
 | |
|                     timeout: DEFAULT_DURATION_MS,
 | |
|                     description: `Ernie's live location`,
 | |
|                     live: true,
 | |
|                     [M_ASSET.name]: {
 | |
|                         type: LocationAssetType.Self,
 | |
|                     },
 | |
|                 }),
 | |
|             );
 | |
|         });
 | |
| 
 | |
|         it("opens error dialog when beacon creation fails", async () => {
 | |
|             // stub logger to keep console clean from expected error
 | |
|             const logSpy = jest.spyOn(logger, "error").mockReturnValue(undefined);
 | |
|             const error = new Error("oh no");
 | |
|             mockClient.unstable_createLiveBeacon.mockRejectedValue(error);
 | |
|             const component = getComponent();
 | |
| 
 | |
|             // advance to location picker
 | |
|             setShareType(component, LocationShareType.Live);
 | |
|             setLocation(component);
 | |
| 
 | |
|             act(() => {
 | |
|                 getSubmitButton(component).at(0).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             await flushPromisesWithFakeTimers();
 | |
|             await flushPromisesWithFakeTimers();
 | |
|             await flushPromisesWithFakeTimers();
 | |
| 
 | |
|             expect(logSpy).toHaveBeenCalledWith("We couldn't start sharing your live location", error);
 | |
|             expect(mocked(Modal).createDialog).toHaveBeenCalledWith(
 | |
|                 QuestionDialog,
 | |
|                 expect.objectContaining({
 | |
|                     button: "Try again",
 | |
|                     description: "Element could not send your location. Please try again later.",
 | |
|                     title: `We couldn't send your location`,
 | |
|                     cancelButton: "Cancel",
 | |
|                 }),
 | |
|             );
 | |
|         });
 | |
| 
 | |
|         it("opens error dialog when beacon creation fails with permission error", async () => {
 | |
|             // stub logger to keep console clean from expected error
 | |
|             const logSpy = jest.spyOn(logger, "error").mockReturnValue(undefined);
 | |
|             const error = { errcode: "M_FORBIDDEN" } as unknown as Error;
 | |
|             mockClient.unstable_createLiveBeacon.mockRejectedValue(error);
 | |
|             const component = getComponent();
 | |
| 
 | |
|             // advance to location picker
 | |
|             setShareType(component, LocationShareType.Live);
 | |
|             setLocation(component);
 | |
| 
 | |
|             act(() => {
 | |
|                 getSubmitButton(component).at(0).simulate("click");
 | |
|                 component.setProps({});
 | |
|             });
 | |
| 
 | |
|             await flushPromisesWithFakeTimers();
 | |
|             await flushPromisesWithFakeTimers();
 | |
|             await flushPromisesWithFakeTimers();
 | |
| 
 | |
|             expect(logSpy).toHaveBeenCalledWith("Insufficient permissions to start sharing your live location", error);
 | |
|             expect(mocked(Modal).createDialog).toHaveBeenCalledWith(
 | |
|                 QuestionDialog,
 | |
|                 expect.objectContaining({
 | |
|                     button: "OK",
 | |
|                     description: "You need to have the right permissions in order to share locations in this room.",
 | |
|                     title: `You don't have permission to share locations`,
 | |
|                     hasCancelButton: false,
 | |
|                 }),
 | |
|             );
 | |
|         });
 | |
|     });
 | |
| });
 | |
| 
 | |
| function enableSettings(settings: string[]) {
 | |
|     mocked(SettingsStore).getValue.mockReturnValue(false);
 | |
|     mocked(SettingsStore).getValue.mockImplementation((settingName: string): any => settings.includes(settingName));
 | |
| }
 |