import { createRequire } from 'node:module'

import test from 'ava'

const require = createRequire(import.meta.url)

const {
  Fib,
  Fib2,
  Fib3,
}: typeof import('../index.js') = require('../index.node')

for (const [index, factory] of [
  () => new Fib(),
  () => Fib2.create(0),
  () => new Fib3(0, 1),
].entries()) {
  test(`should be able to stop a generator #${index}`, (t) => {
    const fib = factory()
    const gen = fib[Symbol.iterator]
    t.is(typeof gen, 'function')
    const iterator = gen()
    t.deepEqual(iterator.next(), {
      done: false,
      value: 1,
    })
    iterator.next()
    iterator.next()
    iterator.next()
    iterator.next()
    t.deepEqual(iterator.next(), {
      done: false,
      value: 8,
    })
    t.deepEqual(iterator.return?.(), {
      done: true,
    })
    t.deepEqual(iterator.next(), {
      done: true,
    })
  })

  test(`should be able to throw to generator #${index}`, (t) => {
    const fib = factory()
    const gen = fib[Symbol.iterator]
    t.is(typeof gen, 'function')
    const iterator = gen()
    t.deepEqual(iterator.next(), {
      done: false,
      value: 1,
    })
    iterator.next()
    iterator.next()
    iterator.next()
    iterator.next()
    t.deepEqual(iterator.next(), {
      done: false,
      value: 8,
    })
    t.throws(() => iterator.throw!(new Error()))
    t.deepEqual(iterator.next(), {
      done: true,
    })
  })
}