Class MockDurableObjectStorage

test/unit/persisters/common/mocks.ts:784–827

// Mock DurableObjectStorage - simple Map-based implementation
class MockDurableObjectStorage {
  private data = new Map<string, any>();

  async get<T>(key: string): Promise<T | undefined>;
  async get<T>(keys: string[]): Promise<Map<string, T>>;
  async get<T>(
    keyOrKeys: string | string[],
  ): Promise<T | undefined | Map<string, T>> {
    if (Array.isArray(keyOrKeys)) {
      const result = new Map<string, T>();
      for (const key of keyOrKeys) {
        const value = this.data.get(key);
        if (value !== undefined) result.set(key, value);
      }
      return result;
    }
    return this.data.get(keyOrKeys);
  }

  async put(entries: Record<string, any>): Promise<void> {
    for (const [key, value] of Object.entries(entries)) {
      this.data.set(key, value);
    }
  }

  async list<T>(options?: {prefix?: string}): Promise<Map<string, T>> {
    const result = new Map<string, T>();
    const prefix = options?.prefix ?? '';
    for (const [key, value] of this.data.entries()) {
      if (key.startsWith(prefix)) {
        result.set(key, value);
      }
    }
    return result;
  }

  async delete(key: string): Promise<boolean> {
    return this.data.delete(key);
  }

  clear(): void {
    this.data.clear();
  }
}

const STORAGE_PREFIX = 'tinybase_';
const T = 't';