@replit/river 0.205.1 → 0.205.2

package/dist/testUtil/index.d.cts

@@ -1,9 +1,9 @@
-import { b as ProvidedClientTransportOptions, c as ProvidedServerTransportOptions, n as ClientHandshakeOptions, C as Connection, r as ServerHandshakeOptions, T as Transport, A as SessionOptions, m as ClientTransportOptions, e as SessionNoConnection, B as SessionBoundSendFn } from '../context-ebf88958.js';
-import { C as ClientTransport } from '../client-c44d1c69.js';
-import { S as ServerTransport } from '../server-b169b234.js';
+import { b as ProvidedClientTransportOptions, c as ProvidedServerTransportOptions, n as ClientHandshakeOptions, C as Connection, r as ServerHandshakeOptions, T as Transport, A as SessionOptions, m as ClientTransportOptions, e as SessionNoConnection, B as SessionBoundSendFn } from '../context-2c4bb66b.js';
+import { C as ClientTransport } from '../client-8ab60722.js';
+import { S as ServerTransport } from '../server-dc261419.js';
 import { c as TransportClientId, f as PartialTransportMessage, b as OpaqueTransportMessage } from '../message-865bba35.js';
 import { Static } from '@sinclair/typebox';
-import { E as BaseErrorSchemaType, u as Readable, X as ReadableIterator, Y as ReadableResult } from '../services-47b312aa.js';
+import { E as BaseErrorSchemaType, u as Readable, X as ReadableIterator, Y as ReadableResult } from '../services-036dcc97.js';
 import NodeWs from 'ws';
 import http from 'node:http';
 import { W as WsLike } from '../wslike-e0b32dd5.js';

package/dist/testUtil/index.d.ts

@@ -1,9 +1,9 @@
-import { b as ProvidedClientTransportOptions, c as ProvidedServerTransportOptions, n as ClientHandshakeOptions, C as Connection, r as ServerHandshakeOptions, T as Transport, A as SessionOptions, m as ClientTransportOptions, e as SessionNoConnection, B as SessionBoundSendFn } from '../context-ebf88958.js';
-import { C as ClientTransport } from '../client-c44d1c69.js';
-import { S as ServerTransport } from '../server-b169b234.js';
+import { b as ProvidedClientTransportOptions, c as ProvidedServerTransportOptions, n as ClientHandshakeOptions, C as Connection, r as ServerHandshakeOptions, T as Transport, A as SessionOptions, m as ClientTransportOptions, e as SessionNoConnection, B as SessionBoundSendFn } from '../context-2c4bb66b.js';
+import { C as ClientTransport } from '../client-8ab60722.js';
+import { S as ServerTransport } from '../server-dc261419.js';
 import { c as TransportClientId, f as PartialTransportMessage, b as OpaqueTransportMessage } from '../message-865bba35.js';
 import { Static } from '@sinclair/typebox';
-import { E as BaseErrorSchemaType, u as Readable, X as ReadableIterator, Y as ReadableResult } from '../services-47b312aa.js';
+import { E as BaseErrorSchemaType, u as Readable, X as ReadableIterator, Y as ReadableResult } from '../services-036dcc97.js';
 import NodeWs from 'ws';
 import http from 'node:http';
 import { W as WsLike } from '../wslike-e0b32dd5.js';

package/dist/testUtil/index.js

@@ -1,19 +1,19 @@
 import {
   ClientTransport
-} from "../chunk-3XCLUGDM.js";
+} from "../chunk-W337YQLT.js";
 import {
   ServerTransport
-} from "../chunk-OVLZDQF4.js";
+} from "../chunk-TP4OJAKI.js";
 import {
   Connection,
   SessionStateGraph,
   defaultClientTransportOptions,
   defaultTransportOptions
-} from "../chunk-Y2N4EJYG.js";
+} from "../chunk-FTOB2HQN.js";
 import {
   currentProtocolVersion,
   getTracer
-} from "../chunk-HNBYBRJ7.js";
+} from "../chunk-BYA34LH3.js";
 import "../chunk-CC7RN7GI.js";
 import "../chunk-AJGIY2UB.js";
 
@@ -116,12 +116,13 @@ var InMemoryConnection = class extends Connection {
   constructor(pipe) {
     super();
     this.conn = pipe;
+    this.conn.allowHalfOpen = false;
     this.conn.on("data", (data) => {
       for (const cb of this.dataListeners) {
         cb(data);
       }
     });
-    this.conn.on("end", () => {
+    this.conn.on("close", () => {
       for (const cb of this.closeListeners) {
         cb();
       }
@@ -141,6 +142,7 @@ var InMemoryConnection = class extends Connection {
   close() {
     setImmediate(() => {
       this.conn.end();
+      this.conn.emit("close");
     });
   }
 };
@@ -206,6 +208,7 @@ function createMockTransportNetwork(opts) {
     simulatePhantomDisconnect() {
       for (const conn of Object.values(connections.get())) {
         conn.serverToClient.pause();
+        conn.clientToServer.pause();
       }
     },
     async restartServer() {

package/dist/testUtil/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../testUtil/index.ts","../../testUtil/observable/observable.ts","../../testUtil/duplex/duplexPair.ts","../../testUtil/fixtures/mockTransport.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { Static } from '@sinclair/typebox';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  currentProtocolVersion,\n} from '../transport/message';\nimport { Transport } from '../transport/transport';\nimport { Readable, ReadableResult, ReadableIterator } from '../router/streams';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport {\n  defaultClientTransportOptions,\n  defaultTransportOptions,\n} from '../transport/options';\nimport { Connection } from '../transport/connection';\nimport { SessionState } from '../transport/sessionStateMachine/common';\nimport { SessionStateGraph } from '../transport/sessionStateMachine/transitions';\nimport { BaseErrorSchemaType } from '../router/errors';\nimport { ClientTransport } from '../transport/client';\nimport { ServerTransport } from '../transport/server';\nimport { getTracer } from '../tracing';\n\nexport {\n  createMockTransportNetwork,\n  InMemoryConnection,\n} from './fixtures/mockTransport';\n\n/**\n * Creates a WebSocket client that connects to a local server at the specified port.\n * This should only be used for testing.\n * @param port - The port number to connect to.\n * @returns A Promise that resolves to a WebSocket instance.\n */\nexport function createLocalWebSocketClient(port: number): WsLike {\n  const sock = new NodeWs(`ws://localhost:${port}`);\n  sock.binaryType = 'arraybuffer';\n\n  return sock;\n}\n\n/**\n * Creates a WebSocket server instance using the provided HTTP server.\n * Only used as helper for testing.\n * @param server - The HTTP server instance to use for the WebSocket server.\n * @returns A Promise that resolves to the created WebSocket server instance.\n */\nexport function createWebSocketServer(server: http.Server) {\n  return new WebSocketServer({ server });\n}\n\n/**\n * Starts listening on the given server and returns the automatically allocated port number.\n * This should only be used for testing.\n * @param server - The http server to listen on.\n * @returns A promise that resolves with the allocated port number.\n * @throws An error if a port cannot be allocated.\n */\nexport function onWsServerReady(server: http.Server): Promise<number> {\n  return new Promise((resolve, reject) => {\n    server.listen(() => {\n      const addr = server.address();\n      if (typeof addr === 'object' && addr) {\n        resolve(addr.port);\n      } else {\n        reject(new Error(\"couldn't find a port to allocate\"));\n      }\n    });\n  });\n}\n\nconst readableIterators = new WeakMap<\n  Readable<unknown, Static<BaseErrorSchemaType>>,\n  ReadableIterator<unknown, Static<BaseErrorSchemaType>>\n>();\n\n/**\n * A safe way to access {@link Readble}'s iterator multiple times in test helpers.\n *\n * If there are other iteration attempts outside of the test helpers\n * (this function, {@link readNextResult}, and {@link isReadableDone})\n * it will throw an error.\n */\nexport function getReadableIterator<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): ReadableIterator<T, E> {\n  let iter = readableIterators.get(readable) as\n    | ReadableIterator<T, E>\n    | undefined;\n\n  if (!iter) {\n    iter = readable[Symbol.asyncIterator]();\n    readableIterators.set(readable, iter);\n  }\n\n  return iter;\n}\n\n/**\n * Retrieves the next value from {@link Readable}, or throws an error if the Readable is done.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function readNextResult<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): Promise<ReadableResult<T, E>> {\n  const res = await getReadableIterator(readable).next();\n\n  if (res.done) {\n    throw new Error('readNext from a done Readable');\n  }\n\n  return res.value;\n}\n\n/**\n * Checks if the readable is done iterating, it consumes an iteration in the process.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function isReadableDone<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n) {\n  const res = await getReadableIterator(readable).next();\n\n  return res.done;\n}\n\nexport function payloadToTransportMessage<Payload>(\n  payload: Payload,\n): PartialTransportMessage<Payload> {\n  return {\n    streamId: 'stream',\n    controlFlags: 0,\n    payload,\n  };\n}\n\nexport function createDummyTransportMessage() {\n  return payloadToTransportMessage({\n    msg: 'cool',\n    test: Math.random(),\n  });\n}\n\n/**\n * Waits for a message on the transport.\n * @param {Transport} t - The transport to listen to.\n * @param filter - An optional filter function to apply to the received messages.\n * @returns A promise that resolves with the payload of the first message that passes the filter.\n */\nexport async function waitForMessage(\n  t: Transport<Connection>,\n  filter?: (msg: OpaqueTransportMessage) => boolean,\n  rejectMismatch?: boolean,\n) {\n  return new Promise((resolve, reject) => {\n    function cleanup() {\n      t.removeEventListener('message', onMessage);\n    }\n\n    function onMessage(msg: OpaqueTransportMessage) {\n      if (!filter || filter(msg)) {\n        cleanup();\n        resolve(msg.payload);\n      } else if (rejectMismatch) {\n        cleanup();\n        reject(new Error('message didnt match the filter'));\n      }\n    }\n\n    t.addEventListener('message', onMessage);\n  });\n}\n\nexport const testingSessionOptions = defaultTransportOptions;\nexport const testingClientSessionOptions = defaultClientTransportOptions;\n\nexport function dummySession() {\n  return SessionStateGraph.entrypoints.NoConnection(\n    'client',\n    'server',\n    {\n      onSessionGracePeriodElapsed: () => {\n        /* noop */\n      },\n    },\n    testingSessionOptions,\n    currentProtocolVersion,\n    getTracer(),\n  );\n}\n\nexport function getClientSendFn(\n  clientTransport: ClientTransport<Connection>,\n  serverTransport: ServerTransport<Connection>,\n) {\n  const session =\n    clientTransport.sessions.get(serverTransport.clientId) ??\n    clientTransport.createUnconnectedSession(serverTransport.clientId);\n\n  return clientTransport.getSessionBoundSendFn(\n    serverTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getServerSendFn(\n  serverTransport: ServerTransport<Connection>,\n  clientTransport: ClientTransport<Connection>,\n) {\n  const session = serverTransport.sessions.get(clientTransport.clientId);\n  if (!session) {\n    throw new Error('session not found');\n  }\n\n  return serverTransport.getSessionBoundSendFn(\n    clientTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getTransportConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): Array<ConnType> {\n  const connections = [];\n  for (const session of transport.sessions.values()) {\n    if (session.state === SessionState.Connected) {\n      connections.push(session.conn);\n    }\n  }\n\n  return connections;\n}\n\nexport function numberOfConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): number {\n  return getTransportConnections(transport).length;\n}\n\nexport function closeAllConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n) {\n  for (const conn of getTransportConnections(transport)) {\n    conn.close();\n  }\n}\n","/**\n * Represents an observable value that can be subscribed to for changes.\n * This should only be used in tests\n * @template T - The type of the value being observed.\n */\nexport class Observable<T> {\n  value: T;\n  private listeners: Set<(val: T) => void>;\n\n  constructor(initialValue: T) {\n    this.value = initialValue;\n    this.listeners = new Set();\n  }\n\n  /**\n   * Gets the current value of the observable.\n   */\n  get() {\n    return this.value;\n  }\n\n  /**\n   * Sets the current value of the observable. All listeners will get an update with this value.\n   * @param newValue - The new value to set.\n   */\n  set(tx: (preValue: T) => T) {\n    const newValue = tx(this.value);\n    this.value = newValue;\n    this.listeners.forEach((listener) => listener(newValue));\n  }\n\n  /**\n   * Subscribes to changes in the observable value.\n   * @param listener - A callback function that will be called when the value changes.\n   * @returns A function that can be called to unsubscribe from further notifications.\n   */\n  observe(listener: (val: T) => void) {\n    this.listeners.add(listener);\n    listener(this.get());\n\n    return () => this.listeners.delete(listener);\n  }\n\n  /**\n   * Returns the number of listeners currently observing the observable\n   */\n  get listenerCount(): number {\n    return this.listeners.size;\n  }\n}\n","import { Duplex } from 'node:stream';\nimport assert from 'assert';\n\nconst kCallback = Symbol('Callback');\nconst kInitOtherSide = Symbol('InitOtherSide');\n\n// yoinked from https://github.com/nodejs/node/blob/c3a7b29e56a5ada6327ebb622ba746d022685742/lib/internal/streams/duplexpair.js#L55\n// but with types\nclass DuplexSide extends Duplex {\n  private otherSide: DuplexSide | null;\n  private [kCallback]: (() => void) | null;\n\n  constructor() {\n    super();\n    this[kCallback] = null;\n    this.otherSide = null;\n  }\n\n  [kInitOtherSide](otherSide: DuplexSide) {\n    if (this.otherSide === null) {\n      this.otherSide = otherSide;\n    }\n  }\n\n  _read() {\n    const callback = this[kCallback];\n    if (callback) {\n      this[kCallback] = null;\n      callback();\n    }\n  }\n\n  _write(\n    chunk: Uint8Array,\n    _encoding: BufferEncoding,\n    callback: (error?: Error | null) => void,\n  ) {\n    assert(this.otherSide !== null);\n    assert(this.otherSide[kCallback] === null);\n    if (chunk.length === 0) {\n      process.nextTick(callback);\n    } else {\n      this.otherSide.push(chunk);\n      this.otherSide[kCallback] = callback;\n    }\n  }\n\n  _final(callback: (error?: Error | null) => void) {\n    this.otherSide?.on('end', callback);\n    this.otherSide?.push(null);\n  }\n}\n\nexport function duplexPair(): [DuplexSide, DuplexSide] {\n  const side0 = new DuplexSide();\n  const side1 = new DuplexSide();\n  side0[kInitOtherSide](side1);\n  side1[kInitOtherSide](side0);\n\n  return [side0, side1];\n}\n","import { TransportClientId } from '../../transport';\nimport { ClientTransport } from '../../transport/client';\nimport { Connection } from '../../transport/connection';\nimport { ServerTransport } from '../../transport/server';\nimport { Observable } from '../observable/observable';\nimport { ProvidedServerTransportOptions } from '../../transport/options';\nimport { TestSetupHelpers, TestTransportOptions } from './transports';\nimport { Duplex } from 'node:stream';\nimport { duplexPair } from '../duplex/duplexPair';\nimport { nanoid } from 'nanoid';\n\nexport class InMemoryConnection extends Connection {\n  conn: Duplex;\n\n  constructor(pipe: Duplex) {\n    super();\n    this.conn = pipe;\n\n    this.conn.on('data', (data: Uint8Array) => {\n      for (const cb of this.dataListeners) {\n        cb(data);\n      }\n    });\n\n    this.conn.on('end', () => {\n      for (const cb of this.closeListeners) {\n        cb();\n      }\n    });\n\n    this.conn.on('error', (err) => {\n      for (const cb of this.errorListeners) {\n        cb(err);\n      }\n    });\n  }\n\n  send(payload: Uint8Array): boolean {\n    setImmediate(() => {\n      this.conn.write(payload);\n    });\n\n    return true;\n  }\n\n  close(): void {\n    setImmediate(() => {\n      this.conn.end();\n    });\n  }\n}\n\ninterface BidiConnection {\n  id: string;\n  clientToServer: Duplex;\n  serverToClient: Duplex;\n  clientId: TransportClientId;\n  serverId: TransportClientId;\n  handled: boolean;\n}\n\n// we construct a network of transports connected by node streams here\n// so that we can test the transport layer without needing to actually\n// use real network/websocket connections\n// this is useful for testing the transport layer in isolation\n// and allows us to control network conditions in a way that would be\n// difficult with real network connections (e.g. simulating a phantom\n// disconnect, .pause() vs .removeAllListeners('data'), congestion,\n// latency, differences in ws implementations between node and browsers, etc.)\nexport function createMockTransportNetwork(\n  opts?: TestTransportOptions,\n): TestSetupHelpers {\n  // conn id -> [client->server, server->client]\n  const connections = new Observable<Record<string, BidiConnection>>({});\n\n  const transports: Array<MockClientTransport | MockServerTransport> = [];\n  class MockClientTransport extends ClientTransport<InMemoryConnection> {\n    async createNewOutgoingConnection(\n      to: TransportClientId,\n    ): Promise<InMemoryConnection> {\n      const [clientToServer, serverToClient] = duplexPair();\n      await new Promise((resolve) => setImmediate(resolve));\n\n      const connId = nanoid();\n      connections.set((prev) => ({\n        ...prev,\n        [connId]: {\n          id: connId,\n          clientToServer,\n          serverToClient,\n          clientId: this.clientId,\n          serverId: to,\n          handled: false,\n        },\n      }));\n\n      return new InMemoryConnection(clientToServer);\n    }\n  }\n\n  class MockServerTransport extends ServerTransport<InMemoryConnection> {\n    subscribeCleanup: () => void;\n\n    constructor(\n      clientId: TransportClientId,\n      options?: ProvidedServerTransportOptions,\n    ) {\n      super(clientId, options);\n\n      this.subscribeCleanup = connections.observe((conns) => {\n        // look for any unhandled connections\n        for (const conn of Object.values(conns)) {\n          // if we've already handled this connection, skip it\n          // or if it's not for us, skip it\n          if (conn.handled || conn.serverId !== this.clientId) {\n            continue;\n          }\n\n          conn.handled = true;\n          const connection = new InMemoryConnection(conn.serverToClient);\n          this.handleConnection(connection);\n        }\n      });\n    }\n\n    close() {\n      this.subscribeCleanup();\n      super.close();\n    }\n  }\n\n  return {\n    getClientTransport: (id, handshakeOptions) => {\n      const clientTransport = new MockClientTransport(id, opts?.client);\n      if (handshakeOptions) {\n        clientTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(clientTransport);\n\n      return clientTransport;\n    },\n    getServerTransport: (id = 'SERVER', handshakeOptions) => {\n      const serverTransport = new MockServerTransport(id, opts?.server);\n      if (handshakeOptions) {\n        serverTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(serverTransport);\n\n      return serverTransport;\n    },\n    simulatePhantomDisconnect() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.pause();\n      }\n    },\n    async restartServer() {\n      for (const transport of transports) {\n        if (transport.clientId !== 'SERVER') continue;\n        transport.close();\n      }\n\n      // kill all connections while we're at it\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.end();\n        conn.clientToServer.end();\n      }\n    },\n    cleanup() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.end();\n        conn.clientToServer.end();\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;;;ACKjC,IAAM,aAAN,MAAoB;AAAA,EACzB;AAAA,EACQ;AAAA,EAER,YAAY,cAAiB;AAC3B,SAAK,QAAQ;AACb,SAAK,YAAY,oBAAI,IAAI;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM;AACJ,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,IAAwB;AAC1B,UAAM,WAAW,GAAG,KAAK,KAAK;AAC9B,SAAK,QAAQ;AACb,SAAK,UAAU,QAAQ,CAAC,aAAa,SAAS,QAAQ,CAAC;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ,UAA4B;AAClC,SAAK,UAAU,IAAI,QAAQ;AAC3B,aAAS,KAAK,IAAI,CAAC;AAEnB,WAAO,MAAM,KAAK,UAAU,OAAO,QAAQ;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,gBAAwB;AAC1B,WAAO,KAAK,UAAU;AAAA,EACxB;AACF;;;ACjDA,SAAS,cAAc;AACvB,OAAO,YAAY;AAEnB,IAAM,YAAY,OAAO,UAAU;AACnC,IAAM,iBAAiB,OAAO,eAAe;AAI7C,IAAM,aAAN,cAAyB,OAAO;AAAA,EACtB;AAAA,EACR,CAAS,SAAS;AAAA,EAElB,cAAc;AACZ,UAAM;AACN,SAAK,SAAS,IAAI;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,CAAC,cAAc,EAAE,WAAuB;AACtC,QAAI,KAAK,cAAc,MAAM;AAC3B,WAAK,YAAY;AAAA,IACnB;AAAA,EACF;AAAA,EAEA,QAAQ;AACN,UAAM,WAAW,KAAK,SAAS;AAC/B,QAAI,UAAU;AACZ,WAAK,SAAS,IAAI;AAClB,eAAS;AAAA,IACX;AAAA,EACF;AAAA,EAEA,OACE,OACA,WACA,UACA;AACA,WAAO,KAAK,cAAc,IAAI;AAC9B,WAAO,KAAK,UAAU,SAAS,MAAM,IAAI;AACzC,QAAI,MAAM,WAAW,GAAG;AACtB,cAAQ,SAAS,QAAQ;AAAA,IAC3B,OAAO;AACL,WAAK,UAAU,KAAK,KAAK;AACzB,WAAK,UAAU,SAAS,IAAI;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,OAAO,UAA0C;AAC/C,SAAK,WAAW,GAAG,OAAO,QAAQ;AAClC,SAAK,WAAW,KAAK,IAAI;AAAA,EAC3B;AACF;AAEO,SAAS,aAAuC;AACrD,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,cAAc,EAAE,KAAK;AAC3B,QAAM,cAAc,EAAE,KAAK;AAE3B,SAAO,CAAC,OAAO,KAAK;AACtB;;;ACnDA,SAAS,cAAc;AAEhB,IAAM,qBAAN,cAAiC,WAAW;AAAA,EACjD;AAAA,EAEA,YAAY,MAAc;AACxB,UAAM;AACN,SAAK,OAAO;AAEZ,SAAK,KAAK,GAAG,QAAQ,CAAC,SAAqB;AACzC,iBAAW,MAAM,KAAK,eAAe;AACnC,WAAG,IAAI;AAAA,MACT;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,OAAO,MAAM;AACxB,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG;AAAA,MACL;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,CAAC,QAAQ;AAC7B,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,KAAK,SAA8B;AACjC,iBAAa,MAAM;AACjB,WAAK,KAAK,MAAM,OAAO;AAAA,IACzB,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEA,QAAc;AACZ,iBAAa,MAAM;AACjB,WAAK,KAAK,IAAI;AAAA,IAChB,CAAC;AAAA,EACH;AACF;AAmBO,SAAS,2BACd,MACkB;AAElB,QAAM,cAAc,IAAI,WAA2C,CAAC,CAAC;AAErE,QAAM,aAA+D,CAAC;AAAA,EACtE,MAAM,4BAA4B,gBAAoC;AAAA,IACpE,MAAM,4BACJ,IAC6B;AAC7B,YAAM,CAAC,gBAAgB,cAAc,IAAI,WAAW;AACpD,YAAM,IAAI,QAAQ,CAAC,YAAY,aAAa,OAAO,CAAC;AAEpD,YAAM,SAAS,OAAO;AACtB,kBAAY,IAAI,CAAC,UAAU;AAAA,QACzB,GAAG;AAAA,QACH,CAAC,MAAM,GAAG;AAAA,UACR,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,UACA,UAAU,KAAK;AAAA,UACf,UAAU;AAAA,UACV,SAAS;AAAA,QACX;AAAA,MACF,EAAE;AAEF,aAAO,IAAI,mBAAmB,cAAc;AAAA,IAC9C;AAAA,EACF;AAAA,EAEA,MAAM,4BAA4B,gBAAoC;AAAA,IACpE;AAAA,IAEA,YACE,UACA,SACA;AACA,YAAM,UAAU,OAAO;AAEvB,WAAK,mBAAmB,YAAY,QAAQ,CAAC,UAAU;AAErD,mBAAW,QAAQ,OAAO,OAAO,KAAK,GAAG;AAGvC,cAAI,KAAK,WAAW,KAAK,aAAa,KAAK,UAAU;AACnD;AAAA,UACF;AAEA,eAAK,UAAU;AACf,gBAAM,aAAa,IAAI,mBAAmB,KAAK,cAAc;AAC7D,eAAK,iBAAiB,UAAU;AAAA,QAClC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IAEA,QAAQ;AACN,WAAK,iBAAiB;AACtB,YAAM,MAAM;AAAA,IACd;AAAA,EACF;AAEA,SAAO;AAAA,IACL,oBAAoB,CAAC,IAAI,qBAAqB;AAC5C,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,oBAAoB,CAAC,KAAK,UAAU,qBAAqB;AACvD,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,4BAA4B;AAC1B,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,MAAM;AAAA,MAC5B;AAAA,IACF;AAAA,IACA,MAAM,gBAAgB;AACpB,iBAAW,aAAa,YAAY;AAClC,YAAI,UAAU,aAAa;AAAU;AACrC,kBAAU,MAAM;AAAA,MAClB;AAGA,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,IAAI;AACxB,aAAK,eAAe,IAAI;AAAA,MAC1B;AAAA,IACF;AAAA,IACA,UAAU;AACR,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,IAAI;AACxB,aAAK,eAAe,IAAI;AAAA,MAC1B;AAAA,IACF;AAAA,EACF;AACF;;;AH9IO,SAAS,2BAA2B,MAAsB;AAC/D,QAAM,OAAO,IAAI,OAAO,kBAAkB,IAAI,EAAE;AAChD,OAAK,aAAa;AAElB,SAAO;AACT;AAQO,SAAS,sBAAsB,QAAqB;AACzD,SAAO,IAAI,gBAAgB,EAAE,OAAO,CAAC;AACvC;AASO,SAAS,gBAAgB,QAAsC;AACpE,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAO,OAAO,MAAM;AAClB,YAAM,OAAO,OAAO,QAAQ;AAC5B,UAAI,OAAO,SAAS,YAAY,MAAM;AACpC,gBAAQ,KAAK,IAAI;AAAA,MACnB,OAAO;AACL,eAAO,IAAI,MAAM,kCAAkC,CAAC;AAAA,MACtD;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AACH;AAEA,IAAM,oBAAoB,oBAAI,QAG5B;AASK,SAAS,oBACd,UACwB;AACxB,MAAI,OAAO,kBAAkB,IAAI,QAAQ;AAIzC,MAAI,CAAC,MAAM;AACT,WAAO,SAAS,OAAO,aAAa,EAAE;AACtC,sBAAkB,IAAI,UAAU,IAAI;AAAA,EACtC;AAEA,SAAO;AACT;AAOA,eAAsB,eACpB,UAC+B;AAC/B,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,MAAI,IAAI,MAAM;AACZ,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO,IAAI;AACb;AAOA,eAAsB,eACpB,UACA;AACA,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,SAAO,IAAI;AACb;AAEO,SAAS,0BACd,SACkC;AAClC,SAAO;AAAA,IACL,UAAU;AAAA,IACV,cAAc;AAAA,IACd;AAAA,EACF;AACF;AAEO,SAAS,8BAA8B;AAC5C,SAAO,0BAA0B;AAAA,IAC/B,KAAK;AAAA,IACL,MAAM,KAAK,OAAO;AAAA,EACpB,CAAC;AACH;AAQA,eAAsB,eACpB,GACA,QACA,gBACA;AACA,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,aAAS,UAAU;AACjB,QAAE,oBAAoB,WAAW,SAAS;AAAA,IAC5C;AAEA,aAAS,UAAU,KAA6B;AAC9C,UAAI,CAAC,UAAU,OAAO,GAAG,GAAG;AAC1B,gBAAQ;AACR,gBAAQ,IAAI,OAAO;AAAA,MACrB,WAAW,gBAAgB;AACzB,gBAAQ;AACR,eAAO,IAAI,MAAM,gCAAgC,CAAC;AAAA,MACpD;AAAA,IACF;AAEA,MAAE,iBAAiB,WAAW,SAAS;AAAA,EACzC,CAAC;AACH;AAEO,IAAM,wBAAwB;AAC9B,IAAM,8BAA8B;AAEpC,SAAS,eAAe;AAC7B,SAAO,kBAAkB,YAAY;AAAA,IACnC;AAAA,IACA;AAAA,IACA;AAAA,MACE,6BAA6B,MAAM;AAAA,MAEnC;AAAA,IACF;AAAA,IACA;AAAA,IACA;AAAA,IACA,UAAU;AAAA,EACZ;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UACJ,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ,KACrD,gBAAgB,yBAAyB,gBAAgB,QAAQ;AAEnE,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UAAU,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ;AACrE,MAAI,CAAC,SAAS;AACZ,UAAM,IAAI,MAAM,mBAAmB;AAAA,EACrC;AAEA,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,wBACd,WACiB;AACjB,QAAM,cAAc,CAAC;AACrB,aAAW,WAAW,UAAU,SAAS,OAAO,GAAG;AACjD,QAAI,QAAQ,uCAAkC;AAC5C,kBAAY,KAAK,QAAQ,IAAI;AAAA,IAC/B;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,oBACd,WACQ;AACR,SAAO,wBAAwB,SAAS,EAAE;AAC5C;AAEO,SAAS,oBACd,WACA;AACA,aAAW,QAAQ,wBAAwB,SAAS,GAAG;AACrD,SAAK,MAAM;AAAA,EACb;AACF;","names":[]}
+{"version":3,"sources":["../../testUtil/index.ts","../../testUtil/observable/observable.ts","../../testUtil/duplex/duplexPair.ts","../../testUtil/fixtures/mockTransport.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { Static } from '@sinclair/typebox';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  currentProtocolVersion,\n} from '../transport/message';\nimport { Transport } from '../transport/transport';\nimport { Readable, ReadableResult, ReadableIterator } from '../router/streams';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport {\n  defaultClientTransportOptions,\n  defaultTransportOptions,\n} from '../transport/options';\nimport { Connection } from '../transport/connection';\nimport { SessionState } from '../transport/sessionStateMachine/common';\nimport { SessionStateGraph } from '../transport/sessionStateMachine/transitions';\nimport { BaseErrorSchemaType } from '../router/errors';\nimport { ClientTransport } from '../transport/client';\nimport { ServerTransport } from '../transport/server';\nimport { getTracer } from '../tracing';\n\nexport {\n  createMockTransportNetwork,\n  InMemoryConnection,\n} from './fixtures/mockTransport';\n\n/**\n * Creates a WebSocket client that connects to a local server at the specified port.\n * This should only be used for testing.\n * @param port - The port number to connect to.\n * @returns A Promise that resolves to a WebSocket instance.\n */\nexport function createLocalWebSocketClient(port: number): WsLike {\n  const sock = new NodeWs(`ws://localhost:${port}`);\n  sock.binaryType = 'arraybuffer';\n\n  return sock;\n}\n\n/**\n * Creates a WebSocket server instance using the provided HTTP server.\n * Only used as helper for testing.\n * @param server - The HTTP server instance to use for the WebSocket server.\n * @returns A Promise that resolves to the created WebSocket server instance.\n */\nexport function createWebSocketServer(server: http.Server) {\n  return new WebSocketServer({ server });\n}\n\n/**\n * Starts listening on the given server and returns the automatically allocated port number.\n * This should only be used for testing.\n * @param server - The http server to listen on.\n * @returns A promise that resolves with the allocated port number.\n * @throws An error if a port cannot be allocated.\n */\nexport function onWsServerReady(server: http.Server): Promise<number> {\n  return new Promise((resolve, reject) => {\n    server.listen(() => {\n      const addr = server.address();\n      if (typeof addr === 'object' && addr) {\n        resolve(addr.port);\n      } else {\n        reject(new Error(\"couldn't find a port to allocate\"));\n      }\n    });\n  });\n}\n\nconst readableIterators = new WeakMap<\n  Readable<unknown, Static<BaseErrorSchemaType>>,\n  ReadableIterator<unknown, Static<BaseErrorSchemaType>>\n>();\n\n/**\n * A safe way to access {@link Readble}'s iterator multiple times in test helpers.\n *\n * If there are other iteration attempts outside of the test helpers\n * (this function, {@link readNextResult}, and {@link isReadableDone})\n * it will throw an error.\n */\nexport function getReadableIterator<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): ReadableIterator<T, E> {\n  let iter = readableIterators.get(readable) as\n    | ReadableIterator<T, E>\n    | undefined;\n\n  if (!iter) {\n    iter = readable[Symbol.asyncIterator]();\n    readableIterators.set(readable, iter);\n  }\n\n  return iter;\n}\n\n/**\n * Retrieves the next value from {@link Readable}, or throws an error if the Readable is done.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function readNextResult<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n): Promise<ReadableResult<T, E>> {\n  const res = await getReadableIterator(readable).next();\n\n  if (res.done) {\n    throw new Error('readNext from a done Readable');\n  }\n\n  return res.value;\n}\n\n/**\n * Checks if the readable is done iterating, it consumes an iteration in the process.\n *\n * Calling semantics are similar to {@link getReadableIterator}\n */\nexport async function isReadableDone<T, E extends Static<BaseErrorSchemaType>>(\n  readable: Readable<T, E>,\n) {\n  const res = await getReadableIterator(readable).next();\n\n  return res.done;\n}\n\nexport function payloadToTransportMessage<Payload>(\n  payload: Payload,\n): PartialTransportMessage<Payload> {\n  return {\n    streamId: 'stream',\n    controlFlags: 0,\n    payload,\n  };\n}\n\nexport function createDummyTransportMessage() {\n  return payloadToTransportMessage({\n    msg: 'cool',\n    test: Math.random(),\n  });\n}\n\n/**\n * Waits for a message on the transport.\n * @param {Transport} t - The transport to listen to.\n * @param filter - An optional filter function to apply to the received messages.\n * @returns A promise that resolves with the payload of the first message that passes the filter.\n */\nexport async function waitForMessage(\n  t: Transport<Connection>,\n  filter?: (msg: OpaqueTransportMessage) => boolean,\n  rejectMismatch?: boolean,\n) {\n  return new Promise((resolve, reject) => {\n    function cleanup() {\n      t.removeEventListener('message', onMessage);\n    }\n\n    function onMessage(msg: OpaqueTransportMessage) {\n      if (!filter || filter(msg)) {\n        cleanup();\n        resolve(msg.payload);\n      } else if (rejectMismatch) {\n        cleanup();\n        reject(new Error('message didnt match the filter'));\n      }\n    }\n\n    t.addEventListener('message', onMessage);\n  });\n}\n\nexport const testingSessionOptions = defaultTransportOptions;\nexport const testingClientSessionOptions = defaultClientTransportOptions;\n\nexport function dummySession() {\n  return SessionStateGraph.entrypoints.NoConnection(\n    'client',\n    'server',\n    {\n      onSessionGracePeriodElapsed: () => {\n        /* noop */\n      },\n    },\n    testingSessionOptions,\n    currentProtocolVersion,\n    getTracer(),\n  );\n}\n\nexport function getClientSendFn(\n  clientTransport: ClientTransport<Connection>,\n  serverTransport: ServerTransport<Connection>,\n) {\n  const session =\n    clientTransport.sessions.get(serverTransport.clientId) ??\n    clientTransport.createUnconnectedSession(serverTransport.clientId);\n\n  return clientTransport.getSessionBoundSendFn(\n    serverTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getServerSendFn(\n  serverTransport: ServerTransport<Connection>,\n  clientTransport: ClientTransport<Connection>,\n) {\n  const session = serverTransport.sessions.get(clientTransport.clientId);\n  if (!session) {\n    throw new Error('session not found');\n  }\n\n  return serverTransport.getSessionBoundSendFn(\n    clientTransport.clientId,\n    session.id,\n  );\n}\n\nexport function getTransportConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): Array<ConnType> {\n  const connections = [];\n  for (const session of transport.sessions.values()) {\n    if (session.state === SessionState.Connected) {\n      connections.push(session.conn);\n    }\n  }\n\n  return connections;\n}\n\nexport function numberOfConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n): number {\n  return getTransportConnections(transport).length;\n}\n\nexport function closeAllConnections<ConnType extends Connection>(\n  transport: Transport<ConnType>,\n) {\n  for (const conn of getTransportConnections(transport)) {\n    conn.close();\n  }\n}\n","/**\n * Represents an observable value that can be subscribed to for changes.\n * This should only be used in tests\n * @template T - The type of the value being observed.\n */\nexport class Observable<T> {\n  value: T;\n  private listeners: Set<(val: T) => void>;\n\n  constructor(initialValue: T) {\n    this.value = initialValue;\n    this.listeners = new Set();\n  }\n\n  /**\n   * Gets the current value of the observable.\n   */\n  get() {\n    return this.value;\n  }\n\n  /**\n   * Sets the current value of the observable. All listeners will get an update with this value.\n   * @param newValue - The new value to set.\n   */\n  set(tx: (preValue: T) => T) {\n    const newValue = tx(this.value);\n    this.value = newValue;\n    this.listeners.forEach((listener) => listener(newValue));\n  }\n\n  /**\n   * Subscribes to changes in the observable value.\n   * @param listener - A callback function that will be called when the value changes.\n   * @returns A function that can be called to unsubscribe from further notifications.\n   */\n  observe(listener: (val: T) => void) {\n    this.listeners.add(listener);\n    listener(this.get());\n\n    return () => this.listeners.delete(listener);\n  }\n\n  /**\n   * Returns the number of listeners currently observing the observable\n   */\n  get listenerCount(): number {\n    return this.listeners.size;\n  }\n}\n","import { Duplex } from 'node:stream';\nimport assert from 'assert';\n\nconst kCallback = Symbol('Callback');\nconst kInitOtherSide = Symbol('InitOtherSide');\n\n// yoinked from https://github.com/nodejs/node/blob/c3a7b29e56a5ada6327ebb622ba746d022685742/lib/internal/streams/duplexpair.js#L55\n// but with types\nclass DuplexSide extends Duplex {\n  private otherSide: DuplexSide | null;\n  private [kCallback]: (() => void) | null;\n\n  constructor() {\n    super();\n    this[kCallback] = null;\n    this.otherSide = null;\n  }\n\n  [kInitOtherSide](otherSide: DuplexSide) {\n    if (this.otherSide === null) {\n      this.otherSide = otherSide;\n    }\n  }\n\n  _read() {\n    const callback = this[kCallback];\n    if (callback) {\n      this[kCallback] = null;\n      callback();\n    }\n  }\n\n  _write(\n    chunk: Uint8Array,\n    _encoding: BufferEncoding,\n    callback: (error?: Error | null) => void,\n  ) {\n    assert(this.otherSide !== null);\n    assert(this.otherSide[kCallback] === null);\n    if (chunk.length === 0) {\n      process.nextTick(callback);\n    } else {\n      this.otherSide.push(chunk);\n      this.otherSide[kCallback] = callback;\n    }\n  }\n\n  _final(callback: (error?: Error | null) => void) {\n    this.otherSide?.on('end', callback);\n    this.otherSide?.push(null);\n  }\n}\n\nexport function duplexPair(): [DuplexSide, DuplexSide] {\n  const side0 = new DuplexSide();\n  const side1 = new DuplexSide();\n  side0[kInitOtherSide](side1);\n  side1[kInitOtherSide](side0);\n\n  return [side0, side1];\n}\n","import { TransportClientId } from '../../transport';\nimport { ClientTransport } from '../../transport/client';\nimport { Connection } from '../../transport/connection';\nimport { ServerTransport } from '../../transport/server';\nimport { Observable } from '../observable/observable';\nimport { ProvidedServerTransportOptions } from '../../transport/options';\nimport { TestSetupHelpers, TestTransportOptions } from './transports';\nimport { Duplex } from 'node:stream';\nimport { duplexPair } from '../duplex/duplexPair';\nimport { nanoid } from 'nanoid';\n\nexport class InMemoryConnection extends Connection {\n  conn: Duplex;\n\n  constructor(pipe: Duplex) {\n    super();\n    this.conn = pipe;\n    this.conn.allowHalfOpen = false;\n\n    this.conn.on('data', (data: Uint8Array) => {\n      for (const cb of this.dataListeners) {\n        cb(data);\n      }\n    });\n\n    this.conn.on('close', () => {\n      for (const cb of this.closeListeners) {\n        cb();\n      }\n    });\n\n    this.conn.on('error', (err) => {\n      for (const cb of this.errorListeners) {\n        cb(err);\n      }\n    });\n  }\n\n  send(payload: Uint8Array): boolean {\n    setImmediate(() => {\n      this.conn.write(payload);\n    });\n\n    return true;\n  }\n\n  close(): void {\n    setImmediate(() => {\n      this.conn.end();\n      this.conn.emit('close');\n    });\n  }\n}\n\ninterface BidiConnection {\n  id: string;\n  clientToServer: Duplex;\n  serverToClient: Duplex;\n  clientId: TransportClientId;\n  serverId: TransportClientId;\n  handled: boolean;\n}\n\n// we construct a network of transports connected by node streams here\n// so that we can test the transport layer without needing to actually\n// use real network/websocket connections\n// this is useful for testing the transport layer in isolation\n// and allows us to control network conditions in a way that would be\n// difficult with real network connections (e.g. simulating a phantom\n// disconnect, .pause() vs .removeAllListeners('data'), congestion,\n// latency, differences in ws implementations between node and browsers, etc.)\nexport function createMockTransportNetwork(\n  opts?: TestTransportOptions,\n): TestSetupHelpers {\n  // conn id -> [client->server, server->client]\n  const connections = new Observable<Record<string, BidiConnection>>({});\n\n  const transports: Array<MockClientTransport | MockServerTransport> = [];\n  class MockClientTransport extends ClientTransport<InMemoryConnection> {\n    async createNewOutgoingConnection(\n      to: TransportClientId,\n    ): Promise<InMemoryConnection> {\n      const [clientToServer, serverToClient] = duplexPair();\n      await new Promise((resolve) => setImmediate(resolve));\n\n      const connId = nanoid();\n      connections.set((prev) => ({\n        ...prev,\n        [connId]: {\n          id: connId,\n          clientToServer,\n          serverToClient,\n          clientId: this.clientId,\n          serverId: to,\n          handled: false,\n        },\n      }));\n\n      return new InMemoryConnection(clientToServer);\n    }\n  }\n\n  class MockServerTransport extends ServerTransport<InMemoryConnection> {\n    subscribeCleanup: () => void;\n\n    constructor(\n      clientId: TransportClientId,\n      options?: ProvidedServerTransportOptions,\n    ) {\n      super(clientId, options);\n\n      this.subscribeCleanup = connections.observe((conns) => {\n        // look for any unhandled connections\n        for (const conn of Object.values(conns)) {\n          // if we've already handled this connection, skip it\n          // or if it's not for us, skip it\n          if (conn.handled || conn.serverId !== this.clientId) {\n            continue;\n          }\n\n          conn.handled = true;\n          const connection = new InMemoryConnection(conn.serverToClient);\n          this.handleConnection(connection);\n        }\n      });\n    }\n\n    close() {\n      this.subscribeCleanup();\n      super.close();\n    }\n  }\n\n  return {\n    getClientTransport: (id, handshakeOptions) => {\n      const clientTransport = new MockClientTransport(id, opts?.client);\n      if (handshakeOptions) {\n        clientTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(clientTransport);\n\n      return clientTransport;\n    },\n    getServerTransport: (id = 'SERVER', handshakeOptions) => {\n      const serverTransport = new MockServerTransport(id, opts?.server);\n      if (handshakeOptions) {\n        serverTransport.extendHandshake(handshakeOptions);\n      }\n\n      transports.push(serverTransport);\n\n      return serverTransport;\n    },\n    simulatePhantomDisconnect() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.pause();\n        conn.clientToServer.pause();\n      }\n    },\n    async restartServer() {\n      for (const transport of transports) {\n        if (transport.clientId !== 'SERVER') continue;\n        transport.close();\n      }\n\n      // kill all connections while we're at it\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.end();\n        conn.clientToServer.end();\n      }\n    },\n    cleanup() {\n      for (const conn of Object.values(connections.get())) {\n        conn.serverToClient.end();\n        conn.clientToServer.end();\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;;;ACKjC,IAAM,aAAN,MAAoB;AAAA,EACzB;AAAA,EACQ;AAAA,EAER,YAAY,cAAiB;AAC3B,SAAK,QAAQ;AACb,SAAK,YAAY,oBAAI,IAAI;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM;AACJ,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,IAAwB;AAC1B,UAAM,WAAW,GAAG,KAAK,KAAK;AAC9B,SAAK,QAAQ;AACb,SAAK,UAAU,QAAQ,CAAC,aAAa,SAAS,QAAQ,CAAC;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ,UAA4B;AAClC,SAAK,UAAU,IAAI,QAAQ;AAC3B,aAAS,KAAK,IAAI,CAAC;AAEnB,WAAO,MAAM,KAAK,UAAU,OAAO,QAAQ;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,gBAAwB;AAC1B,WAAO,KAAK,UAAU;AAAA,EACxB;AACF;;;ACjDA,SAAS,cAAc;AACvB,OAAO,YAAY;AAEnB,IAAM,YAAY,OAAO,UAAU;AACnC,IAAM,iBAAiB,OAAO,eAAe;AAI7C,IAAM,aAAN,cAAyB,OAAO;AAAA,EACtB;AAAA,EACR,CAAS,SAAS;AAAA,EAElB,cAAc;AACZ,UAAM;AACN,SAAK,SAAS,IAAI;AAClB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,CAAC,cAAc,EAAE,WAAuB;AACtC,QAAI,KAAK,cAAc,MAAM;AAC3B,WAAK,YAAY;AAAA,IACnB;AAAA,EACF;AAAA,EAEA,QAAQ;AACN,UAAM,WAAW,KAAK,SAAS;AAC/B,QAAI,UAAU;AACZ,WAAK,SAAS,IAAI;AAClB,eAAS;AAAA,IACX;AAAA,EACF;AAAA,EAEA,OACE,OACA,WACA,UACA;AACA,WAAO,KAAK,cAAc,IAAI;AAC9B,WAAO,KAAK,UAAU,SAAS,MAAM,IAAI;AACzC,QAAI,MAAM,WAAW,GAAG;AACtB,cAAQ,SAAS,QAAQ;AAAA,IAC3B,OAAO;AACL,WAAK,UAAU,KAAK,KAAK;AACzB,WAAK,UAAU,SAAS,IAAI;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,OAAO,UAA0C;AAC/C,SAAK,WAAW,GAAG,OAAO,QAAQ;AAClC,SAAK,WAAW,KAAK,IAAI;AAAA,EAC3B;AACF;AAEO,SAAS,aAAuC;AACrD,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,QAAQ,IAAI,WAAW;AAC7B,QAAM,cAAc,EAAE,KAAK;AAC3B,QAAM,cAAc,EAAE,KAAK;AAE3B,SAAO,CAAC,OAAO,KAAK;AACtB;;;ACnDA,SAAS,cAAc;AAEhB,IAAM,qBAAN,cAAiC,WAAW;AAAA,EACjD;AAAA,EAEA,YAAY,MAAc;AACxB,UAAM;AACN,SAAK,OAAO;AACZ,SAAK,KAAK,gBAAgB;AAE1B,SAAK,KAAK,GAAG,QAAQ,CAAC,SAAqB;AACzC,iBAAW,MAAM,KAAK,eAAe;AACnC,WAAG,IAAI;AAAA,MACT;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,MAAM;AAC1B,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG;AAAA,MACL;AAAA,IACF,CAAC;AAED,SAAK,KAAK,GAAG,SAAS,CAAC,QAAQ;AAC7B,iBAAW,MAAM,KAAK,gBAAgB;AACpC,WAAG,GAAG;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,KAAK,SAA8B;AACjC,iBAAa,MAAM;AACjB,WAAK,KAAK,MAAM,OAAO;AAAA,IACzB,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEA,QAAc;AACZ,iBAAa,MAAM;AACjB,WAAK,KAAK,IAAI;AACd,WAAK,KAAK,KAAK,OAAO;AAAA,IACxB,CAAC;AAAA,EACH;AACF;AAmBO,SAAS,2BACd,MACkB;AAElB,QAAM,cAAc,IAAI,WAA2C,CAAC,CAAC;AAErE,QAAM,aAA+D,CAAC;AAAA,EACtE,MAAM,4BAA4B,gBAAoC;AAAA,IACpE,MAAM,4BACJ,IAC6B;AAC7B,YAAM,CAAC,gBAAgB,cAAc,IAAI,WAAW;AACpD,YAAM,IAAI,QAAQ,CAAC,YAAY,aAAa,OAAO,CAAC;AAEpD,YAAM,SAAS,OAAO;AACtB,kBAAY,IAAI,CAAC,UAAU;AAAA,QACzB,GAAG;AAAA,QACH,CAAC,MAAM,GAAG;AAAA,UACR,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,UACA,UAAU,KAAK;AAAA,UACf,UAAU;AAAA,UACV,SAAS;AAAA,QACX;AAAA,MACF,EAAE;AAEF,aAAO,IAAI,mBAAmB,cAAc;AAAA,IAC9C;AAAA,EACF;AAAA,EAEA,MAAM,4BAA4B,gBAAoC;AAAA,IACpE;AAAA,IAEA,YACE,UACA,SACA;AACA,YAAM,UAAU,OAAO;AAEvB,WAAK,mBAAmB,YAAY,QAAQ,CAAC,UAAU;AAErD,mBAAW,QAAQ,OAAO,OAAO,KAAK,GAAG;AAGvC,cAAI,KAAK,WAAW,KAAK,aAAa,KAAK,UAAU;AACnD;AAAA,UACF;AAEA,eAAK,UAAU;AACf,gBAAM,aAAa,IAAI,mBAAmB,KAAK,cAAc;AAC7D,eAAK,iBAAiB,UAAU;AAAA,QAClC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IAEA,QAAQ;AACN,WAAK,iBAAiB;AACtB,YAAM,MAAM;AAAA,IACd;AAAA,EACF;AAEA,SAAO;AAAA,IACL,oBAAoB,CAAC,IAAI,qBAAqB;AAC5C,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,oBAAoB,CAAC,KAAK,UAAU,qBAAqB;AACvD,YAAM,kBAAkB,IAAI,oBAAoB,IAAI,MAAM,MAAM;AAChE,UAAI,kBAAkB;AACpB,wBAAgB,gBAAgB,gBAAgB;AAAA,MAClD;AAEA,iBAAW,KAAK,eAAe;AAE/B,aAAO;AAAA,IACT;AAAA,IACA,4BAA4B;AAC1B,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,MAAM;AAC1B,aAAK,eAAe,MAAM;AAAA,MAC5B;AAAA,IACF;AAAA,IACA,MAAM,gBAAgB;AACpB,iBAAW,aAAa,YAAY;AAClC,YAAI,UAAU,aAAa;AAAU;AACrC,kBAAU,MAAM;AAAA,MAClB;AAGA,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,IAAI;AACxB,aAAK,eAAe,IAAI;AAAA,MAC1B;AAAA,IACF;AAAA,IACA,UAAU;AACR,iBAAW,QAAQ,OAAO,OAAO,YAAY,IAAI,CAAC,GAAG;AACnD,aAAK,eAAe,IAAI;AACxB,aAAK,eAAe,IAAI;AAAA,MAC1B;AAAA,IACF;AAAA,EACF;AACF;;;AHjJO,SAAS,2BAA2B,MAAsB;AAC/D,QAAM,OAAO,IAAI,OAAO,kBAAkB,IAAI,EAAE;AAChD,OAAK,aAAa;AAElB,SAAO;AACT;AAQO,SAAS,sBAAsB,QAAqB;AACzD,SAAO,IAAI,gBAAgB,EAAE,OAAO,CAAC;AACvC;AASO,SAAS,gBAAgB,QAAsC;AACpE,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAO,OAAO,MAAM;AAClB,YAAM,OAAO,OAAO,QAAQ;AAC5B,UAAI,OAAO,SAAS,YAAY,MAAM;AACpC,gBAAQ,KAAK,IAAI;AAAA,MACnB,OAAO;AACL,eAAO,IAAI,MAAM,kCAAkC,CAAC;AAAA,MACtD;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AACH;AAEA,IAAM,oBAAoB,oBAAI,QAG5B;AASK,SAAS,oBACd,UACwB;AACxB,MAAI,OAAO,kBAAkB,IAAI,QAAQ;AAIzC,MAAI,CAAC,MAAM;AACT,WAAO,SAAS,OAAO,aAAa,EAAE;AACtC,sBAAkB,IAAI,UAAU,IAAI;AAAA,EACtC;AAEA,SAAO;AACT;AAOA,eAAsB,eACpB,UAC+B;AAC/B,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,MAAI,IAAI,MAAM;AACZ,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO,IAAI;AACb;AAOA,eAAsB,eACpB,UACA;AACA,QAAM,MAAM,MAAM,oBAAoB,QAAQ,EAAE,KAAK;AAErD,SAAO,IAAI;AACb;AAEO,SAAS,0BACd,SACkC;AAClC,SAAO;AAAA,IACL,UAAU;AAAA,IACV,cAAc;AAAA,IACd;AAAA,EACF;AACF;AAEO,SAAS,8BAA8B;AAC5C,SAAO,0BAA0B;AAAA,IAC/B,KAAK;AAAA,IACL,MAAM,KAAK,OAAO;AAAA,EACpB,CAAC;AACH;AAQA,eAAsB,eACpB,GACA,QACA,gBACA;AACA,SAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,aAAS,UAAU;AACjB,QAAE,oBAAoB,WAAW,SAAS;AAAA,IAC5C;AAEA,aAAS,UAAU,KAA6B;AAC9C,UAAI,CAAC,UAAU,OAAO,GAAG,GAAG;AAC1B,gBAAQ;AACR,gBAAQ,IAAI,OAAO;AAAA,MACrB,WAAW,gBAAgB;AACzB,gBAAQ;AACR,eAAO,IAAI,MAAM,gCAAgC,CAAC;AAAA,MACpD;AAAA,IACF;AAEA,MAAE,iBAAiB,WAAW,SAAS;AAAA,EACzC,CAAC;AACH;AAEO,IAAM,wBAAwB;AAC9B,IAAM,8BAA8B;AAEpC,SAAS,eAAe;AAC7B,SAAO,kBAAkB,YAAY;AAAA,IACnC;AAAA,IACA;AAAA,IACA;AAAA,MACE,6BAA6B,MAAM;AAAA,MAEnC;AAAA,IACF;AAAA,IACA;AAAA,IACA;AAAA,IACA,UAAU;AAAA,EACZ;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UACJ,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ,KACrD,gBAAgB,yBAAyB,gBAAgB,QAAQ;AAEnE,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,gBACd,iBACA,iBACA;AACA,QAAM,UAAU,gBAAgB,SAAS,IAAI,gBAAgB,QAAQ;AACrE,MAAI,CAAC,SAAS;AACZ,UAAM,IAAI,MAAM,mBAAmB;AAAA,EACrC;AAEA,SAAO,gBAAgB;AAAA,IACrB,gBAAgB;AAAA,IAChB,QAAQ;AAAA,EACV;AACF;AAEO,SAAS,wBACd,WACiB;AACjB,QAAM,cAAc,CAAC;AACrB,aAAW,WAAW,UAAU,SAAS,OAAO,GAAG;AACjD,QAAI,QAAQ,uCAAkC;AAC5C,kBAAY,KAAK,QAAQ,IAAI;AAAA,IAC/B;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,oBACd,WACQ;AACR,SAAO,wBAAwB,SAAS,EAAE;AAC5C;AAEO,SAAS,oBACd,WACA;AACA,aAAW,QAAQ,wBAAwB,SAAS,GAAG;AACrD,SAAK,MAAM;AAAA,EACb;AACF;","names":[]}

package/dist/transport/impls/ws/client.cjs

@@ -499,13 +499,27 @@ var IdentifiedSession = class extends CommonSession {
    * Index of the message we will send next (excluding handshake)
    */
   seq;
+  /**
+   * Last seq we sent over the wire this session (excluding handshake) and retransmissions
+   */
+  seqSent;
   /**
    * Number of unique messages we've received this session (excluding handshake)
    */
   ack;
   sendBuffer;
   constructor(props) {
-    const { id, to, seq, ack, sendBuffer, telemetry, log, protocolVersion } = props;
+    const {
+      id,
+      to,
+      seq,
+      ack,
+      sendBuffer,
+      telemetry,
+      log,
+      protocolVersion,
+      seqSent: messagesSent
+    } = props;
     super(props);
     this.id = id;
     this.to = to;
@@ -515,6 +529,7 @@ var IdentifiedSession = class extends CommonSession {
     this.telemetry = telemetry;
     this.log = log;
     this.protocolVersion = protocolVersion;
+    this.seqSent = messagesSent;
   }
   get loggingMetadata() {
     const metadata = {
@@ -661,7 +676,7 @@ function coerceErrorString(err) {
 }
 
 // package.json
-var version = "0.205.1";
+var version = "0.205.2";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -825,10 +840,23 @@ var SessionConnected = class extends IdentifiedSession {
     this.ack = seq + 1;
     this.heartbeatMisses = 0;
   }
+  assertSendOrdering(constructedMsg) {
+    if (constructedMsg.seq > this.seqSent + 1) {
+      const msg = `invariant violation: would have sent out of order msg (seq: ${constructedMsg.seq}, expected: ${this.seqSent} + 1)`;
+      this.log?.error(msg, {
+        ...this.loggingMetadata,
+        transportMessage: constructedMsg,
+        tags: ["invariant-violation"]
+      });
+      throw new Error(msg);
+    }
+  }
   send(msg) {
     const constructedMsg = this.constructMsg(msg);
+    this.assertSendOrdering(constructedMsg);
     this.sendBuffer.push(constructedMsg);
     this.conn.send(this.options.codec.toBuffer(constructedMsg));
+    this.seqSent = constructedMsg.seq;
     return constructedMsg.id;
   }
   constructor(props) {
@@ -844,7 +872,9 @@ var SessionConnected = class extends IdentifiedSession {
         this.loggingMetadata
       );
       for (const msg of this.sendBuffer) {
+        this.assertSendOrdering(msg);
         this.conn.send(this.options.codec.toBuffer(msg));
+        this.seqSent = msg.seq;
       }
     }
     this.isActivelyHeartbeating = false;
@@ -910,7 +940,7 @@ var SessionConnected = class extends IdentifiedSession {
         );
       } else {
         const reason = `received out-of-order msg, closing connection (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;
-        this.log?.warn(reason, {
+        this.log?.error(reason, {
           ...this.loggingMetadata,
           transportMessage: parsedMsg,
           tags: ["invariant-violation"]
@@ -988,6 +1018,7 @@ function inheritSharedSession(session) {
     to: session.to,
     seq: session.seq,
     ack: session.ack,
+    seqSent: session.seqSent,
     sendBuffer: session.sendBuffer,
     telemetry: session.telemetry,
     options: session.options,
@@ -1015,6 +1046,7 @@ var SessionStateGraph = {
         to,
         seq: 0,
         ack: 0,
+        seqSent: 0,
         graceExpiryTime: Date.now() + options.sessionDisconnectGraceMs,
         sendBuffer,
         telemetry,
@@ -1137,6 +1169,7 @@ var SessionStateGraph = {
           to,
           seq: 0,
           ack: 0,
+          seqSent: 0,
           sendBuffer: [],
           telemetry: createSessionTelemetryInfo(
             pendingSession.tracer,
@@ -1396,7 +1429,7 @@ var Transport = class {
     });
     this.eventDispatcher.dispatchEvent("sessionTransition", {
       state: session.state,
-      session
+      id: session.id
     });
   }
   updateSession(session) {
@@ -1420,7 +1453,7 @@ var Transport = class {
     this.sessions.set(session.to, session);
     this.eventDispatcher.dispatchEvent("sessionTransition", {
       state: session.state,
-      session
+      id: session.id
     });
   }
   deleteSession(session, options) {