jest.setup.ts

import { jest } from '@jest/globals';
import dotenv from 'dotenv';
import { TextEncoder, TextDecoder } from 'util';

// Load test environment variables
dotenv.config({ path: '.env.test' });

// Set test environment
process.env.NODE_ENV = 'test';
process.env.HASS_URL = 'http://localhost:8123';
process.env.HASS_TOKEN = 'test_token';
process.env.CLAUDE_API_KEY = 'test_api_key';
process.env.CLAUDE_MODEL = 'test_model';

// Add TextEncoder and TextDecoder to global scope
Object.defineProperty(global, 'TextEncoder', {
    value: TextEncoder,
    writable: true
});

Object.defineProperty(global, 'TextDecoder', {
    value: TextDecoder,
    writable: true
});

// Configure console for tests
const originalConsole = { ...console };
global.console = {
    ...console,
    log: jest.fn(),
    error: jest.fn(),
    warn: jest.fn(),
    info: jest.fn(),
    debug: jest.fn(),
};

// Increase test timeout
jest.setTimeout(30000);

// Mock WebSocket
jest.mock('ws', () => {
    return {
        WebSocket: jest.fn().mockImplementation(() => ({
            on: jest.fn(),
            send: jest.fn(),
            close: jest.fn(),
            removeAllListeners: jest.fn()
        }))
    };
});

// Mock chalk
const createChalkMock = () => {
    const handler = {
        get(target: any, prop: string) {
            if (prop === 'default') {
                return createChalkMock();
            }
            return typeof prop === 'string' ? createChalkMock() : target[prop];
        },
        apply(target: any, thisArg: any, args: any[]) {
            return args[0];
        }
    };
    return new Proxy(() => { }, handler);
};

jest.mock('chalk', () => createChalkMock());

// Mock ansi-styles
jest.mock('ansi-styles', () => ({}), { virtual: true });

// Mock supports-color
jest.mock('supports-color', () => ({}), { virtual: true });

// Reset mocks between tests
beforeEach(() => {
    jest.clearAllMocks();
});

// Cleanup after tests
afterEach(() => {
    jest.clearAllTimers();
});