passbolt-styleguide/src/react-extension/test/mock/MockStorage.js

Passbolt styleguide contains common styling assets used by the different sites, plugin, etc.

class LocalStorage {
  constructor(changeCallbacks) {
    this.storage = {};
    this.changeCallbacks = changeCallbacks;
  }

  get(keys) {
    return keys.reduce((accumulator, key) => {
      if (this.storage[key]) {
        accumulator[key] = this.storage[key];
      }
      return accumulator;
    }, {});
  }

  set(data) {
    const storageChangeEventValue = {};
    Object.keys(data).forEach(key => {
      this.storage[key] = data[key];
      storageChangeEventValue[key] = {};
      storageChangeEventValue[key].newValue = data[key];
    });
    this.changeCallbacks.forEach(callback => callback(storageChangeEventValue));
  }
}

class MockStorage {
  constructor() {
    this.changeCallbacks = [];
    this.local = new LocalStorage(this.changeCallbacks);
    this.onChanged = {
      addListener: listener => {
        console.debug("browser.storage.onChanged.addListener");
        this.changeCallbacks.push(listener);
      }
    };
  }
}

export default MockStorage;