@restatedev/restate-sdk 1.0.1 → 1.1.1

package/dist/cjs/test/protocol_stream.test.js

@@ -0,0 +1,243 @@
+"use strict";
+/*
+ * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH
+ *
+ * This file is part of the Restate SDK for Node.js/TypeScript,
+ * which is released under the MIT license.
+ *
+ * You can find a copy of the license in file LICENSE in the root
+ * directory of this repository or package, or at
+ * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const protocol_js_1 = require("../src/types/protocol.js");
+const connection_js_1 = require("../src/connection/connection.js");
+const types_js_1 = require("../src/types/types.js");
+const stream = __importStar(require("node:stream/web"));
+const promises_1 = require("timers/promises");
+const promises_js_1 = require("../src/utils/promises.js");
+const vitest_1 = require("vitest");
+const logger_js_1 = require("../src/logger.js");
+// The following test suite is taken from headers.rs
+(0, vitest_1.describe)("Header", () => {
+    (0, vitest_1.it)("Should round trip a custom message", () => {
+        const a = new types_js_1.Header(0xfc00n, 10);
+        const b = types_js_1.Header.fromU64be(a.toU64be());
+        (0, vitest_1.expect)(b).toStrictEqual(a);
+    });
+    (0, vitest_1.it)("invoke_test", () => roundtripTest(newStart(25)));
+    (0, vitest_1.it)("completion_test", () => roundtripTest(newHeader(protocol_js_1.COMPLETION_MESSAGE_TYPE, 22)));
+    (0, vitest_1.it)("completed_get_state", () => roundtripTest(newCompletableEntry(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, true, 0)));
+    (0, vitest_1.it)("not_completed_get_state", () => roundtripTest(newCompletableEntry(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, false, 0)));
+    (0, vitest_1.it)("completed_get_state_with_len", () => roundtripTest(newCompletableEntry(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, true, 10341)));
+    (0, vitest_1.it)("custom_entry", () => roundtripTest(newHeader(0xfc00n, 10341)));
+    (0, vitest_1.it)("custom_entry_with_requires_ack", () => roundtripTest(new types_js_1.Header(0xfc00n, 10341, undefined, true)));
+});
+const rlog = (0, logger_js_1.createRestateConsole)(logger_js_1.defaultLogger, logger_js_1.LogSource.SYSTEM);
+(0, vitest_1.describe)("Restate Streaming Connection", () => {
+    (0, vitest_1.it)("should demonstrate how to write messages and read messages.", async () => {
+        // imagine that the HTTP2 request handler hands to you a bidirectional (duplex in node's lingo)
+        // binary stream, that you can read from and write to.
+        const http2stream = mockHttp2DuplexStream();
+        // the following demonstrates how to use a stream_encoder/decoder to convert
+        // a raw duplex stream to a high-level stream of Restate's protocol messages and headers.
+        const restateStream = new connection_js_1.RestateConnection(rlog, {}, http2stream);
+        // here we need to create a promise for the sake of this test.
+        // this future will be resolved once something is emitted on the stream.
+        const result = new promises_js_1.CompletablePromise();
+        restateStream.pipeToConsumer({
+            handleMessage: (m) => {
+                result.resolve(m);
+                return false;
+            },
+            // eslint-disable-next-line @typescript-eslint/no-empty-function
+            handleInputClosed: () => { },
+            handleStreamError: (error) => result.reject(error),
+        });
+        // now, let's simulate sending a message
+        void restateStream.send(new types_js_1.Message(protocol_js_1.START_MESSAGE_TYPE, new protocol_js_1.StartMessage({
+            id: Buffer.from("abcd"),
+            debugId: "abcd",
+            knownEntries: 1337,
+        }), undefined));
+        // and collect what was written
+        await restateStream.end();
+        const msg = await result.promise;
+        (0, vitest_1.expect)(msg.messageType).toStrictEqual(protocol_js_1.START_MESSAGE_TYPE);
+        (0, vitest_1.expect)(msg.message.knownEntries).toStrictEqual(1337);
+    });
+    (0, vitest_1.it)("should await sending of small data when closing", async () => {
+        const { duplex, processBytes } = mockBackpressuredDuplex(128);
+        const connection = new connection_js_1.RestateConnection(rlog, {}, duplex);
+        // enqueue small data, below watermark, so that 'end' can immediately be written
+        void connection.send(newMessage(10));
+        const done = connection.end();
+        await verifyPromisePending(done);
+        // now let the stream flow
+        processBytes(10 * 1024);
+        // we should be able to finish cleanly
+        await done;
+    });
+    (0, vitest_1.it)("should await sending of larger data when closing", async () => {
+        const { duplex, processBytes } = mockBackpressuredDuplex(128);
+        const connection = new connection_js_1.RestateConnection(rlog, {}, duplex);
+        // enqueue quite some data, before allowing any bytes to flow any
+        void connection.send(newMessage(1024));
+        void connection.send(newMessage(1024));
+        void connection.send(newMessage(1024));
+        const done = connection.end();
+        await verifyPromisePending(done);
+        // now let the stream flow
+        processBytes(10 * 1024);
+        // we should be able to finish cleanly
+        await done;
+    });
+    (0, vitest_1.it)("should not trigger backpressure for small messages", async () => {
+        const { duplex } = mockBackpressuredDuplex(1024);
+        const connection = new connection_js_1.RestateConnection(rlog, {}, duplex);
+        // enqueue a message that is not too large
+        const promise1 = connection.send(newMessage(80));
+        const promise2 = connection.send(newMessage(80));
+        await verifyPromiseResolved(promise1);
+        await verifyPromiseResolved(promise2);
+    });
+    (0, vitest_1.it)("should trigger backpressure for large messages", async () => {
+        const { duplex } = mockBackpressuredDuplex(1024);
+        const connection = new connection_js_1.RestateConnection(rlog, {}, duplex);
+        // this message should get sent immediately because its smaller than 1024
+        const promise1 = connection.send(newMessage(800));
+        await verifyPromiseResolved(promise1);
+        // this one should hang because it takes us over 1024
+        const promise2 = connection.send(newMessage(800));
+        await verifyPromisePending(promise2);
+    });
+    (0, vitest_1.it)("should resolve backpressure promises when the stream flows", async () => {
+        const { duplex, processBytes } = mockBackpressuredDuplex(1024);
+        const connection = new connection_js_1.RestateConnection(rlog, {}, duplex);
+        // this message should get sent immediately because its smaller than 1024
+        const promise1 = connection.send(newMessage(800));
+        await verifyPromiseResolved(promise1);
+        // this one should hang until we let it flow because it takes us over 1024
+        const promise2 = connection.send(newMessage(800));
+        await verifyPromisePending(promise2);
+        // now let the stream flow
+        processBytes(10 * 1024);
+        // this should now complete
+        await promise2;
+    });
+});
+// The following is taken from headers.rs tests
+function newHeader(messageTypeId, length) {
+    return new types_js_1.Header(messageTypeId, length);
+}
+function newStart(length) {
+    return new types_js_1.Header(protocol_js_1.START_MESSAGE_TYPE, length, undefined, undefined);
+}
+function newCompletableEntry(ty, completed, length) {
+    return new types_js_1.Header(ty, length, completed);
+}
+function newMessage(size) {
+    return new types_js_1.Message(protocol_js_1.SET_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.SetStateEntryMessage({
+        key: Buffer.from("abcd"),
+        value: Buffer.alloc(size),
+    }));
+}
+function sameTruthness(a, b) {
+    if (a) {
+        (0, vitest_1.expect)(b).toBeTruthy();
+    }
+    else {
+        (0, vitest_1.expect)(b).toBeFalsy();
+    }
+}
+async function verifyPromisePending(promise) {
+    let complete = false;
+    void promise.then(() => {
+        complete = true;
+    });
+    await (0, promises_1.setTimeout)(0); // let the callbacks and tasks proceed first
+    (0, vitest_1.expect)(complete).toBeFalsy();
+}
+async function verifyPromiseResolved(promise) {
+    let complete = false;
+    void promise.then(() => {
+        complete = true;
+    });
+    await (0, promises_1.setTimeout)(0); // let the callbacks and tasks proceed first
+    (0, vitest_1.expect)(complete).toBeTruthy();
+}
+function roundtripTest(a) {
+    const b = types_js_1.Header.fromU64be(a.toU64be());
+    (0, vitest_1.expect)(b.messageType).toStrictEqual(a.messageType);
+    (0, vitest_1.expect)(b.frameLength).toStrictEqual(a.frameLength);
+    sameTruthness(a.completedFlag, b.completedFlag);
+    sameTruthness(a.requiresAckFlag, b.requiresAckFlag);
+    sameTruthness(a.partialStateFlag, b.partialStateFlag);
+}
+function mockHttp2DuplexStream() {
+    // pipe output into input
+    return new stream.TransformStream();
+}
+function mockBackpressuredDuplex(highWaterMark) {
+    let permittedBytes = 0;
+    const queue = [];
+    const writable = new stream.WritableStream({
+        write: async (chunk) => {
+            if (permittedBytes >= chunk.length) {
+                permittedBytes -= chunk.length;
+            }
+            else {
+                permittedBytes = 0;
+                await new Promise((resolve) => {
+                    queue.push({ bytes: chunk.length - permittedBytes, cb: resolve });
+                });
+            }
+        },
+    }, new stream.ByteLengthQueuingStrategy({ highWaterMark }));
+    function processBytes(numBytes) {
+        while (queue.length > 0 && numBytes > 0) {
+            const next = queue[0];
+            if (next.bytes > numBytes) {
+                next.bytes -= numBytes;
+                numBytes = 0;
+            }
+            else {
+                numBytes -= next.bytes;
+                queue.pop();
+                process.nextTick(next.cb);
+            }
+        }
+        permittedBytes += numBytes;
+    }
+    return {
+        duplex: {
+            readable: new stream.ReadableStream(),
+            writable,
+        },
+        processBytes,
+    };
+}
+//# sourceMappingURL=protocol_stream.test.js.map

package/dist/cjs/test/protocol_stream.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol_stream.test.js","sourceRoot":"","sources":["../../../test/protocol_stream.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,0DAOkC;AAClC,mEAAoE;AACpE,oDAAwD;AACxD,wDAA0C;AAC1C,8CAA6C;AAC7C,0DAA8D;AAC9D,mCAA8C;AAC9C,gDAI0B;AAE1B,oDAAoD;AACpD,IAAA,iBAAQ,EAAC,QAAQ,EAAE,GAAG,EAAE;IACtB,IAAA,WAAE,EAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,CAAC,GAAG,IAAI,iBAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,iBAAM,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAExC,IAAA,eAAM,EAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,aAAa,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAErD,IAAA,WAAE,EAAC,iBAAiB,EAAE,GAAG,EAAE,CACzB,aAAa,CAAC,SAAS,CAAC,qCAAuB,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAEzD,IAAA,WAAE,EAAC,qBAAqB,EAAE,GAAG,EAAE,CAC7B,aAAa,CAAC,mBAAmB,CAAC,0CAA4B,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7E,IAAA,WAAE,EAAC,yBAAyB,EAAE,GAAG,EAAE,CACjC,aAAa,CAAC,mBAAmB,CAAC,0CAA4B,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE9E,IAAA,WAAE,EAAC,8BAA8B,EAAE,GAAG,EAAE,CACtC,aAAa,CACX,mBAAmB,CAAC,0CAA4B,EAAE,IAAI,EAAE,KAAK,CAAC,CAC/D,CAAC,CAAC;IAEL,IAAA,WAAE,EAAC,cAAc,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IAEnE,IAAA,WAAE,EAAC,gCAAgC,EAAE,GAAG,EAAE,CACxC,aAAa,CAAC,IAAI,iBAAM,CAAC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;AAChE,CAAC,CAAC,CAAC;AAEH,MAAM,IAAI,GAAG,IAAA,gCAAoB,EAAC,yBAAa,EAAE,qBAAS,CAAC,MAAM,CAAC,CAAC;AAEnE,IAAA,iBAAQ,EAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,IAAA,WAAE,EAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,+FAA+F;QAC/F,sDAAsD;QACtD,MAAM,WAAW,GAAG,qBAAqB,EAAE,CAAC;QAE5C,4EAA4E;QAC5E,yFAAyF;QACzF,MAAM,aAAa,GAAG,IAAI,iCAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC;QAEnE,8DAA8D;QAC9D,wEAAwE;QACxE,MAAM,MAAM,GAAG,IAAI,gCAAkB,EAAW,CAAC;QAEjD,aAAa,CAAC,cAAc,CAAC;YAC3B,aAAa,EAAE,CAAC,CAAC,EAAE,EAAE;gBACnB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAClB,OAAO,KAAK,CAAC;YACf,CAAC;YAED,gEAAgE;YAChE,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;YAE3B,iBAAiB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;SACnD,CAAC,CAAC;QAEH,wCAAwC;QACxC,KAAK,aAAa,CAAC,IAAI,CACrB,IAAI,kBAAO,CACT,gCAAkB,EAClB,IAAI,0BAAY,CAAC;YACf,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;YACvB,OAAO,EAAE,MAAM;YACf,YAAY,EAAE,IAAI;SACnB,CAAC,EACF,SAAS,CACV,CACF,CAAC;QAEF,+BAA+B;QAC/B,MAAM,aAAa,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC;QAEjC,IAAA,eAAM,EAAC,GAAG,CAAC,WAAW,CAAC,CAAC,aAAa,CAAC,gCAAkB,CAAC,CAAC;QAC1D,IAAA,eAAM,EAAE,GAAG,CAAC,OAAwB,CAAC,YAAY,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAC9D,MAAM,UAAU,GAAG,IAAI,iCAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QAE3D,gFAAgF;QAChF,KAAK,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC;QAE9B,MAAM,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAEjC,0BAA0B;QAC1B,YAAY,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAExB,sCAAsC;QACtC,MAAM,IAAI,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAC9D,MAAM,UAAU,GAAG,IAAI,iCAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QAE3D,iEAAiE;QACjE,KAAK,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QACvC,KAAK,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QACvC,KAAK,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC;QAE9B,MAAM,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAEjC,0BAA0B;QAC1B,YAAY,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAExB,sCAAsC;QACtC,MAAM,IAAI,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,iCAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QAE3D,0CAA0C;QAC1C,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;QAEjD,MAAM,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,EAAE,MAAM,EAAE,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,iCAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QAE3D,yEAAyE;QACzE,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QAEtC,qDAAqD;QACrD,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,UAAU,GAAG,IAAI,iCAAiB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;QAE3D,yEAAyE;QACzE,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACtC,0EAA0E;QAC1E,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAErC,0BAA0B;QAC1B,YAAY,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAExB,2BAA2B;QAC3B,MAAM,QAAQ,CAAC;IACjB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,+CAA+C;AAC/C,SAAS,SAAS,CAAC,aAAqB,EAAE,MAAc;IACtD,OAAO,IAAI,iBAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,QAAQ,CAAC,MAAc;IAC9B,OAAO,IAAI,iBAAM,CAAC,gCAAkB,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;AACtE,CAAC;AAED,SAAS,mBAAmB,CAC1B,EAAU,EACV,SAAkB,EAClB,MAAc;IAEd,OAAO,IAAI,iBAAM,CAAC,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC9B,OAAO,IAAI,kBAAO,CAChB,0CAA4B,EAC5B,IAAI,kCAAoB,CAAC;QACvB,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;QACxB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC;KAC1B,CAAC,CACH,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAO,CAAI,EAAE,CAAI;IACrC,IAAI,CAAC,EAAE,CAAC;QACN,IAAA,eAAM,EAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;IACzB,CAAC;SAAM,CAAC;QACN,IAAA,eAAM,EAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;IACxB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,OAAyB;IAC3D,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;QACrB,QAAQ,GAAG,IAAI,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,MAAM,IAAA,qBAAU,EAAC,CAAC,CAAC,CAAC,CAAC,4CAA4C;IAEjE,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,SAAS,EAAE,CAAC;AAC/B,CAAC;AAED,KAAK,UAAU,qBAAqB,CAAC,OAAyB;IAC5D,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;QACrB,QAAQ,GAAG,IAAI,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,MAAM,IAAA,qBAAU,EAAC,CAAC,CAAC,CAAC,CAAC,4CAA4C;IAEjE,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,aAAa,CAAC,CAAS;IAC9B,MAAM,CAAC,GAAG,iBAAM,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IACxC,IAAA,eAAM,EAAC,CAAC,CAAC,WAAW,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IACnD,IAAA,eAAM,EAAC,CAAC,CAAC,WAAW,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IACnD,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,aAAa,CAAC,CAAC;IAChD,aAAa,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC;IACpD,aAAa,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,qBAAqB;IAI5B,yBAAyB;IACzB,OAAO,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;AACtC,CAAC;AAED,SAAS,uBAAuB,CAAC,aAAqB;IAIpD,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,MAAM,KAAK,GAAsD,EAAE,CAAC;IAEpE,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,cAAc,CACxC;QACE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACrB,IAAI,cAAc,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACnC,cAAc,IAAI,KAAK,CAAC,MAAM,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,cAAc,GAAG,CAAC,CAAC;gBACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBAC5B,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,cAAc,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;gBACpE,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;KACF,EACD,IAAI,MAAM,CAAC,yBAAyB,CAAC,EAAE,aAAa,EAAE,CAAC,CACxD,CAAC;IAEF,SAAS,YAAY,CAAC,QAAgB;QACpC,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAEtB,IAAI,IAAI,CAAC,KAAK,GAAG,QAAQ,EAAE,CAAC;gBAC1B,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC;gBACvB,QAAQ,GAAG,CAAC,CAAC;YACf,CAAC;iBAAM,CAAC;gBACN,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC;gBACvB,KAAK,CAAC,GAAG,EAAE,CAAC;gBACZ,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QACD,cAAc,IAAI,QAAQ,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,MAAM,EAAE;YACN,QAAQ,EAAE,IAAI,MAAM,CAAC,cAAc,EAAE;YACrC,QAAQ;SACT;QACD,YAAY;KACb,CAAC;AACJ,CAAC"}

package/dist/cjs/test/protoutils.d.ts

@@ -0,0 +1,44 @@
+/// <reference types="node" />
+import { Empty } from "@bufbuild/protobuf";
+import { Message } from "../src/types/types.js";
+import { Failure, StartMessage_StateEntry } from "../src/generated/proto/protocol_pb.js";
+import type { JournalErrorContext } from "../src/types/errors.js";
+export type StartMessageOpts = {
+    knownEntries?: number;
+    partialState?: boolean;
+    state?: Buffer[][];
+    key?: string;
+};
+export declare function startMessage({ knownEntries, partialState, state, key, }?: StartMessageOpts): Message;
+export declare function toStateEntries(entries: Buffer[][]): StartMessage_StateEntry[];
+export declare function inputMessage(value: Uint8Array): Message;
+export declare function outputMessage(value?: Uint8Array, failure?: Failure): Message;
+export declare function getStateMessage<T>(key: string, value?: T, empty?: boolean, failure?: Failure): Message;
+export declare function getStateMessageWithEmptyResult(key: string): Message;
+export declare function getStateKeysMessage(value?: Array<string>): Message;
+export declare function setStateMessage<T>(key: string, value: T): Message;
+export declare function clearStateMessage(key: string): Message;
+export declare function sleepMessage(wakeupTime: number, empty?: Empty, failure?: Failure): Message;
+export declare function completionMessage(index: number, value?: any, empty?: boolean, failure?: Failure): Message;
+export declare function completionMessageWithEmpty(index: number): Message;
+export declare function ackMessage(index: number): Message;
+export declare function invokeMessage(serviceName: string, handlerName: string, parameter: Uint8Array, value?: Uint8Array, failure?: Failure, key?: string): Message;
+export declare function backgroundInvokeMessage(serviceName: string, handlerName: string, parameter: Uint8Array, invokeTime?: number, key?: string): Message;
+export declare function sideEffectMessage<T>(value?: T, failure?: Failure, name?: string): Message;
+export declare function awakeableMessage<T>(payload?: T, failure?: Failure): Message;
+export declare function resolveAwakeableMessage<T>(id: string, payload: T): Message;
+export declare function rejectAwakeableMessage(id: string, reason: string): Message;
+export declare function suspensionMessage(entryIndices: number[]): Message;
+export declare function combinatorEntryMessage(combinatorId: number, journalEntriesOrder: number[]): Message;
+export declare function failure(msg: string, code?: number): Failure;
+export declare function greetRequest(myName: string): Uint8Array;
+export declare function greetResponse(myGreeting: string): Uint8Array;
+export declare function errorMessage(failure: Failure, ctx?: JournalErrorContext): Message;
+export declare function checkError(outputMsg: Message, errorMessage: string, code?: number): void;
+export declare function checkJournalMismatchError(outputMsg: Message): void;
+export declare function checkTerminalError(outputMsg: Message, errorMessage: string): void;
+export declare function getAwakeableId(entryIndex: number): string;
+export declare function keyVal(key: string, value: any): Buffer[];
+export declare const END_MESSAGE: Message;
+export declare const CLEAR_ALL_STATE_ENTRY_MESSAGE: Message;
+//# sourceMappingURL=protoutils.d.ts.map

package/dist/cjs/test/protoutils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"protoutils.d.ts","sourceRoot":"","sources":["../../../test/protoutils.ts"],"names":[],"mappings":";AAYA,OAAO,EAAE,KAAK,EAAc,MAAM,oBAAoB,CAAC;AA0CvD,OAAO,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAEhD,OAAO,EACL,OAAO,EAEP,uBAAuB,EACxB,MAAM,uCAAuC,CAAC;AAE/C,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAQlE,MAAM,MAAM,gBAAgB,GAAG;IAC7B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,wBAAgB,YAAY,CAAC,EAC3B,YAAY,EACZ,YAAY,EACZ,KAAK,EACL,GAAG,GACJ,GAAE,gBAAqB,GAAG,OAAO,CAiBjC;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,6BAMjD;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAWvD;AAED,wBAAgB,aAAa,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CA6B5E;AAED,wBAAgB,eAAe,CAAC,CAAC,EAC/B,GAAG,EAAE,MAAM,EACX,KAAK,CAAC,EAAE,CAAC,EACT,KAAK,CAAC,EAAE,OAAO,EACf,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CAqCT;AAED,wBAAgB,8BAA8B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAEnE;AAED,wBAAgB,mBAAmB,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,OAAO,CAmBlE;AAED,wBAAgB,eAAe,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,OAAO,CAQjE;AAED,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAOtD;AAED,wBAAgB,YAAY,CAC1B,UAAU,EAAE,MAAM,EAClB,KAAK,CAAC,EAAE,KAAK,EACb,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CAyBT;AAED,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,MAAM,EAEb,KAAK,CAAC,EAAE,GAAG,EACX,KAAK,CAAC,EAAE,OAAO,EACf,OAAO,CAAC,EAAE,OAAO,GAChB,OAAO,CAkCT;AAED,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAQjE;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAOjD;AAED,wBAAgB,aAAa,CAC3B,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,UAAU,EACrB,KAAK,CAAC,EAAE,UAAU,EAClB,OAAO,CAAC,EAAE,OAAO,EACjB,GAAG,CAAC,EAAE,MAAM,GACX,OAAO,CAkCT;AAED,wBAAgB,uBAAuB,CACrC,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,UAAU,EACrB,UAAU,CAAC,EAAE,MAAM,EACnB,GAAG,CAAC,EAAE,MAAM,GACX,OAAO,CAoBT;AAED,wBAAgB,iBAAiB,CAAC,CAAC,EACjC,KAAK,CAAC,EAAE,CAAC,EACT,OAAO,CAAC,EAAE,OAAO,EACjB,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CA6BT;AAED,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAqB3E;AAED,wBAAgB,uBAAuB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,GAAG,OAAO,CAQ1E;AAED,wBAAgB,sBAAsB,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAW1E;AAED,wBAAgB,iBAAiB,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,OAAO,CAOjE;AAED,wBAAgB,sBAAsB,CACpC,YAAY,EAAE,MAAM,EACpB,mBAAmB,EAAE,MAAM,EAAE,GAC5B,OAAO,CAUT;AAED,wBAAgB,OAAO,CACrB,GAAG,EAAE,MAAM,EACX,IAAI,GAAE,MAA4B,GACjC,OAAO,CAET;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,CAGvD;AAED,wBAAgB,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,UAAU,CAG5D;AAED,wBAAgB,YAAY,CAC1B,OAAO,EAAE,OAAO,EAChB,GAAG,CAAC,EAAE,mBAAmB,GACxB,OAAO,CAaT;AAED,wBAAgB,UAAU,CACxB,SAAS,EAAE,OAAO,EAClB,YAAY,EAAE,MAAM,EACpB,IAAI,GAAE,MAA4B,QAKnC;AAED,wBAAgB,yBAAyB,CAAC,SAAS,EAAE,OAAO,QAM3D;AAED,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,QAK1E;AAED,wBAAgB,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAWzD;AAED,wBAAgB,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,MAAM,EAAE,CAExD;AAED,eAAO,MAAM,WAAW,SAAkD,CAAC;AAC3E,eAAO,MAAM,6BAA6B,SAGzC,CAAC"}

package/dist/cjs/test/protoutils.js

@@ -0,0 +1,359 @@
+"use strict";
+/*
+ * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH
+ *
+ * This file is part of the Restate SDK for Node.js/TypeScript,
+ * which is released under the MIT license.
+ *
+ * You can find a copy of the license in file LICENSE in the root
+ * directory of this repository or package, or at
+ * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CLEAR_ALL_STATE_ENTRY_MESSAGE = exports.END_MESSAGE = exports.keyVal = exports.getAwakeableId = exports.checkTerminalError = exports.checkJournalMismatchError = exports.checkError = exports.errorMessage = exports.greetResponse = exports.greetRequest = exports.failure = exports.combinatorEntryMessage = exports.suspensionMessage = exports.rejectAwakeableMessage = exports.resolveAwakeableMessage = exports.awakeableMessage = exports.sideEffectMessage = exports.backgroundInvokeMessage = exports.invokeMessage = exports.ackMessage = exports.completionMessageWithEmpty = exports.completionMessage = exports.sleepMessage = exports.clearStateMessage = exports.setStateMessage = exports.getStateKeysMessage = exports.getStateMessageWithEmptyResult = exports.getStateMessage = exports.outputMessage = exports.inputMessage = exports.toStateEntries = exports.startMessage = void 0;
+/* istanbul ignore file */
+const protobuf_1 = require("@bufbuild/protobuf");
+const protocol_js_1 = require("../src/types/protocol.js");
+const types_js_1 = require("../src/types/types.js");
+const javascript_pb_js_1 = require("../src/generated/proto/javascript_pb.js");
+const protocol_pb_js_1 = require("../src/generated/proto/protocol_pb.js");
+const utils_js_1 = require("../src/utils/utils.js");
+const errors_js_1 = require("../src/types/errors.js");
+const vitest_1 = require("vitest");
+function startMessage({ knownEntries, partialState, state, key, } = {}) {
+    return new types_js_1.Message(protocol_js_1.START_MESSAGE_TYPE, new protocol_js_1.StartMessage({
+        id: Buffer.from("f311f1fdcb9863f0018bd3400ecd7d69b547204e776218b2", "hex"),
+        debugId: "8xHx_cuYY_AAYvTQA7NfWm1RyBOd2IYsg",
+        knownEntries: knownEntries, // only used for the Lambda case. For bidi streaming, this will be imputed by the testdriver
+        stateMap: toStateEntries(state || []),
+        partialState: partialState !== false,
+        key: key ?? "Till",
+    }), undefined, undefined);
+}
+exports.startMessage = startMessage;
+function toStateEntries(entries) {
+    return (entries.map((el) => new protocol_pb_js_1.StartMessage_StateEntry({ key: el[0], value: el[1] })) || []);
+}
+exports.toStateEntries = toStateEntries;
+function inputMessage(value) {
+    if (value !== undefined) {
+        return new types_js_1.Message(protocol_js_1.INPUT_ENTRY_MESSAGE_TYPE, new protocol_js_1.InputEntryMessage({
+            value,
+        }));
+    }
+    else {
+        throw new Error("Input message needs either a value or a failure set.");
+    }
+}
+exports.inputMessage = inputMessage;
+function outputMessage(value, failure) {
+    if (value !== undefined) {
+        return new types_js_1.Message(protocol_js_1.OUTPUT_ENTRY_MESSAGE_TYPE, new protocol_js_1.OutputEntryMessage({
+            result: { case: "value", value: Buffer.from(value) },
+        }));
+    }
+    else if (failure !== undefined) {
+        return new types_js_1.Message(protocol_js_1.OUTPUT_ENTRY_MESSAGE_TYPE, new protocol_js_1.OutputEntryMessage({
+            result: { case: "failure", value: failure },
+        }));
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.OUTPUT_ENTRY_MESSAGE_TYPE, new protocol_js_1.OutputEntryMessage({
+            result: {
+                case: "failure",
+                value: new protocol_pb_js_1.Failure({
+                    code: 13,
+                    message: `Uncaught exception for invocation id abcd`,
+                }),
+            },
+        }));
+    }
+}
+exports.outputMessage = outputMessage;
+function getStateMessage(key, value, empty, failure) {
+    if (empty === true) {
+        return new types_js_1.Message(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.GetStateEntryMessage({
+            key: Buffer.from(key),
+            result: { case: "empty", value: new protobuf_1.Empty() },
+        }), true);
+    }
+    else if (value !== undefined) {
+        return new types_js_1.Message(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.GetStateEntryMessage({
+            key: Buffer.from(key),
+            result: { case: "value", value: Buffer.from((0, utils_js_1.jsonSerialize)(value)) },
+        }), true);
+    }
+    else if (failure !== undefined) {
+        return new types_js_1.Message(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.GetStateEntryMessage({
+            key: Buffer.from(key),
+            result: { case: "failure", value: failure },
+        }), true);
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.GET_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.GetStateEntryMessage({
+            key: Buffer.from(key),
+        }), false);
+    }
+}
+exports.getStateMessage = getStateMessage;
+function getStateMessageWithEmptyResult(key) {
+    return getStateMessage(key, undefined, true);
+}
+exports.getStateMessageWithEmptyResult = getStateMessageWithEmptyResult;
+function getStateKeysMessage(value) {
+    if (value === undefined) {
+        return new types_js_1.Message(protocol_js_1.GET_STATE_KEYS_ENTRY_MESSAGE_TYPE, new protocol_js_1.GetStateKeysEntryMessage({}), false);
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.GET_STATE_KEYS_ENTRY_MESSAGE_TYPE, new protocol_js_1.GetStateKeysEntryMessage({
+            result: {
+                case: "value",
+                value: { keys: value.map((b) => Buffer.from(b)) },
+            },
+        }), true);
+    }
+}
+exports.getStateKeysMessage = getStateKeysMessage;
+function setStateMessage(key, value) {
+    return new types_js_1.Message(protocol_js_1.SET_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.SetStateEntryMessage({
+        key: Buffer.from(key),
+        value: Buffer.from((0, utils_js_1.jsonSerialize)(value)),
+    }));
+}
+exports.setStateMessage = setStateMessage;
+function clearStateMessage(key) {
+    return new types_js_1.Message(protocol_js_1.CLEAR_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.ClearStateEntryMessage({
+        key: Buffer.from(key),
+    }));
+}
+exports.clearStateMessage = clearStateMessage;
+function sleepMessage(wakeupTime, empty, failure) {
+    if (empty !== undefined) {
+        return new types_js_1.Message(protocol_js_1.SLEEP_ENTRY_MESSAGE_TYPE, new protocol_js_1.SleepEntryMessage({
+            wakeUpTime: protobuf_1.protoInt64.parse(wakeupTime),
+            result: { case: "empty", value: empty },
+        }));
+    }
+    else if (failure !== undefined) {
+        return new types_js_1.Message(protocol_js_1.SLEEP_ENTRY_MESSAGE_TYPE, new protocol_js_1.SleepEntryMessage({
+            wakeUpTime: protobuf_1.protoInt64.parse(wakeupTime),
+            result: { case: "failure", value: failure },
+        }));
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.SLEEP_ENTRY_MESSAGE_TYPE, new protocol_js_1.SleepEntryMessage({
+            wakeUpTime: protobuf_1.protoInt64.parse(wakeupTime),
+        }));
+    }
+}
+exports.sleepMessage = sleepMessage;
+function completionMessage(index, 
+/* eslint-disable @typescript-eslint/no-explicit-any */
+value, empty, failure) {
+    if (value !== undefined) {
+        return new types_js_1.Message(protocol_js_1.COMPLETION_MESSAGE_TYPE, new protocol_js_1.CompletionMessage({
+            entryIndex: index,
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
+            result: { case: "value", value: Buffer.from(value) },
+        }));
+    }
+    else if (empty) {
+        return new types_js_1.Message(protocol_js_1.COMPLETION_MESSAGE_TYPE, new protocol_js_1.CompletionMessage({
+            entryIndex: index,
+            result: { case: "empty", value: new protobuf_1.Empty() },
+        }));
+    }
+    else if (failure != undefined) {
+        return new types_js_1.Message(protocol_js_1.COMPLETION_MESSAGE_TYPE, new protocol_js_1.CompletionMessage({
+            entryIndex: index,
+            result: { case: "failure", value: failure },
+        }));
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.COMPLETION_MESSAGE_TYPE, new protocol_js_1.CompletionMessage({
+            entryIndex: index,
+        }));
+    }
+}
+exports.completionMessage = completionMessage;
+function completionMessageWithEmpty(index) {
+    return new types_js_1.Message(protocol_js_1.COMPLETION_MESSAGE_TYPE, new protocol_js_1.CompletionMessage({
+        entryIndex: index,
+        result: { case: "empty", value: new protobuf_1.Empty() },
+    }));
+}
+exports.completionMessageWithEmpty = completionMessageWithEmpty;
+function ackMessage(index) {
+    return new types_js_1.Message(protocol_js_1.ENTRY_ACK_MESSAGE_TYPE, new protocol_js_1.EntryAckMessage({
+        entryIndex: index,
+    }));
+}
+exports.ackMessage = ackMessage;
+function invokeMessage(serviceName, handlerName, parameter, value, failure, key) {
+    if (value != undefined) {
+        return new types_js_1.Message(protocol_js_1.INVOKE_ENTRY_MESSAGE_TYPE, new protocol_js_1.CallEntryMessage({
+            serviceName: serviceName,
+            handlerName: handlerName,
+            parameter: Buffer.from(parameter),
+            result: { case: "value", value: value },
+            key,
+        }));
+    }
+    else if (failure != undefined) {
+        return new types_js_1.Message(protocol_js_1.INVOKE_ENTRY_MESSAGE_TYPE, new protocol_js_1.CallEntryMessage({
+            serviceName: serviceName,
+            handlerName: handlerName,
+            parameter: Buffer.from(parameter),
+            result: { case: "failure", value: failure },
+            key,
+        }));
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.INVOKE_ENTRY_MESSAGE_TYPE, new protocol_js_1.CallEntryMessage({
+            serviceName: serviceName,
+            handlerName: handlerName,
+            parameter: Buffer.from(parameter),
+            key,
+        }));
+    }
+}
+exports.invokeMessage = invokeMessage;
+function backgroundInvokeMessage(serviceName, handlerName, parameter, invokeTime, key) {
+    return invokeTime
+        ? new types_js_1.Message(protocol_js_1.BACKGROUND_INVOKE_ENTRY_MESSAGE_TYPE, new protocol_js_1.OneWayCallEntryMessage({
+            serviceName: serviceName,
+            handlerName: handlerName,
+            parameter: Buffer.from(parameter),
+            invokeTime: protobuf_1.protoInt64.parse(invokeTime),
+            key,
+        }))
+        : new types_js_1.Message(protocol_js_1.BACKGROUND_INVOKE_ENTRY_MESSAGE_TYPE, new protocol_js_1.OneWayCallEntryMessage({
+            serviceName: serviceName,
+            handlerName: handlerName,
+            parameter: Buffer.from(parameter),
+        }));
+}
+exports.backgroundInvokeMessage = backgroundInvokeMessage;
+function sideEffectMessage(value, failure, name) {
+    if (value !== undefined) {
+        return new types_js_1.Message(protocol_js_1.SIDE_EFFECT_ENTRY_MESSAGE_TYPE, new protocol_pb_js_1.RunEntryMessage({
+            name,
+            result: { case: "value", value: Buffer.from(JSON.stringify(value)) },
+        }), false, true);
+    }
+    else if (failure !== undefined) {
+        return new types_js_1.Message(protocol_js_1.SIDE_EFFECT_ENTRY_MESSAGE_TYPE, new protocol_pb_js_1.RunEntryMessage({
+            name,
+            result: { case: "failure", value: failure },
+        }), false, true);
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.SIDE_EFFECT_ENTRY_MESSAGE_TYPE, new protocol_pb_js_1.RunEntryMessage({ name }), false, true);
+    }
+}
+exports.sideEffectMessage = sideEffectMessage;
+function awakeableMessage(payload, failure) {
+    if (payload) {
+        return new types_js_1.Message(protocol_js_1.AWAKEABLE_ENTRY_MESSAGE_TYPE, new protocol_js_1.AwakeableEntryMessage({
+            result: { case: "value", value: Buffer.from(JSON.stringify(payload)) },
+        }));
+    }
+    else if (failure) {
+        return new types_js_1.Message(protocol_js_1.AWAKEABLE_ENTRY_MESSAGE_TYPE, new protocol_js_1.AwakeableEntryMessage({
+            result: { case: "failure", value: failure },
+        }));
+    }
+    else {
+        return new types_js_1.Message(protocol_js_1.AWAKEABLE_ENTRY_MESSAGE_TYPE, new protocol_js_1.AwakeableEntryMessage());
+    }
+}
+exports.awakeableMessage = awakeableMessage;
+function resolveAwakeableMessage(id, payload) {
+    return new types_js_1.Message(protocol_js_1.COMPLETE_AWAKEABLE_ENTRY_MESSAGE_TYPE, new protocol_js_1.CompleteAwakeableEntryMessage({
+        id: id,
+        result: { case: "value", value: Buffer.from(JSON.stringify(payload)) },
+    }));
+}
+exports.resolveAwakeableMessage = resolveAwakeableMessage;
+function rejectAwakeableMessage(id, reason) {
+    return new types_js_1.Message(protocol_js_1.COMPLETE_AWAKEABLE_ENTRY_MESSAGE_TYPE, new protocol_js_1.CompleteAwakeableEntryMessage({
+        id: id,
+        result: {
+            case: "failure",
+            value: { code: errors_js_1.UNKNOWN_ERROR_CODE, message: reason },
+        },
+    }));
+}
+exports.rejectAwakeableMessage = rejectAwakeableMessage;
+function suspensionMessage(entryIndices) {
+    return new types_js_1.Message(protocol_js_1.SUSPENSION_MESSAGE_TYPE, new protocol_js_1.SuspensionMessage({
+        entryIndexes: entryIndices,
+    }));
+}
+exports.suspensionMessage = suspensionMessage;
+function combinatorEntryMessage(combinatorId, journalEntriesOrder) {
+    return new types_js_1.Message(protocol_js_1.COMBINATOR_ENTRY_MESSAGE, new javascript_pb_js_1.CombinatorEntryMessage({
+        combinatorId,
+        journalEntriesOrder,
+    }), undefined, true);
+}
+exports.combinatorEntryMessage = combinatorEntryMessage;
+function failure(msg, code = errors_js_1.INTERNAL_ERROR_CODE) {
+    return new protocol_pb_js_1.Failure({ code: code, message: msg });
+}
+exports.failure = failure;
+function greetRequest(myName) {
+    const str = JSON.stringify({ name: myName });
+    return Buffer.from(str);
+}
+exports.greetRequest = greetRequest;
+function greetResponse(myGreeting) {
+    const str = JSON.stringify({ greeting: myGreeting });
+    return Buffer.from(str);
+}
+exports.greetResponse = greetResponse;
+function errorMessage(failure, ctx) {
+    return new types_js_1.Message(protocol_js_1.ERROR_MESSAGE_TYPE, new protocol_js_1.ErrorMessage({
+        message: failure.message,
+        code: failure.code,
+        relatedEntryIndex: ctx?.relatedEntryIndex,
+        relatedEntryType: ctx?.relatedEntryType
+            ? Number(ctx.relatedEntryType)
+            : undefined,
+        relatedEntryName: ctx?.relatedEntryName,
+    }));
+}
+exports.errorMessage = errorMessage;
+function checkError(outputMsg, errorMessage, code = errors_js_1.INTERNAL_ERROR_CODE) {
+    (0, vitest_1.expect)(outputMsg.messageType).toEqual(protocol_js_1.ERROR_MESSAGE_TYPE);
+    (0, vitest_1.expect)(outputMsg.message.code).toStrictEqual(code);
+    (0, vitest_1.expect)(outputMsg.message.message).toContain(errorMessage);
+}
+exports.checkError = checkError;
+function checkJournalMismatchError(outputMsg) {
+    checkError(outputMsg, "Journal mismatch: Replayed journal entries did not correspond to the user code. The user code has to be deterministic!", errors_js_1.RestateErrorCodes.JOURNAL_MISMATCH);
+}
+exports.checkJournalMismatchError = checkJournalMismatchError;
+function checkTerminalError(outputMsg, errorMessage) {
+    (0, vitest_1.expect)(outputMsg.messageType).toEqual(protocol_js_1.OUTPUT_ENTRY_MESSAGE_TYPE);
+    const res = outputMsg.message.result;
+    (0, vitest_1.expect)(res.case === "failure" && res.value.message).toContain(errorMessage);
+}
+exports.checkTerminalError = checkTerminalError;
+function getAwakeableId(entryIndex) {
+    const encodedEntryIndex = Buffer.alloc(4 /* Size of u32 */);
+    encodedEntryIndex.writeUInt32BE(entryIndex);
+    return (protocol_js_1.AWAKEABLE_IDENTIFIER_PREFIX +
+        Buffer.concat([
+            Buffer.from("f311f1fdcb9863f0018bd3400ecd7d69b547204e776218b2", "hex"),
+            encodedEntryIndex,
+        ]).toString("base64url"));
+}
+exports.getAwakeableId = getAwakeableId;
+function keyVal(key, value) {
+    return [Buffer.from(key), Buffer.from(JSON.stringify(value))];
+}
+exports.keyVal = keyVal;
+exports.END_MESSAGE = new types_js_1.Message(protocol_js_1.END_MESSAGE_TYPE, new protocol_js_1.EndMessage());
+exports.CLEAR_ALL_STATE_ENTRY_MESSAGE = new types_js_1.Message(protocol_js_1.CLEAR_ALL_STATE_ENTRY_MESSAGE_TYPE, new protocol_js_1.ClearAllStateEntryMessage());
+//# sourceMappingURL=protoutils.js.map