import { expect, test } from 'vitest'

import { poll } from './poll.js'
import { wait } from './wait.js'

test('polls on a given interval', async () => {
	const items: string[] = []
	poll(
		async () => {
			items.push('wagmi')
		},
		{
			interval: 100,
		},
	)

	await wait(450)
	expect(items).toMatchInlineSnapshot(`
    [
      "wagmi",
      "wagmi",
      "wagmi",
      "wagmi",
    ]
  `)
})

test('emitOnBegin', async () => {
	const items: string[] = []
	poll(
		async () => {
			items.push('wagmi')
		},
		{
			emitOnBegin: true,
			interval: 100,
		},
	)

	await wait(500)
	expect(items).toMatchInlineSnapshot(`
    [
      "wagmi",
      "wagmi",
      "wagmi",
      "wagmi",
      "wagmi",
    ]
  `)
})

test('initialWaitTime', async () => {
	const items: string[] = []
	poll(
		async () => {
			items.push('wagmi')
		},
		{
			initialWaitTime: async () => 200,
			interval: 100,
		},
	)

	await wait(500)
	expect(items).toMatchInlineSnapshot(`
    [
      "wagmi",
      "wagmi",
      "wagmi",
    ]
  `)
})

test('stop polling', async () => {
	const items: string[] = []
	const unpoll = poll(
		async () => {
			items.push('wagmi')
		},
		{
			interval: 100,
		},
	)

	await wait(500)
	expect(items).toMatchInlineSnapshot(`
    [
      "wagmi",
      "wagmi",
      "wagmi",
      "wagmi",
    ]
  `)

	unpoll()

	await wait(500)
	expect(items).toMatchInlineSnapshot(`
    [
      "wagmi",
      "wagmi",
      "wagmi",
      "wagmi",
    ]
  `)
})

test('stop polling via callback', async () => {
	const items: string[] = []
	poll(
		async ({ unpoll }) => {
			items.push('wagmi')
			if (items.length === 2) unpoll()
		},
		{
			interval: 100,
		},
	)

	await wait(500)
	expect(items).toMatchInlineSnapshot(`
    [
      "wagmi",
      "wagmi",
    ]
  `)
})