From 2af96281cd374ab33fb4a677fd4af5ff95acb5d1 Mon Sep 17 00:00:00 2001
From: fryorcraken <110212804+fryorcraken@users.noreply.github.com>
Date: Wed, 18 Oct 2023 22:07:11 +1100
Subject: [PATCH] test: ensure filter/light push work with message-encryption
 (#1671)

Co-authored-by: Sasha <118575614+weboko@users.noreply.github.com>
---
 .../tests/tests/filter/subscribe.node.spec.ts | 44 +++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/packages/tests/tests/filter/subscribe.node.spec.ts b/packages/tests/tests/filter/subscribe.node.spec.ts
index 9bde81f753..ce10cc6ca0 100644
--- a/packages/tests/tests/filter/subscribe.node.spec.ts
+++ b/packages/tests/tests/filter/subscribe.node.spec.ts
@@ -6,6 +6,7 @@ import {
 } from "@waku/core";
 import type { IFilterSubscription, LightNode } from "@waku/interfaces";
 import { Protocols } from "@waku/interfaces";
+import { ecies, symmetric } from "@waku/message-encryption";
 import { utf8ToBytes } from "@waku/utils/bytes";
 import { expect } from "chai";
 
@@ -65,6 +66,49 @@ describe("Waku Filter V2: Subscribe", function () {
     expect((await nwaku.messages()).length).to.eq(1);
   });
 
+  it("Subscribe and receive ecies encrypted messages via lightPush", async function () {
+    const privateKey = ecies.generatePrivateKey();
+    const publicKey = ecies.getPublicKey(privateKey);
+    const encoder = ecies.createEncoder({
+      contentTopic: TestContentTopic,
+      publicKey
+    });
+    const decoder = ecies.createDecoder(TestContentTopic, privateKey);
+
+    await subscription.subscribe([decoder], messageCollector.callback);
+
+    await waku.lightPush.send(encoder, messagePayload);
+
+    expect(await messageCollector.waitForMessages(1)).to.eq(true);
+    messageCollector.verifyReceivedMessage(0, {
+      expectedMessageText: messageText,
+      expectedContentTopic: TestContentTopic,
+      expectedVersion: 1
+    });
+    expect((await nwaku.messages()).length).to.eq(1);
+  });
+
+  it("Subscribe and receive symmetrically encrypted messages via lightPush", async function () {
+    const symKey = symmetric.generateSymmetricKey();
+    const encoder = symmetric.createEncoder({
+      contentTopic: TestContentTopic,
+      symKey
+    });
+    const decoder = symmetric.createDecoder(TestContentTopic, symKey);
+
+    await subscription.subscribe([decoder], messageCollector.callback);
+
+    await waku.lightPush.send(encoder, messagePayload);
+
+    expect(await messageCollector.waitForMessages(1)).to.eq(true);
+    messageCollector.verifyReceivedMessage(0, {
+      expectedMessageText: messageText,
+      expectedContentTopic: TestContentTopic,
+      expectedVersion: 1
+    });
+    expect((await nwaku.messages()).length).to.eq(1);
+  });
+
   it("Subscribe and receive messages via waku relay post", async function () {
     await subscription.subscribe([TestDecoder], messageCollector.callback);