Function causesArb

packages/effect/test/utils/cause.ts:5–40

(
  n: number,
  error: fc.Arbitrary<E>,
  defect: fc.Arbitrary<unknown>
)

import * as FiberId from "effect/FiberId"

export const causesArb = <E>(
  n: number,
  error: fc.Arbitrary<E>,
  defect: fc.Arbitrary<unknown>
): fc.Arbitrary<Cause.Cause<E>> => {
  const fiberId: fc.Arbitrary<FiberId.FiberId> = fc.tuple(
    fc.integer(),
    fc.integer()
  ).map(([a, b]) => FiberId.make(a, b))

  const empty = fc.constant(Cause.empty)
  const failure = error.map(Cause.fail)
  const die = defect.map(Cause.die)
  const interrupt = fiberId.map(Cause.interrupt)

  const sequential = (n: number): fc.Arbitrary<Cause.Cause<E>> => {
    return fc.integer({ min: 1, max: n - 1 }).chain((i) =>
      causesN(i).chain((left) => causesN(n - i).map((right) => Cause.sequential(left, right)))
    )
  }

  const parallel = (n: number): fc.Arbitrary<Cause.Cause<E>> => {
    return fc.integer({ min: 1, max: n - 1 }).chain((i) =>
      causesN(i).chain((left) => causesN(n - i).map((right) => Cause.parallel(left, right)))
    )
  }

  const causesN = (n: number): fc.Arbitrary<Cause.Cause<E>> => {
    if (n === 1) {
      return fc.oneof(empty, failure, die, interrupt)
    }
    return fc.oneof(sequential(n), parallel(n))
  }

  return causesN(n)
}

export const causes: fc.Arbitrary<Cause.Cause<string>> = causesArb(
  1,