import { Applied, Late, Of } from "silentium";
import { expect, test, vi } from "vitest";

import { BranchLazy } from "../behaviors/BranchLazy";

test("BranchLazy._main.test", () => {
  const l = Late<number>(2);
  const res = BranchLazy(
    Applied(l, (t) => {
      return t === 2;
    }),
    () => Of("Then branch"),
    () => Of("Else branch"),
  );

  const g = vi.fn();
  res.then(g);
  expect(g).toBeCalledWith("Then branch");

  l.use(1);

  expect(g).toBeCalledWith("Else branch");
});

test("BranchLazy._main.test", async () => {
  const pages = Late([1]);
  const pagination = BranchLazy(
    Applied(pages, (p) => p.length > 1),
    () => Of("many"),
    () => Of("hidden"),
  );

  expect(await pagination).toBe("hidden");

  pages.use([1, 2, 3]);
  expect(await pagination).toBe("many");

  pages.use([1]);
  expect(await pagination).toBe("hidden");

  pages.use([1, 2, 3]);
  expect(await pagination).toBe("many");

  pages.use([]);
  expect(await pagination).toBe("hidden");
});