// Type declarations for @nestjs/websockets mock
declare module '@nestjs/websockets' {
  export function WebSocketGateway(options?: any): ClassDecorator;
  export function WebSocketServer(): PropertyDecorator;
  export function SubscribeMessage(message: string): MethodDecorator;
  export function MessageBody(): ParameterDecorator;
  export function ConnectedSocket(): ParameterDecorator;

  export interface OnGatewayConnection<T = any> {
    handleConnection(client: T, ...args: any[]): any;
  }

  export interface OnGatewayDisconnect<T = any> {
    handleDisconnect(client: T): any;
  }
}
