jest-webextension-mock 3.8.16 → 3.9.0

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # jest-webextension-mock
 
+## 3.9.0
+
+### Minor Changes
+
+- Add support for storage.session
+
 ## 3.8.16
 
 ### Patch Changes

package/__tests__/storage.test.js

@@ -11,7 +11,7 @@ describe('browser.storage', () => {
 });
 
 describe('browser.storage', () => {
-  ['sync', 'local', 'managed'].forEach((type) => {
+  ['sync', 'local', 'managed', 'session'].forEach((type) => {
     describe(type, () => {
       const storage = browser.storage[type];
       describe('get', () => {

package/dist/setup.js

@@ -365,6 +365,48 @@ var storage = {
     }),
     onChanged: createEventListeners()
   },
+  session: {
+    get: jest.fn(function (id, cb) {
+      var result = id === null || id === undefined ? localStore : resolveKey(id, localStore);
+      if (cb !== undefined) {
+        return cb(result);
+      }
+      return Promise.resolve(result);
+    }),
+    getBytesInUse: jest.fn(function (id, cb) {
+      if (cb !== undefined) {
+        return cb(0);
+      }
+      return Promise.resolve(0);
+    }),
+    set: jest.fn(function (payload, cb) {
+      Object.keys(payload).forEach(function (key) {
+        return localStore[key] = payload[key];
+      });
+      if (cb !== undefined) {
+        return cb();
+      }
+      return Promise.resolve();
+    }),
+    remove: jest.fn(function (id, cb) {
+      var keys = typeof id === 'string' ? [id] : id;
+      keys.forEach(function (key) {
+        return delete localStore[key];
+      });
+      if (cb !== undefined) {
+        return cb();
+      }
+      return Promise.resolve();
+    }),
+    clear: jest.fn(function (cb) {
+      localStore = {};
+      if (cb !== undefined) {
+        return cb();
+      }
+      return Promise.resolve();
+    }),
+    onChanged: createEventListeners()
+  },
   managed: {
     get: jest.fn(function (id, cb) {
       var result = id === null || id === undefined ? managedStore : resolveKey(id, managedStore);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jest-webextension-mock",
-  "version": "3.8.16",
+  "version": "3.9.0",
   "description": "Mock the components of a WebExtension",
   "main": "dist/setup.js",
   "module": "src/setup.js",

package/src/storage.js

@@ -3,6 +3,7 @@ import { createEventListeners } from './createEventListeners';
 let syncStore = {};
 let localStore = {};
 let managedStore = {};
+let sessionStore = {};
 
 function resolveKey(key, store) {
   if (typeof key === 'string') {
@@ -104,6 +105,47 @@ export const storage = {
     }),
     onChanged: createEventListeners(),
   },
+  session: {
+    get: jest.fn((id, cb) => {
+      const result =
+        id === null || id === undefined
+          ? localStore
+          : resolveKey(id, localStore);
+      if (cb !== undefined) {
+        return cb(result);
+      }
+      return Promise.resolve(result);
+    }),
+    getBytesInUse: jest.fn((id, cb) => {
+      if (cb !== undefined) {
+        return cb(0);
+      }
+      return Promise.resolve(0);
+    }),
+    set: jest.fn((payload, cb) => {
+      Object.keys(payload).forEach((key) => (localStore[key] = payload[key]));
+      if (cb !== undefined) {
+        return cb();
+      }
+      return Promise.resolve();
+    }),
+    remove: jest.fn((id, cb) => {
+      const keys = typeof id === 'string' ? [id] : id;
+      keys.forEach((key) => delete localStore[key]);
+      if (cb !== undefined) {
+        return cb();
+      }
+      return Promise.resolve();
+    }),
+    clear: jest.fn((cb) => {
+      localStore = {};
+      if (cb !== undefined) {
+        return cb();
+      }
+      return Promise.resolve();
+    }),
+    onChanged: createEventListeners(),
+  },
   managed: {
     get: jest.fn((id, cb) => {
       const result =