@replit/river 0.200.0-rc.2 → 0.200.0-rc.20

package/dist/util/testHelpers.d.cts

@@ -1,13 +1,15 @@
+import { T as Transport, C as Connection, m as SessionOptions, n as ClientTransportOptions, e as SessionNoConnection, o as ServiceContext, S as Session } from '../context-9c907028.js';
 import { Static } from '@sinclair/typebox';
-import { B as BaseErrorSchemaType, R as ReadStream, O as OkResult, E as ErrResult, T as Transport, C as Connection, S as SessionOptions, a as Session, P as PayloadType, b as ProcedureErrorSchemaType, c as Procedure, d as ServiceContext, e as Result, f as OutputReaderErrorSchema, W as WriteStream } from '../services-970f97bb.js';
-import { P as PartialTransportMessage, O as OpaqueTransportMessage } from '../index-10ebd26a.js';
+import { P as PartialTransportMessage, b as OpaqueTransportMessage } from '../message-59fe53e1.js';
 import NodeWs from 'ws';
 import http from 'node:http';
-import net from 'node:net';
+import { B as BaseErrorSchemaType, R as Readable, a as ReadableIterator, b as ReadableResult, P as PayloadType, c as ProcedureErrorSchemaType, d as Procedure, e as Result, f as ReaderErrorSchema, W as Writable } from '../services-aa49a9fb.js';
 import { W as WsLike } from '../wslike-e0b32dd5.js';
+import '@sinclair/typebox/errors';
 import '../types-3e5768ec.js';
 import '@sinclair/typebox/value';
 import '@opentelemetry/api';
+import '../client-75090f07.js';
 
 /**
  * Creates a WebSocket client that connects to a local server at the specified port.
@@ -31,33 +33,26 @@ declare function createWebSocketServer(server: http.Server): NodeWs.Server<typeo
  * @throws An error if a port cannot be allocated.
  */
 declare function onWsServerReady(server: http.Server): Promise<number>;
-declare function onUdsServeReady(server: net.Server, path: string): Promise<void>;
-declare function getIteratorFromStream<T, E extends Static<BaseErrorSchemaType>>(readStream: ReadStream<T, E>): {
-    next(): Promise<{
-        done: true;
-        value: undefined;
-    } | {
-        done: false;
-        value: OkResult<T> | ErrResult<E | {
-            readonly code: "STREAM_DRAINED";
-            readonly message: "Stream was drained";
-        }>;
-    }>;
-};
 /**
- * Retrieves the next value from an async iterable iterator.
- * @param iter The async iterable iterator.
- * @returns A promise that resolves to the next value from the iterator.
+ * A safe way to access {@link Readble}'s iterator multiple times in test helpers.
+ *
+ * If there are other iteration attempts outside of the test helpers
+ * (this function, {@link readNextResult}, and {@link isReadableDone})
+ * it will throw an error.
+ */
+declare function getReadableIterator<T, E extends Static<BaseErrorSchemaType>>(readable: Readable<T, E>): ReadableIterator<T, E>;
+/**
+ * Retrieves the next value from {@link Readable}, or throws an error if the Readable is done.
+ *
+ * Calling semantics are similar to {@link getReadableIterator}
+ */
+declare function readNextResult<T, E extends Static<BaseErrorSchemaType>>(readable: Readable<T, E>): Promise<ReadableResult<T, E>>;
+/**
+ * Checks if the readable is done iterating, it consumes an iteration in the process.
+ *
+ * Calling semantics are similar to {@link getReadableIterator}
  */
-declare function iterNext<T>(iter: {
-    next(): Promise<{
-        done: false;
-        value: T;
-    } | {
-        done: true;
-        value: undefined;
-    }>;
-}): Promise<T>;
+declare function isReadableDone<T, E extends Static<BaseErrorSchemaType>>(readable: Readable<T, E>): Promise<boolean>;
 declare function payloadToTransportMessage<Payload>(payload: Payload): PartialTransportMessage<Payload>;
 declare function createDummyTransportMessage(): PartialTransportMessage<{
     msg: string;
@@ -71,14 +66,22 @@ declare function createDummyTransportMessage(): PartialTransportMessage<{
  */
 declare function waitForMessage(t: Transport<Connection>, filter?: (msg: OpaqueTransportMessage) => boolean, rejectMismatch?: boolean): Promise<unknown>;
 declare const testingSessionOptions: SessionOptions;
-declare function dummySession(): Session<Connection>;
-declare function asClientRpc<State extends object, Init extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'rpc', Init, null, Output, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => Promise<Result<Static<Output>, Static<Err> | Static<typeof OutputReaderErrorSchema>>>;
-declare function asClientStream<State extends object, Init extends PayloadType, Input extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'stream', Init, Input, Output, Err>, init?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): [WriteStream<Static<Input>>, ReadStream<Static<Output>, Static<Err>>];
-declare function asClientSubscription<State extends object, Init extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'subscription', Init, null, Output, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => ReadStream<Static<Output>, Static<Err>>;
-declare function asClientUpload<State extends object, Init extends PayloadType, Input extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'upload', Init, Input, Output, Err>, init?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): [
-    WriteStream<Static<Input>>,
-    () => Promise<Result<Static<Output>, Static<Err>>>
-];
-declare const getUnixSocketPath: () => string;
+declare const testingClientSessionOptions: ClientTransportOptions;
+declare function dummySession(): SessionNoConnection;
+declare function asClientRpc<State extends object, Init extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'rpc', Init, null, Res, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => Promise<Result<Static<Res>, Static<Err> | Static<typeof ReaderErrorSchema>>>;
+declare function asClientStream<State extends object, Init extends PayloadType, Req extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'stream', Init, Req, Res, Err>, reqInit?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): {
+    reqWritable: Writable<Static<Req>>;
+    resReadable: Readable<Static<Res>, Static<Err>>;
+};
+declare function asClientSubscription<State extends object, Init extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'subscription', Init, null, Res, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => {
+    resReadable: Readable<Static<Res>, Static<Err>>;
+};
+declare function asClientUpload<State extends object, Init extends PayloadType, Req extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'upload', Init, Req, Res, Err>, reqInit?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): {
+    reqWritable: Writable<Static<Req>>;
+    finalize: () => Promise<Result<Static<Res>, Static<Err>>>;
+};
+declare function getTransportConnections<ConnType extends Connection>(transport: Transport<ConnType>): Array<ConnType>;
+declare function numberOfConnections<ConnType extends Connection>(transport: Transport<ConnType>): number;
+declare function closeAllConnections<ConnType extends Connection>(transport: Transport<ConnType>): void;
 
-export { asClientRpc, asClientStream, asClientSubscription, asClientUpload, createDummyTransportMessage, createLocalWebSocketClient, createWebSocketServer, dummySession, getIteratorFromStream, getUnixSocketPath, iterNext, onUdsServeReady, onWsServerReady, payloadToTransportMessage, testingSessionOptions, waitForMessage };
+export { asClientRpc, asClientStream, asClientSubscription, asClientUpload, closeAllConnections, createDummyTransportMessage, createLocalWebSocketClient, createWebSocketServer, dummySession, getReadableIterator, getTransportConnections, isReadableDone, numberOfConnections, onWsServerReady, payloadToTransportMessage, readNextResult, testingClientSessionOptions, testingSessionOptions, waitForMessage };

package/dist/util/testHelpers.d.ts

@@ -1,13 +1,15 @@
+import { T as Transport, C as Connection, m as SessionOptions, n as ClientTransportOptions, e as SessionNoConnection, o as ServiceContext, S as Session } from '../context-9c907028.js';
 import { Static } from '@sinclair/typebox';
-import { B as BaseErrorSchemaType, R as ReadStream, O as OkResult, E as ErrResult, T as Transport, C as Connection, S as SessionOptions, a as Session, P as PayloadType, b as ProcedureErrorSchemaType, c as Procedure, d as ServiceContext, e as Result, f as OutputReaderErrorSchema, W as WriteStream } from '../services-970f97bb.js';
-import { P as PartialTransportMessage, O as OpaqueTransportMessage } from '../index-10ebd26a.js';
+import { P as PartialTransportMessage, b as OpaqueTransportMessage } from '../message-59fe53e1.js';
 import NodeWs from 'ws';
 import http from 'node:http';
-import net from 'node:net';
+import { B as BaseErrorSchemaType, R as Readable, a as ReadableIterator, b as ReadableResult, P as PayloadType, c as ProcedureErrorSchemaType, d as Procedure, e as Result, f as ReaderErrorSchema, W as Writable } from '../services-aa49a9fb.js';
 import { W as WsLike } from '../wslike-e0b32dd5.js';
+import '@sinclair/typebox/errors';
 import '../types-3e5768ec.js';
 import '@sinclair/typebox/value';
 import '@opentelemetry/api';
+import '../client-75090f07.js';
 
 /**
  * Creates a WebSocket client that connects to a local server at the specified port.
@@ -31,33 +33,26 @@ declare function createWebSocketServer(server: http.Server): NodeWs.Server<typeo
  * @throws An error if a port cannot be allocated.
  */
 declare function onWsServerReady(server: http.Server): Promise<number>;
-declare function onUdsServeReady(server: net.Server, path: string): Promise<void>;
-declare function getIteratorFromStream<T, E extends Static<BaseErrorSchemaType>>(readStream: ReadStream<T, E>): {
-    next(): Promise<{
-        done: true;
-        value: undefined;
-    } | {
-        done: false;
-        value: OkResult<T> | ErrResult<E | {
-            readonly code: "STREAM_DRAINED";
-            readonly message: "Stream was drained";
-        }>;
-    }>;
-};
 /**
- * Retrieves the next value from an async iterable iterator.
- * @param iter The async iterable iterator.
- * @returns A promise that resolves to the next value from the iterator.
+ * A safe way to access {@link Readble}'s iterator multiple times in test helpers.
+ *
+ * If there are other iteration attempts outside of the test helpers
+ * (this function, {@link readNextResult}, and {@link isReadableDone})
+ * it will throw an error.
+ */
+declare function getReadableIterator<T, E extends Static<BaseErrorSchemaType>>(readable: Readable<T, E>): ReadableIterator<T, E>;
+/**
+ * Retrieves the next value from {@link Readable}, or throws an error if the Readable is done.
+ *
+ * Calling semantics are similar to {@link getReadableIterator}
+ */
+declare function readNextResult<T, E extends Static<BaseErrorSchemaType>>(readable: Readable<T, E>): Promise<ReadableResult<T, E>>;
+/**
+ * Checks if the readable is done iterating, it consumes an iteration in the process.
+ *
+ * Calling semantics are similar to {@link getReadableIterator}
  */
-declare function iterNext<T>(iter: {
-    next(): Promise<{
-        done: false;
-        value: T;
-    } | {
-        done: true;
-        value: undefined;
-    }>;
-}): Promise<T>;
+declare function isReadableDone<T, E extends Static<BaseErrorSchemaType>>(readable: Readable<T, E>): Promise<boolean>;
 declare function payloadToTransportMessage<Payload>(payload: Payload): PartialTransportMessage<Payload>;
 declare function createDummyTransportMessage(): PartialTransportMessage<{
     msg: string;
@@ -71,14 +66,22 @@ declare function createDummyTransportMessage(): PartialTransportMessage<{
  */
 declare function waitForMessage(t: Transport<Connection>, filter?: (msg: OpaqueTransportMessage) => boolean, rejectMismatch?: boolean): Promise<unknown>;
 declare const testingSessionOptions: SessionOptions;
-declare function dummySession(): Session<Connection>;
-declare function asClientRpc<State extends object, Init extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'rpc', Init, null, Output, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => Promise<Result<Static<Output>, Static<Err> | Static<typeof OutputReaderErrorSchema>>>;
-declare function asClientStream<State extends object, Init extends PayloadType, Input extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'stream', Init, Input, Output, Err>, init?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): [WriteStream<Static<Input>>, ReadStream<Static<Output>, Static<Err>>];
-declare function asClientSubscription<State extends object, Init extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'subscription', Init, null, Output, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => ReadStream<Static<Output>, Static<Err>>;
-declare function asClientUpload<State extends object, Init extends PayloadType, Input extends PayloadType, Output extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'upload', Init, Input, Output, Err>, init?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): [
-    WriteStream<Static<Input>>,
-    () => Promise<Result<Static<Output>, Static<Err>>>
-];
-declare const getUnixSocketPath: () => string;
+declare const testingClientSessionOptions: ClientTransportOptions;
+declare function dummySession(): SessionNoConnection;
+declare function asClientRpc<State extends object, Init extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'rpc', Init, null, Res, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => Promise<Result<Static<Res>, Static<Err> | Static<typeof ReaderErrorSchema>>>;
+declare function asClientStream<State extends object, Init extends PayloadType, Req extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'stream', Init, Req, Res, Err>, reqInit?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): {
+    reqWritable: Writable<Static<Req>>;
+    resReadable: Readable<Static<Res>, Static<Err>>;
+};
+declare function asClientSubscription<State extends object, Init extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'subscription', Init, null, Res, Err>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): (msg: Static<Init>) => {
+    resReadable: Readable<Static<Res>, Static<Err>>;
+};
+declare function asClientUpload<State extends object, Init extends PayloadType, Req extends PayloadType, Res extends PayloadType, Err extends ProcedureErrorSchemaType>(state: State, proc: Procedure<State, 'upload', Init, Req, Res, Err>, reqInit?: Static<Init>, extendedContext?: Omit<ServiceContext, 'state'>, session?: Session<Connection>): {
+    reqWritable: Writable<Static<Req>>;
+    finalize: () => Promise<Result<Static<Res>, Static<Err>>>;
+};
+declare function getTransportConnections<ConnType extends Connection>(transport: Transport<ConnType>): Array<ConnType>;
+declare function numberOfConnections<ConnType extends Connection>(transport: Transport<ConnType>): number;
+declare function closeAllConnections<ConnType extends Connection>(transport: Transport<ConnType>): void;
 
-export { asClientRpc, asClientStream, asClientSubscription, asClientUpload, createDummyTransportMessage, createLocalWebSocketClient, createWebSocketServer, dummySession, getIteratorFromStream, getUnixSocketPath, iterNext, onUdsServeReady, onWsServerReady, payloadToTransportMessage, testingSessionOptions, waitForMessage };
+export { asClientRpc, asClientStream, asClientSubscription, asClientUpload, closeAllConnections, createDummyTransportMessage, createLocalWebSocketClient, createWebSocketServer, dummySession, getReadableIterator, getTransportConnections, isReadableDone, numberOfConnections, onWsServerReady, payloadToTransportMessage, readNextResult, testingClientSessionOptions, testingSessionOptions, waitForMessage };

package/dist/util/testHelpers.js

@@ -1,22 +1,23 @@
 import {
   Err,
   Ok,
-  ReadStreamImpl,
+  ReadableImpl,
   UNCAUGHT_ERROR_CODE,
-  WriteStreamImpl
-} from "../chunk-7CKIN3JT.js";
+  WritableImpl
+} from "../chunk-YBCQVIPR.js";
 import {
-  Session,
+  SessionStateGraph,
+  defaultClientTransportOptions,
   defaultTransportOptions
-} from "../chunk-NQWDT6GS.js";
+} from "../chunk-OLWVR5AB.js";
 import {
-  coerceErrorString
-} from "../chunk-S5RL45KH.js";
+  coerceErrorString,
+  currentProtocolVersion
+} from "../chunk-BYCR4VEM.js";
 import "../chunk-4PVU7J25.js";
 
 // util/testHelpers.ts
 import NodeWs, { WebSocketServer } from "ws";
-import { nanoid } from "nanoid";
 function createLocalWebSocketClient(port) {
   const sock = new NodeWs(`ws://localhost:${port}`);
   sock.binaryType = "arraybuffer";
@@ -37,16 +38,25 @@ function onWsServerReady(server) {
     });
   });
 }
-function onUdsServeReady(server, path) {
-  return new Promise((resolve) => {
-    server.listen(path, resolve);
-  });
-}
-function getIteratorFromStream(readStream) {
-  return readStream[Symbol.asyncIterator]();
-}
-async function iterNext(iter) {
-  return await iter.next().then((res) => res.value);
+var readableIterators = /* @__PURE__ */ new WeakMap();
+function getReadableIterator(readable) {
+  let iter = readableIterators.get(readable);
+  if (!iter) {
+    iter = readable[Symbol.asyncIterator]();
+    readableIterators.set(readable, iter);
+  }
+  return iter;
+}
+async function readNextResult(readable) {
+  const res = await getReadableIterator(readable).next();
+  if (res.done) {
+    throw new Error("readNext from a done Readable");
+  }
+  return res.value;
+}
+async function isReadableDone(readable) {
+  const res = await getReadableIterator(readable).next();
+  return res.done;
 }
 function payloadToTransportMessage(payload) {
   return {
@@ -83,113 +93,138 @@ function catchProcError(err) {
   return Err({ code: UNCAUGHT_ERROR_CODE, message: errorMsg });
 }
 var testingSessionOptions = defaultTransportOptions;
+var testingClientSessionOptions = defaultClientTransportOptions;
 function dummySession() {
-  return new Session(
-    void 0,
+  return SessionStateGraph.entrypoints.NoConnection(
     "client",
     "server",
-    testingSessionOptions
+    {
+      onSessionGracePeriodElapsed: () => {
+      }
+    },
+    testingSessionOptions,
+    currentProtocolVersion
   );
 }
 function dummyCtx(state, session, extendedContext) {
   return {
     ...extendedContext,
-    from: session.from,
     state,
+    sessionId: session.id,
+    from: session.from,
     metadata: {},
-    abortController: new AbortController(),
-    clientAbortSignal: new AbortController().signal,
-    onRequestFinished: () => void 0
+    // TODO might wanna hook these up!
+    cancel: () => void 0,
+    signal: new AbortController().signal
   };
 }
 function asClientRpc(state, proc, extendedContext, session = dummySession()) {
   return async (msg) => {
-    return proc.handler(dummyCtx(state, session, extendedContext), msg).catch(catchProcError);
+    return proc.handler({
+      ctx: dummyCtx(state, session, extendedContext),
+      reqInit: msg
+    }).catch(catchProcError);
   };
 }
-function createOutputPipe() {
-  const reader = new ReadStreamImpl(() => {
-    void Promise.resolve().then(() => {
-      writer.triggerCloseRequest();
-    });
-  });
-  const writer = new WriteStreamImpl(
+function createResponsePipe() {
+  const readable = new ReadableImpl();
+  const writable = new WritableImpl(
     (v) => {
-      reader.pushValue(v);
+      readable._pushValue(v);
+    },
+    () => {
+      void Promise.resolve().then(() => {
+        readable._triggerClose();
+      });
     }
   );
-  writer.onClose(() => {
-    void Promise.resolve().then(() => {
-      reader.triggerClose();
-    });
-  });
-  return { reader, writer };
+  return { readable, writable };
 }
-function createInputPipe() {
-  const reader = new ReadStreamImpl(() => {
-    void Promise.resolve().then(() => {
-      writer.triggerCloseRequest();
-    });
-  });
-  const writer = new WriteStreamImpl((v) => {
-    reader.pushValue(Ok(v));
-  });
-  writer.onClose(() => {
-    void Promise.resolve().then(() => {
-      reader.triggerClose();
-    });
-  });
-  return { reader, writer };
-}
-function asClientStream(state, proc, init, extendedContext, session = dummySession()) {
-  const inputPipe = createInputPipe();
-  const outputPipe = createOutputPipe();
-  void proc.handler(
-    dummyCtx(state, session, extendedContext),
-    init ?? {},
-    inputPipe.reader,
-    outputPipe.writer
-  ).catch((err) => outputPipe.writer.write(catchProcError(err)));
-  return [inputPipe.writer, outputPipe.reader];
+function createRequestPipe() {
+  const readable = new ReadableImpl();
+  const writable = new WritableImpl(
+    (v) => {
+      readable._pushValue(Ok(v));
+    },
+    () => {
+      void Promise.resolve().then(() => {
+        readable._triggerClose();
+      });
+    }
+  );
+  return { readable, writable };
+}
+function asClientStream(state, proc, reqInit, extendedContext, session = dummySession()) {
+  const requestPipe = createRequestPipe();
+  const responsePipe = createResponsePipe();
+  void proc.handler({
+    ctx: dummyCtx(state, session, extendedContext),
+    reqInit: reqInit ?? {},
+    reqReadable: requestPipe.readable,
+    resWritable: responsePipe.writable
+  }).catch((err) => responsePipe.writable.write(catchProcError(err)));
+  return {
+    reqWritable: requestPipe.writable,
+    resReadable: responsePipe.readable
+  };
 }
 function asClientSubscription(state, proc, extendedContext, session = dummySession()) {
-  const outputPipe = createOutputPipe();
+  const responsePipe = createResponsePipe();
   return (msg) => {
-    void proc.handler(
-      dummyCtx(state, session, extendedContext),
-      msg,
-      outputPipe.writer
-    ).catch((err) => outputPipe.writer.write(catchProcError(err)));
-    return outputPipe.reader;
+    void proc.handler({
+      ctx: dummyCtx(state, session, extendedContext),
+      reqInit: msg,
+      resWritable: responsePipe.writable
+    }).catch(
+      (err) => responsePipe.writable.write(catchProcError(err))
+    );
+    return { resReadable: responsePipe.readable };
   };
 }
-function asClientUpload(state, proc, init, extendedContext, session = dummySession()) {
-  const inputPipe = createInputPipe();
-  const result = proc.handler(
-    dummyCtx(state, session, extendedContext),
-    init ?? {},
-    inputPipe.reader
-  ).catch(catchProcError);
-  return [inputPipe.writer, () => result];
-}
-var getUnixSocketPath = () => {
-  return process.platform === "win32" ? `\\\\?\\pipe\\${nanoid()}` : `/tmp/${nanoid()}.sock`;
-};
+function asClientUpload(state, proc, reqInit, extendedContext, session = dummySession()) {
+  const requestPipe = createRequestPipe();
+  const result = proc.handler({
+    ctx: dummyCtx(state, session, extendedContext),
+    reqInit: reqInit ?? {},
+    reqReadable: requestPipe.readable
+  }).catch(catchProcError);
+  return { reqWritable: requestPipe.writable, finalize: () => result };
+}
+function getTransportConnections(transport) {
+  const connections = [];
+  for (const session of transport.sessions.values()) {
+    if (session.state === "Connected" /* Connected */) {
+      connections.push(session.conn);
+    }
+  }
+  return connections;
+}
+function numberOfConnections(transport) {
+  return getTransportConnections(transport).length;
+}
+function closeAllConnections(transport) {
+  for (const conn of getTransportConnections(transport)) {
+    conn.close();
+  }
+}
 export {
   asClientRpc,
   asClientStream,
   asClientSubscription,
   asClientUpload,
+  closeAllConnections,
   createDummyTransportMessage,
   createLocalWebSocketClient,
   createWebSocketServer,
   dummySession,
-  getIteratorFromStream,
-  getUnixSocketPath,
-  iterNext,
-  onUdsServeReady,
+  getReadableIterator,
+  getTransportConnections,
+  isReadableDone,
+  numberOfConnections,
   onWsServerReady,
   payloadToTransportMessage,
+  readNextResult,
+  testingClientSessionOptions,
   testingSessionOptions,
   waitForMessage
 };

package/dist/util/testHelpers.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../util/testHelpers.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport {\n  Err,\n  Ok,\n  PayloadType,\n  Procedure,\n  Result,\n  ProcedureErrorSchemaType,\n  InputReaderErrorSchema,\n  OutputReaderErrorSchema,\n  ServiceContext,\n  ProcedureHandlerContext,\n  UNCAUGHT_ERROR_CODE,\n} from '../router';\nimport { Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport net from 'node:net';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n} from '../transport/message';\nimport { coerceErrorString } from './stringify';\nimport { Connection, Session, SessionOptions } from '../transport/session';\nimport { Transport } from '../transport/transport';\nimport {\n  ReadStream,\n  ReadStreamImpl,\n  WriteStream,\n  WriteStreamImpl,\n} from '../router/streams';\nimport { WsLike } from '../transport/impls/ws/wslike';\nimport { defaultTransportOptions } from '../transport/options';\nimport { BaseErrorSchemaType } from '../router/result';\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\nexport function onUdsServeReady(\n  server: net.Server,\n  path: string,\n): Promise<void> {\n  return new Promise<void>((resolve) => {\n    server.listen(path, resolve);\n  });\n}\n\nexport function getIteratorFromStream<T, E extends Static<BaseErrorSchemaType>>(\n  readStream: ReadStream<T, E>,\n) {\n  return readStream[Symbol.asyncIterator]();\n}\n\n/**\n * Retrieves the next value from an async iterable iterator.\n * @param iter The async iterable iterator.\n * @returns A promise that resolves to the next value from the iterator.\n */\nexport async function iterNext<T>(iter: {\n  next(): Promise<\n    | {\n        done: false;\n        value: T;\n      }\n    | {\n        done: true;\n        value: undefined;\n      }\n  >;\n}) {\n  return await iter.next().then((res) => res.value as T);\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\nfunction catchProcError(err: unknown) {\n  const errorMsg = coerceErrorString(err);\n  return Err({ code: UNCAUGHT_ERROR_CODE, message: errorMsg });\n}\n\nexport const testingSessionOptions: SessionOptions = defaultTransportOptions;\n\nexport function dummySession() {\n  return new Session<Connection>(\n    undefined,\n    'client',\n    'server',\n    testingSessionOptions,\n  );\n}\n\nfunction dummyCtx<State>(\n  state: State,\n  session: Session<Connection>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n): ProcedureHandlerContext<State> {\n  return {\n    ...extendedContext,\n    from: session.from,\n    state,\n    metadata: {},\n    abortController: new AbortController(),\n    clientAbortSignal: new AbortController().signal,\n    onRequestFinished: () => undefined,\n  };\n}\n\nexport function asClientRpc<\n  State extends object,\n  Init extends PayloadType,\n  Output extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'rpc', Init, null, Output, Err>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  return async (\n    msg: Static<Init>,\n  ): Promise<\n    Result<Static<Output>, Static<Err> | Static<typeof OutputReaderErrorSchema>>\n  > => {\n    return proc\n      .handler(dummyCtx(state, session, extendedContext), msg)\n      .catch(catchProcError);\n  };\n}\n\nfunction createOutputPipe<\n  Output extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(): {\n  reader: ReadStream<\n    Static<Output>,\n    Static<Err> | Static<typeof OutputReaderErrorSchema>\n  >;\n  writer: WriteStream<Result<Static<Output>, Static<Err>>>;\n} {\n  const reader = new ReadStreamImpl<\n    Static<Output>,\n    Static<Err> | Static<typeof OutputReaderErrorSchema>\n  >(() => {\n    // Make it async to simulate request going over the wire\n    // using promises so that we don't get affected by fake timers.\n    void Promise.resolve().then(() => {\n      writer.triggerCloseRequest();\n    });\n  });\n  const writer = new WriteStreamImpl<Result<Static<Output>, Static<Err>>>(\n    (v) => {\n      reader.pushValue(v);\n    },\n  );\n  writer.onClose(() => {\n    // Make it async to simulate request going over the wire\n    // using promises so that we don't get affected by fake timers.\n    void Promise.resolve().then(() => {\n      reader.triggerClose();\n    });\n  });\n\n  return { reader, writer };\n}\n\nfunction createInputPipe<Input extends PayloadType>(): {\n  reader: ReadStream<Static<Input>, Static<typeof InputReaderErrorSchema>>;\n  writer: WriteStream<Static<Input>>;\n} {\n  const reader = new ReadStreamImpl<\n    Static<Input>,\n    Static<typeof InputReaderErrorSchema>\n  >(() => {\n    // Make it async to simulate request going over the wire\n    // using promises so that we don't get affected by fake timers.\n    void Promise.resolve().then(() => {\n      writer.triggerCloseRequest();\n    });\n  });\n  const writer = new WriteStreamImpl<Static<Input>>((v) => {\n    reader.pushValue(Ok(v));\n  });\n  writer.onClose(() => {\n    // Make it async to simulate request going over the wire\n    // using promises so that we don't get affected by fake timers.\n    void Promise.resolve().then(() => {\n      reader.triggerClose();\n    });\n  });\n\n  return { reader, writer };\n}\n\nexport function asClientStream<\n  State extends object,\n  Init extends PayloadType,\n  Input extends PayloadType,\n  Output extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'stream', Init, Input, Output, Err>,\n  init?: Static<Init>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n): [WriteStream<Static<Input>>, ReadStream<Static<Output>, Static<Err>>] {\n  const inputPipe = createInputPipe<Input>();\n  const outputPipe = createOutputPipe<Output, Err>();\n\n  void proc\n    .handler(\n      dummyCtx(state, session, extendedContext),\n      init ?? {},\n      inputPipe.reader,\n      outputPipe.writer,\n    )\n    .catch((err: unknown) => outputPipe.writer.write(catchProcError(err)));\n\n  return [inputPipe.writer, outputPipe.reader];\n}\n\nexport function asClientSubscription<\n  State extends object,\n  Init extends PayloadType,\n  Output extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'subscription', Init, null, Output, Err>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n): (msg: Static<Init>) => ReadStream<Static<Output>, Static<Err>> {\n  const outputPipe = createOutputPipe<Output, Err>();\n\n  return (msg: Static<Init>) => {\n    void proc\n      .handler(\n        dummyCtx(state, session, extendedContext),\n        msg,\n        outputPipe.writer,\n      )\n      .catch((err: unknown) => outputPipe.writer.write(catchProcError(err)));\n\n    return outputPipe.reader;\n  };\n}\n\nexport function asClientUpload<\n  State extends object,\n  Init extends PayloadType,\n  Input extends PayloadType,\n  Output extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'upload', Init, Input, Output, Err>,\n  init?: Static<Init>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n): [\n  WriteStream<Static<Input>>,\n  () => Promise<Result<Static<Output>, Static<Err>>>,\n] {\n  const inputPipe = createInputPipe<Input>();\n  const result = proc\n    .handler(\n      dummyCtx(state, session, extendedContext),\n      init ?? {},\n      inputPipe.reader,\n    )\n    .catch(catchProcError);\n\n  return [inputPipe.writer, () => result];\n}\n\nexport const getUnixSocketPath = () => {\n  // https://nodejs.org/api/net.html#identifying-paths-for-ipc-connections\n  return process.platform === 'win32'\n    ? `\\\\\\\\?\\\\pipe\\\\${nanoid()}`\n    : `/tmp/${nanoid()}.sock`;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;AAgBxC,SAAS,cAAc;AAyBhB,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;AAEO,SAAS,gBACd,QACA,MACe;AACf,SAAO,IAAI,QAAc,CAAC,YAAY;AACpC,WAAO,OAAO,MAAM,OAAO;AAAA,EAC7B,CAAC;AACH;AAEO,SAAS,sBACd,YACA;AACA,SAAO,WAAW,OAAO,aAAa,EAAE;AAC1C;AAOA,eAAsB,SAAY,MAW/B;AACD,SAAO,MAAM,KAAK,KAAK,EAAE,KAAK,CAAC,QAAQ,IAAI,KAAU;AACvD;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;AAEA,SAAS,eAAe,KAAc;AACpC,QAAM,WAAW,kBAAkB,GAAG;AACtC,SAAO,IAAI,EAAE,MAAM,qBAAqB,SAAS,SAAS,CAAC;AAC7D;AAEO,IAAM,wBAAwC;AAE9C,SAAS,eAAe;AAC7B,SAAO,IAAI;AAAA,IACT;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,SAAS,SACP,OACA,SACA,iBACgC;AAChC,SAAO;AAAA,IACL,GAAG;AAAA,IACH,MAAM,QAAQ;AAAA,IACd;AAAA,IACA,UAAU,CAAC;AAAA,IACX,iBAAiB,IAAI,gBAAgB;AAAA,IACrC,mBAAmB,IAAI,gBAAgB,EAAE;AAAA,IACzC,mBAAmB,MAAM;AAAA,EAC3B;AACF;AAEO,SAAS,YAMd,OACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,SAAO,OACL,QAGG;AACH,WAAO,KACJ,QAAQ,SAAS,OAAO,SAAS,eAAe,GAAG,GAAG,EACtD,MAAM,cAAc;AAAA,EACzB;AACF;AAEA,SAAS,mBASP;AACA,QAAM,SAAS,IAAI,eAGjB,MAAM;AAGN,SAAK,QAAQ,QAAQ,EAAE,KAAK,MAAM;AAChC,aAAO,oBAAoB;AAAA,IAC7B,CAAC;AAAA,EACH,CAAC;AACD,QAAM,SAAS,IAAI;AAAA,IACjB,CAAC,MAAM;AACL,aAAO,UAAU,CAAC;AAAA,IACpB;AAAA,EACF;AACA,SAAO,QAAQ,MAAM;AAGnB,SAAK,QAAQ,QAAQ,EAAE,KAAK,MAAM;AAChC,aAAO,aAAa;AAAA,IACtB,CAAC;AAAA,EACH,CAAC;AAED,SAAO,EAAE,QAAQ,OAAO;AAC1B;AAEA,SAAS,kBAGP;AACA,QAAM,SAAS,IAAI,eAGjB,MAAM;AAGN,SAAK,QAAQ,QAAQ,EAAE,KAAK,MAAM;AAChC,aAAO,oBAAoB;AAAA,IAC7B,CAAC;AAAA,EACH,CAAC;AACD,QAAM,SAAS,IAAI,gBAA+B,CAAC,MAAM;AACvD,WAAO,UAAU,GAAG,CAAC,CAAC;AAAA,EACxB,CAAC;AACD,SAAO,QAAQ,MAAM;AAGnB,SAAK,QAAQ,QAAQ,EAAE,KAAK,MAAM;AAChC,aAAO,aAAa;AAAA,IACtB,CAAC;AAAA,EACH,CAAC;AAED,SAAO,EAAE,QAAQ,OAAO;AAC1B;AAEO,SAAS,eAOd,OACA,MACA,MACA,iBACA,UAA+B,aAAa,GAC2B;AACvE,QAAM,YAAY,gBAAuB;AACzC,QAAM,aAAa,iBAA8B;AAEjD,OAAK,KACF;AAAA,IACC,SAAS,OAAO,SAAS,eAAe;AAAA,IACxC,QAAQ,CAAC;AAAA,IACT,UAAU;AAAA,IACV,WAAW;AAAA,EACb,EACC,MAAM,CAAC,QAAiB,WAAW,OAAO,MAAM,eAAe,GAAG,CAAC,CAAC;AAEvE,SAAO,CAAC,UAAU,QAAQ,WAAW,MAAM;AAC7C;AAEO,SAAS,qBAMd,OACA,MACA,iBACA,UAA+B,aAAa,GACoB;AAChE,QAAM,aAAa,iBAA8B;AAEjD,SAAO,CAAC,QAAsB;AAC5B,SAAK,KACF;AAAA,MACC,SAAS,OAAO,SAAS,eAAe;AAAA,MACxC;AAAA,MACA,WAAW;AAAA,IACb,EACC,MAAM,CAAC,QAAiB,WAAW,OAAO,MAAM,eAAe,GAAG,CAAC,CAAC;AAEvE,WAAO,WAAW;AAAA,EACpB;AACF;AAEO,SAAS,eAOd,OACA,MACA,MACA,iBACA,UAA+B,aAAa,GAI5C;AACA,QAAM,YAAY,gBAAuB;AACzC,QAAM,SAAS,KACZ;AAAA,IACC,SAAS,OAAO,SAAS,eAAe;AAAA,IACxC,QAAQ,CAAC;AAAA,IACT,UAAU;AAAA,EACZ,EACC,MAAM,cAAc;AAEvB,SAAO,CAAC,UAAU,QAAQ,MAAM,MAAM;AACxC;AAEO,IAAM,oBAAoB,MAAM;AAErC,SAAO,QAAQ,aAAa,UACxB,gBAAgB,OAAO,CAAC,KACxB,QAAQ,OAAO,CAAC;AACtB;","names":[]}
+{"version":3,"sources":["../../util/testHelpers.ts"],"sourcesContent":["import NodeWs, { WebSocketServer } from 'ws';\nimport http from 'node:http';\nimport { Err, Ok, Result, BaseErrorSchemaType } from '../router/result';\nimport {\n  ProcedureErrorSchemaType,\n  ReaderErrorSchema,\n  UNCAUGHT_ERROR_CODE,\n  PayloadType,\n  Procedure,\n} from '../router/procedures';\nimport { Static } from '@sinclair/typebox';\nimport {\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  currentProtocolVersion,\n} from '../transport/message';\nimport { coerceErrorString } from './stringify';\nimport { Transport } from '../transport/transport';\nimport {\n  Readable,\n  ReadableImpl,\n  ReadableResult,\n  ReadableIterator,\n  Writable,\n  WritableImpl,\n} from '../router/streams';\nimport { ServiceContext, ProcedureHandlerContext } from '../router/context';\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 {\n  Session,\n  SessionStateGraph,\n} from '../transport/sessionStateMachine/transitions';\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\nfunction catchProcError(err: unknown) {\n  const errorMsg = coerceErrorString(err);\n  return Err({ code: UNCAUGHT_ERROR_CODE, message: errorMsg });\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  );\n}\n\nfunction dummyCtx<State>(\n  state: State,\n  session: Session<Connection>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n): ProcedureHandlerContext<State> {\n  return {\n    ...extendedContext,\n    state,\n    sessionId: session.id,\n    from: session.from,\n    metadata: {},\n    // TODO might wanna hook these up!\n    cancel: () => undefined,\n    signal: new AbortController().signal,\n  };\n}\n\nexport function asClientRpc<\n  State extends object,\n  Init extends PayloadType,\n  Res extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'rpc', Init, null, Res, Err>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n) {\n  return async (\n    msg: Static<Init>,\n  ): Promise<\n    Result<Static<Res>, Static<Err> | Static<typeof ReaderErrorSchema>>\n  > => {\n    return proc\n      .handler({\n        ctx: dummyCtx(state, session, extendedContext),\n        reqInit: msg,\n      })\n      .catch(catchProcError);\n  };\n}\n\nfunction createResponsePipe<\n  Res extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(): {\n  readable: Readable<\n    Static<Res>,\n    Static<Err> | Static<typeof ReaderErrorSchema>\n  >;\n  writable: Writable<Result<Static<Res>, Static<Err>>>;\n} {\n  const readable = new ReadableImpl<\n    Static<Res>,\n    Static<Err> | Static<typeof ReaderErrorSchema>\n  >();\n  const writable = new WritableImpl<Result<Static<Res>, Static<Err>>>(\n    (v) => {\n      readable._pushValue(v);\n    },\n    () => {\n      // Make it async to simulate request going over the wire\n      // using promises so that we don't get affected by fake timers.\n      void Promise.resolve().then(() => {\n        readable._triggerClose();\n      });\n    },\n  );\n\n  return { readable, writable };\n}\n\nfunction createRequestPipe<Req extends PayloadType>(): {\n  readable: Readable<Static<Req>, Static<typeof ReaderErrorSchema>>;\n  writable: Writable<Static<Req>>;\n} {\n  const readable = new ReadableImpl<\n    Static<Req>,\n    Static<typeof ReaderErrorSchema>\n  >();\n  const writable = new WritableImpl<Static<Req>>(\n    (v) => {\n      readable._pushValue(Ok(v));\n    },\n    () => {\n      // Make it async to simulate request going over the wire\n      // using promises so that we don't get affected by fake timers.\n      void Promise.resolve().then(() => {\n        readable._triggerClose();\n      });\n    },\n  );\n\n  return { readable, writable };\n}\n\nexport function asClientStream<\n  State extends object,\n  Init extends PayloadType,\n  Req extends PayloadType,\n  Res extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'stream', Init, Req, Res, Err>,\n  reqInit?: Static<Init>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n): {\n  reqWritable: Writable<Static<Req>>;\n  resReadable: Readable<Static<Res>, Static<Err>>;\n} {\n  const requestPipe = createRequestPipe<Req>();\n  const responsePipe = createResponsePipe<Res, Err>();\n\n  void proc\n    .handler({\n      ctx: dummyCtx(state, session, extendedContext),\n      reqInit: reqInit ?? {},\n      reqReadable: requestPipe.readable,\n      resWritable: responsePipe.writable,\n    })\n    .catch((err: unknown) => responsePipe.writable.write(catchProcError(err)));\n\n  return {\n    reqWritable: requestPipe.writable,\n    resReadable: responsePipe.readable,\n  };\n}\n\nexport function asClientSubscription<\n  State extends object,\n  Init extends PayloadType,\n  Res extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'subscription', Init, null, Res, Err>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n): (msg: Static<Init>) => {\n  resReadable: Readable<Static<Res>, Static<Err>>;\n} {\n  const responsePipe = createResponsePipe<Res, Err>();\n\n  return (msg: Static<Init>) => {\n    void proc\n      .handler({\n        ctx: dummyCtx(state, session, extendedContext),\n        reqInit: msg,\n        resWritable: responsePipe.writable,\n      })\n      .catch((err: unknown) =>\n        responsePipe.writable.write(catchProcError(err)),\n      );\n\n    return { resReadable: responsePipe.readable };\n  };\n}\n\nexport function asClientUpload<\n  State extends object,\n  Init extends PayloadType,\n  Req extends PayloadType,\n  Res extends PayloadType,\n  Err extends ProcedureErrorSchemaType,\n>(\n  state: State,\n  proc: Procedure<State, 'upload', Init, Req, Res, Err>,\n  reqInit?: Static<Init>,\n  extendedContext?: Omit<ServiceContext, 'state'>,\n  session: Session<Connection> = dummySession(),\n): {\n  reqWritable: Writable<Static<Req>>;\n  finalize: () => Promise<Result<Static<Res>, Static<Err>>>;\n} {\n  const requestPipe = createRequestPipe<Req>();\n  const result = proc\n    .handler({\n      ctx: dummyCtx(state, session, extendedContext),\n      reqInit: reqInit ?? {},\n      reqReadable: requestPipe.readable,\n    })\n    .catch(catchProcError);\n\n  return { reqWritable: requestPipe.writable, finalize: () => result };\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"],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,OAAO,UAAU,uBAAuB;AA6CjC,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;AAEA,SAAS,eAAe,KAAc;AACpC,QAAM,WAAW,kBAAkB,GAAG;AACtC,SAAO,IAAI,EAAE,MAAM,qBAAqB,SAAS,SAAS,CAAC;AAC7D;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,EACF;AACF;AAEA,SAAS,SACP,OACA,SACA,iBACgC;AAChC,SAAO;AAAA,IACL,GAAG;AAAA,IACH;AAAA,IACA,WAAW,QAAQ;AAAA,IACnB,MAAM,QAAQ;AAAA,IACd,UAAU,CAAC;AAAA;AAAA,IAEX,QAAQ,MAAM;AAAA,IACd,QAAQ,IAAI,gBAAgB,EAAE;AAAA,EAChC;AACF;AAEO,SAAS,YAMd,OACA,MACA,iBACA,UAA+B,aAAa,GAC5C;AACA,SAAO,OACL,QAGG;AACH,WAAO,KACJ,QAAQ;AAAA,MACP,KAAK,SAAS,OAAO,SAAS,eAAe;AAAA,MAC7C,SAAS;AAAA,IACX,CAAC,EACA,MAAM,cAAc;AAAA,EACzB;AACF;AAEA,SAAS,qBASP;AACA,QAAM,WAAW,IAAI,aAGnB;AACF,QAAM,WAAW,IAAI;AAAA,IACnB,CAAC,MAAM;AACL,eAAS,WAAW,CAAC;AAAA,IACvB;AAAA,IACA,MAAM;AAGJ,WAAK,QAAQ,QAAQ,EAAE,KAAK,MAAM;AAChC,iBAAS,cAAc;AAAA,MACzB,CAAC;AAAA,IACH;AAAA,EACF;AAEA,SAAO,EAAE,UAAU,SAAS;AAC9B;AAEA,SAAS,oBAGP;AACA,QAAM,WAAW,IAAI,aAGnB;AACF,QAAM,WAAW,IAAI;AAAA,IACnB,CAAC,MAAM;AACL,eAAS,WAAW,GAAG,CAAC,CAAC;AAAA,IAC3B;AAAA,IACA,MAAM;AAGJ,WAAK,QAAQ,QAAQ,EAAE,KAAK,MAAM;AAChC,iBAAS,cAAc;AAAA,MACzB,CAAC;AAAA,IACH;AAAA,EACF;AAEA,SAAO,EAAE,UAAU,SAAS;AAC9B;AAEO,SAAS,eAOd,OACA,MACA,SACA,iBACA,UAA+B,aAAa,GAI5C;AACA,QAAM,cAAc,kBAAuB;AAC3C,QAAM,eAAe,mBAA6B;AAElD,OAAK,KACF,QAAQ;AAAA,IACP,KAAK,SAAS,OAAO,SAAS,eAAe;AAAA,IAC7C,SAAS,WAAW,CAAC;AAAA,IACrB,aAAa,YAAY;AAAA,IACzB,aAAa,aAAa;AAAA,EAC5B,CAAC,EACA,MAAM,CAAC,QAAiB,aAAa,SAAS,MAAM,eAAe,GAAG,CAAC,CAAC;AAE3E,SAAO;AAAA,IACL,aAAa,YAAY;AAAA,IACzB,aAAa,aAAa;AAAA,EAC5B;AACF;AAEO,SAAS,qBAMd,OACA,MACA,iBACA,UAA+B,aAAa,GAG5C;AACA,QAAM,eAAe,mBAA6B;AAElD,SAAO,CAAC,QAAsB;AAC5B,SAAK,KACF,QAAQ;AAAA,MACP,KAAK,SAAS,OAAO,SAAS,eAAe;AAAA,MAC7C,SAAS;AAAA,MACT,aAAa,aAAa;AAAA,IAC5B,CAAC,EACA;AAAA,MAAM,CAAC,QACN,aAAa,SAAS,MAAM,eAAe,GAAG,CAAC;AAAA,IACjD;AAEF,WAAO,EAAE,aAAa,aAAa,SAAS;AAAA,EAC9C;AACF;AAEO,SAAS,eAOd,OACA,MACA,SACA,iBACA,UAA+B,aAAa,GAI5C;AACA,QAAM,cAAc,kBAAuB;AAC3C,QAAM,SAAS,KACZ,QAAQ;AAAA,IACP,KAAK,SAAS,OAAO,SAAS,eAAe;AAAA,IAC7C,SAAS,WAAW,CAAC;AAAA,IACrB,aAAa,YAAY;AAAA,EAC3B,CAAC,EACA,MAAM,cAAc;AAEvB,SAAO,EAAE,aAAa,YAAY,UAAU,UAAU,MAAM,OAAO;AACrE;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/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@replit/river",
   "description": "It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!",
-  "version": "0.200.0-rc.2",
+  "version": "0.200.0-rc.20",
   "type": "module",
   "exports": {
     ".": {
@@ -63,14 +63,14 @@
     "@types/ws": "^8.5.5",
     "@typescript-eslint/eslint-plugin": "^7.8.0",
     "@typescript-eslint/parser": "^7.8.0",
-    "@vitest/ui": "^1.3.1",
+    "@vitest/ui": "^2.0.5",
     "eslint": "^8.57.0",
     "eslint-config-prettier": "^9.1.0",
     "eslint-plugin-prettier": "^5.1.3",
     "prettier": "^3.0.0",
     "tsup": "^7.2.0",
     "typescript": "^5.4.5",
-    "vitest": "^1.3.1"
+    "vitest": "^2.0.5"
   },
   "scripts": {
     "check": "tsc --noEmit && npm run format && npm run lint",