import {
  ScreenRevealManager,
  COMPONENT_LOADING_STATE,
} from "../ScreenRevealManager";

describe("ScreenRevealManager explicit component count", () => {
  const components = (count: number) =>
    Array.from({ length: count }, () => ({ component_type: "grid-qb" }));

  it("waits for the number of components it is given, not the computed default", () => {
    const manager = new ScreenRevealManager(components(6), jest.fn(), 6);

    expect(manager.numberOfComponentsWaitToLoadBeforePresent).toBe(6);
  });

  it("never waits for more components than it was handed", () => {
    const manager = new ScreenRevealManager(components(3), jest.fn(), 10);

    expect(manager.numberOfComponentsWaitToLoadBeforePresent).toBe(3);
  });

  it("falls back to the computed default when given no explicit count", () => {
    const manager = new ScreenRevealManager(components(6), jest.fn());

    expect(manager.numberOfComponentsWaitToLoadBeforePresent).toBe(3);
  });
});

describe("ScreenRevealManager", () => {
  const mockCallback = jest.fn();

  beforeEach(() => {
    jest.clearAllMocks();
  });

  it("should initialize with the correct number of components to wait for", () => {
    const componentsToRender: ZappUIComponent[] = [
      { component_type: "component1" },
      { component_type: "component2" },
      { component_type: "component3" },
    ];

    const manager = new ScreenRevealManager(componentsToRender, mockCallback);

    expect(manager.numberOfComponentsWaitToLoadBeforePresent).toBe(3);

    expect(manager.renderingState).toEqual([
      COMPONENT_LOADING_STATE.UNKNOWN,
      COMPONENT_LOADING_STATE.UNKNOWN,
      COMPONENT_LOADING_STATE.UNKNOWN,
    ]);
  });

  it("should call the callback when the required number of components are loaded successfully", () => {
    const componentsToRender: ZappUIComponent[] = [
      { component_type: "component1" },
      { component_type: "component2" },
      { component_type: "component3" },
    ];

    const manager = new ScreenRevealManager(componentsToRender, mockCallback);

    manager.onLoadFinished(0);
    manager.onLoadFinished(1);
    manager.onLoadFinished(2);

    expect(mockCallback).toHaveBeenCalledTimes(1);
  });

  it("should call the callback when the required number of components fail to load", () => {
    const componentsToRender: ZappUIComponent[] = [
      { component_type: "component1" },
      { component_type: "component2" },
      { component_type: "component3" },
    ];

    const manager = new ScreenRevealManager(componentsToRender, mockCallback);

    manager.onLoadFailed(0);
    manager.onLoadFailed(1);
    manager.onLoadFailed(2);

    expect(mockCallback).toHaveBeenCalledTimes(1);
  });

  it("should call the callback when a mix of successful and failed loads meet the required number", () => {
    const componentsToRender: ZappUIComponent[] = [
      { component_type: "component1" },
      { component_type: "component2" },
      { component_type: "component3" },
    ];

    const manager = new ScreenRevealManager(componentsToRender, mockCallback);

    manager.onLoadFinished(0);
    manager.onLoadFailed(1);
    manager.onLoadFinished(2);

    expect(mockCallback).toHaveBeenCalledTimes(1);
  });

  it("should not call the callback if the required number of components are not loaded", () => {
    const componentsToRender: ZappUIComponent[] = [
      { component_type: "component1" },
      { component_type: "component2" },
      { component_type: "component3" },
    ];

    const manager = new ScreenRevealManager(componentsToRender, mockCallback);

    manager.onLoadFinished(0);
    manager.onLoadFailed(1);

    expect(mockCallback).not.toHaveBeenCalled();
  });

  it("should call the callback when the when first component is gallery and it was loaded successfully", () => {
    const componentsToRender: ZappUIComponent[] = [
      { component_type: "gallery-qb" },
      { component_type: "component2" },
      { component_type: "component3" },
    ];

    const manager = new ScreenRevealManager(componentsToRender, mockCallback);

    manager.onLoadFinished(0);

    expect(mockCallback).toHaveBeenCalledTimes(1);
  });
});