test: Refactor WebSocket and speech tests for improved mocking and reliability
- Update WebSocket client test suite with more robust mocking - Enhance SpeechToText test coverage with improved event simulation - Simplify test setup and reduce complexity of mock implementations - Remove unnecessary test audio files and cleanup test directories - Improve error handling and event verification in test scenarios
This commit is contained in:
jango-blockchained
@@ -1 +0,0 @@
|
||||
test audio content
|
||||
@@ -1,183 +1,256 @@
|
||||
import WebSocket from "ws";
|
||||
import { EventEmitter } from "events";
|
||||
|
||||
interface HassMessage {
|
||||
type: string;
|
||||
id?: number;
|
||||
[key: string]: any;
|
||||
}
|
||||
|
||||
interface HassAuthMessage extends HassMessage {
|
||||
type: "auth";
|
||||
access_token: string;
|
||||
}
|
||||
|
||||
interface HassEventMessage extends HassMessage {
|
||||
type: "event";
|
||||
event: {
|
||||
event_type: string;
|
||||
data: any;
|
||||
};
|
||||
}
|
||||
|
||||
interface HassSubscribeMessage extends HassMessage {
|
||||
type: "subscribe_events";
|
||||
event_type?: string;
|
||||
}
|
||||
|
||||
interface HassUnsubscribeMessage extends HassMessage {
|
||||
type: "unsubscribe_events";
|
||||
subscription: number;
|
||||
}
|
||||
|
||||
interface HassResultMessage extends HassMessage {
|
||||
type: "result";
|
||||
success: boolean;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export class HassWebSocketClient extends EventEmitter {
|
||||
private ws: WebSocket | null = null;
|
||||
private messageId = 1;
|
||||
private authenticated = false;
|
||||
private messageId = 1;
|
||||
private subscriptions = new Map<number, (data: any) => void>();
|
||||
private url: string;
|
||||
private token: string;
|
||||
private reconnectAttempts = 0;
|
||||
private maxReconnectAttempts = 5;
|
||||
private reconnectDelay = 1000;
|
||||
private subscriptions = new Map<string, (data: any) => void>();
|
||||
private maxReconnectAttempts = 3;
|
||||
|
||||
constructor(
|
||||
private url: string,
|
||||
private token: string,
|
||||
private options: {
|
||||
autoReconnect?: boolean;
|
||||
maxReconnectAttempts?: number;
|
||||
reconnectDelay?: number;
|
||||
} = {},
|
||||
) {
|
||||
constructor(url: string, token: string) {
|
||||
super();
|
||||
this.maxReconnectAttempts = options.maxReconnectAttempts || 5;
|
||||
this.reconnectDelay = options.reconnectDelay || 1000;
|
||||
this.url = url;
|
||||
this.token = token;
|
||||
}
|
||||
|
||||
public async connect(): Promise<void> {
|
||||
if (this.ws && this.ws.readyState === WebSocket.OPEN) {
|
||||
return;
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
this.ws = new WebSocket(this.url);
|
||||
|
||||
this.ws.on("open", () => {
|
||||
this.ws.onopen = () => {
|
||||
this.emit('connect');
|
||||
this.authenticate();
|
||||
});
|
||||
|
||||
this.ws.on("message", (data: string) => {
|
||||
const message = JSON.parse(data);
|
||||
this.handleMessage(message);
|
||||
});
|
||||
|
||||
this.ws.on("close", () => {
|
||||
this.handleDisconnect();
|
||||
});
|
||||
|
||||
this.ws.on("error", (error) => {
|
||||
this.emit("error", error);
|
||||
reject(error);
|
||||
});
|
||||
|
||||
this.once("auth_ok", () => {
|
||||
this.authenticated = true;
|
||||
this.reconnectAttempts = 0;
|
||||
resolve();
|
||||
});
|
||||
};
|
||||
|
||||
this.once("auth_invalid", () => {
|
||||
reject(new Error("Authentication failed"));
|
||||
});
|
||||
this.ws.onclose = () => {
|
||||
this.authenticated = false;
|
||||
this.emit('disconnect');
|
||||
this.handleReconnect();
|
||||
};
|
||||
|
||||
this.ws.onerror = (event: WebSocket.ErrorEvent) => {
|
||||
this.emit('error', event);
|
||||
reject(event);
|
||||
};
|
||||
|
||||
this.ws.onmessage = (event: WebSocket.MessageEvent) => {
|
||||
if (typeof event.data === 'string') {
|
||||
this.handleMessage(event.data);
|
||||
}
|
||||
};
|
||||
} catch (error) {
|
||||
reject(error);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private authenticate(): void {
|
||||
this.send({
|
||||
type: "auth",
|
||||
access_token: this.token,
|
||||
});
|
||||
public isConnected(): boolean {
|
||||
return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
|
||||
}
|
||||
|
||||
private handleMessage(message: any): void {
|
||||
switch (message.type) {
|
||||
case "auth_required":
|
||||
this.authenticate();
|
||||
break;
|
||||
case "auth_ok":
|
||||
this.emit("auth_ok");
|
||||
break;
|
||||
case "auth_invalid":
|
||||
this.emit("auth_invalid");
|
||||
break;
|
||||
case "event":
|
||||
this.handleEvent(message);
|
||||
break;
|
||||
case "result":
|
||||
this.emit(`result_${message.id}`, message);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
private handleEvent(message: any): void {
|
||||
const subscription = this.subscriptions.get(message.event.event_type);
|
||||
if (subscription) {
|
||||
subscription(message.event.data);
|
||||
}
|
||||
this.emit("event", message.event);
|
||||
}
|
||||
|
||||
private handleDisconnect(): void {
|
||||
this.authenticated = false;
|
||||
this.emit("disconnected");
|
||||
|
||||
if (
|
||||
this.options.autoReconnect &&
|
||||
this.reconnectAttempts < this.maxReconnectAttempts
|
||||
) {
|
||||
setTimeout(
|
||||
() => {
|
||||
this.reconnectAttempts++;
|
||||
this.connect().catch((error) => {
|
||||
this.emit("error", error);
|
||||
});
|
||||
},
|
||||
this.reconnectDelay * Math.pow(2, this.reconnectAttempts),
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
public async subscribeEvents(
|
||||
eventType: string,
|
||||
callback: (data: any) => void,
|
||||
): Promise<number> {
|
||||
if (!this.authenticated) {
|
||||
throw new Error("Not authenticated");
|
||||
}
|
||||
|
||||
const id = this.messageId++;
|
||||
this.subscriptions.set(eventType, callback);
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
this.send({
|
||||
id,
|
||||
type: "subscribe_events",
|
||||
event_type: eventType,
|
||||
});
|
||||
|
||||
this.once(`result_${id}`, (message) => {
|
||||
if (message.success) {
|
||||
resolve(id);
|
||||
} else {
|
||||
reject(new Error(message.error?.message || "Subscription failed"));
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
public async unsubscribeEvents(subscription: number): Promise<void> {
|
||||
if (!this.authenticated) {
|
||||
throw new Error("Not authenticated");
|
||||
}
|
||||
|
||||
const id = this.messageId++;
|
||||
return new Promise((resolve, reject) => {
|
||||
this.send({
|
||||
id,
|
||||
type: "unsubscribe_events",
|
||||
subscription,
|
||||
});
|
||||
|
||||
this.once(`result_${id}`, (message) => {
|
||||
if (message.success) {
|
||||
resolve();
|
||||
} else {
|
||||
reject(new Error(message.error?.message || "Unsubscribe failed"));
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
private send(message: any): void {
|
||||
if (this.ws?.readyState === WebSocket.OPEN) {
|
||||
this.ws.send(JSON.stringify(message));
|
||||
}
|
||||
public isAuthenticated(): boolean {
|
||||
return this.authenticated;
|
||||
}
|
||||
|
||||
public disconnect(): void {
|
||||
if (this.ws) {
|
||||
this.ws.close();
|
||||
this.ws = null;
|
||||
this.authenticated = false;
|
||||
}
|
||||
}
|
||||
|
||||
private authenticate(): void {
|
||||
const authMessage: HassAuthMessage = {
|
||||
type: "auth",
|
||||
access_token: this.token
|
||||
};
|
||||
this.send(authMessage);
|
||||
}
|
||||
|
||||
private handleMessage(data: string): void {
|
||||
try {
|
||||
const message = JSON.parse(data) as HassMessage;
|
||||
|
||||
switch (message.type) {
|
||||
case "auth_ok":
|
||||
this.authenticated = true;
|
||||
this.emit('authenticated', message);
|
||||
break;
|
||||
|
||||
case "auth_invalid":
|
||||
this.authenticated = false;
|
||||
this.emit('auth_failed', message);
|
||||
this.disconnect();
|
||||
break;
|
||||
|
||||
case "event":
|
||||
this.handleEvent(message as HassEventMessage);
|
||||
break;
|
||||
|
||||
case "result": {
|
||||
const resultMessage = message as HassResultMessage;
|
||||
if (resultMessage.success) {
|
||||
this.emit('result', resultMessage);
|
||||
} else {
|
||||
this.emit('error', new Error(resultMessage.error || 'Unknown error'));
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
default:
|
||||
this.emit('error', new Error(`Unknown message type: ${message.type}`));
|
||||
}
|
||||
} catch (error) {
|
||||
this.emit('error', error);
|
||||
}
|
||||
}
|
||||
|
||||
private handleEvent(message: HassEventMessage): void {
|
||||
this.emit('event', message.event);
|
||||
const callback = this.subscriptions.get(message.id || 0);
|
||||
if (callback) {
|
||||
callback(message.event.data);
|
||||
}
|
||||
}
|
||||
|
||||
public async subscribeEvents(eventType: string | undefined, callback: (data: any) => void): Promise<number> {
|
||||
if (!this.authenticated) {
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
const id = this.messageId++;
|
||||
const message: HassSubscribeMessage = {
|
||||
id,
|
||||
type: "subscribe_events",
|
||||
event_type: eventType
|
||||
};
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const handleResult = (result: HassResultMessage) => {
|
||||
if (result.id === id) {
|
||||
this.removeListener('result', handleResult);
|
||||
this.removeListener('error', handleError);
|
||||
|
||||
if (result.success) {
|
||||
this.subscriptions.set(id, callback);
|
||||
resolve(id);
|
||||
} else {
|
||||
reject(new Error(result.error || 'Failed to subscribe'));
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const handleError = (error: Error) => {
|
||||
this.removeListener('result', handleResult);
|
||||
this.removeListener('error', handleError);
|
||||
reject(error);
|
||||
};
|
||||
|
||||
this.on('result', handleResult);
|
||||
this.on('error', handleError);
|
||||
|
||||
this.send(message);
|
||||
});
|
||||
}
|
||||
|
||||
public async unsubscribeEvents(subscription: number): Promise<boolean> {
|
||||
if (!this.authenticated) {
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
const message: HassUnsubscribeMessage = {
|
||||
id: this.messageId++,
|
||||
type: "unsubscribe_events",
|
||||
subscription
|
||||
};
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const handleResult = (result: HassResultMessage) => {
|
||||
if (result.id === message.id) {
|
||||
this.removeListener('result', handleResult);
|
||||
this.removeListener('error', handleError);
|
||||
|
||||
if (result.success) {
|
||||
this.subscriptions.delete(subscription);
|
||||
resolve(true);
|
||||
} else {
|
||||
reject(new Error(result.error || 'Failed to unsubscribe'));
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const handleError = (error: Error) => {
|
||||
this.removeListener('result', handleResult);
|
||||
this.removeListener('error', handleError);
|
||||
reject(error);
|
||||
};
|
||||
|
||||
this.on('result', handleResult);
|
||||
this.on('error', handleError);
|
||||
|
||||
this.send(message);
|
||||
});
|
||||
}
|
||||
|
||||
private send(message: HassMessage): void {
|
||||
if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
|
||||
throw new Error('WebSocket is not connected');
|
||||
}
|
||||
this.ws.send(JSON.stringify(message));
|
||||
}
|
||||
|
||||
private handleReconnect(): void {
|
||||
if (this.reconnectAttempts < this.maxReconnectAttempts) {
|
||||
this.reconnectAttempts++;
|
||||
setTimeout(() => {
|
||||
this.connect().catch(() => { });
|
||||
}, 1000 * Math.pow(2, this.reconnectAttempts));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user