import { strict as assert } from "assert"; import config from "../../../src/config"; import { newTestUser } from "../clientHelper"; import { getMessagesByUserIn } from "../../../src/utils"; import { LogService } from "matrix-bot-sdk"; import { onReactionTo } from "./commandUtils"; describe("Test: The redaction command", function () { it('Mjölnir redacts all of the events sent by a spammer when instructed to by giving their id and a room id.', async function() { this.timeout(20000); // Create a few users and a room. let badUser = await newTestUser(false, "spammer-needs-redacting"); let badUserId = await badUser.getUserId(); const mjolnir = config.RUNTIME.client! let mjolnirUserId = await mjolnir.getUserId(); let moderator = await newTestUser(false, "moderator"); this.moderator = moderator; await moderator.joinRoom(config.managementRoom); let targetRoom = await moderator.createRoom({ invite: [await badUser.getUserId(), mjolnirUserId]}); await moderator.setUserPowerLevel(mjolnirUserId, targetRoom, 100); await badUser.joinRoom(targetRoom); moderator.sendMessage(config.managementRoom, {msgtype: 'm.text.', body: `!mjolnir rooms add ${targetRoom}`}); LogService.debug("redactionTest", `targetRoom: ${targetRoom}, managementRoom: ${config.managementRoom}`); // Sandwich irrelevant messages in bad messages. await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); await Promise.all([...Array(50).keys()].map((i) => moderator.sendMessage(targetRoom, {msgtype: 'm.text.', body: `Irrelevant Message #${i}`}))); for (let i = 0; i < 5; i++) { await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); } await Promise.all([...Array(50).keys()].map((i) => moderator.sendMessage(targetRoom, {msgtype: 'm.text.', body: `Irrelevant Message #${i}`}))); await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); try { moderator.start(); await onReactionTo(moderator, config.managementRoom, '✅', async () => { return await moderator.sendMessage(config.managementRoom, { msgtype: 'm.text', body: `!mjolnir redact ${badUserId} ${targetRoom}` }); }); } finally { moderator.stop(); } await getMessagesByUserIn(moderator, badUserId, targetRoom, 1000, function(events) { events.map(e => { if (e.type === 'm.room.member') { assert.equal(Object.keys(e.content).length, 1, "Only membership should be left on the membership even when it has been redacted.") } else if (Object.keys(e.content).length !== 0) { throw new Error(`This event should have been redacted: ${JSON.stringify(e, null, 2)}`) } }) }); }) it('Mjölnir redacts all of the events sent by a spammer when instructed to by giving their id in multiple rooms.', async function() { this.timeout(20000); // Create a few users and a room. let badUser = await newTestUser(false, "spammer-needs-redacting"); let badUserId = await badUser.getUserId(); const mjolnir = config.RUNTIME.client! let mjolnirUserId = await mjolnir.getUserId(); let moderator = await newTestUser(false, "moderator"); this.moderator = moderator; await moderator.joinRoom(config.managementRoom); let targetRooms: string[] = []; for (let i = 0; i < 5; i++) { let targetRoom = await moderator.createRoom({ invite: [await badUser.getUserId(), mjolnirUserId]}); await moderator.setUserPowerLevel(mjolnirUserId, targetRoom, 100); await badUser.joinRoom(targetRoom); await moderator.sendMessage(config.managementRoom, {msgtype: 'm.text.', body: `!mjolnir rooms add ${targetRoom}`}); targetRooms.push(targetRoom); // Sandwich irrelevant messages in bad messages. await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); await Promise.all([...Array(50).keys()].map((j) => moderator.sendMessage(targetRoom, {msgtype: 'm.text.', body: `Irrelevant Message #${j}`}))); for (let j = 0; j < 5; j++) { await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); } await Promise.all([...Array(50).keys()].map((j) => moderator.sendMessage(targetRoom, {msgtype: 'm.text.', body: `Irrelevant Message #${j}`}))); await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); } try { moderator.start(); await onReactionTo(moderator, config.managementRoom, '✅', async () => { return await moderator.sendMessage(config.managementRoom, { msgtype: 'm.text', body: `!mjolnir redact ${badUserId}` }); }); } finally { moderator.stop(); } targetRooms.map(async targetRoom => { await getMessagesByUserIn(moderator, badUserId, targetRoom, 1000, function(events) { events.map(e => { if (e.type === 'm.room.member') { assert.equal(Object.keys(e.content).length, 1, "Only membership should be left on the membership even when it has been redacted.") } else if (Object.keys(e.content).length !== 0) { throw new Error(`This event should have been redacted: ${JSON.stringify(e, null, 2)}`) } }) }) }); }); it("Redacts a single event when instructed to.", async function () { this.timeout(20000); // Create a few users and a room. let badUser = await newTestUser(false, "spammer-needs-redacting"); const mjolnir = config.RUNTIME.client! let mjolnirUserId = await mjolnir.getUserId(); let moderator = await newTestUser(false, "moderator"); this.moderator = moderator; await moderator.joinRoom(config.managementRoom); let targetRoom = await moderator.createRoom({ invite: [await badUser.getUserId(), mjolnirUserId]}); await moderator.setUserPowerLevel(mjolnirUserId, targetRoom, 100); await badUser.joinRoom(targetRoom); moderator.sendMessage(config.managementRoom, {msgtype: 'm.text.', body: `!mjolnir rooms add ${targetRoom}`}); let eventToRedact = await badUser.sendMessage(targetRoom, {msgtype: 'm.text', body: "Very Bad Stuff"}); try { moderator.start(); await onReactionTo(moderator, config.managementRoom, '✅', async () => { return await moderator.sendMessage(config.managementRoom, {msgtype: 'm.text', body: `!mjolnir redact https://matrix.to/#/${encodeURIComponent(targetRoom)}/${encodeURIComponent(eventToRedact)}`}); }); } finally { moderator.stop(); } let redactedEvent = await moderator.getEvent(targetRoom, eventToRedact); assert.equal(Object.keys(redactedEvent.content).length, 0, "This event should have been redacted"); }) });