From 3c2604b384226c52f8e1c4c48f6a4cb3838455b2 Mon Sep 17 00:00:00 2001
From: Bruno Windels <bruno@windels.cloud>
Date: Mon, 25 Oct 2021 17:33:33 +0200
Subject: [PATCH] test that sessions get free'd correctly

---
 src/matrix/e2ee/megolm/decryption/KeyLoader.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/matrix/e2ee/megolm/decryption/KeyLoader.ts b/src/matrix/e2ee/megolm/decryption/KeyLoader.ts
index 375fb2ac..cf276608 100644
--- a/src/matrix/e2ee/megolm/decryption/KeyLoader.ts
+++ b/src/matrix/e2ee/megolm/decryption/KeyLoader.ts
@@ -368,6 +368,17 @@ export function tests() {
             await loader.useKey(cachedKey, async session => { invocations += 1; });
             assert.equal(loader.size, 1);
             assert.equal(invocations, 2);
+        },
+        "dispose calls free on all sessions": async assert => {
+            instances = 0;
+            const loader = new KeyLoader(olm, PICKLE_KEY, 2);
+            await loader.useKey(new MockRoomKey(roomId, aliceSenderKey, sessionId1, 1), async session => {});
+            await loader.useKey(new MockRoomKey(roomId, aliceSenderKey, sessionId2, 1), async session => {});
+            assert.equal(instances, 2);
+            assert.equal(loader.size, 2);
+            loader.dispose();
+            assert.strictEqual(instances, 0, "instances");
+            assert.strictEqual(loader.size, 0, "loader.size");
         }
     }
 }