livedash-node/tests/unit/database.test.ts

import { describe, it, expect, beforeAll, afterAll } from 'vitest';
import { PrismaClient } from '@prisma/client';

describe('Database Configuration', () => {
  let prisma: PrismaClient;

  beforeAll(() => {
    prisma = new PrismaClient();
  });

  afterAll(async () => {
    await prisma.$disconnect();
  });

  it('should connect to the test database', async () => {
    // Verify we can connect to the database
    const result = await prisma.$queryRaw`SELECT 1 as test`;
    expect(result).toBeDefined();
  });

  it('should use PostgreSQL as the database provider', async () => {
    // Query the database to verify it's PostgreSQL
    const result = await prisma.$queryRaw`SELECT version()` as any[];
    expect(result[0].version).toContain('PostgreSQL');
  });

  it('should be using the test database URL', () => {
    // Verify that DATABASE_URL is set to the test database
    expect(process.env.DATABASE_URL).toBeDefined();
    expect(process.env.DATABASE_URL).toContain('postgresql://');
    
    // If DATABASE_URL_TEST is set, DATABASE_URL should match it (from our test setup)
    if (process.env.DATABASE_URL_TEST) {
      expect(process.env.DATABASE_URL).toBe(process.env.DATABASE_URL_TEST);
    }
  });

  it('should have all required tables', async () => {
    // Verify all our tables exist
    const tables = await prisma.$queryRaw`
      SELECT table_name 
      FROM information_schema.tables 
      WHERE table_schema = 'public' 
      AND table_type = 'BASE TABLE'
      ORDER BY table_name
    ` as any[];

    const tableNames = tables.map(t => t.table_name);
    
    expect(tableNames).toContain('Company');
    expect(tableNames).toContain('User');
    expect(tableNames).toContain('Session');
    expect(tableNames).toContain('SessionImport');
    expect(tableNames).toContain('Message');
    expect(tableNames).toContain('Question');
    expect(tableNames).toContain('SessionQuestion');
    expect(tableNames).toContain('AIProcessingRequest');
  });

  it('should be able to create and query data', async () => {
    // Test basic CRUD operations
    const company = await prisma.company.create({
      data: {
        name: 'Test Company',
        csvUrl: 'https://example.com/test.csv',
      },
    });

    expect(company.id).toBeDefined();
    expect(company.name).toBe('Test Company');

    // Clean up
    await prisma.company.delete({
      where: { id: company.id },
    });
  });
});