react-native-nitro-storage 0.13.0 → 0.15.0

package/src/testing.ts

@@ -1,6 +1,7 @@
 import { createIndexedDBBackend as createIndexedDBBackendFromSubpath } from "./indexeddb-backend";
 import {
   assertAccessControlLevel,
+  assertProtectedDataListener,
   notifyAllListeners,
   notifyKeyListeners,
 } from "./shared";
@@ -23,6 +24,10 @@ import type {
 
 export { StorageScope, AccessControl, BiometricLevel } from "./Storage.types";
 export { isKeychainLockedError } from "./shared";
+export type {
+  StorageScheduledFlushError,
+  StorageScheduledFlushErrorObserver,
+} from "./core/durability";
 export { migrateFromMMKV } from "./migration";
 export { getStorageErrorCode, isStorageError } from "./storage-runtime";
 export {
@@ -80,6 +85,8 @@ export type {
   WebStorageScope,
 } from "./web-storage-backend";
 export type {
+  SecureAccessControlMigrationOptions,
+  SecureAccessControlMigrationResult,
   SetItemConfig,
   SetStorageItem,
   StorageBatchSetItem,
@@ -220,6 +227,7 @@ function buildTestingModule() {
       backend,
       changeSource: "native",
       applyAccessControlOnSecureRawWrite: true,
+      supportsSecureAccessControlMigration: () => true,
       ensureScopeSubscription: (_scope: NonMemoryScope) => {},
       maybeCleanupScopeSubscription: (_scope: NonMemoryScope) => {},
       onWillEmitChanges: () => {},
@@ -233,6 +241,8 @@ function buildTestingModule() {
 
   const core = createStorageCore(buildAdapter);
   const { internals } = core;
+  let protectedDataAvailable = true;
+  const protectedDataListeners = new Set<() => void>();
 
   const storage = {
     ...core.storage,
@@ -240,6 +250,14 @@ function buildTestingModule() {
       assertAccessControlLevel(level);
       internals.setSecureDefaultAccessControl(level);
     },
+    isProtectedDataAvailable: (): boolean => protectedDataAvailable,
+    onProtectedDataAvailable: (listener: () => void): (() => void) => {
+      assertProtectedDataListener(listener);
+      protectedDataListeners.add(listener);
+      return () => {
+        protectedDataListeners.delete(listener);
+      };
+    },
     setSecureWritesAsync: (_enabled: boolean) => {},
     setKeychainAccessGroup: (_group: string) => {},
     getCapabilities: (): StorageCapabilities => ({
@@ -277,8 +295,26 @@ function buildTestingModule() {
     }),
   };
 
+  const setProtectedDataAvailable = (available: boolean): void => {
+    if (typeof available !== "boolean") {
+      throw new TypeError(
+        "NitroStorage: Protected data availability must be a boolean",
+      );
+    }
+    const wasAvailable = protectedDataAvailable;
+    protectedDataAvailable = available;
+    if (available && !wasAvailable) {
+      Array.from(protectedDataListeners).forEach((listener) => {
+        listener();
+      });
+    }
+  };
+
   const reset = (): void => {
+    protectedDataAvailable = true;
+    protectedDataListeners.clear();
     storage.setEventObserver(undefined);
+    storage.setScheduledFlushErrorObserver(undefined);
     storage.setMetricsObserver(undefined);
     storage.resetMetrics();
     storage.clearAll();
@@ -298,8 +334,10 @@ function buildTestingModule() {
     removeBatch: core.removeBatch,
     registerMigration: core.registerMigration,
     migrateToLatest: core.migrateToLatest,
+    migrateSecureAccessControl: core.migrateSecureAccessControl,
     runTransaction: core.runTransaction,
     createSecureAuthStorage: core.createSecureAuthStorage,
+    setProtectedDataAvailable,
     reset,
   };
 }
@@ -319,8 +357,12 @@ export const setBatch = defaultModule.setBatch;
 export const removeBatch = defaultModule.removeBatch;
 export const registerMigration = defaultModule.registerMigration;
 export const migrateToLatest = defaultModule.migrateToLatest;
+export const migrateSecureAccessControl =
+  defaultModule.migrateSecureAccessControl;
 export const runTransaction = defaultModule.runTransaction;
 export const createSecureAuthStorage = defaultModule.createSecureAuthStorage;
+export const setMockProtectedDataAvailable =
+  defaultModule.setProtectedDataAvailable;
 
 export function setWebSecureStorageBackend(
   _backend?: WebSecureStorageBackend,