fix tests

TBD54566975 · Feb 23, 2024 · d699035 · d699035
1 parent 1c9adee
commit d699035
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tests/dwn-server.spec.ts b/tests/dwn-server.spec.ts
@@ -1,3 +1,5 @@
+import type { Dwn } from '@tbd54566975/dwn-sdk-js';
+
 import { expect } from 'chai';
 
 import { config } from '../src/config.js';
@@ -6,12 +8,12 @@ import { getTestDwn } from './test-dwn.js';
 
 describe('DwnServer', function () {
   const dwnServerConfig = { ...config };
+  let dwn: Dwn;
 
-
   it('starts with injected dwn', async function () {
-    const testDwn = await getTestDwn();
+    dwn = await getTestDwn();
 
-    const dwnServer = new DwnServer({ config: dwnServerConfig, dwn: testDwn });
+    const dwnServer = new DwnServer({ config: dwnServerConfig, dwn });
     await dwnServer.start();
 
     dwnServer.stop(() => console.log('server Stop'));
@@ -20,7 +22,9 @@ describe('DwnServer', function () {
 
   describe('webSocketServerEnabled config', function() {
     it('should not return a websocket server if disabled', async function() {
+      dwn = await getTestDwn({ withEvents: true });
       const withoutSocketServer = new DwnServer({
+        dwn,
         config: {
           ...dwnServerConfig,
           webSocketServerEnabled: false,
@@ -35,7 +39,9 @@ describe('DwnServer', function () {
     });
 
     it('should return a websocket server if enabled', async function() {
+      dwn = await getTestDwn({ withEvents: true });
       const withSocketServer = new DwnServer({
+        dwn,
         config: {
           ...dwnServerConfig,
           webSocketServerEnabled: true,