// vitest.setup.ts
/// <reference types="vitest-drizzle-postgres/types" />

import { setupTestDb, useTestDb, cleanupTestDb, teardownTestDb } from 'vitest-drizzle-postgres';
import { beforeAll, beforeEach, afterEach } from 'vitest';

import { db } from './src/db'; // Your existing database connection
import * as schema from './src/schema';

beforeAll(async () => {
  await setupTestDb({
    schema,
    db,
    migrationsFolder: "./migrations", // optional
  });

  return async () => {
    await teardownTestDb();
  }
});

beforeEach(async (ctx) => {
  await useTestDb(ctx);

  return async () => {
    await cleanupTestDb();
  };
});

// user.test.ts
import { describe, test, expect } from 'vitest';
import { users } from './schema';

import { db } from './src/db'; // Your existing database connection

describe('User tests', () => {
  test('should create and find user', async () => {
    const [user] = await db
      .insert(users)
      .values({ name: 'John', email: '[email protected]' })
      .returning();

    expect(user.name).toBe('John');
    
    const foundUsers = await db.select().from(users);
    expect(foundUsers).toHaveLength(1);
  });

  test('should not see data from previous test', async () => {
    const users = await db.select().from(users);
    expect(users).toHaveLength(0); // Clean slate
  });
});