@dxos/client-services 0.6.12-staging.e11e696 → 0.6.12

package/src/packlets/spaces/edge-feed-replicator.ts

@@ -5,19 +5,15 @@
 import { decode as decodeCbor, encode as encodeCbor } from 'cbor-x';
 
 import { Event, Mutex, scheduleMicroTask } from '@dxos/async';
-import { Context, Resource } from '@dxos/context';
+import { Resource, type Context } from '@dxos/context';
 import { type EdgeConnection } from '@dxos/edge-client';
-import { EdgeConnectionClosedError, EdgeIdentityChangedError } from '@dxos/edge-client';
 import { type FeedWrapper } from '@dxos/feed-store';
 import { invariant } from '@dxos/invariant';
 import { PublicKey, type SpaceId } from '@dxos/keys';
-import { log, logInfo } from '@dxos/log';
+import { log } from '@dxos/log';
 import { EdgeService } from '@dxos/protocols';
 import { buf } from '@dxos/protocols/buf';
-import {
-  MessageSchema as RouterMessageSchema,
-  type Message as RouterMessage,
-} from '@dxos/protocols/buf/dxos/edge/messenger_pb';
+import { MessageSchema as RouterMessageSchema } from '@dxos/protocols/buf/dxos/edge/messenger_pb';
 import type { FeedBlock, ProtocolMessage } from '@dxos/protocols/feed-replication';
 import { ComplexMap, arrayToBuffer, bufferToArray, defaultMap, rangeFromTo } from '@dxos/util';
 
@@ -28,10 +24,7 @@ export type EdgeFeedReplicatorParams = {
 
 export class EdgeFeedReplicator extends Resource {
   private readonly _messenger: EdgeConnection;
-
-  @logInfo
   private readonly _spaceId: SpaceId;
-
   private readonly _feeds = new ComplexMap<PublicKey, FeedWrapper<any>>(PublicKey.hash);
 
   private _connectionCtx?: Context = undefined;
@@ -53,10 +46,9 @@ export class EdgeFeedReplicator extends Resource {
   }
 
   protected override async _open(): Promise<void> {
-    log('open');
     // TODO: handle reconnects
     this._ctx.onDispose(
-      this._messenger.addListener((message: RouterMessage) => {
+      this._messenger.addListener(async (message) => {
         if (!message.serviceId) {
           return;
         }
@@ -72,40 +64,21 @@ export class EdgeFeedReplicator extends Resource {
         }
 
         const payload = decodeCbor(message.payload!.value) as ProtocolMessage;
-        log('receive', { from: message.source, feedKey: payload.feedKey, type: payload.type });
+        log.info('recv', { from: message.source, payload });
         this._onMessage(payload);
       }),
     );
 
-    this._messenger.connected.on(this._ctx, async () => {
-      await this._resetConnection();
-      this._startReplication();
-    });
-
-    if (this._messenger.isConnected) {
-      this._startReplication();
+    this._connected = true;
+    this._connectionCtx = this._ctx.derive();
+    for (const feed of this._feeds.values()) {
+      await this._replicateFeed(feed);
     }
   }
 
   protected override async _close(): Promise<void> {
-    log('close');
-    await this._resetConnection();
-  }
-
-  private _startReplication() {
-    this._connected = true;
-    const connectionCtx = this._createConnectionContext();
-    this._connectionCtx = connectionCtx;
-    log('connection context created');
-    scheduleMicroTask(connectionCtx, async () => {
-      for (const feed of this._feeds.values()) {
-        await this._replicateFeed(connectionCtx, feed);
-      }
-    });
-  }
+    this._connected = false;
 
-  private async _resetConnection() {
-    log('resetConnection');
     this._connected = false;
     await this._connectionCtx?.dispose();
     this._connectionCtx = undefined;
@@ -113,11 +86,11 @@ export class EdgeFeedReplicator extends Resource {
   }
 
   async addFeed(feed: FeedWrapper<any>) {
-    log.info('addFeed', { key: feed.key, connected: this._connected, hasConnectionCtx: !!this._connectionCtx });
+    log.info('addFeed', { key: feed.key });
     this._feeds.set(feed.key, feed);
 
-    if (this._connected && this._connectionCtx) {
-      await this._replicateFeed(this._connectionCtx, feed);
+    if (this._connected) {
+      await this._replicateFeed(feed);
     }
   }
 
@@ -125,32 +98,25 @@ export class EdgeFeedReplicator extends Resource {
     return defaultMap(this._pushMutex, key, () => new Mutex());
   }
 
-  private async _replicateFeed(ctx: Context, feed: FeedWrapper<any>) {
-    log('replicateFeed', { key: feed.key });
+  private async _replicateFeed(feed: FeedWrapper<any>) {
+    invariant(this._connectionCtx);
+
     await this._sendMessage({
       type: 'get-metadata',
       feedKey: feed.key.toHex(),
     });
 
-    Event.wrap(feed.core as any, 'append').on(ctx, async () => {
+    Event.wrap(feed.core as any, 'append').on(this._connectionCtx, async () => {
       await this._pushBlocksIfNeeded(feed);
     });
   }
 
   private async _sendMessage(message: ProtocolMessage) {
-    if (!this._connectionCtx) {
-      log.info('message dropped because connection was disposed');
-      return;
-    }
-
-    const logPayload =
-      message.type === 'data' ? { feedKey: message.feedKey, blocks: message.blocks.map((b) => b.index) } : { message };
-    log.info('sending message', logPayload);
+    log.info('sending message', { message });
 
     invariant(message.feedKey);
     const payloadValue = bufferToArray(encodeCbor(message));
 
-    log('send', { type: message.type });
     await this._messenger.send(
       buf.create(RouterMessageSchema, {
         source: {
@@ -164,15 +130,11 @@ export class EdgeFeedReplicator extends Resource {
   }
 
   private _onMessage(message: ProtocolMessage) {
-    if (!this._connectionCtx) {
-      log.warn('received message after connection context was disposed');
-      return;
-    }
-    scheduleMicroTask(this._connectionCtx, async () => {
+    log.info('received message', { message });
+
+    scheduleMicroTask(this._ctx, async () => {
       switch (message.type) {
         case 'metadata': {
-          log.info('received metadata', { message });
-
           const feedKey = PublicKey.fromHex(message.feedKey);
           const feed = this._feeds.get(feedKey);
           if (!feed) {
@@ -198,8 +160,6 @@ export class EdgeFeedReplicator extends Resource {
         }
 
         case 'data': {
-          log.info('received data', { feed: message.feedKey, blocks: message.blocks.map((b) => b.index) });
-
           const feedKey = PublicKey.fromHex(message.feedKey);
           const feed = this._feeds.get(feedKey);
           if (!feed) {
@@ -263,10 +223,9 @@ export class EdgeFeedReplicator extends Resource {
   }
 
   private async _pushBlocksIfNeeded(feed: FeedWrapper<any>) {
-    using _ = await this._getPushMutex(feed.key).acquire();
+    using _guard = await this._getPushMutex(feed.key).acquire();
 
     if (!this._remoteLength.has(feed.key)) {
-      log('blocks not pushed because remote length is unknown');
       return;
     }
 
@@ -275,23 +234,6 @@ export class EdgeFeedReplicator extends Resource {
       await this._pushBlocks(feed, remoteLength, feed.length);
     }
   }
-
-  private _createConnectionContext() {
-    const connectionCtx = new Context({
-      onError: async (err: any) => {
-        if (connectionCtx !== this._connectionCtx) {
-          return;
-        }
-        if (err instanceof EdgeIdentityChangedError || err instanceof EdgeConnectionClosedError) {
-          log('resetting on reconnect');
-          await this._resetConnection();
-        } else {
-          this._ctx.raise(err);
-        }
-      },
-    });
-    return connectionCtx;
-  }
 }
 
 // hypercore requires buffers

package/src/packlets/spaces/epoch-migrations.ts

@@ -4,13 +4,13 @@
 
 import type { AutomergeUrl } from '@dxos/automerge/automerge-repo';
 import { type Context } from '@dxos/context';
-import { migrateDocument } from '@dxos/echo-db';
 import {
   convertLegacyReferences,
   convertLegacySpaceRootDoc,
   findInlineObjectOfType,
+  migrateDocument,
   type EchoHost,
-} from '@dxos/echo-pipeline';
+} from '@dxos/echo-db';
 import { SpaceDocVersion, type SpaceDoc } from '@dxos/echo-protocol';
 import { TYPE_PROPERTIES } from '@dxos/echo-schema';
 import { invariant } from '@dxos/invariant';

package/src/packlets/spaces/notarization-plugin.test.ts

@@ -2,28 +2,27 @@
 // Copyright 2023 DXOS.org
 //
 
-import { onTestFinished, describe, expect, test } from 'vitest';
+import { expect } from 'chai';
 
 import { Context } from '@dxos/context';
 import { CredentialGenerator } from '@dxos/credentials';
 import { MockFeedWriter } from '@dxos/feed-store/testing';
 import { Keyring } from '@dxos/keyring';
-import { SpaceId } from '@dxos/keys';
 import { log } from '@dxos/log';
 import { AdmittedFeed, type Credential } from '@dxos/protocols/proto/dxos/halo/credentials';
 import { TestBuilder, type TestConnection, TestPeer } from '@dxos/teleport/testing';
+import { afterTest, describe, test } from '@dxos/test';
 
-import { NotarizationPlugin, type NotarizationPluginParams } from './notarization-plugin';
+import { NotarizationPlugin } from './notarization-plugin';
 
 class TestAgent extends TestPeer {
   private readonly _ctx = new Context();
 
   feed = new MockFeedWriter<Credential>();
-  notarizationPlugin: NotarizationPlugin;
+  notarizationPlugin = new NotarizationPlugin();
 
-  constructor(params: NotarizationPluginParams) {
+  constructor() {
     super();
-    this.notarizationPlugin = new NotarizationPlugin(params);
     this.feed.written.on(this._ctx, async ([credential]) => {
       log('written to feed', { credential });
       await this.notarizationPlugin.processCredential(credential);
@@ -50,12 +49,10 @@ class TestAgent extends TestPeer {
 describe('NotarizationPlugin', () => {
   test('notarize single credential', async () => {
     const testBuilder = new TestBuilder();
-    onTestFinished(() => testBuilder.destroy());
-
-    const params = { spaceId: SpaceId.random() };
+    afterTest(() => testBuilder.destroy());
 
     // peer0 is there to test retries.
-    const [_peer0, peer1, peer2] = await testBuilder.createPeers({ factory: () => new TestAgent(params) });
+    const [_peer0, peer1, peer2] = await testBuilder.createPeers({ factory: () => new TestAgent() });
     peer1.enableWriting();
 
     peer1.feed.written.on(async ([credential]) => {

package/src/packlets/spaces/notarization-plugin.ts

@@ -2,18 +2,14 @@
 // Copyright 2023 DXOS.org
 //
 
-import { DeferredTask, Event, scheduleTask, sleep, TimeoutError, Trigger, scheduleMicroTask } from '@dxos/async';
-import { type Context, rejectOnDispose, Resource } from '@dxos/context';
-import { type CredentialProcessor, verifyCredential } from '@dxos/credentials';
-import { type EdgeHttpClient } from '@dxos/edge-client';
+import { DeferredTask, Event, scheduleTask, sleep, TimeoutError, Trigger } from '@dxos/async';
+import { Context, rejectOnDispose } from '@dxos/context';
+import { type CredentialProcessor } from '@dxos/credentials';
 import { type FeedWriter } from '@dxos/feed-store';
 import { invariant } from '@dxos/invariant';
 import { PublicKey } from '@dxos/keys';
-import { type SpaceId } from '@dxos/keys';
-import { logInfo, log } from '@dxos/log';
-import { EdgeCallFailedError } from '@dxos/protocols';
+import { log } from '@dxos/log';
 import { schema } from '@dxos/protocols/proto';
-import { type Runtime } from '@dxos/protocols/proto/dxos/config';
 import { type Credential } from '@dxos/protocols/proto/dxos/halo/credentials';
 import { type NotarizationService, type NotarizeRequest } from '@dxos/protocols/proto/dxos/mesh/teleport/notarization';
 import { type ExtensionContext, RpcExtension } from '@dxos/teleport';
@@ -25,18 +21,8 @@ const DEFAULT_SUCCESS_DELAY = 1_000;
 
 const DEFAULT_NOTARIZE_TIMEOUT = 10_000;
 
-const MAX_EDGE_RETRIES = 2;
-
 const WRITER_NOT_SET_ERROR_CODE = 'WRITER_NOT_SET';
 
-const credentialCodec = schema.getCodecForType('dxos.halo.credentials.Credential');
-
-export type NotarizationPluginParams = {
-  spaceId: SpaceId;
-  edgeClient?: EdgeHttpClient;
-  edgeFeatures?: Runtime.Client.EdgeFeatures;
-};
-
 export type NotarizeParams = {
   /**
    * For cancellation.
@@ -67,17 +53,13 @@ export type NotarizeParams = {
    * @default {@link DEFAULT_SUCCESS_DELAY}
    */
   successDelay?: number;
-
-  /**
-   * A random amount of time before making or retrying an edge request to help prevent large bursts of requests.
-   */
-  edgeRetryJitter?: number;
 };
 
 /**
  * See NotarizationService proto.
  */
-export class NotarizationPlugin extends Resource implements CredentialProcessor {
+export class NotarizationPlugin implements CredentialProcessor {
+  private readonly _ctx = new Context();
   private readonly _extensionOpened = new Event();
 
   private _writer: FeedWriter<Credential> | undefined;
@@ -85,30 +67,13 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
   private readonly _processedCredentials = new ComplexSet<PublicKey>(PublicKey.hash);
   private readonly _processCredentialsTriggers = new ComplexMap<PublicKey, Trigger>(PublicKey.hash);
 
-  @logInfo
-  private readonly _spaceId: SpaceId;
-
-  private readonly _edgeClient: EdgeHttpClient | undefined;
-
-  constructor(params: NotarizationPluginParams) {
-    super();
-    this._spaceId = params.spaceId;
-    if (params.edgeClient && params.edgeFeatures?.feedReplicator) {
-      this._edgeClient = params.edgeClient;
-    }
-  }
-
   get hasWriter() {
     return !!this._writer;
   }
 
-  protected override async _open() {
-    if (this._edgeClient && this._writer) {
-      this._notarizePendingEdgeCredentials(this._edgeClient, this._writer);
-    }
-  }
+  async open() {}
 
-  protected override async _close() {
+  async close() {
     await this._ctx.dispose();
   }
 
@@ -121,7 +86,6 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
     timeout = DEFAULT_NOTARIZE_TIMEOUT,
     retryTimeout = DEFAULT_RETRY_TIMEOUT,
     successDelay = DEFAULT_SUCCESS_DELAY,
-    edgeRetryJitter,
   }: NotarizeParams) {
     log('notarize', { credentials });
     invariant(
@@ -139,35 +103,24 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
     });
     opCtx?.onDispose(() => ctx.dispose());
 
+    // Timeout/
     if (timeout !== 0) {
-      this._scheduleTimeout(ctx, errors, timeout);
+      scheduleTask(
+        ctx,
+        () => {
+          log.warn('Notarization timeout', {
+            timeout,
+            peers: Array.from(this._extensions).map((extension) => extension.remotePeerId),
+          });
+          void ctx.dispose();
+          errors.throw(new TimeoutError(timeout, 'Notarization timed out'));
+        },
+        timeout,
+      );
     }
 
     const allNotarized = Promise.all(credentials.map((credential) => this._waitUntilProcessed(credential.id!)));
 
-    this._tryNotarizeCredentialsWithPeers(ctx, credentials, { retryTimeout, successDelay });
-
-    if (this._edgeClient) {
-      this._tryNotarizeCredentialsWithEdge(ctx, this._edgeClient, credentials, {
-        retryTimeout,
-        successDelay,
-        jitter: edgeRetryJitter,
-      });
-    }
-
-    try {
-      await Promise.race([rejectOnDispose(ctx), allNotarized, errors.wait()]);
-      log('done');
-    } finally {
-      await ctx.dispose();
-    }
-  }
-
-  private _tryNotarizeCredentialsWithPeers(
-    ctx: Context,
-    credentials: Credential[],
-    { retryTimeout, successDelay }: NotarizationTimeouts,
-  ) {
     const peersTried = new Set<NotarizationTeleportExtension>();
 
     // Repeatable task that tries to notarize credentials with one of the available peers.
@@ -192,7 +145,6 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
           credentials: credentials.filter((credential) => !this._processedCredentials.has(credential.id!)),
         });
         log('success');
-
         await sleep(successDelay); // wait before trying with a new peer
       } catch (err: any) {
         if (!ctx.disposed && !err.message.includes(WRITER_NOT_SET_ERROR_CODE)) {
@@ -204,31 +156,13 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
 
     notarizeTask.schedule();
     this._extensionOpened.on(ctx, () => notarizeTask.schedule());
-  }
 
-  private _tryNotarizeCredentialsWithEdge(
-    ctx: Context,
-    client: EdgeHttpClient,
-    credentials: Credential[],
-    timeouts: NotarizationTimeouts & { jitter?: number },
-  ) {
-    const encodedCredentials = credentials.map((credential) => {
-      const binary = credentialCodec.encode(credential);
-      return Buffer.from(binary).toString('base64');
-    });
-    scheduleTask(ctx, async () => {
-      try {
-        await client.notarizeCredentials(
-          this._spaceId,
-          { credentials: encodedCredentials },
-          { retry: { count: MAX_EDGE_RETRIES, timeout: timeouts.retryTimeout, jitter: timeouts.jitter } },
-        );
-
-        log('edge notarization success');
-      } catch (error: any) {
-        handleEdgeError(error);
-      }
-    });
+    try {
+      await Promise.race([rejectOnDispose(ctx), allNotarized, errors.wait()]);
+      log('done');
+    } finally {
+      await ctx.dispose();
+    }
   }
 
   /**
@@ -246,44 +180,6 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
   setWriter(writer: FeedWriter<Credential>) {
     invariant(!this._writer, 'Writer already set.');
     this._writer = writer;
-    if (this._edgeClient) {
-      this._notarizePendingEdgeCredentials(this._edgeClient, writer);
-    }
-  }
-
-  /**
-   * The method is used only for adding agent feeds to spaces.
-   * When an agent is created we can admit them into all the existing spaces. In case the operation fails
-   * this method will fix it on the next space open.
-   * Given how rarely this happens there's no need to poll the endpoint.
-   */
-  private _notarizePendingEdgeCredentials(client: EdgeHttpClient, writer: FeedWriter<Credential>) {
-    scheduleMicroTask(this._ctx, async () => {
-      try {
-        const response = await client.getCredentialsForNotarization(this._spaceId, {
-          retry: { count: MAX_EDGE_RETRIES },
-        });
-
-        const credentials = response.awaitingNotarization.credentials;
-        if (!credentials.length) {
-          log('edge did not return credentials for notarization');
-          return;
-        }
-
-        log('got edge credentials for notarization', { count: credentials.length });
-
-        const decodedCredentials = credentials.map((credential) => {
-          const binary = Buffer.from(credential, 'base64');
-          return credentialCodec.decode(binary);
-        });
-
-        await this._notarizeCredentials(writer, decodedCredentials);
-
-        log.info('notarized edge credentials', { count: decodedCredentials.length });
-      } catch (error: any) {
-        handleEdgeError(error);
-      }
-    });
   }
 
   private async _waitUntilProcessed(id: PublicKey) {
@@ -300,20 +196,12 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
     if (!this._writer) {
       throw new Error(WRITER_NOT_SET_ERROR_CODE);
     }
-    await this._notarizeCredentials(this._writer, request.credentials ?? []);
-  }
-
-  private async _notarizeCredentials(writer: FeedWriter<Credential>, credentials: Credential[]) {
-    for (const credential of credentials) {
+    for (const credential of request.credentials ?? []) {
       invariant(credential.id, 'Credential must have an id');
       if (this._processedCredentials.has(credential.id)) {
         continue;
       }
-      const verificationResult = await verifyCredential(credential);
-      if (verificationResult.kind === 'fail') {
-        throw new Error(`Credential verification failed: ${verificationResult.errors.join('\n')}.`);
-      }
-      await writer.write(credential);
+      await this._writer.write(credential);
     }
   }
 
@@ -332,31 +220,8 @@ export class NotarizationPlugin extends Resource implements CredentialProcessor
     });
     return extension;
   }
-
-  private _scheduleTimeout(ctx: Context, errors: Trigger, timeout: number) {
-    scheduleTask(
-      ctx,
-      () => {
-        log.warn('Notarization timeout', {
-          timeout,
-          peers: Array.from(this._extensions).map((extension) => extension.remotePeerId),
-        });
-        void ctx.dispose();
-        errors.throw(new TimeoutError(timeout, 'Notarization timed out'));
-      },
-      timeout,
-    );
-  }
 }
 
-const handleEdgeError = (error: any) => {
-  if (!(error instanceof EdgeCallFailedError) || error.errorData) {
-    log.catch(error);
-  } else {
-    log.info('Edge notarization failure', { reason: error.reason });
-  }
-};
-
 export type NotarizationTeleportExtensionParams = {
   onOpen: () => Promise<void>;
   onClose: () => Promise<void>;
@@ -396,11 +261,6 @@ export class NotarizationTeleportExtension extends RpcExtension<Services, Servic
   }
 }
 
-type NotarizationTimeouts = {
-  retryTimeout: number;
-  successDelay: number;
-};
-
 type Services = {
   NotarizationService: NotarizationService;
 };

package/src/packlets/spaces/spaces-service.test.ts

@@ -2,17 +2,21 @@
 // Copyright 2023 DXOS.org
 //
 
-import { afterEach, onTestFinished, beforeEach, describe, expect, test } from 'vitest';
+import chai, { expect } from 'chai';
+import chaiAsPromised from 'chai-as-promised';
 
 import { Trigger } from '@dxos/async';
 import { Context } from '@dxos/context';
 import { PublicKey } from '@dxos/keys';
 import { type Space, type SpacesService } from '@dxos/protocols/proto/dxos/client/services';
+import { afterEach, afterTest, beforeEach, describe, test } from '@dxos/test';
 
 import { SpacesServiceImpl } from './spaces-service';
 import { type ServiceContext } from '../services';
 import { createServiceContext } from '../testing';
 
+chai.use(chaiAsPromised);
+
 describe('SpacesService', () => {
   let serviceContext: ServiceContext;
   let spacesService: SpacesService;
@@ -32,7 +36,7 @@ describe('SpacesService', () => {
 
   describe('createSpace', () => {
     test('fails if no identity is available', async () => {
-      await expect(spacesService.createSpace()).rejects.toBeInstanceOf(Error);
+      await expect(spacesService.createSpace()).to.be.rejectedWith();
     });
 
     test('creates a new space', async () => {
@@ -52,7 +56,7 @@ describe('SpacesService', () => {
       query.subscribe(({ spaces }) => {
         result.wake(spaces);
       });
-      onTestFinished(() => query.close());
+      afterTest(() => query.close());
       expect(await result.wait()).to.be.length(0);
     });
 
@@ -69,7 +73,7 @@ describe('SpacesService', () => {
       query.subscribe(({ spaces }) => {
         result.wake(spaces);
       });
-      onTestFinished(() => query.close());
+      afterTest(() => query.close());
 
       const spaces = await result.wait();
       expect(spaces).to.be.length(3);
@@ -83,7 +87,7 @@ describe('SpacesService', () => {
       query.subscribe(({ spaces }) => {
         result.wake(spaces);
       });
-      onTestFinished(() => query.close());
+      afterTest(() => query.close());
       expect(await result.wait()).to.be.length(0);
 
       result.reset();

package/src/packlets/storage/storage.ts

@@ -14,6 +14,7 @@ import { getRootPath } from './util';
 // TODO(burdon): Factor out.
 export const createStorageObjects = (config: Runtime.Client.Storage) => {
   const { persistent = false, keyStore, dataStore } = config ?? {};
+
   if (persistent && dataStore === StorageDriver.RAM) {
     throw new InvalidConfigError('RAM storage cannot be used in persistent mode.');
   }

package/src/packlets/system/system-service.test.ts

@@ -2,11 +2,12 @@
 // Copyright 2023 DXOS.org
 //
 
-import { beforeEach, describe, expect, test } from 'vitest';
+import { expect } from 'chai';
 
 import { Event, Trigger } from '@dxos/async';
 import { Config } from '@dxos/config';
 import { type SystemService, SystemStatus, type QueryStatusResponse } from '@dxos/protocols/proto/dxos/client/services';
+import { beforeEach, describe, test } from '@dxos/test';
 
 import { SystemServiceImpl } from './system-service';
 

package/src/packlets/testing/test-builder.ts

@@ -6,7 +6,8 @@ import { type Config } from '@dxos/config';
 import { Context } from '@dxos/context';
 import { createCredentialSignerWithChain, CredentialGenerator } from '@dxos/credentials';
 import { failUndefined } from '@dxos/debug';
-import { EchoHost, MetadataStore, SpaceManager, valueEncoding, MeshEchoReplicator } from '@dxos/echo-pipeline';
+import { EchoHost } from '@dxos/echo-db';
+import { MetadataStore, SpaceManager, valueEncoding, MeshEchoReplicator } from '@dxos/echo-pipeline';
 import { FeedFactory, FeedStore } from '@dxos/feed-store';
 import { Keyring } from '@dxos/keyring';
 import { type LevelDB } from '@dxos/kv-store';
@@ -53,7 +54,7 @@ export const createServiceContext = async ({
   const level = createTestLevel();
   await level.open();
 
-  return new ServiceContext(storage, level, networkManager, signalManager, undefined, undefined, {
+  return new ServiceContext(storage, level, networkManager, signalManager, undefined, {
     invitationConnectionDefaultParams: { controlHeartbeatInterval: 200 },
     ...runtimeParams,
   });