fix(tests): multiple pubsub topics (#1774)

This commit is contained in:
Danish Arora 2024-01-09 13:43:44 +05:30 committed by GitHub
parent c81872ae0e
commit 528803f4c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,5 +1,15 @@
import { LightNode, ShardInfo, SingleShardInfo } from "@waku/interfaces"; import {
import { createEncoder, createLightNode, utf8ToBytes } from "@waku/sdk"; LightNode,
Protocols,
ShardInfo,
SingleShardInfo
} from "@waku/interfaces";
import {
createEncoder,
createLightNode,
utf8ToBytes,
waitForRemotePeer
} from "@waku/sdk";
import { singleShardInfoToPubsubTopic } from "@waku/utils"; import { singleShardInfoToPubsubTopic } from "@waku/utils";
import { expect } from "chai"; import { expect } from "chai";
@ -42,6 +52,8 @@ describe("Static Sharding: Running Nodes", () => {
waku = await createLightNode({ waku = await createLightNode({
shardInfo: shardInfoBothShards shardInfo: shardInfoBothShards
}); });
await waku.dial(await nwaku.getMultiaddrWithId());
await waitForRemotePeer(waku, [Protocols.LightPush]);
const encoder1 = createEncoder({ const encoder1 = createEncoder({
contentTopic: ContentTopic, contentTopic: ContentTopic,
@ -61,8 +73,8 @@ describe("Static Sharding: Running Nodes", () => {
payload: utf8ToBytes("Hello World") payload: utf8ToBytes("Hello World")
}); });
expect(request1.recipients.length).to.eq(0); expect(request1.recipients.length).to.eq(1);
expect(request2.recipients.length).to.eq(0); expect(request2.recipients.length).to.eq(1);
}); });
it("using a protocol with unconfigured pubsub topic should fail", async function () { it("using a protocol with unconfigured pubsub topic should fail", async function () {
@ -119,6 +131,8 @@ describe("Autosharding: Running Nodes", () => {
contentTopics: [ContentTopic, ContentTopic2] contentTopics: [ContentTopic, ContentTopic2]
} }
}); });
await waku.dial(await nwaku.getMultiaddrWithId());
await waitForRemotePeer(waku, [Protocols.LightPush]);
const encoder1 = createEncoder({ const encoder1 = createEncoder({
contentTopic: ContentTopic, contentTopic: ContentTopic,
@ -138,7 +152,7 @@ describe("Autosharding: Running Nodes", () => {
payload: utf8ToBytes("Hello World") payload: utf8ToBytes("Hello World")
}); });
expect(request1.recipients.length).to.eq(0); expect(request1.recipients.length).to.eq(1);
expect(request2.recipients.length).to.eq(0); expect(request2.recipients.length).to.eq(1);
}); });
}); });