diff --git a/extensions/matrix/src/matrix/sdk/crypto-bootstrap.test.ts b/extensions/matrix/src/matrix/sdk/crypto-bootstrap.test.ts index cd637b6fbeb..185b17e8c21 100644 --- a/extensions/matrix/src/matrix/sdk/crypto-bootstrap.test.ts +++ b/extensions/matrix/src/matrix/sdk/crypto-bootstrap.test.ts @@ -1,7 +1,9 @@ -import { beforeEach, describe, expect, it, vi } from "vitest"; +import { beforeEach, describe, expect, it, vi, type Mock } from "vitest"; import { MatrixCryptoBootstrapper, type MatrixCryptoBootstrapperDeps } from "./crypto-bootstrap.js"; import type { MatrixCryptoBootstrapApi, MatrixRawEvent } from "./types.js"; +type BootstrapCrossSigningMock = Mock; + function createBootstrapperDeps() { return { getUserId: vi.fn(async () => "@bot:example.org"), @@ -81,7 +83,7 @@ async function runExplicitSecretStorageRepairScenario(firstError: string) { function expectSecretStorageRepairRetry( deps: ReturnType, crypto: MatrixCryptoBootstrapApi, - bootstrapCrossSigning: ReturnType, + bootstrapCrossSigning: BootstrapCrossSigningMock, ) { expect(deps.recoveryKeyStore.bootstrapSecretStorageWithRecoveryKey).toHaveBeenCalledWith(crypto, { allowSecretStorageRecreateWithoutRecoveryKey: true, @@ -90,7 +92,7 @@ function expectSecretStorageRepairRetry( expect(bootstrapCrossSigning).toHaveBeenCalledTimes(2); } -function createForcedResetHarness(bootstrapCrossSigning: ReturnType) { +function createForcedResetHarness(bootstrapCrossSigning: BootstrapCrossSigningMock) { return createBootstrapperHarness({ bootstrapCrossSigning, isCrossSigningReady: vi.fn(async () => true), @@ -100,7 +102,7 @@ function createForcedResetHarness(bootstrapCrossSigning: ReturnType, + bootstrapCrossSigning: BootstrapCrossSigningMock, params: { setupNewCall: number; totalCalls: number }, ) { expect(bootstrapCrossSigning).toHaveBeenCalledTimes(params.totalCalls);