@ledgerhq/live-common
Version:
Common ground for the Ledger Live apps
58 lines • 2.63 kB
JavaScript
import customLockScreenLoad from "./customLockScreenLoad";
import { DeviceModelId } from "@ledgerhq/devices";
import { lastValueFrom, of } from "rxjs";
import { ManagerNotEnoughSpaceError, StatusCodes, TransportError } from "@ledgerhq/errors";
import { ImageLoadRefusedOnDevice } from "../errors";
const mockTransport = {
send: jest.fn(),
getTraceContext: jest.fn(),
};
jest.mock("../deviceSDK/transports/core", () => ({
withTransport: () => callback => callback({ transportRef: { current: mockTransport } }),
}));
jest.mock("./getDeviceInfo", () => jest.fn(() => of([])));
describe("customLockScreenLoad", () => {
it("should load image on device", async () => {
// given
const request = {
deviceModelId: DeviceModelId.stax,
hexImage: "hello_world",
};
mockTransport.send.mockResolvedValue(Buffer.from([0x42, 0x42, 0x43, 0x90, 0x00]));
// when
const ret = await lastValueFrom(await customLockScreenLoad({ deviceId: "deviceId", request }));
// then
expect(mockTransport.send).toHaveBeenNthCalledWith(1, 0xe0, 0x60, 0x00, 0x00, Buffer.from([0x00, 0x00, 0x00, 0x08]), [StatusCodes.NOT_ENOUGH_SPACE, StatusCodes.USER_REFUSED_ON_DEVICE, StatusCodes.OK]);
expect(mockTransport.send).toHaveBeenNthCalledWith(2, 0xe0, 0x61, 0x00, 0x00, Buffer.from([0x00, 0x00, 0x00, 0x00, 0x90, 0x01, 0xa0, 0x02, 0x21, 0x00, 0x00, 0x00]));
expect(mockTransport.send).toHaveBeenNthCalledWith(3, 0xe0, 0x62, 0x00, 0x00, Buffer.from([]), [0x9000, 0x5501]);
expect(ret).toStrictEqual({
type: "imageLoaded",
imageSize: 1111638928,
imageHash: "424243",
});
});
it.each([
[
"user refused on device",
[0x55, 0x01],
new ImageLoadRefusedOnDevice("5501", { productName: "Ledger Stax" }),
],
["not enough space", [0x51, 0x02], new ManagerNotEnoughSpaceError()],
["unexpected error", [0x42, 0x32], new TransportError("Unexpected device response", "4232")],
])("should return an error if %s", async (_errorStr, statusCode, error) => {
// given
const request = {
deviceModelId: DeviceModelId.stax,
hexImage: "hello_world",
};
mockTransport.send.mockResolvedValue(Buffer.from(statusCode));
// when
try {
await lastValueFrom(await customLockScreenLoad({ deviceId: "nanoX", request }));
}
catch (err) {
expect(err).toStrictEqual(error);
}
});
});
//# sourceMappingURL=customLockScreenLoad.test.js.map