122 lines
4.6 KiB
TypeScript
122 lines
4.6 KiB
TypeScript
/*
|
|
Copyright 2024 New Vector Ltd.
|
|
Copyright 2024 The Matrix.org Foundation C.I.C.
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only
|
|
Please see LICENSE files in the repository root for full details.
|
|
*/
|
|
|
|
import "fake-indexeddb/auto";
|
|
|
|
import { IDBFactory } from "fake-indexeddb";
|
|
import { IndexedDBCryptoStore } from "matrix-js-sdk/src/matrix";
|
|
|
|
import * as StorageManager from "../../src/utils/StorageManager";
|
|
|
|
const LEGACY_CRYPTO_STORE_NAME = "matrix-js-sdk:crypto";
|
|
const RUST_CRYPTO_STORE_NAME = "matrix-js-sdk::matrix-sdk-crypto";
|
|
|
|
describe("StorageManager", () => {
|
|
async function createDB(name: string, withStores: string[] | undefined = undefined): Promise<IDBDatabase> {
|
|
const request = indexedDB.open(name);
|
|
return new Promise((resolve, reject) => {
|
|
request.onupgradeneeded = function (event) {
|
|
const db = request.result;
|
|
if (withStores) {
|
|
withStores.forEach((storeName) => {
|
|
db.createObjectStore(storeName);
|
|
});
|
|
}
|
|
};
|
|
request.onsuccess = function (event) {
|
|
const db = request.result;
|
|
resolve(db);
|
|
};
|
|
request.onerror = function (event) {
|
|
reject(event);
|
|
};
|
|
});
|
|
}
|
|
|
|
async function populateLegacyStore(migrationState: number | undefined) {
|
|
const db = await createDB(LEGACY_CRYPTO_STORE_NAME, [IndexedDBCryptoStore.STORE_ACCOUNT]);
|
|
|
|
if (migrationState) {
|
|
const transaction = db.transaction([IndexedDBCryptoStore.STORE_ACCOUNT], "readwrite");
|
|
const store = transaction.objectStore(IndexedDBCryptoStore.STORE_ACCOUNT);
|
|
store.put(migrationState, "migrationState");
|
|
await new Promise((resolve, reject) => {
|
|
transaction.oncomplete = resolve;
|
|
transaction.onerror = reject;
|
|
});
|
|
}
|
|
}
|
|
|
|
beforeEach(() => {
|
|
global.structuredClone = (v) => JSON.parse(JSON.stringify(v));
|
|
});
|
|
|
|
describe("Crypto store checks", () => {
|
|
async function populateHealthySession() {
|
|
// Storage manager only check for the existence of the `riot-web-sync` store, so just create one.
|
|
await createDB("riot-web-sync");
|
|
}
|
|
|
|
beforeEach(async () => {
|
|
await populateHealthySession();
|
|
// eslint-disable-next-line no-global-assign
|
|
indexedDB = new IDBFactory();
|
|
});
|
|
|
|
it("should not be ok if sync store but no crypto store", async () => {
|
|
const result = await StorageManager.checkConsistency();
|
|
expect(result.healthy).toBe(true);
|
|
expect(result.dataInCryptoStore).toBe(false);
|
|
});
|
|
|
|
it("should be ok if sync store and a rust crypto store", async () => {
|
|
await createDB(RUST_CRYPTO_STORE_NAME);
|
|
|
|
const result = await StorageManager.checkConsistency();
|
|
expect(result.healthy).toBe(true);
|
|
expect(result.dataInCryptoStore).toBe(true);
|
|
});
|
|
|
|
describe("without rust store", () => {
|
|
it("should be ok if there is non migrated legacy crypto store", async () => {
|
|
await populateLegacyStore(undefined);
|
|
|
|
const result = await StorageManager.checkConsistency();
|
|
expect(result.healthy).toBe(true);
|
|
expect(result.dataInCryptoStore).toBe(true);
|
|
});
|
|
|
|
it("should be ok if legacy store in MigrationState `NOT_STARTED`", async () => {
|
|
await populateLegacyStore(0 /* MigrationState.NOT_STARTED*/);
|
|
|
|
const result = await StorageManager.checkConsistency();
|
|
expect(result.healthy).toBe(true);
|
|
expect(result.dataInCryptoStore).toBe(true);
|
|
});
|
|
|
|
it("should not be ok if MigrationState greater than `NOT_STARTED`", async () => {
|
|
await populateLegacyStore(1 /*INITIAL_DATA_MIGRATED*/);
|
|
|
|
const result = await StorageManager.checkConsistency();
|
|
expect(result.healthy).toBe(true);
|
|
expect(result.dataInCryptoStore).toBe(false);
|
|
});
|
|
|
|
it("should not be healthy if no indexeddb", async () => {
|
|
// eslint-disable-next-line no-global-assign
|
|
indexedDB = {} as IDBFactory;
|
|
|
|
const result = await StorageManager.checkConsistency();
|
|
expect(result.healthy).toBe(false);
|
|
|
|
// eslint-disable-next-line no-global-assign
|
|
indexedDB = new IDBFactory();
|
|
});
|
|
});
|
|
});
|
|
});
|