fix null access

matrix-org · Oct 3, 2024 · 2c18370 · 2c18370
1 parent 17d6ad4
commit 2c18370
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/integration/mentionSpamProtectionTest.ts b/test/integration/mentionSpamProtectionTest.ts
@@ -79,7 +79,8 @@ describe("Test: Mention spam protection", function () {
         const mentionDisplaynames = Array.from({length: DEFAULT_MAX_MENTIONS+1}, (_, i) => `Test User ${i}`);
 
         // Pre-set the displayname cache.
-        this.mjolnir.protectionManager.protections.roomDisplaynameCache.set(room, mentionDisplaynames);
+        let protection = this.mjolnir.protectionManager.protections.get('MentionSpam')
+        protection.roomDisplaynameCache.set(room, mentionDisplaynames);
 
         const messageWithTextMentions = await client.sendText(room, mentionUsers.join(' '));
         const messageWithHTMLMentions = await client.sendHtmlText(room, 
@@ -93,7 +94,6 @@ describe("Test: Mention spam protection", function () {
         });
         const messageWithDisplaynameMentions = await client.sendText(room, mentionDisplaynames.join(' '));
 
-
         await delay(500);
 
         const fetchedTextEvent = await client.getEvent(room, messageWithTextMentions);