feat(lib): add tests to moderation events

pubnub · Dec 11, 2023 · 5ae6aba · 5ae6aba
1 parent d35a261
commit 5ae6aba
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 3 deletions.
diff --git a/lib/src/entities/chat.ts b/lib/src/entities/chat.ts
@@ -1095,7 +1095,7 @@ export class Chat {
         channel: userId,
         payload: {
           channelId: channel,
-          action: "restrictionsLifted",
+          restriction: "lifted",
         },
       })
     } else {
@@ -1105,7 +1105,7 @@ export class Chat {
         channel: userId,
         payload: {
           channelId: channel,
-          action: params.ban ? "banned" : "muted",
+          restriction: params.ban ? "banned" : "muted",
         },
       })
     }

diff --git a/lib/src/types.ts b/lib/src/types.ts
@@ -89,7 +89,7 @@ type InviteEventPayload = {
 }
 type ModerationEventPayload = {
   channelId: string
-  action: "muted" | "banned" | "restrictionsLifted"
+  restriction: "muted" | "banned" | "lifted"
 }
 type CustomEventPayload = any
 

diff --git a/lib/tests/channel.test.ts b/lib/tests/channel.test.ts
@@ -1069,4 +1069,63 @@ describe("Channel test", () => {
 
     await publicChannel.delete()
   })
+
+  test("should set (or lift) restrictions on a user", async () => {
+    const moderationEventCallback = jest.fn()
+
+    const removeModerationListener = chat.listenForEvents({
+      channel: chat.currentUser.id,
+      type: "moderation",
+      callback: moderationEventCallback,
+    })
+
+    await chat.setRestrictions(chat.currentUser.id, "some-channel", { mute: true })
+    await sleep(150) // Wait for the message to be sent and cached
+    expect(moderationEventCallback).toHaveBeenCalledWith(
+      expect.objectContaining({
+        payload: {
+          channelId: "PUBNUB_INTERNAL_MODERATION_some-channel",
+          restriction: "muted",
+        },
+      })
+    )
+    moderationEventCallback.mockReset()
+
+    await chat.setRestrictions(chat.currentUser.id, "some-channel", { ban: true })
+    await sleep(150) // Wait for the message to be sent and cached
+    expect(moderationEventCallback).toHaveBeenCalledWith(
+      expect.objectContaining({
+        payload: {
+          channelId: "PUBNUB_INTERNAL_MODERATION_some-channel",
+          restriction: "banned",
+        },
+      })
+    )
+    moderationEventCallback.mockReset()
+
+    await chat.setRestrictions(chat.currentUser.id, "some-channel", { ban: true, mute: true })
+    await sleep(150) // Wait for the message to be sent and cached
+    expect(moderationEventCallback).toHaveBeenCalledWith(
+      expect.objectContaining({
+        payload: {
+          channelId: "PUBNUB_INTERNAL_MODERATION_some-channel",
+          restriction: "banned",
+        },
+      })
+    )
+    moderationEventCallback.mockReset()
+
+    await chat.setRestrictions(chat.currentUser.id, "some-channel", {})
+    await sleep(150) // Wait for the message to be sent and cached
+    expect(moderationEventCallback).toHaveBeenCalledWith(
+      expect.objectContaining({
+        payload: {
+          channelId: "PUBNUB_INTERNAL_MODERATION_some-channel",
+          restriction: "lifted",
+        },
+      })
+    )
+
+    removeModerationListener()
+  })
 })