test

matrix-org · Jul 25, 2024 · 2416566 · 2416566
1 parent a6abe18
commit 2416566
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/test/integration/protectionSettingsTest.ts b/test/integration/protectionSettingsTest.ts
@@ -6,12 +6,16 @@ import { ProtectionSettingValidationError } from "../../src/protections/Protecti
 import { NumberProtectionSetting, StringProtectionSetting, StringListProtectionSetting } from "../../src/protections/ProtectionSettings";
 import { newTestUser, noticeListener } from "./clientHelper";
 import { matrixClient, mjolnir } from "./mjolnirSetupUtils";
+import {MessageIsMedia} from "../../src/protections/MessageIsMedia";
 
 describe("Test: Protection settings", function() {
     let client;
+    let room;
     this.beforeEach(async function () {
         client = await newTestUser(this.config.homeserverUrl, { name: { contains: "protection-settings" }});
         await client.start();
+        room = await client.createRoom();
+        await client.joinRoom(room)
     })
     this.afterEach(async function () {
         await client.stop();
@@ -158,5 +162,33 @@ describe("Test: Protection settings", function() {
             "Changed d0sNrt.test to asd2 (was asd1)"
         )
     });
+    it("Events are checked for new content under media protections", async function() {
+        this.timeout(20000);
+        await client.joinRoom(this.config.managementRoom);
+
+        await this.mjolnir.protectionManager.registerProtection(new MessageIsMedia());
+
+        // send a regular media message to make sure protections are running
+        await client.sendMessage(room, {msgtype: "m.image", body: ""})
+        let reply = () => new Promise((resolve, reject) => {
+            client.on('room.message', noticeListener(this.mjolnir.managementRoomId, (event) => {
+                if (event.content.body.includes("Redacting event")) {
+                    resolve(event);
+                }
+            }))
+        });
+        await reply;
+
+        await client.sendMessage(room, {"m.new_content": {msgtype: "m.image", body: ""}, body: "", msgtype: "m.text"})
+        let reply2 = () => new Promise((resolve, reject) => {
+            client.on('room.message', noticeListener(this.mjolnir.managementRoomId, (event) => {
+                if (event.content.body.includes("Redacting event")) {
+                    resolve(event);
+                }
+            }))
+        });
+        await reply2;
+    });
 });
 
+