pMaps.test.ts

import { sleep } from "bun";
import { sflow } from "./sflow";

it("Infinity concurrent", async () => {
  const runOrder: number[] = [];
  const finishOrder: number[] = [];
  const returnOrder = await sflow([1, 2, 3, 4, 4, 3, 2, 1])
    .pMap(async (e, i) => {
      runOrder.push(e);
      await sleep(e * 20);
      finishOrder.push(e);
      console.log(e, i);
      return e;
    })
    .toArray();
  expect(runOrder).toEqual([1, 2, 3, 4, 4, 3, 2, 1]);
  expect(finishOrder).toEqual([1, 1, 2, 2, 3, 3, 4, 4]);
  expect(returnOrder).toEqual([1, 2, 3, 4, 4, 3, 2, 1]);
});
it("pMaps", async () => {
  const t = Date.now();
  const req = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
  const acc = sflow(req.toReversed())
    .pMap(
      async (n) => {
        await new Promise((r) => setTimeout(r, n * 10));
        return n;
      },
      { concurrency: 3 },
    )
    .toArray();
  expect(acc).resolves.toEqual(req.toReversed());
});