fix(mock-udp): New Node.js API

Author: ffflorian

packages/mock-udp/src/index.ts

@@ -11,7 +11,7 @@ const originalSocketSend = Socket.prototype.send;
 class Scope {
   public _done: boolean;
   public address?: string;
-  public buffer?: Buffer | string | Uint8Array | any[];
+  public buffer?: string | readonly any[] | ArrayBufferLike;
   public length?: number;
   public offset?: number;
   public port?: number;
@@ -25,37 +25,41 @@ class Scope {
   }
 }
 
-function overriddenSocketSend(msg: string | Uint8Array | any[], callback?: SendCallback): void;
-function overriddenSocketSend(msg: string | Uint8Array | any[], port?: number, callback?: SendCallback): void;
 function overriddenSocketSend(
-  msg: string | Uint8Array | any[],
+  msg: string | NodeJS.ArrayBufferView | readonly any[],
   port?: number,
   address?: string,
   callback?: SendCallback
 ): void;
 function overriddenSocketSend(
-  msg: string | Uint8Array | any[],
+  msg: string | NodeJS.ArrayBufferView | readonly any[],
+  port?: number,
+  callback?: SendCallback
+): void;
+function overriddenSocketSend(msg: string | NodeJS.ArrayBufferView | readonly any[], callback?: SendCallback): void;
+function overriddenSocketSend(
+  msg: string | NodeJS.ArrayBufferView,
   offset: number,
   length: number,
+  port?: number,
+  address?: string,
   callback?: SendCallback
 ): void;
 function overriddenSocketSend(
-  msg: string | Uint8Array | any[],
+  msg: string | NodeJS.ArrayBufferView,
   offset: number,
   length: number,
   port?: number,
   callback?: SendCallback
 ): void;
 function overriddenSocketSend(
-  msg: string | Uint8Array | any[],
+  msg: string | NodeJS.ArrayBufferView,
   offset: number,
   length: number,
-  port?: number,
-  address?: string,
   callback?: SendCallback
 ): void;
 function overriddenSocketSend(
-  msg: Buffer | string | Uint8Array | any[],
+  msg: NodeJS.ArrayBufferView | string | readonly any[],
   offsetOrPortOrCallback?: number | SendCallback,
   lengthOrAddressOrCallback?: number | string | SendCallback,
   portOrCallback?: number | SendCallback,
@@ -70,16 +74,17 @@ function overriddenSocketSend(
   const length = hasLengthAndOffset ? (lengthOrAddressOrCallback as number) || 0 : 0;
   const offset = hasLengthAndOffset ? (offsetOrPortOrCallback as number) : 0;
   const port = hasLengthAndOffset ? (portOrCallback as number) : (offsetOrPortOrCallback as number);
+  const message = ArrayBuffer.isView(msg) ? msg.buffer : msg;
 
-  if (offset >= msg.length) {
+  if (offset >= message.slice.length) {
     throw new Error('Offset into buffer too large.');
   }
 
-  if (offset + length > msg.length) {
+  if (offset + length > message.slice.length) {
     throw new Error('Offset + length beyond buffer length.');
   }
 
-  const newBuffer = msg.slice(offset, offset + length);
+  const newBuffer = message.slice(offset, offset + length);
   const host = `${address}:${port}`;
 
   if (intercepts.hasOwnProperty(host)) {