import { sleep } from "bun";
import { sf } from "./index";
import { sflow } from "./sflow";
import { unpromises, unpromisesFn } from "./unpromises";

it("works", async () => {
  const p = getStream();
  expect(p).toBeInstanceOf(Promise);
  expect(await sf(unpromises(p)).toArray()).toEqual([1, 2, 3]);
});

async function getStream() {
  return sflow([1, 2, 3]);
}

it("by async fn", async () => {
  expect(
    await sflow([1, 2, 3])
      .by(
        unpromisesFn(async function (src: sflow<number>) {
          await sleep(100);
          return src.map((x) => x * 2);
        })
      )
      .toArray()
  ).toEqual([2, 4, 6]);
});