From 9b6b4c7622e3bd9dbbee9abe0d0ce95dc341008b Mon Sep 17 00:00:00 2001
From: Valere <bill.carson@valrsoft.com>
Date: Fri, 31 May 2024 15:33:24 +0200
Subject: [PATCH] backup spec | Test when roomId is specified

---
 spec/integ/crypto/megolm-backup.spec.ts | 40 +++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/spec/integ/crypto/megolm-backup.spec.ts b/spec/integ/crypto/megolm-backup.spec.ts
index ec79c34067d..2e0352197cb 100644
--- a/spec/integ/crypto/megolm-backup.spec.ts
+++ b/spec/integ/crypto/megolm-backup.spec.ts
@@ -590,6 +590,46 @@ describe.each(Object.entries(CRYPTO_BACKENDS))("megolm-keys backup (%s)", (backe
             expect(result.imported).toStrictEqual(20);
         });
 
+        it("Should report failures when decryption works but import fails - per room variant", async function () {
+            // @ts-ignore - mock a private method for testing purpose
+            aliceCrypto.importBackedUpRoomKeys = jest
+                .fn()
+                .mockImplementationOnce((keys: IMegolmSessionData[], version: String, opts?: ImportRoomKeysOpts) => {
+                    // report 10 failures to import
+                    opts!.progressCallback!({
+                        stage: "load_keys",
+                        successes: 20,
+                        failures: 10,
+                        total: 30,
+                    });
+                    return Promise.resolve();
+                })
+                // Ok for other chunks
+                .mockResolvedValue(importMockImpl);
+
+            const { response, expectedTotal } = createBackupDownloadResponse([30]);
+            const roomId = Object.keys(response.rooms)[0];
+
+            fetchMock.get(`express:/_matrix/client/v3/room_keys/keys/${roomId}`, response.rooms[roomId]);
+
+            const check = await aliceCrypto.checkKeyBackupAndEnable();
+
+            const progressCallback = jest.fn();
+            const result = await aliceClient.restoreKeyBackupWithRecoveryKey(
+                testData.BACKUP_DECRYPTION_KEY_BASE58,
+                roomId,
+                undefined,
+                check!.backupInfo!,
+                {
+                    progressCallback,
+                },
+            );
+
+            expect(result.total).toStrictEqual(expectedTotal);
+            // A chunk failed to import
+            expect(result.imported).toStrictEqual(20);
+        });
+
         it("recover specific session from backup", async function () {
             fetchMock.get(
                 "express:/_matrix/client/v3/room_keys/keys/:room_id/:session_id",