mjolnir/test/appservice/utils/harness.ts

import path from "path";
import { MjolnirAppService } from "../../../src/appservice/AppService";
import { ensureAliasedRoomExists } from "../../integration/mjolnirSetupUtils";
import { read as configRead, IConfig } from "../../../src/appservice/config/config";
import { PgDataStore } from "../../../src/appservice/datastore";
import { newTestUser } from "../../integration/clientHelper";
import PolicyList from "../../../src/models/PolicyList";
import { CreateEvent, MatrixClient } from "matrix-bot-sdk";

export function readTestConfig(): IConfig {
    return configRead(path.join(__dirname, "../../../src/appservice/config/config.harness.yaml"));
}

export async function setupHarness(): Promise<MjolnirAppService> {
    const config = readTestConfig();
    const utilityUser = await newTestUser(config.homeserver.url, { name: { contains: "utility" }});
    await ensureAliasedRoomExists(utilityUser, config.accessControlList);
    const dataStore = new PgDataStore(config.db.connectionString);
    await dataStore.init();
    const appservice = await MjolnirAppService.makeMjolnirAppService(config, dataStore, "mjolnir-registration.yaml");
    await appservice.start(9000);
    return appservice;
}

export async function isPolicyRoom(user: MatrixClient, roomId: string): Promise<boolean> {
    const createEvent = new CreateEvent(await user.getRoomStateEvent(roomId, "m.room.create", ""));
    return PolicyList.ROOM_TYPE_VARIANTS.includes(createEvent.type);
}