trezor-suite/packages/transport-test/e2e/api/utils.ts

import { error } from './shared';

export const MAGIC = '3f232300';

export const MESSAGES = {
    PING: '01',
    SUCCESS: '02',
    INITIALIZE: '00',
    FEATURES: '11',
} as const;

export const buildMessage = (message: keyof typeof MESSAGES) =>
    Buffer.from(MAGIC + MESSAGES[message], 'hex');

export const assertMessage = (message: Buffer, expected: keyof typeof MESSAGES) => {
    const assertedChunk = message.toString('hex').substring(0, MAGIC.length + 2);
    if (assertedChunk !== `${MAGIC}${MESSAGES[expected]}`) {
        const unexpected = Object.entries(MESSAGES).find(
            ([, v]) => v === assertedChunk.replace(MAGIC, ''),
        )?.[0];
        throw new Error(
            error(`Expected message ${expected} but got ${unexpected || assertedChunk}`),
        );
    }
};
export function assertSuccess(result: any): asserts result is { success: true; payload: any } {
    if (!result.success) {
        throw new Error(error(`${result.error}${result.message ? `: ${result.message}` : ''}`));
    }
}
export function assertFailure(result: any): asserts result is { success: false; error: any } {
    if (result.success) {
        throw new Error(error(`Unexpected success`));
    }
}
export const assertEquals = (a: any, b: any) => {
    const strA = JSON.stringify(a);
    const strB = JSON.stringify(b);
    if (strA !== strB) {
        throw new Error(error(`Expected ${strA} to be equal to ${strB}`));
    }
};