@rivetkit/workflow-engine 2.1.0-rc.1

package/dist/tsup/index.js

@@ -0,0 +1,93 @@
+import {
+  CancelledError,
+  CriticalError,
+  DEFAULT_LOOP_COMMIT_INTERVAL,
+  DEFAULT_LOOP_HISTORY_EVERY,
+  DEFAULT_LOOP_HISTORY_KEEP,
+  DEFAULT_MAX_RETRIES,
+  DEFAULT_RETRY_BACKOFF_BASE,
+  DEFAULT_RETRY_BACKOFF_MAX,
+  DEFAULT_STEP_TIMEOUT,
+  EntryInProgressError,
+  EvictedError,
+  HistoryDivergedError,
+  JoinError,
+  Loop,
+  MessageWaitError,
+  RaceError,
+  RollbackCheckpointError,
+  RollbackError,
+  SleepError,
+  StepExhaustedError,
+  StepFailedError,
+  WorkflowContextImpl,
+  appendLoopIteration,
+  appendName,
+  createEntry,
+  createHistorySnapshot,
+  createStorage,
+  deleteEntriesWithPrefix,
+  emptyLocation,
+  flush,
+  generateId,
+  getEntry,
+  getOrCreateMetadata,
+  isLocationPrefix,
+  isLoopIterationMarker,
+  loadMetadata,
+  loadStorage,
+  locationToKey,
+  locationsEqual,
+  parentLocation,
+  registerName,
+  resolveName,
+  runWorkflow,
+  setEntry
+} from "./chunk-JWHWQBZP.js";
+export {
+  CancelledError,
+  CriticalError,
+  DEFAULT_LOOP_COMMIT_INTERVAL,
+  DEFAULT_LOOP_HISTORY_EVERY,
+  DEFAULT_LOOP_HISTORY_KEEP,
+  DEFAULT_MAX_RETRIES,
+  DEFAULT_RETRY_BACKOFF_BASE,
+  DEFAULT_RETRY_BACKOFF_MAX,
+  DEFAULT_STEP_TIMEOUT,
+  EntryInProgressError,
+  EvictedError,
+  HistoryDivergedError,
+  JoinError,
+  Loop,
+  MessageWaitError,
+  RaceError,
+  RollbackCheckpointError,
+  RollbackError,
+  SleepError,
+  StepExhaustedError,
+  StepFailedError,
+  WorkflowContextImpl,
+  appendLoopIteration,
+  appendName,
+  createEntry,
+  createHistorySnapshot,
+  createStorage,
+  deleteEntriesWithPrefix,
+  emptyLocation,
+  flush,
+  generateId,
+  getEntry,
+  getOrCreateMetadata,
+  isLocationPrefix,
+  isLoopIterationMarker,
+  loadMetadata,
+  loadStorage,
+  locationToKey,
+  locationsEqual,
+  parentLocation,
+  registerName,
+  resolveName,
+  runWorkflow,
+  setEntry
+};
+//# sourceMappingURL=index.js.map

package/dist/tsup/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/tsup/testing.cjs

@@ -0,0 +1,316 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class;
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+var _chunkGJ66YE5Wcjs = require('./chunk-GJ66YE5W.cjs');
+
+// src/testing.ts
+var InMemoryWorkflowMessageDriver = class {
+  #messages = [];
+  #waiters = /* @__PURE__ */ new Set();
+  async addMessage(message) {
+    this.#messages.push(message);
+    this.#notifyWaiters(message.name);
+  }
+  async receiveMessages(opts) {
+    const limitedCount = Math.max(1, opts.count);
+    const nameSet = opts.names && opts.names.length > 0 ? new Set(opts.names) : void 0;
+    const selected = [];
+    for (let i = 0; i < this.#messages.length && selected.length < limitedCount; i++) {
+      const message = this.#messages[i];
+      if (nameSet && !nameSet.has(message.name)) {
+        continue;
+      }
+      selected.push({ message, index: i });
+    }
+    if (selected.length === 0) {
+      return [];
+    }
+    if (!opts.completable) {
+      for (let i = selected.length - 1; i >= 0; i--) {
+        this.#messages.splice(selected[i].index, 1);
+      }
+      return selected.map((entry) => entry.message);
+    }
+    return selected.map((entry) => {
+      const { message } = entry;
+      return {
+        ...message,
+        complete: async () => {
+          await this.completeMessage(message.id);
+        }
+      };
+    });
+  }
+  async completeMessage(messageId) {
+    const index = this.#messages.findIndex((message) => message.id === messageId);
+    if (index !== -1) {
+      this.#messages.splice(index, 1);
+    }
+  }
+  async waitForMessages(messageNames, abortSignal) {
+    if (abortSignal.aborted) {
+      throw new (0, _chunkGJ66YE5Wcjs.EvictedError)();
+    }
+    const nameSet = messageNames.length > 0 ? new Set(messageNames) : void 0;
+    if (this.#messages.some(
+      (message) => nameSet ? nameSet.has(message.name) : true
+    )) {
+      return;
+    }
+    await new Promise((resolve, reject) => {
+      const waiter = {
+        nameSet,
+        resolve: () => {
+          this.#removeWaiter(waiter);
+          resolve();
+        },
+        reject: (error) => {
+          this.#removeWaiter(waiter);
+          reject(error);
+        },
+        abortSignal,
+        onAbort: () => {
+          waiter.reject(new (0, _chunkGJ66YE5Wcjs.EvictedError)());
+        }
+      };
+      abortSignal.addEventListener("abort", waiter.onAbort, { once: true });
+      this.#waiters.add(waiter);
+    });
+  }
+  #removeWaiter(waiter) {
+    if (this.#waiters.delete(waiter)) {
+      waiter.abortSignal.removeEventListener("abort", waiter.onAbort);
+    }
+  }
+  #notifyWaiters(name) {
+    for (const waiter of [...this.#waiters]) {
+      if (waiter.nameSet && !waiter.nameSet.has(name)) {
+        continue;
+      }
+      waiter.resolve();
+    }
+  }
+  clear() {
+    this.#messages = [];
+    for (const waiter of [...this.#waiters]) {
+      waiter.reject(new Error("cleared"));
+    }
+  }
+};
+var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);_class.prototype.__init4.call(this);_class.prototype.__init5.call(this); }
+  // Map from hex-encoded key to { originalKey, value }
+  __init() {this.kv = /* @__PURE__ */ new Map()}
+  __init2() {this.alarms = /* @__PURE__ */ new Map()}
+  #inMemoryMessageDriver = new InMemoryWorkflowMessageDriver();
+  /** Simulated latency per operation (ms) */
+  __init3() {this.latency = 10}
+  /** How often the worker polls for work */
+  __init4() {this.workerPollInterval = 100}
+  __init5() {this.messageDriver = this.#inMemoryMessageDriver}
+  async get(key) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    const entry = this.kv.get(_chunkGJ66YE5Wcjs.keyToHex.call(void 0, key));
+    return _nullishCoalesce((entry == null ? void 0 : entry.value), () => ( null));
+  }
+  async set(key, value) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    this.kv.set(_chunkGJ66YE5Wcjs.keyToHex.call(void 0, key), { key, value });
+  }
+  async delete(key) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    this.kv.delete(_chunkGJ66YE5Wcjs.keyToHex.call(void 0, key));
+  }
+  async deletePrefix(prefix) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    for (const [hexKey, entry] of this.kv) {
+      if (_chunkGJ66YE5Wcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
+        this.kv.delete(hexKey);
+      }
+    }
+  }
+  async list(prefix) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    const results = [];
+    for (const entry of this.kv.values()) {
+      if (_chunkGJ66YE5Wcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
+        results.push({ key: entry.key, value: entry.value });
+      }
+    }
+    return results.sort((a, b) => _chunkGJ66YE5Wcjs.compareKeys.call(void 0, a.key, b.key));
+  }
+  async batch(writes) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    for (const { key, value } of writes) {
+      this.kv.set(_chunkGJ66YE5Wcjs.keyToHex.call(void 0, key), { key, value });
+    }
+  }
+  async setAlarm(workflowId, wakeAt) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    this.alarms.set(workflowId, wakeAt);
+  }
+  async clearAlarm(workflowId) {
+    await _chunkGJ66YE5Wcjs.sleep.call(void 0, this.latency);
+    this.alarms.delete(workflowId);
+  }
+  async waitForMessages(messageNames, abortSignal) {
+    const driver = this.messageDriver;
+    if (driver.waitForMessages) {
+      await driver.waitForMessages(messageNames, abortSignal);
+      return;
+    }
+    while (true) {
+      if (abortSignal.aborted) {
+        throw new (0, _chunkGJ66YE5Wcjs.EvictedError)();
+      }
+      const messages = await this.messageDriver.receiveMessages({
+        names: messageNames.length > 0 ? messageNames : void 0,
+        count: 1,
+        completable: true
+      });
+      if (messages.length > 0) {
+        return;
+      }
+      await _chunkGJ66YE5Wcjs.sleep.call(void 0, Math.max(1, this.latency));
+    }
+  }
+  /**
+   * Get the alarm time for a workflow (for testing).
+   */
+  getAlarm(workflowId) {
+    return this.alarms.get(workflowId);
+  }
+  /**
+   * Check if any alarms are due and return their workflow IDs.
+   */
+  getDueAlarms() {
+    const now = Date.now();
+    const due = [];
+    for (const [workflowId, wakeAt] of this.alarms) {
+      if (wakeAt <= now) {
+        due.push(workflowId);
+      }
+    }
+    return due;
+  }
+  /**
+   * Clear all data (for testing).
+   */
+  clear() {
+    this.kv.clear();
+    this.alarms.clear();
+    this.#inMemoryMessageDriver.clear();
+  }
+  /**
+   * Get a snapshot of all data (for testing/debugging).
+   */
+  snapshot() {
+    const kvSnapshot = {};
+    for (const [hexKey, entry] of this.kv) {
+      kvSnapshot[hexKey] = entry.value;
+    }
+    return {
+      kv: kvSnapshot,
+      alarms: Object.fromEntries(this.alarms)
+    };
+  }
+  /**
+   * Get all hex-encoded keys (for testing).
+   */
+  keys() {
+    return [...this.kv.keys()];
+  }
+}, _class);
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+exports.CancelledError = _chunkGJ66YE5Wcjs.CancelledError; exports.CriticalError = _chunkGJ66YE5Wcjs.CriticalError; exports.DEFAULT_LOOP_COMMIT_INTERVAL = _chunkGJ66YE5Wcjs.DEFAULT_LOOP_COMMIT_INTERVAL; exports.DEFAULT_LOOP_HISTORY_EVERY = _chunkGJ66YE5Wcjs.DEFAULT_LOOP_HISTORY_EVERY; exports.DEFAULT_LOOP_HISTORY_KEEP = _chunkGJ66YE5Wcjs.DEFAULT_LOOP_HISTORY_KEEP; exports.DEFAULT_MAX_RETRIES = _chunkGJ66YE5Wcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkGJ66YE5Wcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkGJ66YE5Wcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkGJ66YE5Wcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkGJ66YE5Wcjs.EntryInProgressError; exports.EvictedError = _chunkGJ66YE5Wcjs.EvictedError; exports.HistoryDivergedError = _chunkGJ66YE5Wcjs.HistoryDivergedError; exports.InMemoryDriver = InMemoryDriver; exports.JoinError = _chunkGJ66YE5Wcjs.JoinError; exports.Loop = _chunkGJ66YE5Wcjs.Loop; exports.MessageWaitError = _chunkGJ66YE5Wcjs.MessageWaitError; exports.RaceError = _chunkGJ66YE5Wcjs.RaceError; exports.RollbackCheckpointError = _chunkGJ66YE5Wcjs.RollbackCheckpointError; exports.RollbackError = _chunkGJ66YE5Wcjs.RollbackError; exports.SleepError = _chunkGJ66YE5Wcjs.SleepError; exports.StepExhaustedError = _chunkGJ66YE5Wcjs.StepExhaustedError; exports.StepFailedError = _chunkGJ66YE5Wcjs.StepFailedError; exports.WorkflowContextImpl = _chunkGJ66YE5Wcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkGJ66YE5Wcjs.appendLoopIteration; exports.appendName = _chunkGJ66YE5Wcjs.appendName; exports.createEntry = _chunkGJ66YE5Wcjs.createEntry; exports.createHistorySnapshot = _chunkGJ66YE5Wcjs.createHistorySnapshot; exports.createStorage = _chunkGJ66YE5Wcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkGJ66YE5Wcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkGJ66YE5Wcjs.emptyLocation; exports.flush = _chunkGJ66YE5Wcjs.flush; exports.generateId = _chunkGJ66YE5Wcjs.generateId; exports.getEntry = _chunkGJ66YE5Wcjs.getEntry; exports.getOrCreateMetadata = _chunkGJ66YE5Wcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkGJ66YE5Wcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkGJ66YE5Wcjs.isLoopIterationMarker; exports.loadMetadata = _chunkGJ66YE5Wcjs.loadMetadata; exports.loadStorage = _chunkGJ66YE5Wcjs.loadStorage; exports.locationToKey = _chunkGJ66YE5Wcjs.locationToKey; exports.locationsEqual = _chunkGJ66YE5Wcjs.locationsEqual; exports.parentLocation = _chunkGJ66YE5Wcjs.parentLocation; exports.registerName = _chunkGJ66YE5Wcjs.registerName; exports.resolveName = _chunkGJ66YE5Wcjs.resolveName; exports.runWorkflow = _chunkGJ66YE5Wcjs.runWorkflow; exports.setEntry = _chunkGJ66YE5Wcjs.setEntry;
+//# sourceMappingURL=testing.cjs.map

package/dist/tsup/testing.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/workflow-engine/dist/tsup/testing.cjs","../../src/testing.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACA;ACrCA,IAAM,8BAAA,EAAN,MAAqE;AAAA,EACpE,CAAA,SAAA,EAAuB,CAAC,CAAA;AAAA,EACxB,CAAA,QAAA,kBAAW,IAAI,GAAA,CAAY,CAAA;AAAA,EAE3B,MAAM,UAAA,CAAW,OAAA,EAAiC;AACjD,IAAA,IAAA,CAAK,CAAA,QAAA,CAAU,IAAA,CAAK,OAAO,CAAA;AAC3B,IAAA,IAAA,CAAK,CAAA,aAAA,CAAe,OAAA,CAAQ,IAAI,CAAA;AAAA,EACjC;AAAA,EAEA,MAAM,eAAA,CAAgB,IAAA,EAIC;AACtB,IAAA,MAAM,aAAA,EAAe,IAAA,CAAK,GAAA,CAAI,CAAA,EAAG,IAAA,CAAK,KAAK,CAAA;AAC3C,IAAA,MAAM,QAAA,EACL,IAAA,CAAK,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,OAAA,EAAS,EAAA,EAC/B,IAAI,GAAA,CAAI,IAAA,CAAK,KAAK,EAAA,EAClB,KAAA,CAAA;AACJ,IAAA,MAAM,SAAA,EAAuD,CAAC,CAAA;AAE9D,IAAA,IAAA,CAAA,IAAS,EAAA,EAAI,CAAA,EAAG,EAAA,EAAI,IAAA,CAAK,CAAA,QAAA,CAAU,OAAA,GAAU,QAAA,CAAS,OAAA,EAAS,YAAA,EAAc,CAAA,EAAA,EAAK;AACjF,MAAA,MAAM,QAAA,EAAU,IAAA,CAAK,CAAA,QAAA,CAAU,CAAC,CAAA;AAChC,MAAA,GAAA,CAAI,QAAA,GAAW,CAAC,OAAA,CAAQ,GAAA,CAAI,OAAA,CAAQ,IAAI,CAAA,EAAG;AAC1C,QAAA,QAAA;AAAA,MACD;AACA,MAAA,QAAA,CAAS,IAAA,CAAK,EAAE,OAAA,EAAS,KAAA,EAAO,EAAE,CAAC,CAAA;AAAA,IACpC;AAEA,IAAA,GAAA,CAAI,QAAA,CAAS,OAAA,IAAW,CAAA,EAAG;AAC1B,MAAA,OAAO,CAAC,CAAA;AAAA,IACT;AAEA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,EAAa;AACtB,MAAA,IAAA,CAAA,IAAS,EAAA,EAAI,QAAA,CAAS,OAAA,EAAS,CAAA,EAAG,EAAA,GAAK,CAAA,EAAG,CAAA,EAAA,EAAK;AAC9C,QAAA,IAAA,CAAK,CAAA,QAAA,CAAU,MAAA,CAAO,QAAA,CAAS,CAAC,CAAA,CAAE,KAAA,EAAO,CAAC,CAAA;AAAA,MAC3C;AACA,MAAA,OAAO,QAAA,CAAS,GAAA,CAAI,CAAC,KAAA,EAAA,GAAU,KAAA,CAAM,OAAO,CAAA;AAAA,IAC7C;AAEA,IAAA,OAAO,QAAA,CAAS,GAAA,CAAI,CAAC,KAAA,EAAA,GAAU;AAC9B,MAAA,MAAM,EAAE,QAAQ,EAAA,EAAI,KAAA;AACpB,MAAA,OAAO;AAAA,QACN,GAAG,OAAA;AAAA,QACH,QAAA,EAAU,MAAA,CAAA,EAAA,GAAY;AACrB,UAAA,MAAM,IAAA,CAAK,eAAA,CAAgB,OAAA,CAAQ,EAAE,CAAA;AAAA,QACtC;AAAA,MACD,CAAA;AAAA,IACD,CAAC,CAAA;AAAA,EACF;AAAA,EAEA,MAAM,eAAA,CAAgB,SAAA,EAAkC;AACvD,IAAA,MAAM,MAAA,EAAQ,IAAA,CAAK,CAAA,QAAA,CAAU,SAAA,CAAU,CAAC,OAAA,EAAA,GAAY,OAAA,CAAQ,GAAA,IAAO,SAAS,CAAA;AAC5E,IAAA,GAAA,CAAI,MAAA,IAAU,CAAA,CAAA,EAAI;AACjB,MAAA,IAAA,CAAK,CAAA,QAAA,CAAU,MAAA,CAAO,KAAA,EAAO,CAAC,CAAA;AAAA,IAC/B;AAAA,EACD;AAAA,EAEA,MAAM,eAAA,CACL,YAAA,EACA,WAAA,EACgB;AAChB,IAAA,GAAA,CAAI,WAAA,CAAY,OAAA,EAAS;AACxB,MAAA,MAAM,IAAI,mCAAA,CAAa,CAAA;AAAA,IACxB;AAEA,IAAA,MAAM,QAAA,EAAU,YAAA,CAAa,OAAA,EAAS,EAAA,EAAI,IAAI,GAAA,CAAI,YAAY,EAAA,EAAI,KAAA,CAAA;AAClE,IAAA,GAAA,CACC,IAAA,CAAK,CAAA,QAAA,CAAU,IAAA;AAAA,MAAK,CAAC,OAAA,EAAA,GACpB,QAAA,EAAU,OAAA,CAAQ,GAAA,CAAI,OAAA,CAAQ,IAAI,EAAA,EAAI;AAAA,IACvC,CAAA,EACC;AACD,MAAA,MAAA;AAAA,IACD;AAEA,IAAA,MAAM,IAAI,OAAA,CAAc,CAAC,OAAA,EAAS,MAAA,EAAA,GAAW;AAC5C,MAAA,MAAM,OAAA,EAAiB;AAAA,QACtB,OAAA;AAAA,QACA,OAAA,EAAS,CAAA,EAAA,GAAM;AACd,UAAA,IAAA,CAAK,CAAA,YAAA,CAAc,MAAM,CAAA;AACzB,UAAA,OAAA,CAAQ,CAAA;AAAA,QACT,CAAA;AAAA,QACA,MAAA,EAAQ,CAAC,KAAA,EAAA,GAAU;AAClB,UAAA,IAAA,CAAK,CAAA,YAAA,CAAc,MAAM,CAAA;AACzB,UAAA,MAAA,CAAO,KAAK,CAAA;AAAA,QACb,CAAA;AAAA,QACA,WAAA;AAAA,QACA,OAAA,EAAS,CAAA,EAAA,GAAM;AACd,UAAA,MAAA,CAAO,MAAA,CAAO,IAAI,mCAAA,CAAa,CAAC,CAAA;AAAA,QACjC;AAAA,MACD,CAAA;AACA,MAAA,WAAA,CAAY,gBAAA,CAAiB,OAAA,EAAS,MAAA,CAAO,OAAA,EAAS,EAAE,IAAA,EAAM,KAAK,CAAC,CAAA;AACpE,MAAA,IAAA,CAAK,CAAA,OAAA,CAAS,GAAA,CAAI,MAAM,CAAA;AAAA,IACzB,CAAC,CAAA;AAAA,EACF;AAAA,EAEA,CAAA,YAAA,CAAc,MAAA,EAAsB;AACnC,IAAA,GAAA,CAAI,IAAA,CAAK,CAAA,OAAA,CAAS,MAAA,CAAO,MAAM,CAAA,EAAG;AACjC,MAAA,MAAA,CAAO,WAAA,CAAY,mBAAA,CAAoB,OAAA,EAAS,MAAA,CAAO,OAAO,CAAA;AAAA,IAC/D;AAAA,EACD;AAAA,EAEA,CAAA,aAAA,CAAe,IAAA,EAAoB;AAClC,IAAA,IAAA,CAAA,MAAW,OAAA,GAAU,CAAC,GAAG,IAAA,CAAK,CAAA,OAAQ,CAAA,EAAG;AACxC,MAAA,GAAA,CAAI,MAAA,CAAO,QAAA,GAAW,CAAC,MAAA,CAAO,OAAA,CAAQ,GAAA,CAAI,IAAI,CAAA,EAAG;AAChD,QAAA,QAAA;AAAA,MACD;AACA,MAAA,MAAA,CAAO,OAAA,CAAQ,CAAA;AAAA,IAChB;AAAA,EACD;AAAA,EAEA,KAAA,CAAA,EAAc;AACb,IAAA,IAAA,CAAK,CAAA,SAAA,EAAY,CAAC,CAAA;AAClB,IAAA,IAAA,CAAA,MAAW,OAAA,GAAU,CAAC,GAAG,IAAA,CAAK,CAAA,OAAQ,CAAA,EAAG;AACxC,MAAA,MAAA,CAAO,MAAA,CAAO,IAAI,KAAA,CAAM,SAAS,CAAC,CAAA;AAAA,IACnC;AAAA,EACD;AACD,CAAA;AAMO,IAAM,eAAA,YAAN,MAA6C;AAAA;AAAA,iBAE3C,GAAA,kBAAK,IAAI,GAAA,CAAoD,EAAA;AAAA,kBAC7D,OAAA,kBAAS,IAAI,GAAA,CAAoB,EAAA;AAAA,EACzC,CAAA,sBAAA,EAAyB,IAAI,6BAAA,CAA8B,CAAA;AAAA;AAAA,kBAG3D,QAAA,EAAU,GAAA;AAAA;AAAA,kBAGV,mBAAA,EAAqB,IAAA;AAAA,kBACrB,cAAA,EAAuC,IAAA,CAAK,CAAA,sBAAA;AAAA,EAE5C,MAAM,GAAA,CAAI,GAAA,EAA6C;AACtD,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,MAAM,MAAA,EAAQ,IAAA,CAAK,EAAA,CAAG,GAAA,CAAI,wCAAA,GAAY,CAAC,CAAA;AACvC,IAAA,wBAAA,CAAO,MAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,KAAA,CAAO,KAAA,CAAA,UAAS,MAAA;AAAA,EACxB;AAAA,EAEA,MAAM,GAAA,CAAI,GAAA,EAAiB,KAAA,EAAkC;AAC5D,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,IAAA,CAAK,EAAA,CAAG,GAAA,CAAI,wCAAA,GAAY,CAAA,EAAG,EAAE,GAAA,EAAK,MAAM,CAAC,CAAA;AAAA,EAC1C;AAAA,EAEA,MAAM,MAAA,CAAO,GAAA,EAAgC;AAC5C,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,IAAA,CAAK,EAAA,CAAG,MAAA,CAAO,wCAAA,GAAY,CAAC,CAAA;AAAA,EAC7B;AAAA,EAEA,MAAM,YAAA,CAAa,MAAA,EAAmC;AACrD,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,IAAA,CAAA,MAAW,CAAC,MAAA,EAAQ,KAAK,EAAA,GAAK,IAAA,CAAK,EAAA,EAAI;AACtC,MAAA,GAAA,CAAI,6CAAA,KAAc,CAAM,GAAA,EAAK,MAAM,CAAA,EAAG;AACrC,QAAA,IAAA,CAAK,EAAA,CAAG,MAAA,CAAO,MAAM,CAAA;AAAA,MACtB;AAAA,IACD;AAAA,EACD;AAAA,EAEA,MAAM,IAAA,CAAK,MAAA,EAAwC;AAClD,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,MAAM,QAAA,EAAqB,CAAC,CAAA;AAC5B,IAAA,IAAA,CAAA,MAAW,MAAA,GAAS,IAAA,CAAK,EAAA,CAAG,MAAA,CAAO,CAAA,EAAG;AACrC,MAAA,GAAA,CAAI,6CAAA,KAAc,CAAM,GAAA,EAAK,MAAM,CAAA,EAAG;AACrC,QAAA,OAAA,CAAQ,IAAA,CAAK,EAAE,GAAA,EAAK,KAAA,CAAM,GAAA,EAAK,KAAA,EAAO,KAAA,CAAM,MAAM,CAAC,CAAA;AAAA,MACpD;AAAA,IACD;AAEA,IAAA,OAAO,OAAA,CAAQ,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,EAAA,GAAM,2CAAA,CAAY,CAAE,GAAA,EAAK,CAAA,CAAE,GAAG,CAAC,CAAA;AAAA,EACxD;AAAA,EAEA,MAAM,KAAA,CAAM,MAAA,EAAkC;AAC7C,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,IAAA,CAAA,MAAW,EAAE,GAAA,EAAK,MAAM,EAAA,GAAK,MAAA,EAAQ;AACpC,MAAA,IAAA,CAAK,EAAA,CAAG,GAAA,CAAI,wCAAA,GAAY,CAAA,EAAG,EAAE,GAAA,EAAK,MAAM,CAAC,CAAA;AAAA,IAC1C;AAAA,EACD;AAAA,EAEA,MAAM,QAAA,CAAS,UAAA,EAAoB,MAAA,EAA+B;AACjE,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,UAAA,EAAY,MAAM,CAAA;AAAA,EACnC;AAAA,EAEA,MAAM,UAAA,CAAW,UAAA,EAAmC;AACnD,IAAA,MAAM,qCAAA,IAAM,CAAK,OAAO,CAAA;AACxB,IAAA,IAAA,CAAK,MAAA,CAAO,MAAA,CAAO,UAAU,CAAA;AAAA,EAC9B;AAAA,EAEA,MAAM,eAAA,CACL,YAAA,EACA,WAAA,EACgB;AAChB,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,aAAA;AAMpB,IAAA,GAAA,CAAI,MAAA,CAAO,eAAA,EAAiB;AAC3B,MAAA,MAAM,MAAA,CAAO,eAAA,CAAgB,YAAA,EAAc,WAAW,CAAA;AACtD,MAAA,MAAA;AAAA,IACD;AAEA,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,GAAA,CAAI,WAAA,CAAY,OAAA,EAAS;AACxB,QAAA,MAAM,IAAI,mCAAA,CAAa,CAAA;AAAA,MACxB;AACA,MAAA,MAAM,SAAA,EAAW,MAAM,IAAA,CAAK,aAAA,CAAc,eAAA,CAAgB;AAAA,QACzD,KAAA,EAAO,YAAA,CAAa,OAAA,EAAS,EAAA,EAAI,aAAA,EAAe,KAAA,CAAA;AAAA,QAChD,KAAA,EAAO,CAAA;AAAA,QACP,WAAA,EAAa;AAAA,MACd,CAAC,CAAA;AACD,MAAA,GAAA,CAAI,QAAA,CAAS,OAAA,EAAS,CAAA,EAAG;AACxB,QAAA,MAAA;AAAA,MACD;AACA,MAAA,MAAM,qCAAA,IAAM,CAAK,GAAA,CAAI,CAAA,EAAG,IAAA,CAAK,OAAO,CAAC,CAAA;AAAA,IACtC;AAAA,EACD;AAAA;AAAA;AAAA;AAAA,EAKA,QAAA,CAAS,UAAA,EAAwC;AAChD,IAAA,OAAO,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,UAAU,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA,EAKA,YAAA,CAAA,EAAyB;AACxB,IAAA,MAAM,IAAA,EAAM,IAAA,CAAK,GAAA,CAAI,CAAA;AACrB,IAAA,MAAM,IAAA,EAAgB,CAAC,CAAA;AACvB,IAAA,IAAA,CAAA,MAAW,CAAC,UAAA,EAAY,MAAM,EAAA,GAAK,IAAA,CAAK,MAAA,EAAQ;AAC/C,MAAA,GAAA,CAAI,OAAA,GAAU,GAAA,EAAK;AAClB,QAAA,GAAA,CAAI,IAAA,CAAK,UAAU,CAAA;AAAA,MACpB;AAAA,IACD;AACA,IAAA,OAAO,GAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKA,KAAA,CAAA,EAAc;AACb,IAAA,IAAA,CAAK,EAAA,CAAG,KAAA,CAAM,CAAA;AACd,IAAA,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,CAAA;AAClB,IAAA,IAAA,CAAK,CAAA,qBAAA,CAAuB,KAAA,CAAM,CAAA;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA,EAKA,QAAA,CAAA,EAGE;AACD,IAAA,MAAM,WAAA,EAAyC,CAAC,CAAA;AAChD,IAAA,IAAA,CAAA,MAAW,CAAC,MAAA,EAAQ,KAAK,EAAA,GAAK,IAAA,CAAK,EAAA,EAAI;AACtC,MAAA,UAAA,CAAW,MAAM,EAAA,EAAI,KAAA,CAAM,KAAA;AAAA,IAC5B;AACA,IAAA,OAAO;AAAA,MACN,EAAA,EAAI,UAAA;AAAA,MACJ,MAAA,EAAQ,MAAA,CAAO,WAAA,CAAY,IAAA,CAAK,MAAM;AAAA,IACvC,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA,EAKA,IAAA,CAAA,EAAiB;AAChB,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,EAAA,CAAG,IAAA,CAAK,CAAC,CAAA;AAAA,EAC1B;AACD,UAAA;ADpBA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,0sFAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/workflow-engine/dist/tsup/testing.cjs","sourcesContent":[null,"import type { EngineDriver, KVEntry, KVWrite } from \"./driver.js\";\nimport { EvictedError } from \"./errors.js\";\nimport { compareKeys, keyStartsWith, keyToHex } from \"./keys.js\";\nimport type { Message, WorkflowMessageDriver } from \"./types.js\";\nimport { sleep } from \"./utils.js\";\n\ninterface Waiter {\n\tnameSet?: Set<string>;\n\tresolve: () => void;\n\treject: (error: Error) => void;\n\tabortSignal: AbortSignal;\n\tonAbort: () => void;\n}\n\nclass InMemoryWorkflowMessageDriver implements WorkflowMessageDriver {\n\t#messages: Message[] = [];\n\t#waiters = new Set<Waiter>();\n\n\tasync addMessage(message: Message): Promise<void> {\n\t\tthis.#messages.push(message);\n\t\tthis.#notifyWaiters(message.name);\n\t}\n\n\tasync receiveMessages(opts: {\n\t\tnames?: readonly string[];\n\t\tcount: number;\n\t\tcompletable: boolean;\n\t}): Promise<Message[]> {\n\t\tconst limitedCount = Math.max(1, opts.count);\n\t\tconst nameSet =\n\t\t\topts.names && opts.names.length > 0\n\t\t\t\t? new Set(opts.names)\n\t\t\t\t: undefined;\n\t\tconst selected: Array<{ message: Message; index: number }> = [];\n\n\t\tfor (let i = 0; i < this.#messages.length && selected.length < limitedCount; i++) {\n\t\t\tconst message = this.#messages[i];\n\t\t\tif (nameSet && !nameSet.has(message.name)) {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tselected.push({ message, index: i });\n\t\t}\n\n\t\tif (selected.length === 0) {\n\t\t\treturn [];\n\t\t}\n\n\t\tif (!opts.completable) {\n\t\t\tfor (let i = selected.length - 1; i >= 0; i--) {\n\t\t\t\tthis.#messages.splice(selected[i].index, 1);\n\t\t\t}\n\t\t\treturn selected.map((entry) => entry.message);\n\t\t}\n\n\t\treturn selected.map((entry) => {\n\t\t\tconst { message } = entry;\n\t\t\treturn {\n\t\t\t\t...message,\n\t\t\t\tcomplete: async () => {\n\t\t\t\t\tawait this.completeMessage(message.id);\n\t\t\t\t},\n\t\t\t};\n\t\t});\n\t}\n\n\tasync completeMessage(messageId: string): Promise<void> {\n\t\tconst index = this.#messages.findIndex((message) => message.id === messageId);\n\t\tif (index !== -1) {\n\t\t\tthis.#messages.splice(index, 1);\n\t\t}\n\t}\n\n\tasync waitForMessages(\n\t\tmessageNames: string[],\n\t\tabortSignal: AbortSignal,\n\t): Promise<void> {\n\t\tif (abortSignal.aborted) {\n\t\t\tthrow new EvictedError();\n\t\t}\n\n\t\tconst nameSet = messageNames.length > 0 ? new Set(messageNames) : undefined;\n\t\tif (\n\t\t\tthis.#messages.some((message) =>\n\t\t\t\tnameSet ? nameSet.has(message.name) : true,\n\t\t\t)\n\t\t) {\n\t\t\treturn;\n\t\t}\n\n\t\tawait new Promise<void>((resolve, reject) => {\n\t\t\tconst waiter: Waiter = {\n\t\t\t\tnameSet,\n\t\t\t\tresolve: () => {\n\t\t\t\t\tthis.#removeWaiter(waiter);\n\t\t\t\t\tresolve();\n\t\t\t\t},\n\t\t\t\treject: (error) => {\n\t\t\t\t\tthis.#removeWaiter(waiter);\n\t\t\t\t\treject(error);\n\t\t\t\t},\n\t\t\t\tabortSignal,\n\t\t\t\tonAbort: () => {\n\t\t\t\t\twaiter.reject(new EvictedError());\n\t\t\t\t},\n\t\t\t};\n\t\t\tabortSignal.addEventListener(\"abort\", waiter.onAbort, { once: true });\n\t\t\tthis.#waiters.add(waiter);\n\t\t});\n\t}\n\n\t#removeWaiter(waiter: Waiter): void {\n\t\tif (this.#waiters.delete(waiter)) {\n\t\t\twaiter.abortSignal.removeEventListener(\"abort\", waiter.onAbort);\n\t\t}\n\t}\n\n\t#notifyWaiters(name: string): void {\n\t\tfor (const waiter of [...this.#waiters]) {\n\t\t\tif (waiter.nameSet && !waiter.nameSet.has(name)) {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\twaiter.resolve();\n\t\t}\n\t}\n\n\tclear(): void {\n\t\tthis.#messages = [];\n\t\tfor (const waiter of [...this.#waiters]) {\n\t\t\twaiter.reject(new Error(\"cleared\"));\n\t\t}\n\t}\n}\n\n/**\n * In-memory implementation of EngineDriver for testing.\n * Uses binary keys (Uint8Array) with hex encoding for internal Map storage.\n */\nexport class InMemoryDriver implements EngineDriver {\n\t// Map from hex-encoded key to { originalKey, value }\n\tprivate kv = new Map<string, { key: Uint8Array; value: Uint8Array }>();\n\tprivate alarms = new Map<string, number>();\n\t#inMemoryMessageDriver = new InMemoryWorkflowMessageDriver();\n\n\t/** Simulated latency per operation (ms) */\n\tlatency = 10;\n\n\t/** How often the worker polls for work */\n\tworkerPollInterval = 100;\n\tmessageDriver: WorkflowMessageDriver = this.#inMemoryMessageDriver;\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\tawait sleep(this.latency);\n\t\tconst entry = this.kv.get(keyToHex(key));\n\t\treturn entry?.value ?? null;\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait sleep(this.latency);\n\t\tthis.kv.set(keyToHex(key), { key, value });\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tawait sleep(this.latency);\n\t\tthis.kv.delete(keyToHex(key));\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tawait sleep(this.latency);\n\t\tfor (const [hexKey, entry] of this.kv) {\n\t\t\tif (keyStartsWith(entry.key, prefix)) {\n\t\t\t\tthis.kv.delete(hexKey);\n\t\t\t}\n\t\t}\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\tawait sleep(this.latency);\n\t\tconst results: KVEntry[] = [];\n\t\tfor (const entry of this.kv.values()) {\n\t\t\tif (keyStartsWith(entry.key, prefix)) {\n\t\t\t\tresults.push({ key: entry.key, value: entry.value });\n\t\t\t}\n\t\t}\n\t\t// Sort by key lexicographically\n\t\treturn results.sort((a, b) => compareKeys(a.key, b.key));\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tawait sleep(this.latency);\n\t\tfor (const { key, value } of writes) {\n\t\t\tthis.kv.set(keyToHex(key), { key, value });\n\t\t}\n\t}\n\n\tasync setAlarm(workflowId: string, wakeAt: number): Promise<void> {\n\t\tawait sleep(this.latency);\n\t\tthis.alarms.set(workflowId, wakeAt);\n\t}\n\n\tasync clearAlarm(workflowId: string): Promise<void> {\n\t\tawait sleep(this.latency);\n\t\tthis.alarms.delete(workflowId);\n\t}\n\n\tasync waitForMessages(\n\t\tmessageNames: string[],\n\t\tabortSignal: AbortSignal,\n\t): Promise<void> {\n\t\tconst driver = this.messageDriver as WorkflowMessageDriver & {\n\t\t\twaitForMessages?: (\n\t\t\t\tmessageNames: string[],\n\t\t\t\tabortSignal: AbortSignal,\n\t\t\t) => Promise<void>;\n\t\t};\n\t\tif (driver.waitForMessages) {\n\t\t\tawait driver.waitForMessages(messageNames, abortSignal);\n\t\t\treturn;\n\t\t}\n\n\t\twhile (true) {\n\t\t\tif (abortSignal.aborted) {\n\t\t\t\tthrow new EvictedError();\n\t\t\t}\n\t\t\tconst messages = await this.messageDriver.receiveMessages({\n\t\t\t\tnames: messageNames.length > 0 ? messageNames : undefined,\n\t\t\t\tcount: 1,\n\t\t\t\tcompletable: true,\n\t\t\t});\n\t\t\tif (messages.length > 0) {\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tawait sleep(Math.max(1, this.latency));\n\t\t}\n\t}\n\n\t/**\n\t * Get the alarm time for a workflow (for testing).\n\t */\n\tgetAlarm(workflowId: string): number | undefined {\n\t\treturn this.alarms.get(workflowId);\n\t}\n\n\t/**\n\t * Check if any alarms are due and return their workflow IDs.\n\t */\n\tgetDueAlarms(): string[] {\n\t\tconst now = Date.now();\n\t\tconst due: string[] = [];\n\t\tfor (const [workflowId, wakeAt] of this.alarms) {\n\t\t\tif (wakeAt <= now) {\n\t\t\t\tdue.push(workflowId);\n\t\t\t}\n\t\t}\n\t\treturn due;\n\t}\n\n\t/**\n\t * Clear all data (for testing).\n\t */\n\tclear(): void {\n\t\tthis.kv.clear();\n\t\tthis.alarms.clear();\n\t\tthis.#inMemoryMessageDriver.clear();\n\t}\n\n\t/**\n\t * Get a snapshot of all data (for testing/debugging).\n\t */\n\tsnapshot(): {\n\t\tkv: Record<string, Uint8Array>;\n\t\talarms: Record<string, number>;\n\t} {\n\t\tconst kvSnapshot: Record<string, Uint8Array> = {};\n\t\tfor (const [hexKey, entry] of this.kv) {\n\t\t\tkvSnapshot[hexKey] = entry.value;\n\t\t}\n\t\treturn {\n\t\t\tkv: kvSnapshot,\n\t\t\talarms: Object.fromEntries(this.alarms),\n\t\t};\n\t}\n\n\t/**\n\t * Get all hex-encoded keys (for testing).\n\t */\n\tkeys(): string[] {\n\t\treturn [...this.kv.keys()];\n\t}\n}\n\n// Re-export main exports for convenience\nexport * from \"./index.js\";\n"]}

package/dist/tsup/testing.d.cts

@@ -0,0 +1,52 @@
+import { EngineDriver, WorkflowMessageDriver, KVEntry, KVWrite } from './index.cjs';
+export { BranchConfig, BranchOutput, BranchStatus, BranchStatusType, CancelledError, CriticalError, DEFAULT_LOOP_COMMIT_INTERVAL, DEFAULT_LOOP_HISTORY_EVERY, DEFAULT_LOOP_HISTORY_KEEP, DEFAULT_MAX_RETRIES, DEFAULT_RETRY_BACKOFF_BASE, DEFAULT_RETRY_BACKOFF_MAX, DEFAULT_STEP_TIMEOUT, Entry, EntryInProgressError, EntryKind, EntryKindType, EntryMetadata, EntryStatus, EvictedError, History, HistoryDivergedError, JoinEntry, JoinError, Location, Loop, LoopConfig, LoopEntry, LoopIterationMarker, LoopResult, Message, MessageEntry, MessageWaitError, NameIndex, PathSegment, RaceEntry, RaceError, RemovedEntry, RollbackCheckpointEntry, RollbackCheckpointError, RollbackContextInterface, RollbackError, RunWorkflowOptions, SleepEntry, SleepError, SleepState, StepConfig, StepEntry, StepExhaustedError, StepFailedError, Storage, WorkflowContextImpl, WorkflowContextInterface, WorkflowEntryMetadataSnapshot, WorkflowFunction, WorkflowHandle, WorkflowHistoryEntry, WorkflowHistorySnapshot, WorkflowQueue, WorkflowQueueMessage, WorkflowQueueNextOptions, WorkflowResult, WorkflowRunMode, WorkflowState, appendLoopIteration, appendName, createEntry, createHistorySnapshot, createStorage, deleteEntriesWithPrefix, emptyLocation, flush, generateId, getEntry, getOrCreateMetadata, isLocationPrefix, isLoopIterationMarker, loadMetadata, loadStorage, locationToKey, locationsEqual, parentLocation, registerName, resolveName, runWorkflow, setEntry } from './index.cjs';
+import 'pino';
+
+/**
+ * In-memory implementation of EngineDriver for testing.
+ * Uses binary keys (Uint8Array) with hex encoding for internal Map storage.
+ */
+declare class InMemoryDriver implements EngineDriver {
+    #private;
+    private kv;
+    private alarms;
+    /** Simulated latency per operation (ms) */
+    latency: number;
+    /** How often the worker polls for work */
+    workerPollInterval: number;
+    messageDriver: WorkflowMessageDriver;
+    get(key: Uint8Array): Promise<Uint8Array | null>;
+    set(key: Uint8Array, value: Uint8Array): Promise<void>;
+    delete(key: Uint8Array): Promise<void>;
+    deletePrefix(prefix: Uint8Array): Promise<void>;
+    list(prefix: Uint8Array): Promise<KVEntry[]>;
+    batch(writes: KVWrite[]): Promise<void>;
+    setAlarm(workflowId: string, wakeAt: number): Promise<void>;
+    clearAlarm(workflowId: string): Promise<void>;
+    waitForMessages(messageNames: string[], abortSignal: AbortSignal): Promise<void>;
+    /**
+     * Get the alarm time for a workflow (for testing).
+     */
+    getAlarm(workflowId: string): number | undefined;
+    /**
+     * Check if any alarms are due and return their workflow IDs.
+     */
+    getDueAlarms(): string[];
+    /**
+     * Clear all data (for testing).
+     */
+    clear(): void;
+    /**
+     * Get a snapshot of all data (for testing/debugging).
+     */
+    snapshot(): {
+        kv: Record<string, Uint8Array>;
+        alarms: Record<string, number>;
+    };
+    /**
+     * Get all hex-encoded keys (for testing).
+     */
+    keys(): string[];
+}
+
+export { EngineDriver, InMemoryDriver, KVEntry, KVWrite, WorkflowMessageDriver };

package/dist/tsup/testing.d.ts

@@ -0,0 +1,52 @@
+import { EngineDriver, WorkflowMessageDriver, KVEntry, KVWrite } from './index.js';
+export { BranchConfig, BranchOutput, BranchStatus, BranchStatusType, CancelledError, CriticalError, DEFAULT_LOOP_COMMIT_INTERVAL, DEFAULT_LOOP_HISTORY_EVERY, DEFAULT_LOOP_HISTORY_KEEP, DEFAULT_MAX_RETRIES, DEFAULT_RETRY_BACKOFF_BASE, DEFAULT_RETRY_BACKOFF_MAX, DEFAULT_STEP_TIMEOUT, Entry, EntryInProgressError, EntryKind, EntryKindType, EntryMetadata, EntryStatus, EvictedError, History, HistoryDivergedError, JoinEntry, JoinError, Location, Loop, LoopConfig, LoopEntry, LoopIterationMarker, LoopResult, Message, MessageEntry, MessageWaitError, NameIndex, PathSegment, RaceEntry, RaceError, RemovedEntry, RollbackCheckpointEntry, RollbackCheckpointError, RollbackContextInterface, RollbackError, RunWorkflowOptions, SleepEntry, SleepError, SleepState, StepConfig, StepEntry, StepExhaustedError, StepFailedError, Storage, WorkflowContextImpl, WorkflowContextInterface, WorkflowEntryMetadataSnapshot, WorkflowFunction, WorkflowHandle, WorkflowHistoryEntry, WorkflowHistorySnapshot, WorkflowQueue, WorkflowQueueMessage, WorkflowQueueNextOptions, WorkflowResult, WorkflowRunMode, WorkflowState, appendLoopIteration, appendName, createEntry, createHistorySnapshot, createStorage, deleteEntriesWithPrefix, emptyLocation, flush, generateId, getEntry, getOrCreateMetadata, isLocationPrefix, isLoopIterationMarker, loadMetadata, loadStorage, locationToKey, locationsEqual, parentLocation, registerName, resolveName, runWorkflow, setEntry } from './index.js';
+import 'pino';
+
+/**
+ * In-memory implementation of EngineDriver for testing.
+ * Uses binary keys (Uint8Array) with hex encoding for internal Map storage.
+ */
+declare class InMemoryDriver implements EngineDriver {
+    #private;
+    private kv;
+    private alarms;
+    /** Simulated latency per operation (ms) */
+    latency: number;
+    /** How often the worker polls for work */
+    workerPollInterval: number;
+    messageDriver: WorkflowMessageDriver;
+    get(key: Uint8Array): Promise<Uint8Array | null>;
+    set(key: Uint8Array, value: Uint8Array): Promise<void>;
+    delete(key: Uint8Array): Promise<void>;
+    deletePrefix(prefix: Uint8Array): Promise<void>;
+    list(prefix: Uint8Array): Promise<KVEntry[]>;
+    batch(writes: KVWrite[]): Promise<void>;
+    setAlarm(workflowId: string, wakeAt: number): Promise<void>;
+    clearAlarm(workflowId: string): Promise<void>;
+    waitForMessages(messageNames: string[], abortSignal: AbortSignal): Promise<void>;
+    /**
+     * Get the alarm time for a workflow (for testing).
+     */
+    getAlarm(workflowId: string): number | undefined;
+    /**
+     * Check if any alarms are due and return their workflow IDs.
+     */
+    getDueAlarms(): string[];
+    /**
+     * Clear all data (for testing).
+     */
+    clear(): void;
+    /**
+     * Get a snapshot of all data (for testing/debugging).
+     */
+    snapshot(): {
+        kv: Record<string, Uint8Array>;
+        alarms: Record<string, number>;
+    };
+    /**
+     * Get all hex-encoded keys (for testing).
+     */
+    keys(): string[];
+}
+
+export { EngineDriver, InMemoryDriver, KVEntry, KVWrite, WorkflowMessageDriver };