@dxos/echo-pipeline 0.3.2-main.ce2190e → 0.3.2-main.d77d2bc

package/dist/lib/node/testing/index.cjs

@@ -30,6 +30,8 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // packages/core/echo/echo-pipeline/src/testing/index.ts
 var testing_exports = {};
 __export(testing_exports, {
+  DatabaseTestBuilder: () => DatabaseTestBuilder,
+  DatabaseTestPeer: () => DatabaseTestPeer,
   MemoryNetworkManagerProvider: () => MemoryNetworkManagerProvider,
   TestAgent: () => TestAgent,
   TestAgentBuilder: () => TestAgentBuilder,
@@ -285,23 +287,12 @@ var DatabaseHost = class {
 };
 
 // packages/core/echo/echo-pipeline/src/db-host/snapshot-manager.ts
-var import_async = require("@dxos/async");
 var import_context2 = require("@dxos/context");
 var import_keys = require("@dxos/keys");
 var import_protocols2 = require("@dxos/protocols");
 var import_blob = require("@dxos/protocols/proto/dxos/echo/blob");
-function _ts_decorate(decorators, target, key, desc) {
-  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
-  if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
-    r = Reflect.decorate(decorators, target, key, desc);
-  else
-    for (var i = decorators.length - 1; i >= 0; i--)
-      if (d = decorators[i])
-        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
-  return c > 3 && r && Object.defineProperty(target, key, r), r;
-}
 var SpaceSnapshot = import_protocols2.schema.getCodecForType("dxos.echo.snapshot.SpaceSnapshot");
-var SnapshotManager = class SnapshotManager2 {
+var SnapshotManager = class {
   constructor(_snapshotStore, _blobStore, _blobSync) {
     this._snapshotStore = _snapshotStore;
     this._blobStore = _blobStore;
@@ -330,9 +321,6 @@ var SnapshotManager = class SnapshotManager2 {
     return import_keys.PublicKey.from(id).toHex();
   }
 };
-SnapshotManager = _ts_decorate([
-  (0, import_async.trackLeaks)("open", "close")
-], SnapshotManager);
 
 // packages/core/echo/echo-pipeline/src/db-host/snapshot-store.ts
 var import_crypto = require("@dxos/crypto");
@@ -487,14 +475,14 @@ var DataServiceImpl = class {
 
 // packages/core/echo/echo-pipeline/src/metadata/metadata-store.ts
 var import_crc_32 = __toESM(require("crc-32"));
-var import_async2 = require("@dxos/async");
+var import_async = require("@dxos/async");
 var import_invariant4 = require("@dxos/invariant");
 var import_keys3 = require("@dxos/keys");
 var import_log3 = require("@dxos/log");
 var import_protocols4 = require("@dxos/protocols");
 var import_services = require("@dxos/protocols/proto/dxos/client/services");
 var import_util3 = require("@dxos/util");
-function _ts_decorate2(decorators, target, key, desc) {
+function _ts_decorate(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -520,7 +508,7 @@ var MetadataStore = class {
     this._metadata = emptyEchoMetadata();
     this._spaceLargeMetadata = new import_util3.ComplexMap(import_keys3.PublicKey.hash);
     this._metadataFile = void 0;
-    this.update = new import_async2.Event();
+    this.update = new import_async.Event();
   }
   get metadata() {
     return this._metadata;
@@ -778,19 +766,19 @@ var MetadataStore = class {
     await this.flush();
   }
 };
-_ts_decorate2([
-  import_async2.synchronized
+_ts_decorate([
+  import_async.synchronized
 ], MetadataStore.prototype, "load", null);
-_ts_decorate2([
-  import_async2.synchronized
+_ts_decorate([
+  import_async.synchronized
 ], MetadataStore.prototype, "_save", null);
-_ts_decorate2([
-  import_async2.synchronized
+_ts_decorate([
+  import_async.synchronized
 ], MetadataStore.prototype, "_saveSpaceLargeMetadata", null);
 var fromBytesInt32 = (buf) => buf.readInt32LE(0);
 
 // packages/core/echo/echo-pipeline/src/space/auth.ts
-var import_async3 = require("@dxos/async");
+var import_async2 = require("@dxos/async");
 var import_context3 = require("@dxos/context");
 var import_crypto2 = require("@dxos/crypto");
 var import_invariant5 = require("@dxos/invariant");
@@ -848,7 +836,7 @@ var AuthExtension = class extends import_teleport.RpcExtension {
   }
   async onOpen(context) {
     await super.onOpen(context);
-    (0, import_async3.scheduleTask)(this._ctx, async () => {
+    (0, import_async2.scheduleTask)(this._ctx, async () => {
       try {
         const challenge = (0, import_crypto2.randomBytes)(32);
         const { credential } = await this.rpc.AuthService.authenticate({
@@ -873,7 +861,7 @@ var AuthExtension = class extends import_teleport.RpcExtension {
             "'credential not verified'"
           ]
         });
-        (0, import_async3.runInContext)(this._ctx, () => this._authParams.onAuthSuccess());
+        (0, import_async2.runInContext)(this._ctx, () => this._authParams.onAuthSuccess());
       } catch (err) {
         (0, import_log4.log)("auth failed", err, {
           F: __dxlog_file5,
@@ -897,7 +885,7 @@ var AuthExtension = class extends import_teleport.RpcExtension {
 };
 
 // packages/core/echo/echo-pipeline/src/space/space.ts
-var import_async8 = require("@dxos/async");
+var import_async7 = require("@dxos/async");
 var import_invariant9 = require("@dxos/invariant");
 var import_log10 = require("@dxos/log");
 var import_credentials4 = require("@dxos/protocols/proto/dxos/halo/credentials");
@@ -905,7 +893,7 @@ var import_tracing3 = require("@dxos/tracing");
 var import_util7 = require("@dxos/util");
 
 // packages/core/echo/echo-pipeline/src/space/control-pipeline.ts
-var import_async6 = require("@dxos/async");
+var import_async5 = require("@dxos/async");
 var import_context5 = require("@dxos/context");
 var import_credentials = require("@dxos/credentials");
 var import_keys5 = require("@dxos/keys");
@@ -916,7 +904,7 @@ var import_tracing = require("@dxos/tracing");
 var import_util5 = require("@dxos/util");
 
 // packages/core/echo/echo-pipeline/src/pipeline/pipeline.ts
-var import_async5 = require("@dxos/async");
+var import_async4 = require("@dxos/async");
 var import_context4 = require("@dxos/context");
 var import_debug3 = require("@dxos/debug");
 var import_feed_store = require("@dxos/feed-store");
@@ -957,11 +945,11 @@ var createMessageSelector = (timeframeClock) => {
 };
 
 // packages/core/echo/echo-pipeline/src/pipeline/timeframe-clock.ts
-var import_async4 = require("@dxos/async");
+var import_async3 = require("@dxos/async");
 var import_debug2 = require("@dxos/debug");
 var import_log6 = require("@dxos/log");
 var import_timeframe = require("@dxos/timeframe");
-function _ts_decorate3(decorators, target, key, desc) {
+function _ts_decorate2(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -983,7 +971,7 @@ var startAfter = (timeframe) => timeframe.frames().map(([feedKey, index]) => ({
 var TimeframeClock = class {
   constructor(_timeframe = new import_timeframe.Timeframe()) {
     this._timeframe = _timeframe;
-    this.update = new import_async4.Event();
+    this.update = new import_async3.Event();
     this._pendingTimeframe = _timeframe;
   }
   /**
@@ -1045,12 +1033,12 @@ var TimeframeClock = class {
     });
   }
 };
-_ts_decorate3([
+_ts_decorate2([
   (0, import_debug2.timed)(5e3)
 ], TimeframeClock.prototype, "waitUntilReached", null);
 
 // packages/core/echo/echo-pipeline/src/pipeline/pipeline.ts
-function _ts_decorate4(decorators, target, key, desc) {
+function _ts_decorate3(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -1067,7 +1055,7 @@ var PipelineState = class {
     this._timeframeClock = _timeframeClock;
     this._ctx = new import_context4.Context();
     this.timeframeUpdate = this._timeframeClock.update;
-    this.stalled = new import_async5.Event();
+    this.stalled = new import_async4.Event();
     this._startTimeframe = new import_timeframe2.Timeframe();
     this._reachedTarget = false;
   }
@@ -1141,7 +1129,7 @@ var PipelineState = class {
           done = true;
           this._reachedTarget = true;
         }),
-        (0, import_async5.sleep)(timeout).then(() => {
+        (0, import_async4.sleepWithContext)(this._ctx, timeout).then(() => {
           if (done) {
             return;
           }
@@ -1170,8 +1158,8 @@ var Pipeline = class {
     // External state accessor.
     this._state = new PipelineState(this._feeds, this._timeframeClock);
     // Waits for the message consumer to process the message and yield control back to the pipeline.
-    this._processingTrigger = new import_async5.Trigger().wake();
-    this._pauseTrigger = new import_async5.Trigger().wake();
+    this._processingTrigger = new import_async4.Trigger().wake();
+    this._pauseTrigger = new import_async4.Trigger().wake();
     // Pending downloads.
     this._downloads = new import_util4.ComplexMap((value) => import_keys4.PublicKey.hash(value.key));
     this._isStopping = false;
@@ -1448,24 +1436,24 @@ var Pipeline = class {
     }
   }
 };
-_ts_decorate4([
-  import_async5.synchronized
+_ts_decorate3([
+  import_async4.synchronized
 ], Pipeline.prototype, "start", null);
-_ts_decorate4([
-  import_async5.synchronized
+_ts_decorate3([
+  import_async4.synchronized
 ], Pipeline.prototype, "stop", null);
-_ts_decorate4([
-  import_async5.synchronized
+_ts_decorate3([
+  import_async4.synchronized
 ], Pipeline.prototype, "setCursor", null);
-_ts_decorate4([
-  import_async5.synchronized
+_ts_decorate3([
+  import_async4.synchronized
 ], Pipeline.prototype, "pause", null);
-_ts_decorate4([
-  import_async5.synchronized
+_ts_decorate3([
+  import_async4.synchronized
 ], Pipeline.prototype, "unpause", null);
 
 // packages/core/echo/echo-pipeline/src/space/control-pipeline.ts
-function _ts_decorate5(decorators, target, key, desc) {
+function _ts_decorate4(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -1486,8 +1474,8 @@ var ControlPipeline = class ControlPipeline2 {
     this.onFeedAdmitted = new import_util5.Callback();
     this._usage = new import_tracing.TimeUsageCounter();
     this._mutations = new import_tracing.TimeSeriesCounter();
-    this._snapshotTask = new import_async6.DeferredTask(this._ctx, async () => {
-      await (0, import_async6.sleep)(CONTROL_PIPELINE_SNAPSHOT_DELAY);
+    this._snapshotTask = new import_async5.DeferredTask(this._ctx, async () => {
+      await (0, import_async5.sleepWithContext)(this._ctx, CONTROL_PIPELINE_SNAPSHOT_DELAY);
       await this._saveSnapshot();
     });
     this._spaceKey = spaceKey;
@@ -1500,7 +1488,7 @@ var ControlPipeline = class ControlPipeline2 {
         key: info.key
       }, {
         F: __dxlog_file9,
-        L: 73,
+        L: 74,
         S: this,
         C: (f, a) => f(...a)
       });
@@ -1512,7 +1500,7 @@ var ControlPipeline = class ControlPipeline2 {
           } catch (err) {
             import_log8.log.catch(err, void 0, {
               F: __dxlog_file9,
-              L: 82,
+              L: 83,
               S: this,
               C: (f, a) => f(...a)
             });
@@ -1541,16 +1529,16 @@ var ControlPipeline = class ControlPipeline2 {
       tf: snapshot?.timeframe
     }, {
       F: __dxlog_file9,
-      L: 109,
+      L: 110,
       S: this,
       C: (f, a) => f(...a)
     });
     if (USE_SNAPSHOTS && snapshot) {
       await this._processSnapshot(snapshot);
     }
-    (0, import_log8.log)("starting...", void 0, {
+    import_log8.log.info("starting...", void 0, {
       F: __dxlog_file9,
-      L: 114,
+      L: 115,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1560,7 +1548,7 @@ var ControlPipeline = class ControlPipeline2 {
     await this._pipeline.start();
     (0, import_log8.log)("started", void 0, {
       F: __dxlog_file9,
-      L: 120,
+      L: 121,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1577,7 +1565,7 @@ var ControlPipeline = class ControlPipeline2 {
           message
         }, {
           F: __dxlog_file9,
-          L: 133,
+          L: 134,
           S: this,
           C: (f, a) => f(...a)
         });
@@ -1599,7 +1587,7 @@ var ControlPipeline = class ControlPipeline2 {
       snapshot
     }, {
       F: __dxlog_file9,
-      L: 149,
+      L: 150,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1614,7 +1602,7 @@ var ControlPipeline = class ControlPipeline2 {
       } catch (err) {
         import_log8.log.catch(err, void 0, {
           F: __dxlog_file9,
-          L: 162,
+          L: 163,
           S: this,
           C: (f, a) => f(...a)
         });
@@ -1628,7 +1616,7 @@ var ControlPipeline = class ControlPipeline2 {
       seq: msg.seq
     }, {
       F: __dxlog_file9,
-      L: 172,
+      L: 173,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1643,7 +1631,7 @@ var ControlPipeline = class ControlPipeline2 {
           msg
         }, {
           F: __dxlog_file9,
-          L: 181,
+          L: 182,
           S: this,
           C: (f, a) => f(...a)
         });
@@ -1660,9 +1648,9 @@ var ControlPipeline = class ControlPipeline2 {
     }
   }
   async stop() {
-    (0, import_log8.log)("stopping...", void 0, {
+    import_log8.log.warn("stopping...", void 0, {
       F: __dxlog_file9,
-      L: 201,
+      L: 202,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1671,7 +1659,7 @@ var ControlPipeline = class ControlPipeline2 {
     await this._saveTargetTimeframe(this._pipeline.state.timeframe);
     (0, import_log8.log)("stopped", void 0, {
       F: __dxlog_file9,
-      L: 205,
+      L: 206,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1684,36 +1672,37 @@ var ControlPipeline = class ControlPipeline2 {
     } catch (err) {
       (0, import_log8.log)(err, void 0, {
         F: __dxlog_file9,
-        L: 214,
+        L: 215,
         S: this,
         C: (f, a) => f(...a)
       });
     }
   }
 };
-_ts_decorate5([
+_ts_decorate4([
   import_tracing.trace.metricsCounter()
 ], ControlPipeline.prototype, "_usage", void 0);
-_ts_decorate5([
+_ts_decorate4([
   import_tracing.trace.metricsCounter()
 ], ControlPipeline.prototype, "_mutations", void 0);
-_ts_decorate5([
+_ts_decorate4([
   import_tracing.trace.span({
     showInBrowserTimeline: true
   })
 ], ControlPipeline.prototype, "start", null);
-_ts_decorate5([
+_ts_decorate4([
   import_tracing.trace.span()
 ], ControlPipeline.prototype, "_consumePipeline", null);
-_ts_decorate5([
+_ts_decorate4([
   import_tracing.trace.span()
 ], ControlPipeline.prototype, "_processMessage", null);
-ControlPipeline = _ts_decorate5([
-  import_tracing.trace.resource()
+ControlPipeline = _ts_decorate4([
+  import_tracing.trace.resource(),
+  (0, import_async5.trackLeaks)("start", "stop")
 ], ControlPipeline);
 
 // packages/core/echo/echo-pipeline/src/space/data-pipeline.ts
-var import_async7 = require("@dxos/async");
+var import_async6 = require("@dxos/async");
 var import_context6 = require("@dxos/context");
 var import_credentials3 = require("@dxos/credentials");
 var import_echo_db3 = require("@dxos/echo-db");
@@ -1723,7 +1712,7 @@ var import_protocols6 = require("@dxos/protocols");
 var import_timeframe4 = require("@dxos/timeframe");
 var import_tracing2 = require("@dxos/tracing");
 var import_util6 = require("@dxos/util");
-function _ts_decorate6(decorators, target, key, desc) {
+function _ts_decorate5(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -1752,7 +1741,7 @@ var DataPipeline = class DataPipeline2 {
     this._mutations = new import_tracing2.TimeSeriesCounter();
     this.currentEpoch = void 0;
     this.appliedEpoch = void 0;
-    this.onNewEpoch = new import_async7.Event();
+    this.onNewEpoch = new import_async6.Event();
   }
   get isOpen() {
     return this._isOpen;
@@ -1815,7 +1804,7 @@ var DataPipeline = class DataPipeline2 {
     this.databaseHost = new DatabaseHost(feedWriter, () => this._flush());
     this.itemManager = new import_echo_db3.ItemManager(this._params.modelFactory);
     await this.databaseHost.open(this.itemManager, this._params.modelFactory);
-    (0, import_async7.scheduleTask)(this._ctx, async () => {
+    (0, import_async6.scheduleTask)(this._ctx, async () => {
       await this._consumePipeline();
     });
     this._isOpen = true;
@@ -1934,7 +1923,7 @@ var DataPipeline = class DataPipeline2 {
       span.end();
       if (++messageCounter > 1e3) {
         messageCounter = 0;
-        await (0, import_async7.sleep)(1);
+        await (0, import_async6.sleep)(1);
       }
     }
   }
@@ -2014,7 +2003,7 @@ var DataPipeline = class DataPipeline2 {
       }
     });
     this._epochCtx = ctx;
-    (0, import_async7.scheduleTask)(ctx, async () => {
+    (0, import_async6.scheduleTask)(ctx, async () => {
       if (!this._isOpen) {
         return;
       }
@@ -2128,31 +2117,31 @@ var DataPipeline = class DataPipeline2 {
     await this._params.metadataStore.flush();
   }
 };
-_ts_decorate6([
+_ts_decorate5([
   import_tracing2.trace.metricsCounter()
 ], DataPipeline.prototype, "_usage", void 0);
-_ts_decorate6([
+_ts_decorate5([
   import_tracing2.trace.metricsCounter()
 ], DataPipeline.prototype, "_mutations", void 0);
-_ts_decorate6([
-  import_async7.synchronized
+_ts_decorate5([
+  import_async6.synchronized
 ], DataPipeline.prototype, "open", null);
-_ts_decorate6([
-  import_async7.synchronized
+_ts_decorate5([
+  import_async6.synchronized
 ], DataPipeline.prototype, "close", null);
-_ts_decorate6([
-  import_async7.synchronized
+_ts_decorate5([
+  import_async6.synchronized
 ], DataPipeline.prototype, "_processEpoch", null);
-_ts_decorate6([
-  import_async7.synchronized
+_ts_decorate5([
+  import_async6.synchronized
 ], DataPipeline.prototype, "createEpoch", null);
-DataPipeline = _ts_decorate6([
-  (0, import_async7.trackLeaks)("open", "close"),
+DataPipeline = _ts_decorate5([
+  (0, import_async6.trackLeaks)("open", "close"),
   import_tracing2.trace.resource()
 ], DataPipeline);
 
 // packages/core/echo/echo-pipeline/src/space/space.ts
-function _ts_decorate7(decorators, target, key, desc) {
+function _ts_decorate6(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2165,9 +2154,9 @@ function _ts_decorate7(decorators, target, key, desc) {
 var __dxlog_file11 = "/home/runner/work/dxos/dxos/packages/core/echo/echo-pipeline/src/space/space.ts";
 var Space = class Space2 {
   constructor(params) {
-    this._addFeedLock = new import_async8.Lock();
+    this._addFeedLock = new import_async7.Lock();
     this.onCredentialProcessed = new import_util7.Callback();
-    this.stateUpdate = new import_async8.Event();
+    this.stateUpdate = new import_async7.Event();
     this._isOpen = false;
     (0, import_invariant9.invariant)(params.spaceKey && params.feedProvider, void 0, {
       F: __dxlog_file11,
@@ -2381,27 +2370,27 @@ var Space = class Space2 {
     await this._dataPipeline.open();
   }
 };
-_ts_decorate7([
+_ts_decorate6([
   import_log10.logInfo,
   import_tracing3.trace.info()
 ], Space.prototype, "key", null);
-_ts_decorate7([
-  import_async8.synchronized,
+_ts_decorate6([
+  import_async7.synchronized,
   import_tracing3.trace.span()
 ], Space.prototype, "open", null);
-_ts_decorate7([
-  import_async8.synchronized
+_ts_decorate6([
+  import_async7.synchronized
 ], Space.prototype, "close", null);
-_ts_decorate7([
-  import_async8.synchronized
+_ts_decorate6([
+  import_async7.synchronized
 ], Space.prototype, "initializeDataPipeline", null);
-Space = _ts_decorate7([
-  (0, import_async8.trackLeaks)("open", "close"),
+Space = _ts_decorate6([
+  (0, import_async7.trackLeaks)("open", "close"),
   import_tracing3.trace.resource()
 ], Space);
 
 // packages/core/echo/echo-pipeline/src/space/space-manager.ts
-var import_async9 = require("@dxos/async");
+var import_async8 = require("@dxos/async");
 var import_debug4 = require("@dxos/debug");
 var import_keys7 = require("@dxos/keys");
 var import_log12 = require("@dxos/log");
@@ -2417,7 +2406,7 @@ var import_teleport2 = require("@dxos/teleport");
 var import_teleport_extension_object_sync = require("@dxos/teleport-extension-object-sync");
 var import_teleport_extension_replicator = require("@dxos/teleport-extension-replicator");
 var import_util8 = require("@dxos/util");
-function _ts_decorate8(decorators, target, key, desc) {
+function _ts_decorate7(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2535,10 +2524,10 @@ var SpaceProtocol = class {
     };
   }
 };
-_ts_decorate8([
+_ts_decorate7([
   import_log11.logInfo
 ], SpaceProtocol.prototype, "_topic", void 0);
-_ts_decorate8([
+_ts_decorate7([
   import_log11.logInfo
 ], SpaceProtocol.prototype, "_ownPeerKey", null);
 var AuthStatus;
@@ -2607,15 +2596,15 @@ var SpaceProtocolSession = class {
     await this._teleport.abort();
   }
 };
-_ts_decorate8([
+_ts_decorate7([
   import_log11.logInfo
 ], SpaceProtocolSession.prototype, "_wireParams", void 0);
-_ts_decorate8([
+_ts_decorate7([
   import_log11.logInfo
 ], SpaceProtocolSession.prototype, "authStatus", null);
 
 // packages/core/echo/echo-pipeline/src/space/space-manager.ts
-function _ts_decorate9(decorators, target, key, desc) {
+function _ts_decorate8(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2698,14 +2687,14 @@ var SpaceManager = class SpaceManager2 {
     return space;
   }
 };
-_ts_decorate9([
-  import_async9.synchronized
+_ts_decorate8([
+  import_async8.synchronized
 ], SpaceManager.prototype, "open", null);
-_ts_decorate9([
-  import_async9.synchronized
+_ts_decorate8([
+  import_async8.synchronized
 ], SpaceManager.prototype, "close", null);
-SpaceManager = _ts_decorate9([
-  (0, import_async9.trackLeaks)("open", "close")
+SpaceManager = _ts_decorate8([
+  (0, import_async8.trackLeaks)("open", "close")
 ], SpaceManager);
 
 // packages/core/echo/echo-pipeline/src/testing/test-agent-builder.ts
@@ -2882,7 +2871,7 @@ var TestAgent = class {
 };
 
 // packages/core/echo/echo-pipeline/src/testing/util.ts
-var import_async10 = require("@dxos/async");
+var import_async9 = require("@dxos/async");
 var import_document_model2 = require("@dxos/document-model");
 var import_echo_db4 = require("@dxos/echo-db");
 var import_testing2 = require("@dxos/feed-store/testing");
@@ -2944,10 +2933,252 @@ var testLocalDatabase = async (create, check = create) => {
       ]
     }
   });
-  await (0, import_async10.asyncTimeout)(check.databaseHost._itemDemuxer.mutation.waitForCondition(() => check.itemManager.entities.has(objectId)), 2e3);
+  await (0, import_async9.asyncTimeout)(check.databaseHost._itemDemuxer.mutation.waitForCondition(() => check.itemManager.entities.has(objectId)), 2e3);
+};
+
+// packages/core/echo/echo-pipeline/src/testing/database-test-rig.ts
+var import_async10 = require("@dxos/async");
+var import_document_model3 = require("@dxos/document-model");
+var import_echo_db5 = require("@dxos/echo-db");
+var import_invariant10 = require("@dxos/invariant");
+var import_keys10 = require("@dxos/keys");
+var import_model_factory3 = require("@dxos/model-factory");
+var import_protocols8 = require("@dxos/protocols");
+var import_text_model = require("@dxos/text-model");
+var import_timeframe6 = require("@dxos/timeframe");
+var import_util11 = require("@dxos/util");
+var __dxlog_file14 = "/home/runner/work/dxos/dxos/packages/core/echo/echo-pipeline/src/testing/database-test-rig.ts";
+var SPACE_KEY = import_keys10.PublicKey.random();
+var DatabaseTestBuilder = class {
+  constructor() {
+    this.peers = new import_util11.ComplexMap(import_keys10.PublicKey.hash);
+  }
+  async createPeer() {
+    const peer = new DatabaseTestPeer(this);
+    this.peers.set(peer.key, peer);
+    await peer.open();
+    return peer;
+  }
+};
+var DatabaseTestPeer = class {
+  constructor(rig) {
+    this.rig = rig;
+    this.modelFactory = new import_model_factory3.ModelFactory().registerModel(import_document_model3.DocumentModel).registerModel(import_text_model.TextModel);
+    this.key = import_keys10.PublicKey.random();
+    this.feedMessages = [];
+    this.snapshots = /* @__PURE__ */ new Map();
+    this.confirmed = -1;
+    this.timeframe = new import_timeframe6.Timeframe();
+    this._onConfirm = new import_async10.Event();
+    this._writes = /* @__PURE__ */ new Set();
+  }
+  async open() {
+    this.hostItems = new import_echo_db5.ItemManager(this.modelFactory);
+    this.host = new DatabaseHost({
+      write: async (message, { afterWrite }) => {
+        const seq = this.feedMessages.push({
+          timeframe: this.timeframe,
+          payload: {
+            data: message
+          }
+        }) - 1;
+        const request = {
+          receipt: {
+            seq,
+            feedKey: this.key
+          },
+          options: {
+            afterWrite
+          },
+          trigger: new import_async10.Trigger()
+        };
+        this._writes.add(request);
+        await request.trigger.wait();
+        return request.receipt;
+      }
+    }, async () => {
+    });
+    await this.host.open(this.hostItems, this.modelFactory);
+    if (this.snapshot) {
+      this.host._itemDemuxer.restoreFromSnapshot(this.snapshot.database);
+    }
+    this.items = new import_echo_db5.ItemManager(this.modelFactory);
+    this.proxy = new import_echo_db5.DatabaseProxy({
+      service: this.host.createDataServiceHost(),
+      itemManager: this.items,
+      spaceKey: SPACE_KEY
+    });
+    await this.proxy.open(this.modelFactory);
+  }
+  /**
+  * Confirm mutations written to the local feed.
+  * @param seq Sequence number of the mutation to confirm. If not specified, all mutations will be confirmed.
+  */
+  async confirm(seq) {
+    this.confirmed = seq ?? this.feedMessages.length - 1;
+    this._onConfirm.emit();
+    for (const request of [
+      ...this._writes
+    ]) {
+      if (this.confirmed >= request.receipt.seq) {
+        this._writes.delete(request);
+        await request.options.afterWrite?.(request.receipt);
+        request.trigger.wake();
+      }
+    }
+    this._processMessages(import_timeframe6.Timeframe.merge(this.timeframe, new import_timeframe6.Timeframe([
+      [
+        this.key,
+        this.confirmed
+      ]
+    ])));
+  }
+  /**
+  * Replicate the database to the specified timeframe.
+  * @param to Timeframe to replicate to. If not specified, the database will be replicated to the latest timeframe (based on all other peers).
+  */
+  replicate(to) {
+    const toTimeframe = import_timeframe6.Timeframe.merge(to ?? new import_timeframe6.Timeframe(Array.from(this.rig.peers.values()).map((peer) => [
+      peer.key,
+      peer.confirmed
+    ])), this.timeframe);
+    toTimeframe.set(this.key, this.confirmed);
+    this._processMessages(toTimeframe);
+  }
+  /**
+  * Reload data from the feed. Wipes unconfirmed mutations.
+  */
+  async reload() {
+    await this.open();
+    const timeframe = this.timeframe;
+    this.timeframe = this.snapshot?.timeframe ?? new import_timeframe6.Timeframe();
+    this._processMessages(timeframe);
+  }
+  /**
+  * Create snapshot and use it for the next reload.
+  */
+  makeSnapshot() {
+    this.snapshot = {
+      spaceKey: SPACE_KEY.asUint8Array(),
+      database: this.host.createSnapshot(),
+      timeframe: this.timeframe
+    };
+    return this.snapshot;
+  }
+  createEpoch(mockSnapshot) {
+    const snapshot = this.makeSnapshot();
+    mockSnapshot && (snapshot.database = mockSnapshot);
+    const snapshotCid = import_keys10.PublicKey.from(import_protocols8.schema.getCodecForType("dxos.echo.snapshot.SpaceSnapshot").encode(snapshot)).toHex();
+    this.snapshots.set(snapshotCid, snapshot);
+    const epoch = {
+      previousId: import_keys10.PublicKey.random(),
+      timeframe: this.timeframe,
+      number: this.currentEpoch ? this.currentEpoch.number + 1 : 0,
+      snapshotCid
+    };
+    this.currentEpoch = epoch;
+    this.host._itemDemuxer.restoreFromSnapshot(snapshot.database);
+  }
+  /**
+  * Gets all candidate messages according to the current timeframe.
+  * Does not take into account the current snapshot, timeframe dependencies, or the confirmed, or replicated state.
+  */
+  _getHeads() {
+    return Array.from(this.rig.peers.values()).map((peer) => {
+      const seq = this.timeframe.get(peer.key) ?? -1;
+      const message = peer.feedMessages[seq + 1];
+      return message && {
+        feedKey: peer.key,
+        seq: seq + 1,
+        data: message
+      };
+    }).filter(import_util11.isNotNullOrUndefined);
+  }
+  _processMessages(to) {
+    let run = true;
+    while (run) {
+      run = false;
+      const heads = this._getHeads();
+      for (const candidate of heads) {
+        const toSeq = to.get(candidate.feedKey) ?? -1;
+        if (toSeq < candidate.seq) {
+          continue;
+        }
+        const snapshotSeq = this.snapshot?.timeframe?.get(candidate.feedKey) ?? -1;
+        if (candidate.seq <= snapshotSeq) {
+          continue;
+        }
+        if (!import_timeframe6.Timeframe.dependencies(candidate.data.timeframe, this.timeframe).isEmpty()) {
+          continue;
+        }
+        run = true;
+        this.host.echoProcessor({
+          batch: candidate.data.payload.data.batch,
+          meta: {
+            feedKey: candidate.feedKey,
+            seq: candidate.seq,
+            memberKey: candidate.feedKey,
+            timeframe: candidate.data.timeframe
+          }
+        });
+        this.timeframe = import_timeframe6.Timeframe.merge(this.timeframe, new import_timeframe6.Timeframe([
+          [
+            candidate.feedKey,
+            candidate.seq
+          ]
+        ]));
+      }
+    }
+  }
+  getModel(id) {
+    const item = this.items.getItem(id);
+    if (!item) {
+      return;
+    }
+    (0, import_invariant10.invariant)(item.modelMeta, void 0, {
+      F: __dxlog_file14,
+      L: 262,
+      S: this,
+      A: [
+        "item.modelMeta",
+        ""
+      ]
+    });
+    const ModelConstructor = this.modelFactory.getModel(item.modelMeta.type)?.constructor;
+    (0, import_invariant10.invariant)(ModelConstructor, void 0, {
+      F: __dxlog_file14,
+      L: 264,
+      S: this,
+      A: [
+        "ModelConstructor",
+        ""
+      ]
+    });
+    const model = new ModelConstructor(item.modelMeta, item.id, () => item.state, async (mutation) => {
+      (0, import_invariant10.invariant)(item.modelMeta, void 0, {
+        F: __dxlog_file14,
+        L: 271,
+        S: this,
+        A: [
+          "item.modelMeta",
+          ""
+        ]
+      });
+      this.proxy.mutate((0, import_echo_db5.createModelMutation)(id, (0, import_echo_db5.encodeModelMutation)(item.modelMeta, mutation)));
+      return {
+        feedKey: import_keys10.PublicKey.from("00"),
+        seq: 0,
+        waitToBeProcessed: () => Promise.resolve()
+      };
+    });
+    model.initialize();
+    return model;
+  }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  DatabaseTestBuilder,
+  DatabaseTestPeer,
   MemoryNetworkManagerProvider,
   TestAgent,
   TestAgentBuilder,