ice/tests/setup.ts

import { Database } from 'sqlite3';
import fs from 'fs';
import path from 'path';

// Setup test environment
process.env.NODE_ENV = 'test';
process.env.ADMIN_PASSWORD = 'test_admin_password';
process.env.MAPBOX_ACCESS_TOKEN = 'pk.test_token_here';

// Test database paths
export const TEST_DB_PATH = path.join(__dirname, 'test_icewatch.db');
export const TEST_PROFANITY_DB_PATH = path.join(__dirname, 'test_profanity.db');

// Helper function to create test database
export const createTestDatabase = (): Promise<Database> => {
  return new Promise((resolve, reject) => {
    const db = new Database(':memory:', (err) => {
      if (err) {
        reject(err);
        return;
      }
      
      // Create locations table
      db.run(`
        CREATE TABLE IF NOT EXISTS locations (
          id INTEGER PRIMARY KEY AUTOINCREMENT,
          address TEXT NOT NULL,
          latitude REAL,
          longitude REAL,
          timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
          description TEXT,
          persistent INTEGER DEFAULT 0,
          created_at DATETIME DEFAULT CURRENT_TIMESTAMP
        )
      `, (err) => {
        if (err) {
          reject(err);
          return;
        }
        resolve(db);
      });
    });
  });
};

// Helper function to create test profanity database
export const createTestProfanityDatabase = (): Promise<Database> => {
  return new Promise((resolve, reject) => {
    const db = new Database(':memory:', (err) => {
      if (err) {
        reject(err);
        return;
      }
      
      // Create profanity_words table
      db.run(`
        CREATE TABLE IF NOT EXISTS profanity_words (
          id INTEGER PRIMARY KEY AUTOINCREMENT,
          word TEXT NOT NULL UNIQUE,
          severity TEXT NOT NULL,
          category TEXT NOT NULL,
          created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
          created_by TEXT DEFAULT 'system'
        )
      `, (err) => {
        if (err) {
          reject(err);
          return;
        }
        resolve(db);
      });
    });
  });
};

// Cleanup function for tests
export const cleanupTestDatabases = () => {
  // Clean up any test database files
  [TEST_DB_PATH, TEST_PROFANITY_DB_PATH].forEach(dbPath => {
    if (fs.existsSync(dbPath)) {
      fs.unlinkSync(dbPath);
    }
  });
};

// Global test cleanup
afterAll(() => {
  cleanupTestDatabases();
});

// Console override for cleaner test output
const originalConsoleLog = console.log;
const originalConsoleError = console.error;
const originalConsoleWarn = console.warn;

beforeAll(() => {
  // Suppress console output during tests unless running in verbose mode
  if (!process.env.VERBOSE_TESTS) {
    console.log = jest.fn();
    console.error = jest.fn();
    console.warn = jest.fn();
  }
});

afterAll(() => {
  // Restore console functions
  console.log = originalConsoleLog;
  console.error = originalConsoleError;
  console.warn = originalConsoleWarn;
});