import { MockProxy, mock } from "jest-mock-extended"; import { Subject } from "rxjs"; import { AbstractStorageService, StorageUpdate, } from "../src/platform/abstractions/storage.service"; import { StorageOptions } from "../src/platform/models/domain/storage-options"; export class FakeStorageService implements AbstractStorageService { private store: Record; private updatesSubject = new Subject(); private _valuesRequireDeserialization = false; /** * Returns a mock of a {@see AbstractStorageService} for asserting the expected * amount of calls. It is not recommended to use this to mock implementations as * they are not respected. */ mock: MockProxy; constructor(initial?: Record) { this.store = initial ?? {}; this.mock = mock(); } /** * Updates the internal store for this fake implementation, this bypasses any mock calls * or updates to the {@link updates$} observable. * @param store */ internalUpdateStore(store: Record) { this.store = store; } internalUpdateValuesRequireDeserialization(value: boolean) { this._valuesRequireDeserialization = value; } get valuesRequireDeserialization(): boolean { return this._valuesRequireDeserialization; } get updates$() { return this.updatesSubject.asObservable(); } get(key: string, options?: StorageOptions): Promise { this.mock.get(key, options); const value = this.store[key] as T; return Promise.resolve(value); } has(key: string, options?: StorageOptions): Promise { this.mock.has(key, options); return Promise.resolve(this.store[key] != null); } save(key: string, obj: T, options?: StorageOptions): Promise { this.mock.save(key, options); this.store[key] = obj; this.updatesSubject.next({ key: key, updateType: "save" }); return Promise.resolve(); } remove(key: string, options?: StorageOptions): Promise { this.mock.remove(key, options); delete this.store[key]; this.updatesSubject.next({ key: key, updateType: "remove" }); return Promise.resolve(); } }