Merge branch 'main' into bump-v102.0.0-beta.0

streamr-dev · Oct 24, 2024 · 765a69a · 765a69a
2 parents f985d5c + be66d55
commit 765a69a
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/test/end-to-end.test.ts b/test/end-to-end.test.ts
@@ -3,7 +3,7 @@ import 'reflect-metadata'
 import { DhtAddress, NodeType, randomDhtAddress, toDhtAddress, toDhtAddressRaw } from '@streamr/dht'
 import StreamrClient, { NetworkNodeType, PeerDescriptor, StreamID, StreamPermission, StreamrClientConfig } from '@streamr/sdk'
 import { NetworkNode, createNetworkNode } from '@streamr/trackerless-network'
-import { StreamPartID, setAbortableInterval, toStreamPartID, waitForCondition } from '@streamr/utils'
+import { StreamPartID, collect, setAbortableInterval, toStreamPartID, waitForCondition } from '@streamr/utils'
 import { sample, uniq, without } from 'lodash'
 import Container from 'typedi'
 import { CONFIG_TOKEN } from '../src/Config'
@@ -63,11 +63,9 @@ const createClientConfig = (entryPointPeerDescriptor: PeerDescriptor): StreamrCl
     }
 }
 
-const createClient = (privateKey: string, entryPointPeerDescriptor: PeerDescriptor) => {
+const createClient = (privateKey: string | undefined, entryPointPeerDescriptor: PeerDescriptor) => {
     return new StreamrClient({
-        auth: {
-            privateKey
-        },
+        auth: (privateKey !== undefined) ? { privateKey } : undefined,
         ...createClientConfig(entryPointPeerDescriptor)
     })
 }
@@ -177,6 +175,16 @@ describe('end-to-end', () => {
         }))
     }
 
+    const waitForTheGraphToIndex = async (streamIds: StreamID[]): Promise<void> => {
+        const client = createClient(undefined, entryPoint.getPeerDescriptor())
+        for (const streamId of streamIds) {
+            await waitForCondition(async () => {
+                const streams = await collect(client.searchStreams(streamId, undefined))
+                return streams.length > 0
+            }, 5000, 500)
+        }
+    }
+
     beforeAll(async () => {
         entryPoint = await startEntryPoint()
         const config = {
@@ -221,6 +229,7 @@ describe('end-to-end', () => {
         await startPublisherAndSubscriberForStream(privateStream.id, publishingAbortControler.signal)
         const publicStream = await createTestStream(true)
         await startPublisherAndSubscriberForStream(publicStream.id, publishingAbortControler.signal)
+        await waitForTheGraphToIndex([privateStream.id, publicStream.id])
 
         crawler = Container.get(Crawler)
         await crawler.start(1)