import { assert, describe, expect } from 'vitest'
import { createTaskCollector, getCurrentSuite } from 'vitest/suite';

let shouldBeTrueAfterTest: boolean = false;

const wrappedInLogs = createTaskCollector(function(
  this: object,
  name: string,
  fn: () => void,
  timeout: number
) {
  const handler = async () => {
    shouldBeTrueAfterTest = false;
    const ret = await fn();
    assert(shouldBeTrueAfterTest);
    return ret;
  }
  getCurrentSuite().task(name, {
    ...this,
    meta: {
      ignoreme: true,
    },
    handler,
    timeout
  })
})

describe('suite name', () => {
  wrappedInLogs('foo', () => {
    assert.equal(Math.sqrt(4), 2)
    shouldBeTrueAfterTest = true;
  })

  wrappedInLogs('bar', () => {
    expect(1 + 1).eq(2)
    shouldBeTrueAfterTest = true;
  })

  wrappedInLogs('snapshot', () => {
    expect({ foo: 'bar' }).toMatchSnapshot()
    shouldBeTrueAfterTest = true;
  })
})

type User = {
  id: number;
  email: string;
}

type UserContext = {
  user: User
}

const userTest = wrappedInLogs.extend({
  user: async ({}, use) => {
    await use({
      id: 1,
      email: 'foo@example.com'
    })
  }
})

describe('suite name', () => {
  userTest('foo', () => {
    assert.equal(Math.sqrt(4), 2)
  })

  userTest('bar', () => {
    expect(1 + 1).eq(2)
  })

  userTest('snapshot', () => {
    expect({ foo: 'bar' }).toMatchSnapshot()
  })
})