@streamr/dht 100.0.0-pretestnet.3 → 100.0.0-pretestnet.6

package/test/integration/StoreAndDelete.test.ts

@@ -1,6 +1,6 @@
 import { LatencyType, Simulator } from '../../src/connection/simulator/Simulator'
 import { DhtNode } from '../../src/dht/DhtNode'
-import { NodeType, PeerDescriptor } from '../../src/proto/packages/dht/protos/DhtRpc'
+import { PeerDescriptor } from '../../src/proto/packages/dht/protos/DhtRpc'
 import { createMockConnectionDhtNode, waitConnectionManagersReadyForTesting } from '../utils/utils'
 import { PeerID } from '../../src/helpers/PeerID'
 import { areEqualPeerDescriptors } from '../../src/helpers/peerIdFromPeerDescriptor'
@@ -10,7 +10,7 @@ describe('Storing data in DHT', () => {
     let entryPoint: DhtNode
     let nodes: DhtNode[]
     let entrypointDescriptor: PeerDescriptor
-    const simulator = new Simulator(LatencyType.RANDOM)
+    const simulator = new Simulator(LatencyType.REAL)
     const NUM_NODES = 5
     const MAX_CONNECTIONS = 5
     const K = 4
@@ -27,10 +27,7 @@ describe('Storing data in DHT', () => {
             undefined, K, MAX_CONNECTIONS)
         nodes.push(entryPoint)
         nodeIndicesById[entryPoint.getNodeId().toKey()] = 0
-        entrypointDescriptor = {
-            kademliaId: entryPoint.getNodeId().value,
-            type: NodeType.NODEJS
-        }
+        entrypointDescriptor = entryPoint.getLocalPeerDescriptor()
         nodes.push(entryPoint)
         for (let i = 1; i < NUM_NODES; i++) {
             const nodeId = `${i}`

package/test/integration/WebrtcConnectionManagement.test.ts

@@ -200,7 +200,7 @@ describe('WebRTC Connection Management', () => {
 
         msg.targetDescriptor = peerDescriptor2
         manager1.send(msg).catch((e) => {
-            expect(e.code).toEqual('CONNECTION_FAILED')
+            expect(e.code).toEqual('SEND_FAILED')
         })
 
         // @ts-expect-error private field

package/test/utils/utils.ts

@@ -57,14 +57,20 @@ export const createMockConnectionDhtNode = async (
     }
     const mockConnectionManager = new SimulatorTransport(peerDescriptor, simulator)
     await mockConnectionManager.start()
-    const node = new DhtNode({
+    const opts = {
         peerDescriptor: peerDescriptor,
         transport: mockConnectionManager,
         numberOfNodesPerKBucket,
         maxConnections: maxConnections,
         dhtJoinTimeout,
         rpcRequestTimeout: 5000
-    })
+    }
+    const node = new class extends DhtNode {
+        async stop(): Promise<void> {
+            await super.stop()
+            await mockConnectionManager.stop()
+        }
+    }(opts)
     await node.start()
     return node
 }