59cbd2552b
- Created detailed test suite for Home Assistant WebSocket client - Implemented tests for WebSocket connection, authentication, and error handling - Added comprehensive test coverage for HassInstanceImpl API methods - Mocked WebSocket and fetch to simulate various connection scenarios - Covered authentication, state retrieval, service calls, and environment configuration - Improved test infrastructure for Home Assistant integration
84 lines
1.9 KiB
TypeScript
84 lines
1.9 KiB
TypeScript
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();
|
|
});
|