Draupnir/test/integration/commands/shutdownCommandTest.ts

import { strict as assert } from "assert";

import { newTestUser } from "../clientHelper";

describe("Test: shutdown command", function() {
    let client;
    this.beforeEach(async function () {
        client = await newTestUser(this.config.homeserverUrl, { name: { contains: "shutdown-command" }});
        await client.start();
    })
    this.afterEach(async function () {
        await client.stop();
    })
    it("Mjolnir asks synapse to shut down a channel", async function() {
        this.timeout(20000);
        const badRoom = await client.createRoom();
        await client.joinRoom(this.mjolnir.managementRoomId);

        let reply1 = new Promise(async (resolve, reject) => {
            const msgid = await client.sendMessage(this.mjolnir.managementRoomId, {msgtype: "m.text", body: `!mjolnir shutdown room ${badRoom} closure test`});
            client.on('room.event', (roomId, event) => {
                if (
                    roomId === this.mjolnir.managementRoomId
                    && event?.type === "m.reaction"
                    && event.sender === this.mjolnir.client.userId
                    && event.content?.["m.relates_to"]?.event_id === msgid
                ) {
                    resolve(event);
                }
            });
        });

        const reply2 = new Promise((resolve, reject) => {
            this.mjolnir.client.on('room.event', (roomId, event) => {
                if (
                    roomId !== this.mjolnir.managementRoomId
                    && roomId !== badRoom
                    && event?.type === "m.room.message"
                    && event.sender === this.mjolnir.client.userId
                    && event.content?.body === "closure test"
                ) {
                    resolve(event);
                }
            });
        });

        await reply1
        await reply2

        await assert.rejects(client.joinRoom(badRoom), (e: any) => {
            assert.equal(e.statusCode, 403);
            assert.equal(e.body.error, "This room has been blocked on this server");
            return true;
        });
    });
});