/* 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'; import { act } from 'react-dom/test-utils'; import { mount } from 'enzyme'; import { Room, Beacon, BeaconEvent } from 'matrix-js-sdk/src/matrix'; import { logger } from 'matrix-js-sdk/src/logger'; import RoomLiveShareWarning from '../../../../src/components/views/beacon/RoomLiveShareWarning'; import { OwnBeaconStore, OwnBeaconStoreEvent } from '../../../../src/stores/OwnBeaconStore'; import { advanceDateAndTime, findByTestId, getMockClientWithEventEmitter, makeBeaconInfoEvent, mockGeolocation, resetAsyncStoreWithClient, setupAsyncStoreWithClient, } from '../../../test-utils'; jest.useFakeTimers(); describe('', () => { const aliceId = '@alice:server.org'; const room1Id = '$room1:server.org'; const room2Id = '$room2:server.org'; const room3Id = '$room3:server.org'; const mockClient = getMockClientWithEventEmitter({ getVisibleRooms: jest.fn().mockReturnValue([]), getUserId: jest.fn().mockReturnValue(aliceId), unstable_setLiveBeacon: jest.fn().mockResolvedValue({ event_id: '1' }), sendEvent: jest.fn(), }); // 14.03.2022 16:15 const now = 1647270879403; const MINUTE_MS = 60000; const HOUR_MS = 3600000; // mock the date so events are stable for snapshots etc jest.spyOn(global.Date, 'now').mockReturnValue(now); const room1Beacon1 = makeBeaconInfoEvent(aliceId, room1Id, { isLive: true, timeout: HOUR_MS, }, '$0'); const room2Beacon1 = makeBeaconInfoEvent(aliceId, room2Id, { isLive: true, timeout: HOUR_MS }, '$1'); const room2Beacon2 = makeBeaconInfoEvent(aliceId, room2Id, { isLive: true, timeout: HOUR_MS * 12 }, '$2'); const room3Beacon1 = makeBeaconInfoEvent(aliceId, room3Id, { isLive: true, timeout: HOUR_MS }, '$3'); // make fresh rooms every time // as we update room state const makeRoomsWithStateEvents = (stateEvents = []): [Room, Room] => { const room1 = new Room(room1Id, mockClient, aliceId); const room2 = new Room(room2Id, mockClient, aliceId); room1.currentState.setStateEvents(stateEvents); room2.currentState.setStateEvents(stateEvents); mockClient.getVisibleRooms.mockReturnValue([room1, room2]); return [room1, room2]; }; const makeOwnBeaconStore = async () => { const store = OwnBeaconStore.instance; await setupAsyncStoreWithClient(store, mockClient); return store; }; const defaultProps = { roomId: room1Id, }; const getComponent = (props = {}) => { let component; // component updates on render // wrap in act act(() => { component = mount(); }); return component; }; beforeEach(() => { mockGeolocation(); jest.spyOn(global.Date, 'now').mockReturnValue(now); mockClient.unstable_setLiveBeacon.mockClear(); }); afterEach(async () => { await resetAsyncStoreWithClient(OwnBeaconStore.instance); }); afterAll(() => { jest.spyOn(global.Date, 'now').mockRestore(); }); const getExpiryText = wrapper => findByTestId(wrapper, 'room-live-share-expiry').text(); it('renders nothing when user has no live beacons at all', async () => { await makeOwnBeaconStore(); const component = getComponent(); expect(component.html()).toBe(null); }); it('renders nothing when user has no live beacons in room', async () => { await act(async () => { await makeRoomsWithStateEvents([room2Beacon1]); await makeOwnBeaconStore(); }); const component = getComponent({ roomId: room1Id }); expect(component.html()).toBe(null); }); it('does not render when geolocation is not working', async () => { jest.spyOn(logger, 'error').mockImplementation(() => { }); // @ts-ignore navigator.geolocation = undefined; await act(async () => { await makeRoomsWithStateEvents([room1Beacon1, room2Beacon1, room2Beacon2]); await makeOwnBeaconStore(); }); const component = getComponent({ roomId: room1Id }); // beacons have generated ids that break snapshots // assert on html expect(component.html()).toBeNull(); }); describe('when user has live beacons and geolocation is available', () => { beforeEach(async () => { await act(async () => { await makeRoomsWithStateEvents([room1Beacon1, room2Beacon1, room2Beacon2]); await makeOwnBeaconStore(); }); }); it('renders correctly with one live beacon in room', () => { const component = getComponent({ roomId: room1Id }); // beacons have generated ids that break snapshots // assert on html expect(component.html()).toMatchSnapshot(); }); it('renders correctly with two live beacons in room', () => { const component = getComponent({ roomId: room2Id }); // beacons have generated ids that break snapshots // assert on html expect(component.html()).toMatchSnapshot(); // later expiry displayed expect(getExpiryText(component)).toEqual('12h left'); }); it('removes itself when user stops having live beacons', async () => { const component = getComponent({ roomId: room1Id }); // started out rendered expect(component.html()).toBeTruthy(); // time travel until room1Beacon1 is expired act(() => { advanceDateAndTime(HOUR_MS + 1); }); act(() => { mockClient.emit(BeaconEvent.LivenessChange, false, new Beacon(room1Beacon1)); component.setProps({}); }); expect(component.html()).toBe(null); }); it('removes itself when user stops monitoring live position', async () => { const component = getComponent({ roomId: room1Id }); // started out rendered expect(component.html()).toBeTruthy(); act(() => { // cheat to clear this // @ts-ignore OwnBeaconStore.instance.clearPositionWatch = undefined; OwnBeaconStore.instance.emit(OwnBeaconStoreEvent.MonitoringLivePosition); component.setProps({}); }); expect(component.html()).toBe(null); }); it('renders when user adds a live beacon', async () => { const component = getComponent({ roomId: room3Id }); // started out not rendered expect(component.html()).toBeFalsy(); act(() => { mockClient.emit(BeaconEvent.New, room3Beacon1, new Beacon(room3Beacon1)); component.setProps({}); }); expect(component.html()).toBeTruthy(); }); it('updates beacon time left periodically', () => { const component = getComponent({ roomId: room1Id }); expect(getExpiryText(component)).toEqual('1h left'); act(() => { advanceDateAndTime(MINUTE_MS * 25); }); expect(getExpiryText(component)).toEqual('35m left'); }); it('clears expiry time interval on unmount', () => { const clearIntervalSpy = jest.spyOn(global, 'clearInterval'); const component = getComponent({ roomId: room1Id }); expect(getExpiryText(component)).toEqual('1h left'); act(() => { component.unmount(); }); expect(clearIntervalSpy).toHaveBeenCalled(); }); describe('stopping beacons', () => { it('stops beacon on stop sharing click', () => { const component = getComponent({ roomId: room2Id }); act(() => { findByTestId(component, 'room-live-share-stop-sharing').at(0).simulate('click'); component.setProps({}); }); expect(mockClient.unstable_setLiveBeacon).toHaveBeenCalledTimes(2); expect(component.find('Spinner').length).toBeTruthy(); expect(findByTestId(component, 'room-live-share-stop-sharing').at(0).props().disabled).toBeTruthy(); }); it('displays again with correct state after stopping a beacon', () => { // make sure the loading state is reset correctly after removing a beacon const component = getComponent({ roomId: room1Id }); // stop the beacon act(() => { findByTestId(component, 'room-live-share-stop-sharing').at(0).simulate('click'); }); // time travel until room1Beacon1 is expired act(() => { advanceDateAndTime(HOUR_MS + 1); }); act(() => { mockClient.emit(BeaconEvent.LivenessChange, false, new Beacon(room1Beacon1)); }); const newLiveBeacon = makeBeaconInfoEvent(aliceId, room1Id, { isLive: true }); act(() => { mockClient.emit(BeaconEvent.New, newLiveBeacon, new Beacon(newLiveBeacon)); }); // button not disabled and expiry time shown expect(findByTestId(component, 'room-live-share-stop-sharing').at(0).props().disabled).toBeFalsy(); expect(findByTestId(component, 'room-live-share-expiry').text()).toEqual('1h left'); }); }); }); });