From d18a5152efc61302e0d70a0a8b2611507b47f872 Mon Sep 17 00:00:00 2001
From: Franck Royer <franck@status.im>
Date: Mon, 25 Jul 2022 17:11:36 +1000
Subject: [PATCH] fix(test): check messages on custom pubsub topic

---
 src/lib/waku_light_push/index.node.spec.ts | 2 +-
 src/test_utils/nwaku.ts                    | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/lib/waku_light_push/index.node.spec.ts b/src/lib/waku_light_push/index.node.spec.ts
index d67a59c223..1ea42c49a5 100644
--- a/src/lib/waku_light_push/index.node.spec.ts
+++ b/src/lib/waku_light_push/index.node.spec.ts
@@ -90,7 +90,7 @@ describe("Waku Light Push [node only]", () => {
     dbg("Waiting for message to show in nwaku");
     while (msgs.length === 0) {
       await delay(200);
-      msgs = await nwaku.messages();
+      msgs = await nwaku.messages(customPubSubTopic);
     }
 
     expect(msgs[0].contentTopic).to.equal(message.contentTopic);
diff --git a/src/test_utils/nwaku.ts b/src/test_utils/nwaku.ts
index 8633a3fc45..7b6c7acc35 100644
--- a/src/test_utils/nwaku.ts
+++ b/src/test_utils/nwaku.ts
@@ -217,7 +217,7 @@ export class Nwaku {
     ]);
   }
 
-  async messages(): Promise<WakuMessage[]> {
+  async messages(pubsubTopic?: string): Promise<WakuMessage[]> {
     this.checkProcess();
 
     const isDefined = (msg: WakuMessage | undefined): msg is WakuMessage => {
@@ -226,7 +226,7 @@ export class Nwaku {
 
     const protoMsgs = await this.rpcCall<proto.WakuMessage[]>(
       "get_waku_v2_relay_v1_messages",
-      [DefaultPubSubTopic]
+      [pubsubTopic ?? DefaultPubSubTopic]
     );
 
     const msgs = await Promise.all(