From 3bc0439fd218d7c2360454cf09b8df87115d7caa Mon Sep 17 00:00:00 2001
From: Florian Duros <florian.duros@ormaz.fr>
Date: Mon, 14 Oct 2024 17:08:42 +0200
Subject: [PATCH] Replace `MatrixClient.checkSecretStorageKey` by
 `MatrixClient.SecretStorage.checkKey` (#142)

---
 src/SecurityManager.ts                              |  2 +-
 .../dialogs/security/AccessSecretStorageDialog.tsx  |  2 +-
 .../dialogs/AccessSecretStorageDialog-test.tsx      | 13 +++++--------
 test/test-utils/test-utils.ts                       |  1 +
 4 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/src/SecurityManager.ts b/src/SecurityManager.ts
index fb73e25389..a89427e5f3 100644
--- a/src/SecurityManager.ts
+++ b/src/SecurityManager.ts
@@ -121,7 +121,7 @@ async function getSecretStorageKey({
             keyInfo,
             checkPrivateKey: async (input: KeyParams): Promise<boolean> => {
                 const key = await inputToKey(input);
-                return MatrixClientPeg.safeGet().checkSecretStorageKey(key, keyInfo);
+                return MatrixClientPeg.safeGet().secretStorage.checkKey(key, keyInfo);
             },
         },
         /* className= */ undefined,
diff --git a/src/components/views/dialogs/security/AccessSecretStorageDialog.tsx b/src/components/views/dialogs/security/AccessSecretStorageDialog.tsx
index 0c4e875607..412d88c704 100644
--- a/src/components/views/dialogs/security/AccessSecretStorageDialog.tsx
+++ b/src/components/views/dialogs/security/AccessSecretStorageDialog.tsx
@@ -102,7 +102,7 @@ export default class AccessSecretStorageDialog extends React.PureComponent<IProp
         try {
             const cli = MatrixClientPeg.safeGet();
             const decodedKey = decodeRecoveryKey(this.state.recoveryKey);
-            const correct = await cli.checkSecretStorageKey(decodedKey, this.props.keyInfo);
+            const correct = await cli.secretStorage.checkKey(decodedKey, this.props.keyInfo);
             this.setState({
                 recoveryKeyValid: true,
                 recoveryKeyCorrect: correct,
diff --git a/test/components/views/dialogs/AccessSecretStorageDialog-test.tsx b/test/components/views/dialogs/AccessSecretStorageDialog-test.tsx
index c07808ac11..517d3e619b 100644
--- a/test/components/views/dialogs/AccessSecretStorageDialog-test.tsx
+++ b/test/components/views/dialogs/AccessSecretStorageDialog-test.tsx
@@ -10,15 +10,14 @@ import React, { ComponentProps } from "react";
 import { SecretStorage, MatrixClient } from "matrix-js-sdk/src/matrix";
 import { act, fireEvent, render, screen } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
-import { Mocked } from "jest-mock";
 
-import { getMockClientWithEventEmitter, mockPlatformPeg } from "../../../test-utils";
+import { mockPlatformPeg, stubClient } from "../../../test-utils";
 import AccessSecretStorageDialog from "../../../../src/components/views/dialogs/security/AccessSecretStorageDialog";
 
 const securityKey = "EsTc WKmb ivvk jLS7 Y1NH 5CcQ mP1E JJwj B3Fd pFWm t4Dp dbyu";
 
 describe("AccessSecretStorageDialog", () => {
-    let mockClient: Mocked<MatrixClient>;
+    let mockClient: MatrixClient;
 
     const defaultProps: ComponentProps<typeof AccessSecretStorageDialog> = {
         keyInfo: {} as any,
@@ -57,13 +56,11 @@ describe("AccessSecretStorageDialog", () => {
     });
 
     beforeEach(() => {
-        mockClient = getMockClientWithEventEmitter({
-            checkSecretStorageKey: jest.fn(),
-        });
+        mockClient = stubClient();
     });
 
     it("Closes the dialog when the form is submitted with a valid key", async () => {
-        mockClient.checkSecretStorageKey.mockResolvedValue(true);
+        jest.spyOn(mockClient.secretStorage, "checkKey").mockResolvedValue(true);
 
         const onFinished = jest.fn();
         const checkPrivateKey = jest.fn().mockResolvedValue(true);
@@ -85,7 +82,7 @@ describe("AccessSecretStorageDialog", () => {
         const checkPrivateKey = jest.fn().mockResolvedValue(true);
         renderComponent({ onFinished, checkPrivateKey });
 
-        mockClient.checkSecretStorageKey.mockImplementation(() => {
+        jest.spyOn(mockClient.secretStorage, "checkKey").mockImplementation(() => {
             throw new Error("invalid key");
         });
 
diff --git a/test/test-utils/test-utils.ts b/test/test-utils/test-utils.ts
index ebfc6b221b..2f534c7bd7 100644
--- a/test/test-utils/test-utils.ts
+++ b/test/test-utils/test-utils.ts
@@ -109,6 +109,7 @@ export function createTestClient(): MatrixClient {
         secretStorage: {
             get: jest.fn(),
             isStored: jest.fn().mockReturnValue(false),
+            checkKey: jest.fn().mockResolvedValue(false),
         },
 
         store: {