@restatedev/restate-sdk-testcontainers 1.13.0 → 1.14.1

package/dist/index.d.cts

@@ -10,5 +10,5 @@ import { LogMetadata, LogSource, LoggerContext, LoggerTransport, RestateLogLevel
 import { DefaultServiceOptions, RestateEndpoint, RestateEndpointBase } from "./restate-sdk/src/endpoint.cjs";
 import { EndpointOptions } from "./restate-sdk/src/endpoint/types.cjs";
 import "./restate-sdk/src/index.cjs";
-import { RestateContainer, RestateTestEnvironment, StateProxy, TestEnvironmentOptions } from "./restate_test_environment.cjs";
-export { type DefaultServiceOptions, type Duration, type EndpointOptions, type Hooks, type HooksProvider, type Interceptor, type InvocationId, type JournalValueCodec, type LogMetadata, type LogSource, type LoggerContext, type LoggerTransport, type ObjectOptions, type Request, RestateContainer, type RestateEndpoint, type RestateEndpointBase, type RestateError, type RestateLogLevel, RestateTestEnvironment, type RetryPolicy, type Serde, type ServiceDefinition, type ServiceOptions, StateProxy, type Target, type TerminalError, type TestEnvironmentOptions, type TypedState, type UntypedState, type VirtualObjectDefinition, type WorkflowDefinition, type WorkflowOptions };
+import { RestateContainer, RestateTestEnvironment, ServiceEndpointAccess, StateProxy, TestEnvironmentOptions, TestEnvironmentStorage } from "./restate_test_environment.cjs";
+export { type DefaultServiceOptions, type Duration, type EndpointOptions, type Hooks, type HooksProvider, type Interceptor, type InvocationId, type JournalValueCodec, type LogMetadata, type LogSource, type LoggerContext, type LoggerTransport, type ObjectOptions, type Request, RestateContainer, type RestateEndpoint, type RestateEndpointBase, type RestateError, type RestateLogLevel, RestateTestEnvironment, type RetryPolicy, type Serde, type ServiceDefinition, type ServiceEndpointAccess, type ServiceOptions, StateProxy, type Target, type TerminalError, type TestEnvironmentOptions, type TestEnvironmentStorage, type TypedState, type UntypedState, type VirtualObjectDefinition, type WorkflowDefinition, type WorkflowOptions };

package/dist/index.d.ts

@@ -10,5 +10,5 @@ import { LogMetadata, LogSource, LoggerContext, LoggerTransport, RestateLogLevel
 import { DefaultServiceOptions, RestateEndpoint, RestateEndpointBase } from "./restate-sdk/src/endpoint.js";
 import { EndpointOptions } from "./restate-sdk/src/endpoint/types.js";
 import "./restate-sdk/src/index.js";
-import { RestateContainer, RestateTestEnvironment, StateProxy, TestEnvironmentOptions } from "./restate_test_environment.js";
-export { type DefaultServiceOptions, type Duration, type EndpointOptions, type Hooks, type HooksProvider, type Interceptor, type InvocationId, type JournalValueCodec, type LogMetadata, type LogSource, type LoggerContext, type LoggerTransport, type ObjectOptions, type Request, RestateContainer, type RestateEndpoint, type RestateEndpointBase, type RestateError, type RestateLogLevel, RestateTestEnvironment, type RetryPolicy, type Serde, type ServiceDefinition, type ServiceOptions, StateProxy, type Target, type TerminalError, type TestEnvironmentOptions, type TypedState, type UntypedState, type VirtualObjectDefinition, type WorkflowDefinition, type WorkflowOptions };
+import { RestateContainer, RestateTestEnvironment, ServiceEndpointAccess, StateProxy, TestEnvironmentOptions, TestEnvironmentStorage } from "./restate_test_environment.js";
+export { type DefaultServiceOptions, type Duration, type EndpointOptions, type Hooks, type HooksProvider, type Interceptor, type InvocationId, type JournalValueCodec, type LogMetadata, type LogSource, type LoggerContext, type LoggerTransport, type ObjectOptions, type Request, RestateContainer, type RestateEndpoint, type RestateEndpointBase, type RestateError, type RestateLogLevel, RestateTestEnvironment, type RetryPolicy, type Serde, type ServiceDefinition, type ServiceEndpointAccess, type ServiceOptions, StateProxy, type Target, type TerminalError, type TestEnvironmentOptions, type TestEnvironmentStorage, type TypedState, type UntypedState, type VirtualObjectDefinition, type WorkflowDefinition, type WorkflowOptions };

package/dist/restate-sdk-core/src/serde_api.d.cts

@@ -1,8 +1,106 @@
 //#region ../restate-sdk-core/src/serde_api.d.ts
+/**
+ * Serializer/deserializer pair for any Restate-managed value of type `T` —
+ * handler inputs and outputs, service state, side-effect results,
+ * awakeables, durable promises, and so on.
+ *
+ * A `Serde<T>` is responsible for two conversions:
+ *
+ * - **Wire format**: `serialize` / `deserialize` convert a value `T` to and
+ *   from the raw bytes that flow over the network and get stored in the
+ *   journal.
+ * - **JSON preview (optional)**: `preview.toJsonString` /
+ *   `preview.fromJsonString` convert a value `T` to and from a JSON string,
+ *   used by tooling to render and edit values that humans can read.
+ *
+ * It also advertises metadata (`contentType`, `jsonSchema`) that surfaces in
+ * service discovery.
+ */
 interface Serde<T> {
+  /**
+   * MIME type of the bytes produced by `serialize` (and accepted by
+   * `deserialize`).
+   */
   contentType?: string;
+  /**
+   * Optional JSON Schema describing the logical shape of `T`. Surfaced in
+   * discovery so that tooling can generate forms, auto-complete payloads,
+   * etc. Has no effect on serialization at runtime.
+   */
   jsonSchema?: object;
+  /**
+   * Optional bridge between the serde's wire format and a JSON
+   * representation that humans (and tooling like the Restate UI) can read
+   * and edit.
+   *
+   * Only useful for serdes whose wire format isn't already plain JSON —
+   * protobuf, MessagePack, length-prefixed binary, JSON variants that need
+   * post-processing for bigints/dates, etc. Both methods may be async if the
+   * conversion needs I/O.
+   *
+   * ### Preview flow
+   *
+   * Together with `serialize` / `deserialize`, the four functions chain in
+   * both directions between a JSON string and on-the-wire bytes:
+   *
+   * ```text
+   * JSON string  ──fromJsonString──►  T  ──serialize──►  wire bytes
+   * wire bytes   ──deserialize────►  T  ──toJsonString──►  JSON string
+   * ```
+   *
+   * The split keeps `serialize` / `deserialize` responsible for the full
+   * on-the-wire format; `preview` only handles the JSON ↔ `T` bridge.
+   *
+   * @example
+   * ```ts
+   * // Wire format is protobuf; the protobuf-es runtime already ships
+   * // `toJsonString` / `fromJsonString`, so preview is a direct pass-through.
+   * import {
+   *   fromBinary,
+   *   fromJsonString,
+   *   toBinary,
+   *   toJsonString,
+   * } from "@bufbuild/protobuf";
+   * import { PersonSchema, type Person } from "./person_pb.js";
+   *
+   * const personSerde: Serde<Person> = {
+   *   contentType: "application/protobuf",
+   *   serialize(message) { return toBinary(PersonSchema, message); },
+   *   deserialize(bytes) { return fromBinary(PersonSchema, bytes); },
+   *   preview: {
+   *     toJsonString: (v) => toJsonString(PersonSchema, v),
+   *     fromJsonString: (j) => fromJsonString(PersonSchema, j),
+   *   },
+   * };
+   * ```
+   */
+  preview?: {
+    /**
+     * Convert a value into a JSON string for display or editing by humans.
+     * The returned string must round-trip through `fromJsonString` back to
+     * the same logical value.
+     */
+    toJsonString(value: T): Promise<string> | string;
+    /**
+     * Parse a human-supplied JSON string back into a value of type `T`.
+     * Should throw (or reject) if `json` is invalid for this serde.
+     */
+    fromJsonString(json: string): Promise<T> | T;
+  };
+  /**
+   * Convert a value of type `T` into its on-the-wire bytes. This is the
+   * format Restate sends between services and persists in the journal.
+   *
+   * Must be the inverse of `deserialize` — `deserialize(serialize(v))`
+   * should produce a value equivalent to `v`.
+   */
   serialize(value: T): Uint8Array;
+  /**
+   * Convert on-the-wire bytes back into a value of type `T`. Must be the
+   * inverse of `serialize`.
+   *
+   * May throw if `data` doesn't conform to the expected wire format.
+   */
   deserialize(data: Uint8Array): T;
 }
 //#endregion

package/dist/restate-sdk-core/src/serde_api.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"serde_api.d.cts","names":[],"sources":["../../../../restate-sdk-core/src/serde_api.ts"],"sourcesContent":[],"mappings":";AAmBiB,UAAA,KAAK,CAAA,CAAA,CAAA,CAAA;EAIH,WAAA,CAAA,EAAA,MAAA;EAAI,UAAA,CAAA,EAAA,MAAA;EAEH,SAAA,CAAA,KAAA,EAFD,CAEC,CAAA,EAFG,UAEH;EAAa,WAAA,CAAA,IAAA,EAAb,UAAa,CAAA,EAAA,CAAA"}
+{"version":3,"file":"serde_api.d.cts","names":[],"sources":["../../../../restate-sdk-core/src/serde_api.ts"],"sourcesContent":[],"mappings":";AAoCA;;;;;;;;;;;;;;;;;UAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;wBAkEO,IAAI;;;;;kCAMM,QAAQ,KAAK;;;;;;;;;mBAU5B,IAAI;;;;;;;oBAQH,aAAa"}

package/dist/restate-sdk-core/src/serde_api.d.ts

@@ -1,8 +1,106 @@
 //#region ../restate-sdk-core/src/serde_api.d.ts
+/**
+ * Serializer/deserializer pair for any Restate-managed value of type `T` —
+ * handler inputs and outputs, service state, side-effect results,
+ * awakeables, durable promises, and so on.
+ *
+ * A `Serde<T>` is responsible for two conversions:
+ *
+ * - **Wire format**: `serialize` / `deserialize` convert a value `T` to and
+ *   from the raw bytes that flow over the network and get stored in the
+ *   journal.
+ * - **JSON preview (optional)**: `preview.toJsonString` /
+ *   `preview.fromJsonString` convert a value `T` to and from a JSON string,
+ *   used by tooling to render and edit values that humans can read.
+ *
+ * It also advertises metadata (`contentType`, `jsonSchema`) that surfaces in
+ * service discovery.
+ */
 interface Serde<T> {
+  /**
+   * MIME type of the bytes produced by `serialize` (and accepted by
+   * `deserialize`).
+   */
   contentType?: string;
+  /**
+   * Optional JSON Schema describing the logical shape of `T`. Surfaced in
+   * discovery so that tooling can generate forms, auto-complete payloads,
+   * etc. Has no effect on serialization at runtime.
+   */
   jsonSchema?: object;
+  /**
+   * Optional bridge between the serde's wire format and a JSON
+   * representation that humans (and tooling like the Restate UI) can read
+   * and edit.
+   *
+   * Only useful for serdes whose wire format isn't already plain JSON —
+   * protobuf, MessagePack, length-prefixed binary, JSON variants that need
+   * post-processing for bigints/dates, etc. Both methods may be async if the
+   * conversion needs I/O.
+   *
+   * ### Preview flow
+   *
+   * Together with `serialize` / `deserialize`, the four functions chain in
+   * both directions between a JSON string and on-the-wire bytes:
+   *
+   * ```text
+   * JSON string  ──fromJsonString──►  T  ──serialize──►  wire bytes
+   * wire bytes   ──deserialize────►  T  ──toJsonString──►  JSON string
+   * ```
+   *
+   * The split keeps `serialize` / `deserialize` responsible for the full
+   * on-the-wire format; `preview` only handles the JSON ↔ `T` bridge.
+   *
+   * @example
+   * ```ts
+   * // Wire format is protobuf; the protobuf-es runtime already ships
+   * // `toJsonString` / `fromJsonString`, so preview is a direct pass-through.
+   * import {
+   *   fromBinary,
+   *   fromJsonString,
+   *   toBinary,
+   *   toJsonString,
+   * } from "@bufbuild/protobuf";
+   * import { PersonSchema, type Person } from "./person_pb.js";
+   *
+   * const personSerde: Serde<Person> = {
+   *   contentType: "application/protobuf",
+   *   serialize(message) { return toBinary(PersonSchema, message); },
+   *   deserialize(bytes) { return fromBinary(PersonSchema, bytes); },
+   *   preview: {
+   *     toJsonString: (v) => toJsonString(PersonSchema, v),
+   *     fromJsonString: (j) => fromJsonString(PersonSchema, j),
+   *   },
+   * };
+   * ```
+   */
+  preview?: {
+    /**
+     * Convert a value into a JSON string for display or editing by humans.
+     * The returned string must round-trip through `fromJsonString` back to
+     * the same logical value.
+     */
+    toJsonString(value: T): Promise<string> | string;
+    /**
+     * Parse a human-supplied JSON string back into a value of type `T`.
+     * Should throw (or reject) if `json` is invalid for this serde.
+     */
+    fromJsonString(json: string): Promise<T> | T;
+  };
+  /**
+   * Convert a value of type `T` into its on-the-wire bytes. This is the
+   * format Restate sends between services and persists in the journal.
+   *
+   * Must be the inverse of `deserialize` — `deserialize(serialize(v))`
+   * should produce a value equivalent to `v`.
+   */
   serialize(value: T): Uint8Array;
+  /**
+   * Convert on-the-wire bytes back into a value of type `T`. Must be the
+   * inverse of `serialize`.
+   *
+   * May throw if `data` doesn't conform to the expected wire format.
+   */
   deserialize(data: Uint8Array): T;
 }
 //#endregion

package/dist/restate-sdk-core/src/serde_api.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"serde_api.d.ts","names":[],"sources":["../../../../restate-sdk-core/src/serde_api.ts"],"sourcesContent":[],"mappings":";AAmBiB,UAAA,KAAK,CAAA,CAAA,CAAA,CAAA;EAIH,WAAA,CAAA,EAAA,MAAA;EAAI,UAAA,CAAA,EAAA,MAAA;EAEH,SAAA,CAAA,KAAA,EAFD,CAEC,CAAA,EAFG,UAEH;EAAa,WAAA,CAAA,IAAA,EAAb,UAAa,CAAA,EAAA,CAAA"}
+{"version":3,"file":"serde_api.d.ts","names":[],"sources":["../../../../restate-sdk-core/src/serde_api.ts"],"sourcesContent":[],"mappings":";AAoCA;;;;;;;;;;;;;;;;;UAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;wBAkEO,IAAI;;;;;kCAMM,QAAQ,KAAK;;;;;;;;;mBAU5B,IAAI;;;;;;;oBAQH,aAAa"}

package/dist/restate_test_environment.cjs

@@ -9,6 +9,10 @@ let http2 = require("http2");
 http2 = require_rolldown_runtime.__toESM(http2);
 
 //#region src/restate_test_environment.ts
+const DEFAULT_START_OPTIONS = {
+	serviceEndpointAccess: "docker-host",
+	storage: "memory"
+};
 /**
 * Custom wait strategy that waits for Restate partitions to be ready by
 * executing a SQL query against the admin API. This ensures all partitions
@@ -51,12 +55,7 @@ var PartitionsReadyWaitStrategy = class {
 	}
 };
 async function prepareRestateEndpoint(param) {
-	let handler;
-	if (typeof param === "function") {
-		const restateEndpoint = (0, __restatedev_restate_sdk.endpoint)();
-		param(restateEndpoint);
-		handler = restateEndpoint.http2Handler();
-	} else handler = (0, __restatedev_restate_sdk.createEndpointHandler)(param);
+	const handler = (0, __restatedev_restate_sdk.createEndpointHandler)(param);
 	const restateHttpServer = http2.createServer(handler);
 	await new Promise((resolve, reject) => {
 		restateHttpServer.listen(0).once("listening", resolve).once("error", reject);
@@ -65,20 +64,26 @@ async function prepareRestateEndpoint(param) {
 	console.info(`Restate container listening on port ${restateServerPort}`);
 	return restateHttpServer;
 }
-async function prepareRestateTestContainer(restateServerPort, restateContainerFactory) {
-	const restateContainer = restateContainerFactory().withExposedPorts(8080, 9070).withWaitStrategy(testcontainers.Wait.forAll([
+async function prepareRestateTestContainer(restateServerPort, restateContainerFactory, options) {
+	let restateContainer = restateContainerFactory().withExposedPorts(8080, 9070).withWaitStrategy(testcontainers.Wait.forAll([
 		testcontainers.Wait.forHttp("/restate/health", 8080),
 		testcontainers.Wait.forHttp("/health", 9070),
 		new PartitionsReadyWaitStrategy()
 	]));
-	await testcontainers.TestContainers.exposeHostPorts(restateServerPort);
+	if (options.storage === "memory") restateContainer = restateContainer.withTmpFs({ "/restate-data": "rw" });
+	const serviceEndpointHost = options.serviceEndpointAccess === "docker-host" ? "host.docker.internal" : "host.testcontainers.internal";
+	if (options.serviceEndpointAccess === "docker-host") restateContainer = restateContainer.withExtraHosts([{
+		host: "host.docker.internal",
+		ipAddress: "host-gateway"
+	}]);
+	else await testcontainers.TestContainers.exposeHostPorts(restateServerPort);
 	const startedRestateContainer = await restateContainer.start();
 	try {
-		console.info(`Registering services at http://host.testcontainers.internal:${restateServerPort}...`);
+		console.info(`Registering services at http://${serviceEndpointHost}:${restateServerPort}...`);
 		const res = await fetch(`http://${startedRestateContainer.getHost()}:${startedRestateContainer.getMappedPort(9070)}/deployments`, {
 			method: "POST",
 			headers: { "Content-Type": "application/json" },
-			body: JSON.stringify({ uri: `http://host.testcontainers.internal:${restateServerPort}` })
+			body: JSON.stringify({ uri: `http://${serviceEndpointHost}:${restateServerPort}` })
 		});
 		if (!res.ok) {
 			const badResponse = await res.text();
@@ -110,18 +115,22 @@ var RestateTestEnvironment = class RestateTestEnvironment {
 		await this.startedRestateContainer.stop();
 		this.startedRestateHttpServer.close();
 	}
-	static async start(param, restateContainerFactory) {
+	static async start(options, restateContainerFactory) {
 		let containerFactory;
 		if (restateContainerFactory) containerFactory = restateContainerFactory;
-		else if (typeof param !== "function") containerFactory = () => {
+		else if (options.container) containerFactory = options.container;
+		else containerFactory = () => {
 			const container = new RestateContainer();
-			if (param.alwaysReplay) container.alwaysReplay();
-			if (param.disableRetries) container.disableRetries();
+			if (options.alwaysReplay) container.alwaysReplay();
+			if (options.disableRetries) container.disableRetries();
 			return container;
 		};
-		else containerFactory = () => new RestateContainer();
-		const startedRestateHttpServer = typeof param === "function" ? await prepareRestateEndpoint(param) : await prepareRestateEndpoint(param);
-		return new RestateTestEnvironment(startedRestateHttpServer, await prepareRestateTestContainer(startedRestateHttpServer.address().port, containerFactory));
+		const resolvedStartOptions = {
+			serviceEndpointAccess: options.serviceEndpointAccess ?? DEFAULT_START_OPTIONS.serviceEndpointAccess,
+			storage: options.storage ?? DEFAULT_START_OPTIONS.storage
+		};
+		const startedRestateHttpServer = await prepareRestateEndpoint(options);
+		return new RestateTestEnvironment(startedRestateHttpServer, await prepareRestateTestContainer(startedRestateHttpServer.address().port, containerFactory, resolvedStartOptions));
 	}
 };
 var RestateContainer = class extends testcontainers.GenericContainer {

package/dist/restate_test_environment.d.cts

@@ -1,14 +1,50 @@
 import { VirtualObjectDefinition, WorkflowDefinition } from "./restate-sdk-core/src/core.cjs";
 import { Serde } from "./restate-sdk-core/src/serde_api.cjs";
 import { TypedState, UntypedState } from "./restate-sdk/src/context.cjs";
-import { RestateEndpoint } from "./restate-sdk/src/endpoint.cjs";
 import { EndpointOptions } from "./restate-sdk/src/endpoint/types.cjs";
 import "./restate-sdk/src/index.cjs";
 import * as http2 from "http2";
 import { GenericContainer, StartedTestContainer } from "testcontainers";
 
 //#region src/restate_test_environment.d.ts
+type ServiceEndpointAccess = "testcontainers" | "docker-host";
+type TestEnvironmentStorage = "disk" | "memory";
 interface TestEnvironmentOptions extends EndpointOptions {
+  /**
+   * Factory for the Restate container used by the test environment.
+   *
+   * Use this to customize the Restate image or container settings before the
+   * helper applies its own ports, wait strategy, networking, and storage
+   * configuration.
+   *
+   * For backwards compatibility, the second `start(options, factory)` argument
+   * is still supported and takes precedence over this option.
+   *
+   * @defaultValue `() => new RestateContainer()`
+   */
+  container?: () => GenericContainer;
+  /**
+   * Controls how the Restate container reaches the SDK service endpoint running
+   * on the test host.
+   *
+   * - `"testcontainers"` exposes the host port through Testcontainers and
+   *   registers `http://host.testcontainers.internal:<port>`.
+   * - `"docker-host"` skips Testcontainers port exposure, adds
+   *   `host.docker.internal:host-gateway` to the Restate container, and
+   *   registers `http://host.docker.internal:<port>`.
+   *
+   * @defaultValue `"docker-host"`
+   */
+  serviceEndpointAccess?: ServiceEndpointAccess;
+  /**
+   * Controls where Restate stores container data.
+   *
+   * - `"disk"` keeps the current Testcontainers/Docker storage behavior.
+   * - `"memory"` mounts `/restate-data` as tmpfs for faster disposable tests.
+   *
+   * @defaultValue `"memory"`
+   */
+  storage?: TestEnvironmentStorage;
   /**
    * Forces restate-server to always replay on a suspension point.
    * This is useful to hunt non-deterministic bugs that might prevent
@@ -30,15 +66,6 @@ declare class RestateTestEnvironment {
   adminAPIBaseUrl(): string;
   stateOf<TState extends TypedState = UntypedState>(service: VirtualObjectDefinition<string, unknown> | WorkflowDefinition<string, unknown>, key: string): StateProxy<TState>;
   stop(): Promise<void>;
-  /**
-   *
-   * @deprecated Please use {@link TestEnvironmentOptions} instead of this.
-   * @example
-   * ```
-   * RestateTestEnvironment.start({ services: [mysService] })
-   * ```
-   */
-  static start(mountServicesFn: (server: RestateEndpoint) => void, restateContainerFactory?: () => GenericContainer): Promise<RestateTestEnvironment>;
   static start(options: TestEnvironmentOptions, restateContainerFactory?: () => GenericContainer): Promise<RestateTestEnvironment>;
 }
 declare class RestateContainer extends GenericContainer {
@@ -69,5 +96,5 @@ declare class StateProxy<TState extends TypedState> {
   private setAllRaw;
 }
 //#endregion
-export { RestateContainer, RestateTestEnvironment, StateProxy, TestEnvironmentOptions };
+export { RestateContainer, RestateTestEnvironment, ServiceEndpointAccess, StateProxy, TestEnvironmentOptions, TestEnvironmentStorage };
 //# sourceMappingURL=restate_test_environment.d.cts.map

package/dist/restate_test_environment.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"restate_test_environment.d.cts","names":[],"sources":["../src/restate_test_environment.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UAkNiB,sBAAA,SAA+B;;;;;;;EAA/B;AAgBjB;;;;EAGsC,cAAA,CAAA,EAAA,OAAA;;AAgBO,cAnBhC,sBAAA,CAmBgC;EAErC,SAAA,wBAAA,EAnB+B,KAAA,CAAM,WAmBrC;EACA,SAAA,uBAAA,EAnB8B,oBAmB9B;EAEQ,WAAA,CAAA,wBAAA,EAtBuB,KAAA,CAAM,WAsB7B,EAAA,uBAAA,EArBsB,oBAqBtB;EAAX,OAAA,CAAA,CAAA,EAAA,MAAA;EAIc,eAAA,CAAA,CAAA,EAAA,MAAA;EAcW,OAAA,CAAA,eAvBE,UAuBF,GAvBe,YAuBf,CAAA,CAAA,OAAA,EArBtB,uBAqBsB,CAAA,MAAA,EAAA,OAAA,CAAA,GApBtB,kBAoBsB,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,GAAA,EAAA,MAAA,CAAA,EAlBzB,UAkByB,CAlBd,MAkBc,CAAA;EACM,IAAA,CAAA,CAAA,EAfjB,OAeiB,CAAA,IAAA,CAAA;EACvB;;;;;;;AA0Cb;EA8Ba,OAAA,KAAA,CAAU,eAAA,EAAA,CAAA,MAAA,EA1EO,eA0EP,EAAA,GAAA,IAAA,EAAA,uBAAA,CAAA,EAAA,GAAA,GAzEa,gBAyEb,CAAA,EAxElB,OAwEkB,CAxEV,sBAwEU,CAAA;EAAgB,OAAA,KAAA,CAAA,OAAA,EAtE1B,sBAsE0B,EAAA,uBAAA,CAAA,EAAA,GAAA,GArEH,gBAqEG,CAAA,EApElC,OAoEkC,CApE1B,sBAoE0B,CAAA;;AAS7B,cAvCG,gBAAA,SAAyB,gBAAA,CAuC5B;EAAe,WAAA,CAAA,OAAA,CAAA,EAAA,MAAA;EAAwB;;;;;EACe,YAAA,CAAA,CAAA,EAAA,IAAA;EAApD;;;;;EACgD,cAAA,CAAA,CAAA,EAAA,IAAA;;AAkCtB,cA7CzB,UA6CyB,CAAA,eA7CC,UA6CD,CAAA,CAAA;EAEhC,QAAA,eAAA;EAAe,QAAA,OAAA;EACX,QAAA,UAAA;EAAc,WAAA,CAAA,eAAA,EAAA,MAAA,EAAA,OAAA,EAAA,MAAA,EAAA,UAAA,EAAA,MAAA;EACd,GAAA,CAAA,MAAA,EAAA,aAAA,MAzCoC,MAyCpC,GAAA,MAAA,CAAA,CAAA,IAAA,EAxCA,MAwCA,SAxCe,YAwCf,GAAA,MAAA,GAxCuC,IAwCvC,EAAA,KAAA,CAAA,EAvCE,KAuCF,CAvCQ,MAuCR,SAvCuB,YAuCvB,GAvCsC,MAuCtC,GAvC+C,MAuC/C,CAvCsD,IAuCtD,CAAA,CAAA,CAAA,EAtCL,OAsCK,CAAA,CAtCI,MAsCJ,SAtCmB,YAsCnB,GAtCkC,MAsClC,GAtC2C,MAsC3C,CAtCkD,IAsClD,CAAA,CAAA,GAAA,IAAA,CAAA;EAAa,MAAA,CAAA,gBAJe,UAIf,CAAA,CAAA,KAAA,CAAA,EAHX,KAGW,CAFjB,MAEiB,SAFF,YAEE,GADb,OACa,CAAA,MADC,OACD,CAAA,GAAb,MAAa,CAAA,MAAA,MAAA,CAAA,CAAA,CAAA,EAElB,OAFkB,CAEV,MAFU,SAEK,YAFL,GAEoB,OAFpB,GAE8B,MAF9B,CAAA;EAHX,QAAA,SAAA;EAKC,GAAA,CAAA,MAAA,EAAA,aAAA,MA0CiC,MA1CjC,GAAA,MAAA,CAAA,CAAA,IAAA,EA2CH,MA3CG,SA2CY,YA3CZ,GAAA,MAAA,GA2CoC,IA3CpC,EAAA,KAAA,EA4CF,MA5CE,SA4Ca,YA5Cb,GA4C4B,MA5C5B,GA4CqC,MA5CrC,CA4C4C,IA5C5C,CAAA,EAAA,KAAA,CAAA,EA6CD,KA7CC,CA6CK,MA7CL,SA6CoB,YA7CpB,GA6CmC,MA7CnC,GA6C4C,MA7C5C,CA6CmD,IA7CnD,CAAA,CAAA,CAAA,EA8CR,OA9CQ,CAAA,IAAA,CAAA;EAAe,MAAA,CAAA,gBA4DU,UA5DV,CAAA,CAAA,MAAA,EA6DhB,MA7DgB,SA6DD,YA7DC,GA6Dc,OA7Dd,GA6DwB,MA7DxB,EAAA,KAAA,CAAA,EA8DhB,KA9DgB,CA+DtB,MA/DsB,SA+DP,YA/DO,GAgElB,OAhEkB,CAAA,MAgEJ,OAhEI,CAAA,GAiElB,MAjEkB,CAAA,MAiEL,MAjEK,CAAA,CAAA,CAAA,EAkEvB,OAlEuB,CAAA,IAAA,CAAA;EAAe,QAAA,SAAA"}
+{"version":3,"file":"restate_test_environment.d.cts","names":[],"sources":["../src/restate_test_environment.ts"],"sourcesContent":[],"mappings":";;;;;;;;;KAkCY,qBAAA;KAEA,sBAAA;UAyLK,sBAAA,SAA+B;;;;AA3LhD;AAEA;AAyLA;;;;;;AAsDA;EAEuC,SAAM,CAAA,EAAA,GAAA,GA3CzB,gBA2CyB;EACP;;;;;;;;;;;;EAiCzB,qBAAA,CAAA,EA/Da,qBA+Db;EAAR;;AAsCL;AA8BA;;;;;EASiD,OAAA,CAAA,EAlIrC,sBAkIqC;EAC/B;;;;;EAAN,YAAA,CAAA,EAAA,OAAA;EACE;;;;;EAAT,cAAA,CAAA,EAAA,OAAA;;AAoCC,cAvJO,sBAAA,CAuJP;EAAe,SAAA,wBAAA,EArJkB,KAAA,CAAM,WAqJxB;EACX,SAAA,uBAAA,EArJ4B,oBAqJ5B;EAAc,WAAA,CAAA,wBAAA,EAtJe,KAAA,CAAM,WAsJrB,EAAA,uBAAA,EArJc,oBAqJd;EACd,OAAA,CAAA,CAAA,EAAA,MAAA;EAAa,eAAA,CAAA,CAAA,EAAA,MAAA;EAHX,OAAA,CAAA,eAnIoB,UAmIpB,GAnIiC,YAmIjC,CAAA,CAAA,OAAA,EAjIJ,uBAiII,CAAA,MAAA,EAAA,OAAA,CAAA,GAhIJ,kBAgII,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,GAAA,EAAA,MAAA,CAAA,EA9HP,UA8HO,CA9HI,MA8HJ,CAAA;EAKC,IAAA,CAAA,CAAA,EA/HM,OA+HN,CAAA,IAAA,CAAA;EAAe,OAAA,KAAA,CAAA,OAAA,EAzHf,sBAyHe,EAAA,uBAAA,CAAA,EAAA,GAAA,GAxHQ,gBAwHR,CAAA,EAvHvB,OAuHuB,CAvHf,sBAuHe,CAAA;;AAAyB,cAjFxC,gBAAA,SAAyB,gBAAA,CAiFe;EAAhD,WAAA,CAAA,OAAA,CAAA,EAAA,MAAA;EA0CyC;;;;;EAEpB,YAAA,CAAA,CAAA,EAAA,IAAA;EAAe;;;;;EACO,cAAA,CAAA,CAAA,EAAA,IAAA;;AAAgB,cAhGnD,UAgGmD,CAAA,eAhGzB,UAgGyB,CAAA,CAAA;EAApD,QAAA,eAAA;EACP,QAAA,OAAA;EAciC,QAAA,UAAA;EAC1B,WAAA,CAAA,eAAA,EAAA,MAAA,EAAA,OAAA,EAAA,MAAA,EAAA,UAAA,EAAA,MAAA;EAAe,GAAA,CAAA,MAAA,EAAA,aAAA,MAxGmB,MAwGnB,GAAA,MAAA,CAAA,CAAA,IAAA,EAvGjB,MAuGiB,SAvGF,YAuGE,GAAA,MAAA,GAvGsB,IAuGtB,EAAA,KAAA,CAAA,EAtGf,KAsGe,CAtGT,MAsGS,SAtGM,YAsGN,GAtGqB,MAsGrB,GAtG8B,MAsG9B,CAtGqC,IAsGrC,CAAA,CAAA,CAAA,EArGtB,OAqGsB,CAAA,CArGb,MAqGa,SArGE,YAqGF,GArGiB,MAqGjB,GArG0B,MAqG1B,CArGiC,IAqGjC,CAAA,CAAA,GAAA,IAAA,CAAA;EAAe,MAAA,CAAA,gBAnEJ,UAmEI,CAAA,CAAA,KAAA,CAAA,EAlE9B,KAkE8B,CAjEpC,MAiEoC,SAjErB,YAiEqB,GAhEhC,OAgEgC,CAAA,MAhElB,OAgEkB,CAAA,GA/DhC,MA+DgC,CAAA,MA/DnB,MA+DmB,CAAA,CAAA,CAAA,EA7DrC,OA6DqC,CA7D7B,MA6D6B,SA7Dd,YA6Dc,GA7DC,OA6DD,GA7DW,MA6DX,CAAA;EAAU,QAAA,SAAA;EAE9C,GAAA,CAAA,MAAA,EAAA,aAAA,MArBwC,MAqBxC,GAAA,MAAA,CAAA,CAAA,IAAA,EApBI,MAoBJ,SApBmB,YAoBnB,GAAA,MAAA,GApB2C,IAoB3C,EAAA,KAAA,EAnBK,MAmBL,SAnBoB,YAmBpB,GAnBmC,MAmBnC,GAnB4C,MAmB5C,CAnBmD,IAmBnD,CAAA,EAAA,KAAA,CAAA,EAlBM,KAkBN,CAlBY,MAkBZ,SAlB2B,YAkB3B,GAlB0C,MAkB1C,GAlBmD,MAkBnD,CAlB0D,IAkB1D,CAAA,CAAA,CAAA,EAjBD,OAiBC,CAAA,IAAA,CAAA;EAAe,MAAA,CAAA,gBAHiB,UAGjB,CAAA,CAAA,MAAA,EAFT,MAES,SAFM,YAEN,GAFqB,OAErB,GAF+B,MAE/B,EAAA,KAAA,CAAA,EADT,KACS,CAAf,MAAe,SAAA,YAAA,GACX,OADW,CAAA,MACG,OADH,CAAA,GAEX,MAFW,CAAA,MAEE,MAFF,CAAA,CAAA,CAAA,EAGhB,OAHgB,CAAA,IAAA,CAAA;EACX,QAAA,SAAA"}

package/dist/restate_test_environment.d.ts

@@ -1,14 +1,50 @@
 import { VirtualObjectDefinition, WorkflowDefinition } from "./restate-sdk-core/src/core.js";
 import { Serde } from "./restate-sdk-core/src/serde_api.js";
 import { TypedState, UntypedState } from "./restate-sdk/src/context.js";
-import { RestateEndpoint } from "./restate-sdk/src/endpoint.js";
 import { EndpointOptions } from "./restate-sdk/src/endpoint/types.js";
 import "./restate-sdk/src/index.js";
 import { GenericContainer, StartedTestContainer } from "testcontainers";
 import * as http2 from "http2";
 
 //#region src/restate_test_environment.d.ts
+type ServiceEndpointAccess = "testcontainers" | "docker-host";
+type TestEnvironmentStorage = "disk" | "memory";
 interface TestEnvironmentOptions extends EndpointOptions {
+  /**
+   * Factory for the Restate container used by the test environment.
+   *
+   * Use this to customize the Restate image or container settings before the
+   * helper applies its own ports, wait strategy, networking, and storage
+   * configuration.
+   *
+   * For backwards compatibility, the second `start(options, factory)` argument
+   * is still supported and takes precedence over this option.
+   *
+   * @defaultValue `() => new RestateContainer()`
+   */
+  container?: () => GenericContainer;
+  /**
+   * Controls how the Restate container reaches the SDK service endpoint running
+   * on the test host.
+   *
+   * - `"testcontainers"` exposes the host port through Testcontainers and
+   *   registers `http://host.testcontainers.internal:<port>`.
+   * - `"docker-host"` skips Testcontainers port exposure, adds
+   *   `host.docker.internal:host-gateway` to the Restate container, and
+   *   registers `http://host.docker.internal:<port>`.
+   *
+   * @defaultValue `"docker-host"`
+   */
+  serviceEndpointAccess?: ServiceEndpointAccess;
+  /**
+   * Controls where Restate stores container data.
+   *
+   * - `"disk"` keeps the current Testcontainers/Docker storage behavior.
+   * - `"memory"` mounts `/restate-data` as tmpfs for faster disposable tests.
+   *
+   * @defaultValue `"memory"`
+   */
+  storage?: TestEnvironmentStorage;
   /**
    * Forces restate-server to always replay on a suspension point.
    * This is useful to hunt non-deterministic bugs that might prevent
@@ -30,15 +66,6 @@ declare class RestateTestEnvironment {
   adminAPIBaseUrl(): string;
   stateOf<TState extends TypedState = UntypedState>(service: VirtualObjectDefinition<string, unknown> | WorkflowDefinition<string, unknown>, key: string): StateProxy<TState>;
   stop(): Promise<void>;
-  /**
-   *
-   * @deprecated Please use {@link TestEnvironmentOptions} instead of this.
-   * @example
-   * ```
-   * RestateTestEnvironment.start({ services: [mysService] })
-   * ```
-   */
-  static start(mountServicesFn: (server: RestateEndpoint) => void, restateContainerFactory?: () => GenericContainer): Promise<RestateTestEnvironment>;
   static start(options: TestEnvironmentOptions, restateContainerFactory?: () => GenericContainer): Promise<RestateTestEnvironment>;
 }
 declare class RestateContainer extends GenericContainer {
@@ -69,5 +96,5 @@ declare class StateProxy<TState extends TypedState> {
   private setAllRaw;
 }
 //#endregion
-export { RestateContainer, RestateTestEnvironment, StateProxy, TestEnvironmentOptions };
+export { RestateContainer, RestateTestEnvironment, ServiceEndpointAccess, StateProxy, TestEnvironmentOptions, TestEnvironmentStorage };
 //# sourceMappingURL=restate_test_environment.d.ts.map

package/dist/restate_test_environment.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"restate_test_environment.d.ts","names":[],"sources":["../src/restate_test_environment.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UAkNiB,sBAAA,SAA+B;;;;;;;EAA/B;AAgBjB;;;;EAGsC,cAAA,CAAA,EAAA,OAAA;;AAgBO,cAnBhC,sBAAA,CAmBgC;EAErC,SAAA,wBAAA,EAnB+B,KAAA,CAAM,WAmBrC;EACA,SAAA,uBAAA,EAnB8B,oBAmB9B;EAEQ,WAAA,CAAA,wBAAA,EAtBuB,KAAA,CAAM,WAsB7B,EAAA,uBAAA,EArBsB,oBAqBtB;EAAX,OAAA,CAAA,CAAA,EAAA,MAAA;EAIc,eAAA,CAAA,CAAA,EAAA,MAAA;EAcW,OAAA,CAAA,eAvBE,UAuBF,GAvBe,YAuBf,CAAA,CAAA,OAAA,EArBtB,uBAqBsB,CAAA,MAAA,EAAA,OAAA,CAAA,GApBtB,kBAoBsB,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,GAAA,EAAA,MAAA,CAAA,EAlBzB,UAkByB,CAlBd,MAkBc,CAAA;EACM,IAAA,CAAA,CAAA,EAfjB,OAeiB,CAAA,IAAA,CAAA;EACvB;;;;;;;AA0Cb;EA8Ba,OAAA,KAAA,CAAU,eAAA,EAAA,CAAA,MAAA,EA1EO,eA0EP,EAAA,GAAA,IAAA,EAAA,uBAAA,CAAA,EAAA,GAAA,GAzEa,gBAyEb,CAAA,EAxElB,OAwEkB,CAxEV,sBAwEU,CAAA;EAAgB,OAAA,KAAA,CAAA,OAAA,EAtE1B,sBAsE0B,EAAA,uBAAA,CAAA,EAAA,GAAA,GArEH,gBAqEG,CAAA,EApElC,OAoEkC,CApE1B,sBAoE0B,CAAA;;AAS7B,cAvCG,gBAAA,SAAyB,gBAAA,CAuC5B;EAAe,WAAA,CAAA,OAAA,CAAA,EAAA,MAAA;EAAwB;;;;;EACe,YAAA,CAAA,CAAA,EAAA,IAAA;EAApD;;;;;EACgD,cAAA,CAAA,CAAA,EAAA,IAAA;;AAkCtB,cA7CzB,UA6CyB,CAAA,eA7CC,UA6CD,CAAA,CAAA;EAEhC,QAAA,eAAA;EAAe,QAAA,OAAA;EACX,QAAA,UAAA;EAAc,WAAA,CAAA,eAAA,EAAA,MAAA,EAAA,OAAA,EAAA,MAAA,EAAA,UAAA,EAAA,MAAA;EACd,GAAA,CAAA,MAAA,EAAA,aAAA,MAzCoC,MAyCpC,GAAA,MAAA,CAAA,CAAA,IAAA,EAxCA,MAwCA,SAxCe,YAwCf,GAAA,MAAA,GAxCuC,IAwCvC,EAAA,KAAA,CAAA,EAvCE,KAuCF,CAvCQ,MAuCR,SAvCuB,YAuCvB,GAvCsC,MAuCtC,GAvC+C,MAuC/C,CAvCsD,IAuCtD,CAAA,CAAA,CAAA,EAtCL,OAsCK,CAAA,CAtCI,MAsCJ,SAtCmB,YAsCnB,GAtCkC,MAsClC,GAtC2C,MAsC3C,CAtCkD,IAsClD,CAAA,CAAA,GAAA,IAAA,CAAA;EAAa,MAAA,CAAA,gBAJe,UAIf,CAAA,CAAA,KAAA,CAAA,EAHX,KAGW,CAFjB,MAEiB,SAFF,YAEE,GADb,OACa,CAAA,MADC,OACD,CAAA,GAAb,MAAa,CAAA,MAAA,MAAA,CAAA,CAAA,CAAA,EAElB,OAFkB,CAEV,MAFU,SAEK,YAFL,GAEoB,OAFpB,GAE8B,MAF9B,CAAA;EAHX,QAAA,SAAA;EAKC,GAAA,CAAA,MAAA,EAAA,aAAA,MA0CiC,MA1CjC,GAAA,MAAA,CAAA,CAAA,IAAA,EA2CH,MA3CG,SA2CY,YA3CZ,GAAA,MAAA,GA2CoC,IA3CpC,EAAA,KAAA,EA4CF,MA5CE,SA4Ca,YA5Cb,GA4C4B,MA5C5B,GA4CqC,MA5CrC,CA4C4C,IA5C5C,CAAA,EAAA,KAAA,CAAA,EA6CD,KA7CC,CA6CK,MA7CL,SA6CoB,YA7CpB,GA6CmC,MA7CnC,GA6C4C,MA7C5C,CA6CmD,IA7CnD,CAAA,CAAA,CAAA,EA8CR,OA9CQ,CAAA,IAAA,CAAA;EAAe,MAAA,CAAA,gBA4DU,UA5DV,CAAA,CAAA,MAAA,EA6DhB,MA7DgB,SA6DD,YA7DC,GA6Dc,OA7Dd,GA6DwB,MA7DxB,EAAA,KAAA,CAAA,EA8DhB,KA9DgB,CA+DtB,MA/DsB,SA+DP,YA/DO,GAgElB,OAhEkB,CAAA,MAgEJ,OAhEI,CAAA,GAiElB,MAjEkB,CAAA,MAiEL,MAjEK,CAAA,CAAA,CAAA,EAkEvB,OAlEuB,CAAA,IAAA,CAAA;EAAe,QAAA,SAAA"}
+{"version":3,"file":"restate_test_environment.d.ts","names":[],"sources":["../src/restate_test_environment.ts"],"sourcesContent":[],"mappings":";;;;;;;;;KAkCY,qBAAA;KAEA,sBAAA;UAyLK,sBAAA,SAA+B;;;;AA3LhD;AAEA;AAyLA;;;;;;AAsDA;EAEuC,SAAM,CAAA,EAAA,GAAA,GA3CzB,gBA2CyB;EACP;;;;;;;;;;;;EAiCzB,qBAAA,CAAA,EA/Da,qBA+Db;EAAR;;AAsCL;AA8BA;;;;;EASiD,OAAA,CAAA,EAlIrC,sBAkIqC;EAC/B;;;;;EAAN,YAAA,CAAA,EAAA,OAAA;EACE;;;;;EAAT,cAAA,CAAA,EAAA,OAAA;;AAoCC,cAvJO,sBAAA,CAuJP;EAAe,SAAA,wBAAA,EArJkB,KAAA,CAAM,WAqJxB;EACX,SAAA,uBAAA,EArJ4B,oBAqJ5B;EAAc,WAAA,CAAA,wBAAA,EAtJe,KAAA,CAAM,WAsJrB,EAAA,uBAAA,EArJc,oBAqJd;EACd,OAAA,CAAA,CAAA,EAAA,MAAA;EAAa,eAAA,CAAA,CAAA,EAAA,MAAA;EAHX,OAAA,CAAA,eAnIoB,UAmIpB,GAnIiC,YAmIjC,CAAA,CAAA,OAAA,EAjIJ,uBAiII,CAAA,MAAA,EAAA,OAAA,CAAA,GAhIJ,kBAgII,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,GAAA,EAAA,MAAA,CAAA,EA9HP,UA8HO,CA9HI,MA8HJ,CAAA;EAKC,IAAA,CAAA,CAAA,EA/HM,OA+HN,CAAA,IAAA,CAAA;EAAe,OAAA,KAAA,CAAA,OAAA,EAzHf,sBAyHe,EAAA,uBAAA,CAAA,EAAA,GAAA,GAxHQ,gBAwHR,CAAA,EAvHvB,OAuHuB,CAvHf,sBAuHe,CAAA;;AAAyB,cAjFxC,gBAAA,SAAyB,gBAAA,CAiFe;EAAhD,WAAA,CAAA,OAAA,CAAA,EAAA,MAAA;EA0CyC;;;;;EAEpB,YAAA,CAAA,CAAA,EAAA,IAAA;EAAe;;;;;EACO,cAAA,CAAA,CAAA,EAAA,IAAA;;AAAgB,cAhGnD,UAgGmD,CAAA,eAhGzB,UAgGyB,CAAA,CAAA;EAApD,QAAA,eAAA;EACP,QAAA,OAAA;EAciC,QAAA,UAAA;EAC1B,WAAA,CAAA,eAAA,EAAA,MAAA,EAAA,OAAA,EAAA,MAAA,EAAA,UAAA,EAAA,MAAA;EAAe,GAAA,CAAA,MAAA,EAAA,aAAA,MAxGmB,MAwGnB,GAAA,MAAA,CAAA,CAAA,IAAA,EAvGjB,MAuGiB,SAvGF,YAuGE,GAAA,MAAA,GAvGsB,IAuGtB,EAAA,KAAA,CAAA,EAtGf,KAsGe,CAtGT,MAsGS,SAtGM,YAsGN,GAtGqB,MAsGrB,GAtG8B,MAsG9B,CAtGqC,IAsGrC,CAAA,CAAA,CAAA,EArGtB,OAqGsB,CAAA,CArGb,MAqGa,SArGE,YAqGF,GArGiB,MAqGjB,GArG0B,MAqG1B,CArGiC,IAqGjC,CAAA,CAAA,GAAA,IAAA,CAAA;EAAe,MAAA,CAAA,gBAnEJ,UAmEI,CAAA,CAAA,KAAA,CAAA,EAlE9B,KAkE8B,CAjEpC,MAiEoC,SAjErB,YAiEqB,GAhEhC,OAgEgC,CAAA,MAhElB,OAgEkB,CAAA,GA/DhC,MA+DgC,CAAA,MA/DnB,MA+DmB,CAAA,CAAA,CAAA,EA7DrC,OA6DqC,CA7D7B,MA6D6B,SA7Dd,YA6Dc,GA7DC,OA6DD,GA7DW,MA6DX,CAAA;EAAU,QAAA,SAAA;EAE9C,GAAA,CAAA,MAAA,EAAA,aAAA,MArBwC,MAqBxC,GAAA,MAAA,CAAA,CAAA,IAAA,EApBI,MAoBJ,SApBmB,YAoBnB,GAAA,MAAA,GApB2C,IAoB3C,EAAA,KAAA,EAnBK,MAmBL,SAnBoB,YAmBpB,GAnBmC,MAmBnC,GAnB4C,MAmB5C,CAnBmD,IAmBnD,CAAA,EAAA,KAAA,CAAA,EAlBM,KAkBN,CAlBY,MAkBZ,SAlB2B,YAkB3B,GAlB0C,MAkB1C,GAlBmD,MAkBnD,CAlB0D,IAkB1D,CAAA,CAAA,CAAA,EAjBD,OAiBC,CAAA,IAAA,CAAA;EAAe,MAAA,CAAA,gBAHiB,UAGjB,CAAA,CAAA,MAAA,EAFT,MAES,SAFM,YAEN,GAFqB,OAErB,GAF+B,MAE/B,EAAA,KAAA,CAAA,EADT,KACS,CAAf,MAAe,SAAA,YAAA,GACX,OADW,CAAA,MACG,OADH,CAAA,GAEX,MAFW,CAAA,MAEE,MAFF,CAAA,CAAA,CAAA,EAGhB,OAHgB,CAAA,IAAA,CAAA;EACX,QAAA,SAAA"}

package/dist/restate_test_environment.js

@@ -1,9 +1,13 @@
-import { createEndpointHandler, endpoint, serde } from "@restatedev/restate-sdk";
+import { createEndpointHandler, serde } from "@restatedev/restate-sdk";
 import { GenericContainer, TestContainers, Wait, getContainerRuntimeClient } from "testcontainers";
 import { tableFromIPC } from "apache-arrow";
 import * as http2 from "http2";
 
 //#region src/restate_test_environment.ts
+const DEFAULT_START_OPTIONS = {
+	serviceEndpointAccess: "docker-host",
+	storage: "memory"
+};
 /**
 * Custom wait strategy that waits for Restate partitions to be ready by
 * executing a SQL query against the admin API. This ensures all partitions
@@ -46,12 +50,7 @@ var PartitionsReadyWaitStrategy = class {
 	}
 };
 async function prepareRestateEndpoint(param) {
-	let handler;
-	if (typeof param === "function") {
-		const restateEndpoint = endpoint();
-		param(restateEndpoint);
-		handler = restateEndpoint.http2Handler();
-	} else handler = createEndpointHandler(param);
+	const handler = createEndpointHandler(param);
 	const restateHttpServer = http2.createServer(handler);
 	await new Promise((resolve, reject) => {
 		restateHttpServer.listen(0).once("listening", resolve).once("error", reject);
@@ -60,20 +59,26 @@ async function prepareRestateEndpoint(param) {
 	console.info(`Restate container listening on port ${restateServerPort}`);
 	return restateHttpServer;
 }
-async function prepareRestateTestContainer(restateServerPort, restateContainerFactory) {
-	const restateContainer = restateContainerFactory().withExposedPorts(8080, 9070).withWaitStrategy(Wait.forAll([
+async function prepareRestateTestContainer(restateServerPort, restateContainerFactory, options) {
+	let restateContainer = restateContainerFactory().withExposedPorts(8080, 9070).withWaitStrategy(Wait.forAll([
 		Wait.forHttp("/restate/health", 8080),
 		Wait.forHttp("/health", 9070),
 		new PartitionsReadyWaitStrategy()
 	]));
-	await TestContainers.exposeHostPorts(restateServerPort);
+	if (options.storage === "memory") restateContainer = restateContainer.withTmpFs({ "/restate-data": "rw" });
+	const serviceEndpointHost = options.serviceEndpointAccess === "docker-host" ? "host.docker.internal" : "host.testcontainers.internal";
+	if (options.serviceEndpointAccess === "docker-host") restateContainer = restateContainer.withExtraHosts([{
+		host: "host.docker.internal",
+		ipAddress: "host-gateway"
+	}]);
+	else await TestContainers.exposeHostPorts(restateServerPort);
 	const startedRestateContainer = await restateContainer.start();
 	try {
-		console.info(`Registering services at http://host.testcontainers.internal:${restateServerPort}...`);
+		console.info(`Registering services at http://${serviceEndpointHost}:${restateServerPort}...`);
 		const res = await fetch(`http://${startedRestateContainer.getHost()}:${startedRestateContainer.getMappedPort(9070)}/deployments`, {
 			method: "POST",
 			headers: { "Content-Type": "application/json" },
-			body: JSON.stringify({ uri: `http://host.testcontainers.internal:${restateServerPort}` })
+			body: JSON.stringify({ uri: `http://${serviceEndpointHost}:${restateServerPort}` })
 		});
 		if (!res.ok) {
 			const badResponse = await res.text();
@@ -105,18 +110,22 @@ var RestateTestEnvironment = class RestateTestEnvironment {
 		await this.startedRestateContainer.stop();
 		this.startedRestateHttpServer.close();
 	}
-	static async start(param, restateContainerFactory) {
+	static async start(options, restateContainerFactory) {
 		let containerFactory;
 		if (restateContainerFactory) containerFactory = restateContainerFactory;
-		else if (typeof param !== "function") containerFactory = () => {
+		else if (options.container) containerFactory = options.container;
+		else containerFactory = () => {
 			const container = new RestateContainer();
-			if (param.alwaysReplay) container.alwaysReplay();
-			if (param.disableRetries) container.disableRetries();
+			if (options.alwaysReplay) container.alwaysReplay();
+			if (options.disableRetries) container.disableRetries();
 			return container;
 		};
-		else containerFactory = () => new RestateContainer();
-		const startedRestateHttpServer = typeof param === "function" ? await prepareRestateEndpoint(param) : await prepareRestateEndpoint(param);
-		return new RestateTestEnvironment(startedRestateHttpServer, await prepareRestateTestContainer(startedRestateHttpServer.address().port, containerFactory));
+		const resolvedStartOptions = {
+			serviceEndpointAccess: options.serviceEndpointAccess ?? DEFAULT_START_OPTIONS.serviceEndpointAccess,
+			storage: options.storage ?? DEFAULT_START_OPTIONS.storage
+		};
+		const startedRestateHttpServer = await prepareRestateEndpoint(options);
+		return new RestateTestEnvironment(startedRestateHttpServer, await prepareRestateTestContainer(startedRestateHttpServer.address().port, containerFactory, resolvedStartOptions));
 	}
 };
 var RestateContainer = class extends GenericContainer {

package/dist/restate_test_environment.js.map

@@ -1 +1 @@
-{"version":3,"file":"restate_test_environment.js","names":["handler: (\n    request: http2.Http2ServerRequest,\n    response: http2.Http2ServerResponse\n  ) => void","startedRestateHttpServer: http2.Http2Server","startedRestateContainer: StartedTestContainer","containerFactory: () => GenericContainer","adminAPIBaseUrl: string","service: string","serviceKey: string","serde","value"],"sources":["../src/restate_test_environment.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport {\n  endpoint,\n  createEndpointHandler,\n  serde,\n} from \"@restatedev/restate-sdk\";\nimport type {\n  TypedState,\n  UntypedState,\n  Serde,\n  RestateEndpoint,\n  VirtualObjectDefinition,\n  WorkflowDefinition,\n  EndpointOptions,\n} from \"@restatedev/restate-sdk\";\n\nimport {\n  GenericContainer,\n  type StartedTestContainer,\n  TestContainers,\n  Wait,\n  type WaitStrategy,\n  type BoundPorts,\n  getContainerRuntimeClient,\n} from \"testcontainers\";\nimport { tableFromIPC } from \"apache-arrow\";\nimport * as http2 from \"http2\";\nimport type * as net from \"net\";\n\n/**\n * Custom wait strategy that waits for Restate partitions to be ready by\n * executing a SQL query against the admin API. This ensures all partitions\n * are initialized and queryable before the container is considered ready.\n */\nclass PartitionsReadyWaitStrategy implements WaitStrategy {\n  private startupTimeoutMs = 60_000;\n  private startupTimeoutSet = false;\n  private readonly port: number;\n  private readonly pollIntervalMs: number;\n\n  constructor(port = 9070, pollIntervalMs = 200) {\n    this.port = port;\n    this.pollIntervalMs = pollIntervalMs;\n  }\n\n  public withStartupTimeout(startupTimeoutMs: number): this {\n    this.startupTimeoutMs = startupTimeoutMs;\n    this.startupTimeoutSet = true;\n    return this;\n  }\n\n  public isStartupTimeoutSet(): boolean {\n    return this.startupTimeoutSet;\n  }\n\n  public getStartupTimeout(): number {\n    return this.startupTimeoutMs;\n  }\n\n  public async waitUntilReady(\n    container: { id: string },\n    boundPorts: BoundPorts\n  ): Promise<void> {\n    const client = await getContainerRuntimeClient();\n    const host = client.info.containerRuntime.host;\n    const mappedPort = boundPorts.getBinding(this.port);\n    const adminUrl = `http://${host}:${mappedPort}`;\n\n    const startTime = Date.now();\n\n    while (Date.now() - startTime < this.startupTimeoutMs) {\n      try {\n        const res = await fetch(`${adminUrl}/query`, {\n          method: \"POST\",\n          headers: { \"Content-Type\": \"application/json\" },\n          body: JSON.stringify({\n            query: \"SELECT count(1) FROM sys_invocation\",\n          }),\n        });\n\n        if (res.ok) {\n          // Partitions are ready\n          return;\n        }\n      } catch {\n        // Ignore errors, keep polling\n      }\n\n      await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));\n    }\n\n    throw new Error(\n      `Restate partitions not ready after ${this.startupTimeoutMs}ms`\n    );\n  }\n}\n\n// Prepare the restate server\nasync function prepareRestateEndpoint(\n  param: (server: RestateEndpoint) => void\n): Promise<http2.Http2Server>;\nasync function prepareRestateEndpoint(\n  param: EndpointOptions\n): Promise<http2.Http2Server>;\nasync function prepareRestateEndpoint(\n  param: EndpointOptions | ((server: RestateEndpoint) => void)\n): Promise<http2.Http2Server> {\n  // Prepare RestateServer\n  let handler: (\n    request: http2.Http2ServerRequest,\n    response: http2.Http2ServerResponse\n  ) => void;\n  if (typeof param === \"function\") {\n    const restateEndpoint = endpoint();\n    param(restateEndpoint);\n    handler = restateEndpoint.http2Handler();\n  } else {\n    handler = createEndpointHandler(param);\n  }\n\n  // Start HTTP2 server on random port\n  const restateHttpServer = http2.createServer(handler);\n  await new Promise((resolve, reject) => {\n    restateHttpServer\n      .listen(0)\n      .once(\"listening\", resolve)\n      .once(\"error\", reject);\n  });\n  const restateServerPort = (restateHttpServer.address() as net.AddressInfo)\n    .port;\n  console.info(`Restate container listening on port ${restateServerPort}`);\n\n  return restateHttpServer;\n}\n\n// Prepare the restate testcontainer\nasync function prepareRestateTestContainer(\n  restateServerPort: number,\n  restateContainerFactory: () => GenericContainer\n): Promise<StartedTestContainer> {\n  const restateContainer = restateContainerFactory()\n    // Expose ports\n    .withExposedPorts(8080, 9070)\n    // Wait start on health checks and partition readiness\n    .withWaitStrategy(\n      Wait.forAll([\n        Wait.forHttp(\"/restate/health\", 8080),\n        Wait.forHttp(\"/health\", 9070),\n        new PartitionsReadyWaitStrategy(),\n      ])\n    );\n\n  // This MUST be executed before starting the restate container\n  // Expose host port to access the restate server\n  await TestContainers.exposeHostPorts(restateServerPort);\n\n  // Start restate container\n  const startedRestateContainer = await restateContainer.start();\n\n  // From now on, if something fails, stop the container to cleanup the environment\n  try {\n    console.info(\n      `Registering services at http://host.testcontainers.internal:${restateServerPort}...`\n    );\n\n    // Register this service endpoint\n    const res = await fetch(\n      `http://${startedRestateContainer.getHost()}:${startedRestateContainer.getMappedPort(\n        9070\n      )}/deployments`,\n      {\n        method: \"POST\",\n        headers: {\n          \"Content-Type\": \"application/json\",\n        },\n        body: JSON.stringify({\n          // See https://node.testcontainers.org/features/networking/#expose-host-ports-to-container\n          uri: `http://host.testcontainers.internal:${restateServerPort}`,\n        }),\n      }\n    );\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(\n        `Error ${res.status} during registration: ${badResponse}`\n      );\n    }\n\n    const resp = (await res.json()) as { services: { name: string }[] };\n    console.info(\n      \"Registered services:\",\n      resp?.services?.map((s) => s.name)\n    );\n    return startedRestateContainer;\n  } catch (e) {\n    await startedRestateContainer.stop();\n    throw e;\n  }\n}\n\nexport interface TestEnvironmentOptions extends EndpointOptions {\n  /**\n   * Forces restate-server to always replay on a suspension point.\n   * This is useful to hunt non-deterministic bugs that might prevent\n   * your code from replaying correctly.\n   */\n  alwaysReplay?: boolean;\n\n  /**\n   * Disables retries in the restate-server invoker.\n   * This is useful in tests so that failures surface immediately\n   * instead of hanging through retry backoff.\n   */\n  disableRetries?: boolean;\n}\n\nexport class RestateTestEnvironment {\n  constructor(\n    readonly startedRestateHttpServer: http2.Http2Server,\n    readonly startedRestateContainer: StartedTestContainer\n  ) {}\n\n  public baseUrl(): string {\n    return `http://${this.startedRestateContainer.getHost()}:${this.startedRestateContainer.getMappedPort(\n      8080\n    )}`;\n  }\n\n  public adminAPIBaseUrl(): string {\n    return `http://${this.startedRestateContainer.getHost()}:${this.startedRestateContainer.getMappedPort(\n      9070\n    )}`;\n  }\n\n  // Create a handle that allows read/write of state under a given Virtual Object/Workflow key.\n  public stateOf<TState extends TypedState = UntypedState>(\n    service:\n      | VirtualObjectDefinition<string, unknown>\n      | WorkflowDefinition<string, unknown>,\n    key: string\n  ): StateProxy<TState> {\n    return new StateProxy(this.adminAPIBaseUrl(), service.name, key);\n  }\n\n  public async stop() {\n    await this.startedRestateContainer.stop();\n    this.startedRestateHttpServer.close();\n  }\n\n  /**\n   *\n   * @deprecated Please use {@link TestEnvironmentOptions} instead of this.\n   * @example\n   * ```\n   * RestateTestEnvironment.start({ services: [mysService] })\n   * ```\n   */\n  public static async start(\n    mountServicesFn: (server: RestateEndpoint) => void,\n    restateContainerFactory?: () => GenericContainer\n  ): Promise<RestateTestEnvironment>;\n  public static async start(\n    options: TestEnvironmentOptions,\n    restateContainerFactory?: () => GenericContainer\n  ): Promise<RestateTestEnvironment>;\n  public static async start(\n    param: TestEnvironmentOptions | ((server: RestateEndpoint) => void),\n    restateContainerFactory?: () => GenericContainer\n  ): Promise<RestateTestEnvironment> {\n    let containerFactory: () => GenericContainer;\n    if (restateContainerFactory) {\n      containerFactory = restateContainerFactory;\n    } else if (typeof param !== \"function\") {\n      containerFactory = () => {\n        const container = new RestateContainer();\n        if (param.alwaysReplay) {\n          container.alwaysReplay();\n        }\n        if (param.disableRetries) {\n          container.disableRetries();\n        }\n        return container;\n      };\n    } else {\n      containerFactory = () => new RestateContainer();\n    }\n\n    const startedRestateHttpServer =\n      typeof param === \"function\"\n        ? await prepareRestateEndpoint(param)\n        : await prepareRestateEndpoint(param);\n    const startedRestateContainer = await prepareRestateTestContainer(\n      (startedRestateHttpServer.address() as net.AddressInfo).port,\n      containerFactory\n    );\n    return new RestateTestEnvironment(\n      startedRestateHttpServer,\n      startedRestateContainer\n    );\n  }\n}\n\nexport class RestateContainer extends GenericContainer {\n  constructor(version = \"latest\") {\n    super(`docker.io/restatedev/restate:${version}`);\n  }\n\n  /**\n   * Forces restate-server to always replay on a suspension point.\n   * This is useful to hunt non-deterministic bugs that might prevent\n   * your code from replaying correctly.\n   */\n  alwaysReplay(): this {\n    this.withEnvironment({\n      RESTATE_WORKER__INVOKER__INACTIVITY_TIMEOUT: \"0s\",\n    });\n    return this;\n  }\n\n  /**\n   * Disables retries in the restate-server invoker.\n   * This is useful in tests so that failures surface immediately\n   * instead of hanging through retry backoff.\n   */\n  disableRetries(): this {\n    this.withEnvironment({\n      RESTATE_DEFAULT_RETRY_POLICY__MAX_ATTEMPTS: \"1\",\n      RESTATE_DEFAULT_RETRY_POLICY__ON_MAX_ATTEMPTS: \"kill\",\n    });\n    return this;\n  }\n}\nexport class StateProxy<TState extends TypedState> {\n  constructor(\n    private adminAPIBaseUrl: string,\n    private service: string,\n    private serviceKey: string\n  ) {}\n\n  // Read a single value from state under a given Virtual Object or Workflow key\n  public async get<TValue, TKey extends keyof TState = string>(\n    name: TState extends UntypedState ? string : TKey,\n    serde?: Serde<TState extends UntypedState ? TValue : TState[TKey]>\n  ): Promise<(TState extends UntypedState ? TValue : TState[TKey]) | null> {\n    serde = serde ?? defaultSerde();\n\n    const res = await fetch(`${this.adminAPIBaseUrl}/query`, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n      },\n      body: JSON.stringify({\n        query: `SELECT value from state where service_name = '${\n          this.service\n        }' and service_key = '${this.serviceKey}' and key = '${String(name)}';`,\n      }),\n    });\n\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(`Error ${res.status} during read state: ${badResponse}`);\n    }\n\n    // eslint-disable-next-line @typescript-eslint/await-thenable\n    const table = (await tableFromIPC(res.body)).toArray() as {\n      key: string;\n      value: Uint8Array;\n    }[];\n\n    if (table.length === 0) {\n      return null;\n    }\n\n    return serde.deserialize(table[0]!.value);\n  }\n\n  // Read all values from state under a given Virtual Object or Workflow key\n  public async getAll<TValues extends TypedState>(\n    serde?: Serde<\n      TState extends UntypedState\n        ? TValues[keyof TValues]\n        : TState[keyof TState]\n    >\n  ): Promise<TState extends UntypedState ? TValues : TState> {\n    serde = serde ?? defaultSerde();\n\n    const items = await this.getAllRaw();\n\n    return Object.fromEntries(\n      items.map(({ key, value }) => {\n        return [key, serde.deserialize(value)];\n      })\n    ) as TState extends UntypedState ? TValues : TState;\n  }\n\n  private async getAllRaw(): Promise<{ key: string; value: Uint8Array }[]> {\n    const res = await fetch(`${this.adminAPIBaseUrl}/query`, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n      },\n      body: JSON.stringify({\n        query: `SELECT key, value from state where service_name = '${this.service}' and service_key = '${this.serviceKey}';`,\n      }),\n    });\n\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(`Error ${res.status} during read state: ${badResponse}`);\n    }\n\n    // eslint-disable-next-line @typescript-eslint/await-thenable\n    const table = (await tableFromIPC(res.body)).toArray() as {\n      key: string;\n      value: Uint8Array;\n    }[];\n\n    return table;\n  }\n\n  // Asynchronously set a single value from state under a given Virtual Object or Workflow key.\n  // This will first read all values, then insert the update and submit the new set of values to Restate;\n  // as such it is possible to overwrite changes that happened between the read and the mutation being applied.\n  // A successful return from this function does not imply that the set has finished, only that the mutation\n  // was submitted to Restate for processing.\n  public async set<TValue, TKey extends keyof TState = string>(\n    name: TState extends UntypedState ? string : TKey,\n    value: TState extends UntypedState ? TValue : TState[TKey],\n    serde?: Serde<TState extends UntypedState ? TValue : TState[TKey]>\n  ): Promise<void> {\n    serde = serde ?? defaultSerde();\n    const serialisedValue = serde.serialize(value);\n\n    const items = await this.getAllRaw();\n\n    items.push({ key: String(name), value: serialisedValue });\n\n    await this.setAllRaw(items.map(({ key, value }) => [key, value]));\n  }\n\n  // Asynchronously set all state values under a given Virtual Object or Workflow key.\n  // A successful return from this function does not imply that the set has finished,\n  // only that the mutation was submitted to Restate for processing.\n  public async setAll<TValues extends TypedState>(\n    values: TState extends UntypedState ? TValues : TState,\n    serde?: Serde<\n      TState extends UntypedState\n        ? TValues[keyof TValues]\n        : TState[keyof TState]\n    >\n  ) {\n    serde = serde ?? defaultSerde();\n\n    return this.setAllRaw(\n      Object.entries<\n        TState extends UntypedState\n          ? TValues[keyof TValues]\n          : TState[keyof TState]\n      >(values).map(([key, value]) => {\n        return [key, serde.serialize(value)];\n      })\n    );\n  }\n\n  private async setAllRaw(\n    entries: [key: string, value: Uint8Array][],\n    version?: string\n  ) {\n    const res = await fetch(\n      `${this.adminAPIBaseUrl}/services/${this.service}/state`,\n      {\n        method: \"POST\",\n        headers: {\n          \"Content-Type\": \"application/json\",\n        },\n        body: JSON.stringify(\n          {\n            version,\n            object_key: this.serviceKey,\n            new_state: Object.fromEntries(entries),\n          },\n          (key, value) => {\n            if (value instanceof Uint8Array) {\n              return Array.from(value);\n            } else {\n              // eslint-disable-next-line @typescript-eslint/no-unsafe-return\n              return value;\n            }\n          }\n        ),\n      }\n    );\n\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(\n        `Error ${res.status} during modify state: ${badResponse}`\n      );\n    }\n  }\n}\n\nexport const defaultSerde = <T>(): Serde<T> => {\n  return serde.json as Serde<T>;\n};\n"],"mappings":";;;;;;;;;;;AA4CA,IAAM,8BAAN,MAA0D;CACxD,AAAQ,mBAAmB;CAC3B,AAAQ,oBAAoB;CAC5B,AAAiB;CACjB,AAAiB;CAEjB,YAAY,OAAO,MAAM,iBAAiB,KAAK;AAC7C,OAAK,OAAO;AACZ,OAAK,iBAAiB;;CAGxB,AAAO,mBAAmB,kBAAgC;AACxD,OAAK,mBAAmB;AACxB,OAAK,oBAAoB;AACzB,SAAO;;CAGT,AAAO,sBAA+B;AACpC,SAAO,KAAK;;CAGd,AAAO,oBAA4B;AACjC,SAAO,KAAK;;CAGd,MAAa,eACX,WACA,YACe;EAIf,MAAM,WAAW,WAHF,MAAM,2BAA2B,EAC5B,KAAK,iBAAiB,KAEV,GADb,WAAW,WAAW,KAAK,KAAK;EAGnD,MAAM,YAAY,KAAK,KAAK;AAE5B,SAAO,KAAK,KAAK,GAAG,YAAY,KAAK,kBAAkB;AACrD,OAAI;AASF,SARY,MAAM,MAAM,GAAG,SAAS,SAAS;KAC3C,QAAQ;KACR,SAAS,EAAE,gBAAgB,oBAAoB;KAC/C,MAAM,KAAK,UAAU,EACnB,OAAO,uCACR,CAAC;KACH,CAAC,EAEM,GAEN;WAEI;AAIR,SAAM,IAAI,SAAS,YAAY,WAAW,SAAS,KAAK,eAAe,CAAC;;AAG1E,QAAM,IAAI,MACR,sCAAsC,KAAK,iBAAiB,IAC7D;;;AAWL,eAAe,uBACb,OAC4B;CAE5B,IAAIA;AAIJ,KAAI,OAAO,UAAU,YAAY;EAC/B,MAAM,kBAAkB,UAAU;AAClC,QAAM,gBAAgB;AACtB,YAAU,gBAAgB,cAAc;OAExC,WAAU,sBAAsB,MAAM;CAIxC,MAAM,oBAAoB,MAAM,aAAa,QAAQ;AACrD,OAAM,IAAI,SAAS,SAAS,WAAW;AACrC,oBACG,OAAO,EAAE,CACT,KAAK,aAAa,QAAQ,CAC1B,KAAK,SAAS,OAAO;GACxB;CACF,MAAM,oBAAqB,kBAAkB,SAAS,CACnD;AACH,SAAQ,KAAK,uCAAuC,oBAAoB;AAExE,QAAO;;AAIT,eAAe,4BACb,mBACA,yBAC+B;CAC/B,MAAM,mBAAmB,yBAAyB,CAE/C,iBAAiB,MAAM,KAAK,CAE5B,iBACC,KAAK,OAAO;EACV,KAAK,QAAQ,mBAAmB,KAAK;EACrC,KAAK,QAAQ,WAAW,KAAK;EAC7B,IAAI,6BAA6B;EAClC,CAAC,CACH;AAIH,OAAM,eAAe,gBAAgB,kBAAkB;CAGvD,MAAM,0BAA0B,MAAM,iBAAiB,OAAO;AAG9D,KAAI;AACF,UAAQ,KACN,+DAA+D,kBAAkB,KAClF;EAGD,MAAM,MAAM,MAAM,MAChB,UAAU,wBAAwB,SAAS,CAAC,GAAG,wBAAwB,cACrE,KACD,CAAC,eACF;GACE,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UAAU,EAEnB,KAAK,uCAAuC,qBAC7C,CAAC;GACH,CACF;AACD,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MACR,SAAS,IAAI,OAAO,wBAAwB,cAC7C;;EAGH,MAAM,OAAQ,MAAM,IAAI,MAAM;AAC9B,UAAQ,KACN,wBACA,MAAM,UAAU,KAAK,MAAM,EAAE,KAAK,CACnC;AACD,SAAO;UACA,GAAG;AACV,QAAM,wBAAwB,MAAM;AACpC,QAAM;;;AAoBV,IAAa,yBAAb,MAAa,uBAAuB;CAClC,YACE,AAASC,0BACT,AAASC,yBACT;EAFS;EACA;;CAGX,AAAO,UAAkB;AACvB,SAAO,UAAU,KAAK,wBAAwB,SAAS,CAAC,GAAG,KAAK,wBAAwB,cACtF,KACD;;CAGH,AAAO,kBAA0B;AAC/B,SAAO,UAAU,KAAK,wBAAwB,SAAS,CAAC,GAAG,KAAK,wBAAwB,cACtF,KACD;;CAIH,AAAO,QACL,SAGA,KACoB;AACpB,SAAO,IAAI,WAAW,KAAK,iBAAiB,EAAE,QAAQ,MAAM,IAAI;;CAGlE,MAAa,OAAO;AAClB,QAAM,KAAK,wBAAwB,MAAM;AACzC,OAAK,yBAAyB,OAAO;;CAmBvC,aAAoB,MAClB,OACA,yBACiC;EACjC,IAAIC;AACJ,MAAI,wBACF,oBAAmB;WACV,OAAO,UAAU,WAC1B,0BAAyB;GACvB,MAAM,YAAY,IAAI,kBAAkB;AACxC,OAAI,MAAM,aACR,WAAU,cAAc;AAE1B,OAAI,MAAM,eACR,WAAU,gBAAgB;AAE5B,UAAO;;MAGT,0BAAyB,IAAI,kBAAkB;EAGjD,MAAM,2BACJ,OAAO,UAAU,aACb,MAAM,uBAAuB,MAAM,GACnC,MAAM,uBAAuB,MAAM;AAKzC,SAAO,IAAI,uBACT,0BAL8B,MAAM,4BACnC,yBAAyB,SAAS,CAAqB,MACxD,iBACD,CAIA;;;AAIL,IAAa,mBAAb,cAAsC,iBAAiB;CACrD,YAAY,UAAU,UAAU;AAC9B,QAAM,gCAAgC,UAAU;;;;;;;CAQlD,eAAqB;AACnB,OAAK,gBAAgB,EACnB,6CAA6C,MAC9C,CAAC;AACF,SAAO;;;;;;;CAQT,iBAAuB;AACrB,OAAK,gBAAgB;GACnB,4CAA4C;GAC5C,+CAA+C;GAChD,CAAC;AACF,SAAO;;;AAGX,IAAa,aAAb,MAAmD;CACjD,YACE,AAAQC,iBACR,AAAQC,SACR,AAAQC,YACR;EAHQ;EACA;EACA;;CAIV,MAAa,IACX,MACA,SACuE;AACvE,YAAQC,WAAS,cAAc;EAE/B,MAAM,MAAM,MAAM,MAAM,GAAG,KAAK,gBAAgB,SAAS;GACvD,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UAAU,EACnB,OAAO,iDACL,KAAK,QACN,uBAAuB,KAAK,WAAW,eAAe,OAAO,KAAK,CAAC,KACrE,CAAC;GACH,CAAC;AAEF,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MAAM,SAAS,IAAI,OAAO,sBAAsB,cAAc;;EAI1E,MAAM,SAAS,MAAM,aAAa,IAAI,KAAK,EAAE,SAAS;AAKtD,MAAI,MAAM,WAAW,EACnB,QAAO;AAGT,SAAOA,QAAM,YAAY,MAAM,GAAI,MAAM;;CAI3C,MAAa,OACX,SAKyD;AACzD,YAAQA,WAAS,cAAc;EAE/B,MAAM,QAAQ,MAAM,KAAK,WAAW;AAEpC,SAAO,OAAO,YACZ,MAAM,KAAK,EAAE,KAAK,YAAY;AAC5B,UAAO,CAAC,KAAKA,QAAM,YAAY,MAAM,CAAC;IACtC,CACH;;CAGH,MAAc,YAA2D;EACvE,MAAM,MAAM,MAAM,MAAM,GAAG,KAAK,gBAAgB,SAAS;GACvD,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UAAU,EACnB,OAAO,sDAAsD,KAAK,QAAQ,uBAAuB,KAAK,WAAW,KAClH,CAAC;GACH,CAAC;AAEF,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MAAM,SAAS,IAAI,OAAO,sBAAsB,cAAc;;AAS1E,UALe,MAAM,aAAa,IAAI,KAAK,EAAE,SAAS;;CAaxD,MAAa,IACX,MACA,OACA,SACe;AACf,YAAQA,WAAS,cAAc;EAC/B,MAAM,kBAAkBA,QAAM,UAAU,MAAM;EAE9C,MAAM,QAAQ,MAAM,KAAK,WAAW;AAEpC,QAAM,KAAK;GAAE,KAAK,OAAO,KAAK;GAAE,OAAO;GAAiB,CAAC;AAEzD,QAAM,KAAK,UAAU,MAAM,KAAK,EAAE,KAAK,qBAAY,CAAC,KAAKC,QAAM,CAAC,CAAC;;CAMnE,MAAa,OACX,QACA,SAKA;AACA,YAAQD,WAAS,cAAc;AAE/B,SAAO,KAAK,UACV,OAAO,QAIL,OAAO,CAAC,KAAK,CAAC,KAAK,WAAW;AAC9B,UAAO,CAAC,KAAKA,QAAM,UAAU,MAAM,CAAC;IACpC,CACH;;CAGH,MAAc,UACZ,SACA,SACA;EACA,MAAM,MAAM,MAAM,MAChB,GAAG,KAAK,gBAAgB,YAAY,KAAK,QAAQ,SACjD;GACE,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UACT;IACE;IACA,YAAY,KAAK;IACjB,WAAW,OAAO,YAAY,QAAQ;IACvC,GACA,KAAK,UAAU;AACd,QAAI,iBAAiB,WACnB,QAAO,MAAM,KAAK,MAAM;QAGxB,QAAO;KAGZ;GACF,CACF;AAED,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MACR,SAAS,IAAI,OAAO,wBAAwB,cAC7C;;;;AAKP,MAAa,qBAAkC;AAC7C,QAAO,MAAM"}
+{"version":3,"file":"restate_test_environment.js","names":["DEFAULT_START_OPTIONS: ResolvedTestEnvironmentOptions","handler: (\n    request: http2.Http2ServerRequest,\n    response: http2.Http2ServerResponse\n  ) => void","startedRestateHttpServer: http2.Http2Server","startedRestateContainer: StartedTestContainer","containerFactory: () => GenericContainer","resolvedStartOptions: ResolvedTestEnvironmentOptions","adminAPIBaseUrl: string","service: string","serviceKey: string","serde","value"],"sources":["../src/restate_test_environment.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport { createEndpointHandler, serde } from \"@restatedev/restate-sdk\";\nimport type {\n  TypedState,\n  UntypedState,\n  Serde,\n  VirtualObjectDefinition,\n  WorkflowDefinition,\n  EndpointOptions,\n} from \"@restatedev/restate-sdk\";\n\nimport {\n  GenericContainer,\n  type StartedTestContainer,\n  TestContainers,\n  Wait,\n  type WaitStrategy,\n  type BoundPorts,\n  getContainerRuntimeClient,\n} from \"testcontainers\";\nimport { tableFromIPC } from \"apache-arrow\";\nimport * as http2 from \"http2\";\nimport type * as net from \"net\";\n\nexport type ServiceEndpointAccess = \"testcontainers\" | \"docker-host\";\n\nexport type TestEnvironmentStorage = \"disk\" | \"memory\";\n\ninterface ResolvedTestEnvironmentOptions {\n  serviceEndpointAccess: ServiceEndpointAccess;\n  storage: TestEnvironmentStorage;\n}\n\nconst DEFAULT_START_OPTIONS: ResolvedTestEnvironmentOptions = {\n  serviceEndpointAccess: \"docker-host\",\n  storage: \"memory\",\n};\n\n/**\n * Custom wait strategy that waits for Restate partitions to be ready by\n * executing a SQL query against the admin API. This ensures all partitions\n * are initialized and queryable before the container is considered ready.\n */\nclass PartitionsReadyWaitStrategy implements WaitStrategy {\n  private startupTimeoutMs = 60_000;\n  private startupTimeoutSet = false;\n  private readonly port: number;\n  private readonly pollIntervalMs: number;\n\n  constructor(port = 9070, pollIntervalMs = 200) {\n    this.port = port;\n    this.pollIntervalMs = pollIntervalMs;\n  }\n\n  public withStartupTimeout(startupTimeoutMs: number): this {\n    this.startupTimeoutMs = startupTimeoutMs;\n    this.startupTimeoutSet = true;\n    return this;\n  }\n\n  public isStartupTimeoutSet(): boolean {\n    return this.startupTimeoutSet;\n  }\n\n  public getStartupTimeout(): number {\n    return this.startupTimeoutMs;\n  }\n\n  public async waitUntilReady(\n    container: { id: string },\n    boundPorts: BoundPorts\n  ): Promise<void> {\n    const client = await getContainerRuntimeClient();\n    const host = client.info.containerRuntime.host;\n    const mappedPort = boundPorts.getBinding(this.port);\n    const adminUrl = `http://${host}:${mappedPort}`;\n\n    const startTime = Date.now();\n\n    while (Date.now() - startTime < this.startupTimeoutMs) {\n      try {\n        const res = await fetch(`${adminUrl}/query`, {\n          method: \"POST\",\n          headers: { \"Content-Type\": \"application/json\" },\n          body: JSON.stringify({\n            query: \"SELECT count(1) FROM sys_invocation\",\n          }),\n        });\n\n        if (res.ok) {\n          // Partitions are ready\n          return;\n        }\n      } catch {\n        // Ignore errors, keep polling\n      }\n\n      await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));\n    }\n\n    throw new Error(\n      `Restate partitions not ready after ${this.startupTimeoutMs}ms`\n    );\n  }\n}\n\n// Prepare the restate server\nasync function prepareRestateEndpoint(\n  param: EndpointOptions\n): Promise<http2.Http2Server> {\n  // Prepare RestateServer\n  const handler: (\n    request: http2.Http2ServerRequest,\n    response: http2.Http2ServerResponse\n  ) => void = createEndpointHandler(param);\n\n  // Start HTTP2 server on random port\n  const restateHttpServer = http2.createServer(handler);\n  await new Promise((resolve, reject) => {\n    restateHttpServer\n      .listen(0)\n      .once(\"listening\", resolve)\n      .once(\"error\", reject);\n  });\n  const restateServerPort = (restateHttpServer.address() as net.AddressInfo)\n    .port;\n  console.info(`Restate container listening on port ${restateServerPort}`);\n\n  return restateHttpServer;\n}\n\n// Prepare the restate testcontainer\nasync function prepareRestateTestContainer(\n  restateServerPort: number,\n  restateContainerFactory: () => GenericContainer,\n  options: ResolvedTestEnvironmentOptions\n): Promise<StartedTestContainer> {\n  let restateContainer = restateContainerFactory()\n    // Expose ports\n    .withExposedPorts(8080, 9070)\n    // Wait start on health checks and partition readiness\n    .withWaitStrategy(\n      Wait.forAll([\n        Wait.forHttp(\"/restate/health\", 8080),\n        Wait.forHttp(\"/health\", 9070),\n        new PartitionsReadyWaitStrategy(),\n      ])\n    );\n\n  if (options.storage === \"memory\") {\n    restateContainer = restateContainer.withTmpFs({ \"/restate-data\": \"rw\" });\n  }\n\n  const serviceEndpointHost =\n    options.serviceEndpointAccess === \"docker-host\"\n      ? \"host.docker.internal\"\n      : \"host.testcontainers.internal\";\n\n  if (options.serviceEndpointAccess === \"docker-host\") {\n    restateContainer = restateContainer.withExtraHosts([\n      { host: \"host.docker.internal\", ipAddress: \"host-gateway\" },\n    ]);\n  } else {\n    // This MUST be executed before starting the restate container.\n    // Expose host port to access the restate server.\n    await TestContainers.exposeHostPorts(restateServerPort);\n  }\n\n  // Start restate container\n  const startedRestateContainer = await restateContainer.start();\n\n  // From now on, if something fails, stop the container to cleanup the environment\n  try {\n    console.info(\n      `Registering services at http://${serviceEndpointHost}:${restateServerPort}...`\n    );\n\n    // Register this service endpoint\n    const res = await fetch(\n      `http://${startedRestateContainer.getHost()}:${startedRestateContainer.getMappedPort(\n        9070\n      )}/deployments`,\n      {\n        method: \"POST\",\n        headers: {\n          \"Content-Type\": \"application/json\",\n        },\n        body: JSON.stringify({\n          uri: `http://${serviceEndpointHost}:${restateServerPort}`,\n        }),\n      }\n    );\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(\n        `Error ${res.status} during registration: ${badResponse}`\n      );\n    }\n\n    const resp = (await res.json()) as { services: { name: string }[] };\n    console.info(\n      \"Registered services:\",\n      resp?.services?.map((s) => s.name)\n    );\n    return startedRestateContainer;\n  } catch (e) {\n    await startedRestateContainer.stop();\n    throw e;\n  }\n}\n\nexport interface TestEnvironmentOptions extends EndpointOptions {\n  /**\n   * Factory for the Restate container used by the test environment.\n   *\n   * Use this to customize the Restate image or container settings before the\n   * helper applies its own ports, wait strategy, networking, and storage\n   * configuration.\n   *\n   * For backwards compatibility, the second `start(options, factory)` argument\n   * is still supported and takes precedence over this option.\n   *\n   * @defaultValue `() => new RestateContainer()`\n   */\n  container?: () => GenericContainer;\n\n  /**\n   * Controls how the Restate container reaches the SDK service endpoint running\n   * on the test host.\n   *\n   * - `\"testcontainers\"` exposes the host port through Testcontainers and\n   *   registers `http://host.testcontainers.internal:<port>`.\n   * - `\"docker-host\"` skips Testcontainers port exposure, adds\n   *   `host.docker.internal:host-gateway` to the Restate container, and\n   *   registers `http://host.docker.internal:<port>`.\n   *\n   * @defaultValue `\"docker-host\"`\n   */\n  serviceEndpointAccess?: ServiceEndpointAccess;\n\n  /**\n   * Controls where Restate stores container data.\n   *\n   * - `\"disk\"` keeps the current Testcontainers/Docker storage behavior.\n   * - `\"memory\"` mounts `/restate-data` as tmpfs for faster disposable tests.\n   *\n   * @defaultValue `\"memory\"`\n   */\n  storage?: TestEnvironmentStorage;\n\n  /**\n   * Forces restate-server to always replay on a suspension point.\n   * This is useful to hunt non-deterministic bugs that might prevent\n   * your code from replaying correctly.\n   */\n  alwaysReplay?: boolean;\n\n  /**\n   * Disables retries in the restate-server invoker.\n   * This is useful in tests so that failures surface immediately\n   * instead of hanging through retry backoff.\n   */\n  disableRetries?: boolean;\n}\n\nexport class RestateTestEnvironment {\n  constructor(\n    readonly startedRestateHttpServer: http2.Http2Server,\n    readonly startedRestateContainer: StartedTestContainer\n  ) {}\n\n  public baseUrl(): string {\n    return `http://${this.startedRestateContainer.getHost()}:${this.startedRestateContainer.getMappedPort(\n      8080\n    )}`;\n  }\n\n  public adminAPIBaseUrl(): string {\n    return `http://${this.startedRestateContainer.getHost()}:${this.startedRestateContainer.getMappedPort(\n      9070\n    )}`;\n  }\n\n  // Create a handle that allows read/write of state under a given Virtual Object/Workflow key.\n  public stateOf<TState extends TypedState = UntypedState>(\n    service:\n      | VirtualObjectDefinition<string, unknown>\n      | WorkflowDefinition<string, unknown>,\n    key: string\n  ): StateProxy<TState> {\n    return new StateProxy(this.adminAPIBaseUrl(), service.name, key);\n  }\n\n  public async stop() {\n    await this.startedRestateContainer.stop();\n    this.startedRestateHttpServer.close();\n  }\n\n  public static async start(\n    options: TestEnvironmentOptions,\n    restateContainerFactory?: () => GenericContainer\n  ): Promise<RestateTestEnvironment> {\n    let containerFactory: () => GenericContainer;\n    if (restateContainerFactory) {\n      containerFactory = restateContainerFactory;\n    } else if (options.container) {\n      containerFactory = options.container;\n    } else {\n      containerFactory = () => {\n        const container = new RestateContainer();\n        if (options.alwaysReplay) {\n          container.alwaysReplay();\n        }\n        if (options.disableRetries) {\n          container.disableRetries();\n        }\n        return container;\n      };\n    }\n    const resolvedStartOptions: ResolvedTestEnvironmentOptions = {\n      serviceEndpointAccess:\n        options.serviceEndpointAccess ??\n        DEFAULT_START_OPTIONS.serviceEndpointAccess,\n      storage: options.storage ?? DEFAULT_START_OPTIONS.storage,\n    };\n\n    const startedRestateHttpServer = await prepareRestateEndpoint(options);\n    const startedRestateContainer = await prepareRestateTestContainer(\n      (startedRestateHttpServer.address() as net.AddressInfo).port,\n      containerFactory,\n      resolvedStartOptions\n    );\n    return new RestateTestEnvironment(\n      startedRestateHttpServer,\n      startedRestateContainer\n    );\n  }\n}\n\nexport class RestateContainer extends GenericContainer {\n  constructor(version = \"latest\") {\n    super(`docker.io/restatedev/restate:${version}`);\n  }\n\n  /**\n   * Forces restate-server to always replay on a suspension point.\n   * This is useful to hunt non-deterministic bugs that might prevent\n   * your code from replaying correctly.\n   */\n  alwaysReplay(): this {\n    this.withEnvironment({\n      RESTATE_WORKER__INVOKER__INACTIVITY_TIMEOUT: \"0s\",\n    });\n    return this;\n  }\n\n  /**\n   * Disables retries in the restate-server invoker.\n   * This is useful in tests so that failures surface immediately\n   * instead of hanging through retry backoff.\n   */\n  disableRetries(): this {\n    this.withEnvironment({\n      RESTATE_DEFAULT_RETRY_POLICY__MAX_ATTEMPTS: \"1\",\n      RESTATE_DEFAULT_RETRY_POLICY__ON_MAX_ATTEMPTS: \"kill\",\n    });\n    return this;\n  }\n}\nexport class StateProxy<TState extends TypedState> {\n  constructor(\n    private adminAPIBaseUrl: string,\n    private service: string,\n    private serviceKey: string\n  ) {}\n\n  // Read a single value from state under a given Virtual Object or Workflow key\n  public async get<TValue, TKey extends keyof TState = string>(\n    name: TState extends UntypedState ? string : TKey,\n    serde?: Serde<TState extends UntypedState ? TValue : TState[TKey]>\n  ): Promise<(TState extends UntypedState ? TValue : TState[TKey]) | null> {\n    serde = serde ?? defaultSerde();\n\n    const res = await fetch(`${this.adminAPIBaseUrl}/query`, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n      },\n      body: JSON.stringify({\n        query: `SELECT value from state where service_name = '${\n          this.service\n        }' and service_key = '${this.serviceKey}' and key = '${String(name)}';`,\n      }),\n    });\n\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(`Error ${res.status} during read state: ${badResponse}`);\n    }\n\n    // eslint-disable-next-line @typescript-eslint/await-thenable\n    const table = (await tableFromIPC(res.body)).toArray() as {\n      key: string;\n      value: Uint8Array;\n    }[];\n\n    if (table.length === 0) {\n      return null;\n    }\n\n    return serde.deserialize(table[0]!.value);\n  }\n\n  // Read all values from state under a given Virtual Object or Workflow key\n  public async getAll<TValues extends TypedState>(\n    serde?: Serde<\n      TState extends UntypedState\n        ? TValues[keyof TValues]\n        : TState[keyof TState]\n    >\n  ): Promise<TState extends UntypedState ? TValues : TState> {\n    serde = serde ?? defaultSerde();\n\n    const items = await this.getAllRaw();\n\n    return Object.fromEntries(\n      items.map(({ key, value }) => {\n        return [key, serde.deserialize(value)];\n      })\n    ) as TState extends UntypedState ? TValues : TState;\n  }\n\n  private async getAllRaw(): Promise<{ key: string; value: Uint8Array }[]> {\n    const res = await fetch(`${this.adminAPIBaseUrl}/query`, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n      },\n      body: JSON.stringify({\n        query: `SELECT key, value from state where service_name = '${this.service}' and service_key = '${this.serviceKey}';`,\n      }),\n    });\n\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(`Error ${res.status} during read state: ${badResponse}`);\n    }\n\n    // eslint-disable-next-line @typescript-eslint/await-thenable\n    const table = (await tableFromIPC(res.body)).toArray() as {\n      key: string;\n      value: Uint8Array;\n    }[];\n\n    return table;\n  }\n\n  // Asynchronously set a single value from state under a given Virtual Object or Workflow key.\n  // This will first read all values, then insert the update and submit the new set of values to Restate;\n  // as such it is possible to overwrite changes that happened between the read and the mutation being applied.\n  // A successful return from this function does not imply that the set has finished, only that the mutation\n  // was submitted to Restate for processing.\n  public async set<TValue, TKey extends keyof TState = string>(\n    name: TState extends UntypedState ? string : TKey,\n    value: TState extends UntypedState ? TValue : TState[TKey],\n    serde?: Serde<TState extends UntypedState ? TValue : TState[TKey]>\n  ): Promise<void> {\n    serde = serde ?? defaultSerde();\n    const serialisedValue = serde.serialize(value);\n\n    const items = await this.getAllRaw();\n\n    items.push({ key: String(name), value: serialisedValue });\n\n    await this.setAllRaw(items.map(({ key, value }) => [key, value]));\n  }\n\n  // Asynchronously set all state values under a given Virtual Object or Workflow key.\n  // A successful return from this function does not imply that the set has finished,\n  // only that the mutation was submitted to Restate for processing.\n  public async setAll<TValues extends TypedState>(\n    values: TState extends UntypedState ? TValues : TState,\n    serde?: Serde<\n      TState extends UntypedState\n        ? TValues[keyof TValues]\n        : TState[keyof TState]\n    >\n  ) {\n    serde = serde ?? defaultSerde();\n\n    return this.setAllRaw(\n      Object.entries<\n        TState extends UntypedState\n          ? TValues[keyof TValues]\n          : TState[keyof TState]\n      >(values).map(([key, value]) => {\n        return [key, serde.serialize(value)];\n      })\n    );\n  }\n\n  private async setAllRaw(\n    entries: [key: string, value: Uint8Array][],\n    version?: string\n  ) {\n    const res = await fetch(\n      `${this.adminAPIBaseUrl}/services/${this.service}/state`,\n      {\n        method: \"POST\",\n        headers: {\n          \"Content-Type\": \"application/json\",\n        },\n        body: JSON.stringify(\n          {\n            version,\n            object_key: this.serviceKey,\n            new_state: Object.fromEntries(entries),\n          },\n          (key, value) => {\n            if (value instanceof Uint8Array) {\n              return Array.from(value);\n            } else {\n              // eslint-disable-next-line @typescript-eslint/no-unsafe-return\n              return value;\n            }\n          }\n        ),\n      }\n    );\n\n    if (!res.ok) {\n      const badResponse = await res.text();\n      throw new Error(\n        `Error ${res.status} during modify state: ${badResponse}`\n      );\n    }\n  }\n}\n\nexport const defaultSerde = <T>(): Serde<T> => {\n  return serde.json as Serde<T>;\n};\n"],"mappings":";;;;;;AA2CA,MAAMA,wBAAwD;CAC5D,uBAAuB;CACvB,SAAS;CACV;;;;;;AAOD,IAAM,8BAAN,MAA0D;CACxD,AAAQ,mBAAmB;CAC3B,AAAQ,oBAAoB;CAC5B,AAAiB;CACjB,AAAiB;CAEjB,YAAY,OAAO,MAAM,iBAAiB,KAAK;AAC7C,OAAK,OAAO;AACZ,OAAK,iBAAiB;;CAGxB,AAAO,mBAAmB,kBAAgC;AACxD,OAAK,mBAAmB;AACxB,OAAK,oBAAoB;AACzB,SAAO;;CAGT,AAAO,sBAA+B;AACpC,SAAO,KAAK;;CAGd,AAAO,oBAA4B;AACjC,SAAO,KAAK;;CAGd,MAAa,eACX,WACA,YACe;EAIf,MAAM,WAAW,WAHF,MAAM,2BAA2B,EAC5B,KAAK,iBAAiB,KAEV,GADb,WAAW,WAAW,KAAK,KAAK;EAGnD,MAAM,YAAY,KAAK,KAAK;AAE5B,SAAO,KAAK,KAAK,GAAG,YAAY,KAAK,kBAAkB;AACrD,OAAI;AASF,SARY,MAAM,MAAM,GAAG,SAAS,SAAS;KAC3C,QAAQ;KACR,SAAS,EAAE,gBAAgB,oBAAoB;KAC/C,MAAM,KAAK,UAAU,EACnB,OAAO,uCACR,CAAC;KACH,CAAC,EAEM,GAEN;WAEI;AAIR,SAAM,IAAI,SAAS,YAAY,WAAW,SAAS,KAAK,eAAe,CAAC;;AAG1E,QAAM,IAAI,MACR,sCAAsC,KAAK,iBAAiB,IAC7D;;;AAKL,eAAe,uBACb,OAC4B;CAE5B,MAAMC,UAGM,sBAAsB,MAAM;CAGxC,MAAM,oBAAoB,MAAM,aAAa,QAAQ;AACrD,OAAM,IAAI,SAAS,SAAS,WAAW;AACrC,oBACG,OAAO,EAAE,CACT,KAAK,aAAa,QAAQ,CAC1B,KAAK,SAAS,OAAO;GACxB;CACF,MAAM,oBAAqB,kBAAkB,SAAS,CACnD;AACH,SAAQ,KAAK,uCAAuC,oBAAoB;AAExE,QAAO;;AAIT,eAAe,4BACb,mBACA,yBACA,SAC+B;CAC/B,IAAI,mBAAmB,yBAAyB,CAE7C,iBAAiB,MAAM,KAAK,CAE5B,iBACC,KAAK,OAAO;EACV,KAAK,QAAQ,mBAAmB,KAAK;EACrC,KAAK,QAAQ,WAAW,KAAK;EAC7B,IAAI,6BAA6B;EAClC,CAAC,CACH;AAEH,KAAI,QAAQ,YAAY,SACtB,oBAAmB,iBAAiB,UAAU,EAAE,iBAAiB,MAAM,CAAC;CAG1E,MAAM,sBACJ,QAAQ,0BAA0B,gBAC9B,yBACA;AAEN,KAAI,QAAQ,0BAA0B,cACpC,oBAAmB,iBAAiB,eAAe,CACjD;EAAE,MAAM;EAAwB,WAAW;EAAgB,CAC5D,CAAC;KAIF,OAAM,eAAe,gBAAgB,kBAAkB;CAIzD,MAAM,0BAA0B,MAAM,iBAAiB,OAAO;AAG9D,KAAI;AACF,UAAQ,KACN,kCAAkC,oBAAoB,GAAG,kBAAkB,KAC5E;EAGD,MAAM,MAAM,MAAM,MAChB,UAAU,wBAAwB,SAAS,CAAC,GAAG,wBAAwB,cACrE,KACD,CAAC,eACF;GACE,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UAAU,EACnB,KAAK,UAAU,oBAAoB,GAAG,qBACvC,CAAC;GACH,CACF;AACD,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MACR,SAAS,IAAI,OAAO,wBAAwB,cAC7C;;EAGH,MAAM,OAAQ,MAAM,IAAI,MAAM;AAC9B,UAAQ,KACN,wBACA,MAAM,UAAU,KAAK,MAAM,EAAE,KAAK,CACnC;AACD,SAAO;UACA,GAAG;AACV,QAAM,wBAAwB,MAAM;AACpC,QAAM;;;AA0DV,IAAa,yBAAb,MAAa,uBAAuB;CAClC,YACE,AAASC,0BACT,AAASC,yBACT;EAFS;EACA;;CAGX,AAAO,UAAkB;AACvB,SAAO,UAAU,KAAK,wBAAwB,SAAS,CAAC,GAAG,KAAK,wBAAwB,cACtF,KACD;;CAGH,AAAO,kBAA0B;AAC/B,SAAO,UAAU,KAAK,wBAAwB,SAAS,CAAC,GAAG,KAAK,wBAAwB,cACtF,KACD;;CAIH,AAAO,QACL,SAGA,KACoB;AACpB,SAAO,IAAI,WAAW,KAAK,iBAAiB,EAAE,QAAQ,MAAM,IAAI;;CAGlE,MAAa,OAAO;AAClB,QAAM,KAAK,wBAAwB,MAAM;AACzC,OAAK,yBAAyB,OAAO;;CAGvC,aAAoB,MAClB,SACA,yBACiC;EACjC,IAAIC;AACJ,MAAI,wBACF,oBAAmB;WACV,QAAQ,UACjB,oBAAmB,QAAQ;MAE3B,0BAAyB;GACvB,MAAM,YAAY,IAAI,kBAAkB;AACxC,OAAI,QAAQ,aACV,WAAU,cAAc;AAE1B,OAAI,QAAQ,eACV,WAAU,gBAAgB;AAE5B,UAAO;;EAGX,MAAMC,uBAAuD;GAC3D,uBACE,QAAQ,yBACR,sBAAsB;GACxB,SAAS,QAAQ,WAAW,sBAAsB;GACnD;EAED,MAAM,2BAA2B,MAAM,uBAAuB,QAAQ;AAMtE,SAAO,IAAI,uBACT,0BAN8B,MAAM,4BACnC,yBAAyB,SAAS,CAAqB,MACxD,kBACA,qBACD,CAIA;;;AAIL,IAAa,mBAAb,cAAsC,iBAAiB;CACrD,YAAY,UAAU,UAAU;AAC9B,QAAM,gCAAgC,UAAU;;;;;;;CAQlD,eAAqB;AACnB,OAAK,gBAAgB,EACnB,6CAA6C,MAC9C,CAAC;AACF,SAAO;;;;;;;CAQT,iBAAuB;AACrB,OAAK,gBAAgB;GACnB,4CAA4C;GAC5C,+CAA+C;GAChD,CAAC;AACF,SAAO;;;AAGX,IAAa,aAAb,MAAmD;CACjD,YACE,AAAQC,iBACR,AAAQC,SACR,AAAQC,YACR;EAHQ;EACA;EACA;;CAIV,MAAa,IACX,MACA,SACuE;AACvE,YAAQC,WAAS,cAAc;EAE/B,MAAM,MAAM,MAAM,MAAM,GAAG,KAAK,gBAAgB,SAAS;GACvD,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UAAU,EACnB,OAAO,iDACL,KAAK,QACN,uBAAuB,KAAK,WAAW,eAAe,OAAO,KAAK,CAAC,KACrE,CAAC;GACH,CAAC;AAEF,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MAAM,SAAS,IAAI,OAAO,sBAAsB,cAAc;;EAI1E,MAAM,SAAS,MAAM,aAAa,IAAI,KAAK,EAAE,SAAS;AAKtD,MAAI,MAAM,WAAW,EACnB,QAAO;AAGT,SAAOA,QAAM,YAAY,MAAM,GAAI,MAAM;;CAI3C,MAAa,OACX,SAKyD;AACzD,YAAQA,WAAS,cAAc;EAE/B,MAAM,QAAQ,MAAM,KAAK,WAAW;AAEpC,SAAO,OAAO,YACZ,MAAM,KAAK,EAAE,KAAK,YAAY;AAC5B,UAAO,CAAC,KAAKA,QAAM,YAAY,MAAM,CAAC;IACtC,CACH;;CAGH,MAAc,YAA2D;EACvE,MAAM,MAAM,MAAM,MAAM,GAAG,KAAK,gBAAgB,SAAS;GACvD,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UAAU,EACnB,OAAO,sDAAsD,KAAK,QAAQ,uBAAuB,KAAK,WAAW,KAClH,CAAC;GACH,CAAC;AAEF,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MAAM,SAAS,IAAI,OAAO,sBAAsB,cAAc;;AAS1E,UALe,MAAM,aAAa,IAAI,KAAK,EAAE,SAAS;;CAaxD,MAAa,IACX,MACA,OACA,SACe;AACf,YAAQA,WAAS,cAAc;EAC/B,MAAM,kBAAkBA,QAAM,UAAU,MAAM;EAE9C,MAAM,QAAQ,MAAM,KAAK,WAAW;AAEpC,QAAM,KAAK;GAAE,KAAK,OAAO,KAAK;GAAE,OAAO;GAAiB,CAAC;AAEzD,QAAM,KAAK,UAAU,MAAM,KAAK,EAAE,KAAK,qBAAY,CAAC,KAAKC,QAAM,CAAC,CAAC;;CAMnE,MAAa,OACX,QACA,SAKA;AACA,YAAQD,WAAS,cAAc;AAE/B,SAAO,KAAK,UACV,OAAO,QAIL,OAAO,CAAC,KAAK,CAAC,KAAK,WAAW;AAC9B,UAAO,CAAC,KAAKA,QAAM,UAAU,MAAM,CAAC;IACpC,CACH;;CAGH,MAAc,UACZ,SACA,SACA;EACA,MAAM,MAAM,MAAM,MAChB,GAAG,KAAK,gBAAgB,YAAY,KAAK,QAAQ,SACjD;GACE,QAAQ;GACR,SAAS,EACP,gBAAgB,oBACjB;GACD,MAAM,KAAK,UACT;IACE;IACA,YAAY,KAAK;IACjB,WAAW,OAAO,YAAY,QAAQ;IACvC,GACA,KAAK,UAAU;AACd,QAAI,iBAAiB,WACnB,QAAO,MAAM,KAAK,MAAM;QAGxB,QAAO;KAGZ;GACF,CACF;AAED,MAAI,CAAC,IAAI,IAAI;GACX,MAAM,cAAc,MAAM,IAAI,MAAM;AACpC,SAAM,IAAI,MACR,SAAS,IAAI,OAAO,wBAAwB,cAC7C;;;;AAKP,MAAa,qBAAkC;AAC7C,QAAO,MAAM"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@restatedev/restate-sdk-testcontainers",
-  "version": "1.13.0",
+  "version": "1.14.1",
   "description": "Typescript SDK for Restate",
   "author": "Restate Developers",
   "email": "code@restate.dev",
@@ -34,8 +34,8 @@
   "dependencies": {
     "apache-arrow": "^18.0.0",
     "testcontainers": "^11.12.0",
-    "@restatedev/restate-sdk": "1.13.0",
-    "@restatedev/restate-sdk-clients": "1.13.0"
+    "@restatedev/restate-sdk": "1.14.1",
+    "@restatedev/restate-sdk-clients": "1.14.1"
   },
   "devDependencies": {},
   "scripts": {