mirror of https://github.com/vector-im/riot-web
				
				
				
			
		
			
				
	
	
		
			111 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			111 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2023 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 { render, RenderResult, screen } from "@testing-library/react";
 | 
						|
import userEvent from "@testing-library/user-event";
 | 
						|
import { mocked, Mocked } from "jest-mock";
 | 
						|
import { CryptoApi, DeviceVerificationStatus, IMyDevice, MatrixClient } from "matrix-js-sdk/src/matrix";
 | 
						|
import { DeviceInfo } from "matrix-js-sdk/src/crypto/deviceinfo";
 | 
						|
 | 
						|
import dis from "../../src/dispatcher/dispatcher";
 | 
						|
import { showToast } from "../../src/toasts/UnverifiedSessionToast";
 | 
						|
import { filterConsole, flushPromises, stubClient } from "../test-utils";
 | 
						|
import ToastContainer from "../../src/components/structures/ToastContainer";
 | 
						|
import { Action } from "../../src/dispatcher/actions";
 | 
						|
import DeviceListener from "../../src/DeviceListener";
 | 
						|
 | 
						|
describe("UnverifiedSessionToast", () => {
 | 
						|
    const otherDevice: IMyDevice = {
 | 
						|
        device_id: "ABC123",
 | 
						|
    };
 | 
						|
    const otherDeviceInfo = new DeviceInfo(otherDevice.device_id);
 | 
						|
    let client: Mocked<MatrixClient>;
 | 
						|
    let renderResult: RenderResult;
 | 
						|
 | 
						|
    filterConsole("Dismissing unverified sessions: ABC123");
 | 
						|
 | 
						|
    beforeAll(() => {
 | 
						|
        client = mocked(stubClient());
 | 
						|
        client.getDevice.mockImplementation(async (deviceId: string) => {
 | 
						|
            if (deviceId === otherDevice.device_id) {
 | 
						|
                return otherDevice;
 | 
						|
            }
 | 
						|
 | 
						|
            throw new Error(`Unknown device ${deviceId}`);
 | 
						|
        });
 | 
						|
        client.getStoredDevice.mockImplementation((userId: string, deviceId: string) => {
 | 
						|
            if (deviceId === otherDevice.device_id) {
 | 
						|
                return otherDeviceInfo;
 | 
						|
            }
 | 
						|
 | 
						|
            return null;
 | 
						|
        });
 | 
						|
        client.getCrypto.mockReturnValue({
 | 
						|
            getDeviceVerificationStatus: jest
 | 
						|
                .fn()
 | 
						|
                .mockResolvedValue(new DeviceVerificationStatus({ crossSigningVerified: true })),
 | 
						|
        } as unknown as CryptoApi);
 | 
						|
        jest.spyOn(dis, "dispatch");
 | 
						|
        jest.spyOn(DeviceListener.sharedInstance(), "dismissUnverifiedSessions");
 | 
						|
    });
 | 
						|
 | 
						|
    beforeEach(() => {
 | 
						|
        renderResult = render(<ToastContainer />);
 | 
						|
    });
 | 
						|
 | 
						|
    describe("when rendering the toast", () => {
 | 
						|
        beforeEach(async () => {
 | 
						|
            showToast(otherDevice.device_id);
 | 
						|
            await flushPromises();
 | 
						|
        });
 | 
						|
 | 
						|
        const itShouldDismissTheDevice = () => {
 | 
						|
            it("should dismiss the device", () => {
 | 
						|
                expect(DeviceListener.sharedInstance().dismissUnverifiedSessions).toHaveBeenCalledWith([
 | 
						|
                    otherDevice.device_id,
 | 
						|
                ]);
 | 
						|
            });
 | 
						|
        };
 | 
						|
 | 
						|
        it("should render as expected", () => {
 | 
						|
            expect(renderResult.baseElement).toMatchSnapshot();
 | 
						|
        });
 | 
						|
 | 
						|
        describe("and confirming the login", () => {
 | 
						|
            beforeEach(async () => {
 | 
						|
                await userEvent.click(screen.getByRole("button", { name: "Yes, it was me" }));
 | 
						|
            });
 | 
						|
 | 
						|
            itShouldDismissTheDevice();
 | 
						|
        });
 | 
						|
 | 
						|
        describe("and dismissing the login", () => {
 | 
						|
            beforeEach(async () => {
 | 
						|
                await userEvent.click(screen.getByRole("button", { name: "No" }));
 | 
						|
            });
 | 
						|
 | 
						|
            itShouldDismissTheDevice();
 | 
						|
 | 
						|
            it("should show the device settings", () => {
 | 
						|
                expect(dis.dispatch).toHaveBeenCalledWith({
 | 
						|
                    action: Action.ViewUserDeviceSettings,
 | 
						|
                });
 | 
						|
            });
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 |