2022-03-18 10:11:23 +00:00
|
|
|
import { strict as assert } from "assert";
|
|
|
|
|
|
|
|
import { Mjolnir } from "../../src/Mjolnir";
|
|
|
|
import { IProtection } from "../../src/protections/IProtection";
|
|
|
|
import { newTestUser, noticeListener } from "./clientHelper";
|
|
|
|
import { matrixClient, mjolnir } from "./mjolnirSetupUtils";
|
|
|
|
import { ConsequenceType, Consequence } from "../../src/protections/consequence";
|
|
|
|
|
|
|
|
describe("Test: standard consequences", function() {
|
|
|
|
let badUser;
|
|
|
|
let goodUser;
|
|
|
|
this.beforeEach(async function () {
|
2022-08-16 14:51:18 +00:00
|
|
|
badUser = await newTestUser(this.config.homeserverUrl, { name: { contains: "standard-consequences" }});
|
|
|
|
goodUser = await newTestUser(this.config.homeserverUrl, { name: { contains: "standard-consequences" }});
|
2022-03-18 10:11:23 +00:00
|
|
|
await badUser.start();
|
|
|
|
await goodUser.start();
|
|
|
|
})
|
|
|
|
this.afterEach(async function () {
|
|
|
|
await badUser.stop();
|
|
|
|
await goodUser.stop();
|
|
|
|
})
|
|
|
|
it("Mjolnir applies a standard consequence redaction", async function() {
|
|
|
|
this.timeout(20000);
|
|
|
|
|
|
|
|
let protectedRoomId = await this.mjolnir.client.createRoom({ invite: [await badUser.getUserId()] });
|
|
|
|
await badUser.joinRoom(this.mjolnir.managementRoomId);
|
|
|
|
await badUser.joinRoom(protectedRoomId);
|
|
|
|
await this.mjolnir.addProtectedRoom(protectedRoomId);
|
|
|
|
|
|
|
|
await this.mjolnir.registerProtection(new class implements IProtection {
|
|
|
|
name = "JY2TPN";
|
|
|
|
description = "A test protection";
|
|
|
|
settings = { };
|
|
|
|
handleEvent = async (mjolnir: Mjolnir, roomId: string, event: any) => {
|
|
|
|
if (event.content.body === "ngmWkF") {
|
|
|
|
return new Consequence(ConsequenceType.redact, "asd");
|
|
|
|
}
|
|
|
|
};
|
|
|
|
});
|
|
|
|
await this.mjolnir.enableProtection("JY2TPN");
|
|
|
|
|
|
|
|
let reply = new Promise(async (resolve, reject) => {
|
|
|
|
const messageId = await badUser.sendMessage(protectedRoomId, {msgtype: "m.text", body: "ngmWkF"});
|
|
|
|
let redaction;
|
|
|
|
badUser.on('room.event', (roomId, event) => {
|
|
|
|
if (
|
|
|
|
roomId === protectedRoomId
|
|
|
|
&& event?.type === "m.room.redaction"
|
|
|
|
&& event.redacts === messageId
|
|
|
|
) {
|
|
|
|
redaction = event
|
|
|
|
}
|
|
|
|
if (
|
|
|
|
roomId === this.mjolnir.managementRoomId
|
|
|
|
&& event?.type === "m.room.message"
|
|
|
|
&& event?.content?.body?.startsWith("protection JY2TPN enacting redact against ")
|
|
|
|
&& redaction !== undefined
|
|
|
|
) {
|
|
|
|
resolve([redaction, event])
|
|
|
|
}
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
const [eventRedact, eventMessage] = await reply
|
|
|
|
});
|
|
|
|
it("Mjolnir applies a standard consequence ban", async function() {
|
|
|
|
this.timeout(20000);
|
|
|
|
|
|
|
|
let protectedRoomId = await this.mjolnir.client.createRoom({ invite: [await badUser.getUserId()] });
|
|
|
|
await badUser.joinRoom(this.mjolnir.managementRoomId);
|
|
|
|
await badUser.joinRoom(protectedRoomId);
|
|
|
|
await this.mjolnir.addProtectedRoom(protectedRoomId);
|
|
|
|
|
|
|
|
await this.mjolnir.registerProtection(new class implements IProtection {
|
|
|
|
name = "0LxMTy";
|
|
|
|
description = "A test protection";
|
|
|
|
settings = { };
|
|
|
|
handleEvent = async (mjolnir: Mjolnir, roomId: string, event: any) => {
|
|
|
|
if (event.content.body === "7Uga3d") {
|
|
|
|
return new Consequence(ConsequenceType.ban, "asd");
|
|
|
|
}
|
|
|
|
};
|
|
|
|
});
|
|
|
|
await this.mjolnir.enableProtection("0LxMTy");
|
|
|
|
|
|
|
|
let reply = new Promise(async (resolve, reject) => {
|
|
|
|
const messageId = await badUser.sendMessage(protectedRoomId, {msgtype: "m.text", body: "7Uga3d"});
|
|
|
|
let ban;
|
|
|
|
badUser.on('room.leave', (roomId, event) => {
|
|
|
|
if (
|
|
|
|
roomId === protectedRoomId
|
|
|
|
&& event?.type === "m.room.member"
|
|
|
|
&& event.content?.membership === "ban"
|
|
|
|
&& event.state_key === badUser.userId
|
|
|
|
) {
|
|
|
|
ban = event;
|
|
|
|
}
|
|
|
|
});
|
|
|
|
badUser.on('room.event', (roomId, event) => {
|
|
|
|
if (
|
|
|
|
roomId === this.mjolnir.managementRoomId
|
|
|
|
&& event?.type === "m.room.message"
|
|
|
|
&& event?.content?.body?.startsWith("protection 0LxMTy enacting ban against ")
|
|
|
|
&& ban !== undefined
|
|
|
|
) {
|
|
|
|
resolve([ban, event])
|
|
|
|
}
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
const [eventBan, eventMessage] = await reply
|
|
|
|
});
|
|
|
|
it("Mjolnir doesn't ban a good user", async function() {
|
|
|
|
this.timeout(20000);
|
|
|
|
|
2022-05-03 10:20:40 +00:00
|
|
|
let protectedRoomId = await this.mjolnir.client.createRoom({ invite: [await goodUser.getUserId(), await badUser.getUserId()] });
|
2022-03-18 10:11:23 +00:00
|
|
|
await badUser.joinRoom(protectedRoomId);
|
|
|
|
await goodUser.joinRoom(protectedRoomId);
|
|
|
|
await this.mjolnir.addProtectedRoom(protectedRoomId);
|
|
|
|
|
|
|
|
await this.mjolnir.registerProtection(new class implements IProtection {
|
|
|
|
name = "95B1Cr";
|
|
|
|
description = "A test protection";
|
|
|
|
settings = { };
|
|
|
|
handleEvent = async (mjolnir: Mjolnir, roomId: string, event: any) => {
|
|
|
|
if (event.content.body === "8HUnwb") {
|
|
|
|
return new Consequence(ConsequenceType.ban, "asd");
|
|
|
|
}
|
|
|
|
};
|
|
|
|
});
|
|
|
|
await this.mjolnir.enableProtection("95B1Cr");
|
|
|
|
|
|
|
|
let reply = new Promise(async (resolve, reject) => {
|
|
|
|
this.mjolnir.client.on('room.message', async (roomId, event) => {
|
|
|
|
if (event?.content?.body === "SUwvFT") {
|
|
|
|
await badUser.sendMessage(protectedRoomId, {msgtype: "m.text", body: "8HUnwb"});
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
this.mjolnir.client.on('room.event', (roomId, event) => {
|
|
|
|
if (
|
|
|
|
roomId === protectedRoomId
|
|
|
|
&& event?.type === "m.room.member"
|
|
|
|
&& event.content?.membership === "ban"
|
|
|
|
) {
|
|
|
|
if (event.state_key === goodUser.userId) {
|
|
|
|
reject("good user has been banned");
|
|
|
|
} else if (event.state_key === badUser.userId) {
|
|
|
|
resolve(null);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
});
|
|
|
|
});
|
|
|
|
await goodUser.sendMessage(protectedRoomId, {msgtype: "m.text", body: "SUwvFT"});
|
|
|
|
|
|
|
|
await reply
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|