122 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			122 lines
		
	
	
		
			3.9 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
 | 
						|
Please see LICENSE files in the repository root for full details.
 | 
						|
*/
 | 
						|
 | 
						|
import { MatrixEvent } from "matrix-js-sdk/src/matrix";
 | 
						|
 | 
						|
import BasePlatform from "../../../src/BasePlatform";
 | 
						|
import { IConfigOptions } from "../../../src/IConfigOptions";
 | 
						|
import { getDeviceClientInformation, recordClientInformation } from "../../../src/utils/device/clientInformation";
 | 
						|
import { getMockClientWithEventEmitter } from "../../test-utils";
 | 
						|
import { DEFAULTS } from "../../../src/SdkConfig";
 | 
						|
import { DeepReadonly } from "../../../src/@types/common";
 | 
						|
 | 
						|
describe("recordClientInformation()", () => {
 | 
						|
    const deviceId = "my-device-id";
 | 
						|
    const version = "1.2.3";
 | 
						|
    const isElectron = window.electron;
 | 
						|
 | 
						|
    const mockClient = getMockClientWithEventEmitter({
 | 
						|
        getDeviceId: jest.fn().mockReturnValue(deviceId),
 | 
						|
        setAccountData: jest.fn(),
 | 
						|
    });
 | 
						|
 | 
						|
    const sdkConfig: DeepReadonly<IConfigOptions> = {
 | 
						|
        ...DEFAULTS,
 | 
						|
        brand: "Test Brand",
 | 
						|
        element_call: { url: "", use_exclusively: false, brand: "Element Call" },
 | 
						|
    };
 | 
						|
 | 
						|
    const platform = {
 | 
						|
        getAppVersion: jest.fn().mockResolvedValue(version),
 | 
						|
    } as unknown as BasePlatform;
 | 
						|
 | 
						|
    beforeEach(() => {
 | 
						|
        jest.clearAllMocks();
 | 
						|
        window.electron = false;
 | 
						|
    });
 | 
						|
 | 
						|
    afterAll(() => {
 | 
						|
        // restore global
 | 
						|
        window.electron = isElectron;
 | 
						|
    });
 | 
						|
 | 
						|
    it("saves client information without url for electron clients", async () => {
 | 
						|
        window.electron = true;
 | 
						|
 | 
						|
        await recordClientInformation(mockClient, sdkConfig, platform);
 | 
						|
 | 
						|
        expect(mockClient.setAccountData).toHaveBeenCalledWith(`io.element.matrix_client_information.${deviceId}`, {
 | 
						|
            name: sdkConfig.brand,
 | 
						|
            version,
 | 
						|
            url: undefined,
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    it("saves client information with url for non-electron clients", async () => {
 | 
						|
        await recordClientInformation(mockClient, sdkConfig, platform);
 | 
						|
 | 
						|
        expect(mockClient.setAccountData).toHaveBeenCalledWith(`io.element.matrix_client_information.${deviceId}`, {
 | 
						|
            name: sdkConfig.brand,
 | 
						|
            version,
 | 
						|
            url: "localhost",
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 | 
						|
 | 
						|
describe("getDeviceClientInformation()", () => {
 | 
						|
    const deviceId = "my-device-id";
 | 
						|
 | 
						|
    const mockClient = getMockClientWithEventEmitter({
 | 
						|
        getAccountData: jest.fn(),
 | 
						|
    });
 | 
						|
 | 
						|
    beforeEach(() => {
 | 
						|
        jest.resetAllMocks();
 | 
						|
    });
 | 
						|
 | 
						|
    it("returns an empty object when no event exists for the device", () => {
 | 
						|
        expect(getDeviceClientInformation(mockClient, deviceId)).toEqual({});
 | 
						|
 | 
						|
        expect(mockClient.getAccountData).toHaveBeenCalledWith(`io.element.matrix_client_information.${deviceId}`);
 | 
						|
    });
 | 
						|
 | 
						|
    it("returns client information for the device", () => {
 | 
						|
        const eventContent = {
 | 
						|
            name: "Element Web",
 | 
						|
            version: "1.2.3",
 | 
						|
            url: "test.com",
 | 
						|
        };
 | 
						|
        const event = new MatrixEvent({
 | 
						|
            type: `io.element.matrix_client_information.${deviceId}`,
 | 
						|
            content: eventContent,
 | 
						|
        });
 | 
						|
        mockClient.getAccountData.mockReturnValue(event);
 | 
						|
        expect(getDeviceClientInformation(mockClient, deviceId)).toEqual(eventContent);
 | 
						|
    });
 | 
						|
 | 
						|
    it("excludes values with incorrect types", () => {
 | 
						|
        const eventContent = {
 | 
						|
            extraField: "hello",
 | 
						|
            name: "Element Web",
 | 
						|
            // wrong format
 | 
						|
            version: { value: "1.2.3" },
 | 
						|
            url: "test.com",
 | 
						|
        };
 | 
						|
        const event = new MatrixEvent({
 | 
						|
            type: `io.element.matrix_client_information.${deviceId}`,
 | 
						|
            content: eventContent,
 | 
						|
        });
 | 
						|
        mockClient.getAccountData.mockReturnValue(event);
 | 
						|
        // invalid fields excluded
 | 
						|
        expect(getDeviceClientInformation(mockClient, deviceId)).toEqual({
 | 
						|
            name: eventContent.name,
 | 
						|
            url: eventContent.url,
 | 
						|
        });
 | 
						|
    });
 | 
						|
});
 |