Draupnir/apps/draupnir/test/unit/stores/userRestrictionAuditLogTest.ts

// SPDX-FileCopyrightText: 2025 Gnuxie <Gnuxie@protonmail.com>
//
// SPDX-License-Identifier: Apache-2.0

import Database from "better-sqlite3";
import { BetterSqliteOptions } from "../../../src/backingstore/better-sqlite3/BetterSqliteStore";
import {
  describePolicyRule,
  LiteralPolicyRule,
  parsePolicyRule,
  PolicyRuleType,
  randomRoomID,
  randomUserID,
  Recommendation,
} from "matrix-protection-suite";
import expect from "expect";
import { AccountRestriction } from "matrix-protection-suite-for-matrix-bot-sdk";
import { SqliteUserRestrictionAuditLog } from "../../../src/protections/HomeserverUserPolicyApplication/SqliteUserRestrictionAuditLog";

describe("UserAuditLog test", function () {
  const options = { path: ":memory:" } satisfies BetterSqliteOptions;
  const db = new Database(options.path);
  db.pragma("FOREIGN_KEYS = ON");
  const store = new SqliteUserRestrictionAuditLog(db);
  it("Can logged suspended users", async function () {
    const bannedUser = randomUserID();
    const policyRoom = randomRoomID([]);
    const moderator = randomUserID();
    expect(
      (await store.isUserRestricted(bannedUser)).expect(
        "Should be able to query if user is suspended"
      )
    ).toBe(false);
    const ban = parsePolicyRule(
      describePolicyRule({
        room_id: policyRoom.toRoomIDOrAlias(),
        entity: bannedUser,
        reason: "spam",
        recommendation: Recommendation.Ban,
        type: PolicyRuleType.User,
      }) as never
    ).expect("Should be able to parse the policy rule.");
    (
      await store.recordUserRestriction(
        bannedUser,
        AccountRestriction.Suspended,
        {
          sender: moderator,
          rule: ban as LiteralPolicyRule,
        }
      )
    ).expect("Should be able to takedown a room");
    expect(
      (await store.isUserRestricted(bannedUser)).expect(
        "Should be able to query if user is suspended"
      )
    ).toBe(true);
    // now unsuspend them
    (await store.unrestrictUser(bannedUser, moderator)).expect(
      "Should be able to unsuspend a user"
    );
    expect(
      (await store.isUserRestricted(bannedUser)).expect(
        "Should be able to query if user is suspended"
      )
    ).toBe(false);
  });
  it("Can log suspended users even without a policy (when a command is used)", async function () {
    const bannedUser = randomUserID();
    const moderator = randomUserID();
    (
      await store.recordUserRestriction(
        bannedUser,
        AccountRestriction.Suspended,
        {
          sender: moderator,
          rule: null,
        }
      )
    ).expect("To be able to do this");
    expect(
      (await store.isUserRestricted(bannedUser)).expect(
        "Should be able to query if user is suspended"
      )
    ).toBe(true);
  });
});