import { mock, MockProxy } from "jest-mock-extended";

// eslint-disable-next-line import/no-restricted-paths -- Needed to print log messages
import { LogService } from "../platform/abstractions/log.service";
// eslint-disable-next-line import/no-restricted-paths -- Needed to interface with storage locations
import { AbstractStorageService } from "../platform/abstractions/storage.service";

import { CURRENT_VERSION, currentVersion, migrate } from "./migrate";
import { MigrationBuilder } from "./migration-builder";

jest.mock("./migration-builder", () => {
  return {
    MigrationBuilder: {
      create: jest.fn().mockReturnThis(),
    },
  };
});

describe("migrate", () => {
  it("should not run migrations if state is empty", async () => {
    const storage = mock<AbstractStorageService>();
    const logService = mock<LogService>();
    storage.get.mockReturnValueOnce(null);
    await migrate(storage, logService);
    expect(MigrationBuilder.create).not.toHaveBeenCalled();
  });

  it("should set to current version if state is empty", async () => {
    const storage = mock<AbstractStorageService>();
    const logService = mock<LogService>();
    storage.get.mockReturnValueOnce(null);
    await migrate(storage, logService);
    expect(storage.save).toHaveBeenCalledWith("stateVersion", CURRENT_VERSION);
  });
});

describe("currentVersion", () => {
  let storage: MockProxy<AbstractStorageService>;
  let logService: MockProxy<LogService>;

  beforeEach(() => {
    storage = mock();
    logService = mock();
  });

  it("should return -1 if no version", async () => {
    storage.get.mockReturnValueOnce(null);
    expect(await currentVersion(storage, logService)).toEqual(-1);
  });

  it("should return version", async () => {
    storage.get.calledWith("stateVersion").mockReturnValueOnce(1 as any);
    expect(await currentVersion(storage, logService)).toEqual(1);
  });

  it("should return version from global", async () => {
    storage.get.calledWith("stateVersion").mockReturnValueOnce(null);
    storage.get.calledWith("global").mockReturnValueOnce({ stateVersion: 1 } as any);
    expect(await currentVersion(storage, logService)).toEqual(1);
  });

  it("should prefer root version to global", async () => {
    storage.get.calledWith("stateVersion").mockReturnValue(1 as any);
    storage.get.calledWith("global").mockReturnValue({ stateVersion: 2 } as any);
    expect(await currentVersion(storage, logService)).toEqual(1);
  });
});