import { shouldUpdateBootloaderForClassicAndMini } from '../src/api/firmware/bootloaderHelper';

import type { IVersionArray } from '../src';

// Add mock for config module
jest.mock('../src/data/config', () => ({
  getSDKVersion: () => '1.0.0',
  DEFAULT_DOMAIN: 'https://jssdk.onekey.so/1.0.0/',
}));

const fixtures = [
  {
    description: 'version: 2.10.0 -> 2.11.0, boot 1.9.0 -> boot 2.0.0, No need to upgrade',
    currentVersion: '2.10.0',
    bootloaderVersion: '1.9.0',
    willUpdateFirmware: '2.11.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: false,
  },
  {
    description: 'version: 2.10.0 -> 3.0.0, boot 1.9.0 -> boot 2.0.0, Should upgrade bootloader',
    currentVersion: '2.10.0',
    bootloaderVersion: '1.9.0',
    willUpdateFirmware: '3.0.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: true,
  },
  {
    description: 'version: 2.10.0 -> 3.0.0, boot 2.0.0 -> boot 2.0.0, No need to upgrade',
    currentVersion: '2.10.0',
    bootloaderVersion: '2.0.0',
    willUpdateFirmware: '3.0.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: false,
  },
  {
    description: 'version: 3.0.0 -> 3.0.0, boot 2.0.0 -> boot 2.0.0, No need to upgrade',
    currentVersion: '3.0.0',
    bootloaderVersion: '2.0.0',
    willUpdateFirmware: '3.0.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: false,
  },
  {
    description: 'version: 3.0.0 -> 3.1.0, boot 2.0.0 -> boot 2.0.0, No need to upgrade',
    currentVersion: '3.0.0',
    bootloaderVersion: '2.0.0',
    willUpdateFirmware: '3.1.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: false,
  },
  {
    description: 'version: 2.11.0 -> 3.1.0, boot 1.9.0 -> boot 2.0.0, Should upgrade bootloader',
    currentVersion: '2.11.0',
    bootloaderVersion: '2.0.0',
    willUpdateFirmware: '3.1.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: false,
  },
  {
    description: 'version: 2.11.0 -> 3.1.0, boot 2.0.0 -> boot 2.0.0, No need to upgrade',
    currentVersion: '2.11.0',
    bootloaderVersion: '2.0.0',
    willUpdateFirmware: '3.1.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: false,
  },
  {
    description: 'version: 3.1.0 -> 3.2.0, boot 1.9.0 -> boot 2.0.0, Should upgrade bootloader',
    currentVersion: '3.1.0',
    bootloaderVersion: '1.9.0',
    willUpdateFirmware: '3.2.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: true,
  },
  {
    description: 'version: 3.1.0 -> 3.3.0, boot 2.0.0 -> boot 2.0.0, No need to upgrade',
    currentVersion: '3.1.0',
    bootloaderVersion: '1.9.0',
    willUpdateFirmware: '3.2.0',
    targetBootloaderVersion: [2, 0, 0],
    bootloaderRelatedFirmwareVersion: [3, 0, 0],
    shouldUpdateBootloader: true,
  },
];

describe('CheckBootloaderReleast', () => {
  fixtures.forEach(data => {
    test(data.description, () => {
      const {
        currentVersion,
        bootloaderVersion,
        willUpdateFirmware,
        targetBootloaderVersion,
        bootloaderRelatedFirmwareVersion,
      } = data;
      const shouldUpdateBootloader = shouldUpdateBootloaderForClassicAndMini({
        currentVersion,
        bootloaderVersion,
        willUpdateFirmware,
        targetBootloaderVersion: targetBootloaderVersion as IVersionArray,
        bootloaderRelatedFirmwareVersion: bootloaderRelatedFirmwareVersion as IVersionArray,
      });
      expect(shouldUpdateBootloader).toBe(data.shouldUpdateBootloader);
    });
  });
});