import { PrexStorage } from './PrexStorage';

export class MockStorage implements PrexStorage {
  private store: Record<string, any> = {};

  async getItem<T>(key: string): Promise<T | null> {
    return this.store[key] || null;
  }

  async setItem<T>(key: string, value: T): Promise<void> {
    this.store[key] = value;
  }

  async removeItem(key: string): Promise<void> {
    delete this.store[key];
  }

  async getItemFromSessionStorage<T>(key: string): Promise<T | null> {
    return this.store[key] || null;
  }

  async setItemToSessionStorage<T>(key: string, value: T): Promise<void> {
    this.store[key] = value;
  }

  async removeItemFromSessionStorage(key: string): Promise<void> {
    delete this.store[key];
  }
}