Make setup function callable to reset mocks between tests

__setups__/localstorage.js

@@ -1 +1 @@
-require('../src/setup');
+require('../src/index');

__tests__/setup.test.js

@@ -1,5 +1,3 @@
-import { LocalStorage } from '../src/localstorage';
-
 describe('setup', () => {
   const orignalImpGlobsl = {};
 
@@ -33,15 +31,15 @@ describe('setup', () => {
 
   ['_localStorage', '_sessionStorage'].forEach(gKey => {
     it(`[${gKey}] should define a property on the global object with writable false`, () => {
-      require('../src/setup');
+      require('../src/index');
       expect(global[gKey.replace('_', '')].constructor.name).toBe(
         'LocalStorage'
       );
     });
 
     it(`[${gKey}] should define a property on the global object with writable false`, () => {
       global[gKey] = true;
-      require('../src/setup');
+      require('../src/index');
       let e;
       try {
         global[`_${gKey.replace('_', '')}`] = 'blah';
@@ -51,4 +49,20 @@ describe('setup', () => {
       expect(e).toBeDefined();
     });
   });
+
+  it('setup should be callable to manually restore mocks', () => {
+    const { setup } = require('../src/setup');
+    const KEY = 'foo';
+    const VALUE = 'bar';
+
+    setup();
+    localStorage.setItem(KEY, VALUE);
+    expect(localStorage.getItem(KEY)).toBe(VALUE);
+    jest.resetAllMocks();
+    localStorage.setItem(KEY, VALUE);
+    expect(localStorage.getItem(KEY)).toBe(undefined);
+    setup();
+    localStorage.setItem(KEY, VALUE);
+    expect(localStorage.getItem(KEY)).toBe(VALUE);
+  });
 });

package.json

@@ -2,8 +2,8 @@
   "name": "jest-localstorage-mock",
   "version": "2.4.0",
   "description": "Auto mock all localstorage and sessionstorage APIs for your Jest tests",
-  "main": "dist/setup.js",
-  "module": "src/setup.js",
+  "main": "dist/index.js",
+  "module": "src/index.js",
   "author": "Bryan Clark <[email protected]> (https://twitter.com/clarkbw)",
   "maintainers": [
     {

rollup.config.js

@@ -1,18 +1,35 @@
 import resolve from 'rollup-plugin-node-resolve';
 import babel from 'rollup-plugin-babel';
 
-export default {
-  input: 'src/setup.js',
-  plugins: [
-    resolve(),
-    babel({
-      exclude: 'node_modules/**',
-      babelrc: false,
-    }),
-  ],
-  output: {
-    file: 'dist/setup.js',
-    format: 'cjs',
-    sourcemap: true,
+export default [
+  {
+    input: 'src/index.js',
+    plugins: [
+      resolve(),
+      babel({
+        exclude: 'node_modules/**',
+        babelrc: false,
+      }),
+    ],
+    output: {
+      file: 'dist/index.js',
+      format: 'cjs',
+      sourcemap: true,
+    },
   },
-};
+  {
+    input: 'src/setup.js',
+    plugins: [
+      resolve(),
+      babel({
+        exclude: 'node_modules/**',
+        babelrc: false,
+      }),
+    ],
+    output: {
+      file: 'dist/setup.js',
+      format: 'cjs',
+      sourcemap: true,
+    },
+  },
+];

src/index.js

@@ -0,0 +1,2 @@
+import { setup } from './setup';
+setup();

src/setup.js

@@ -1,19 +1,21 @@
 import { LocalStorage } from './localstorage';
 
-if (typeof global._localStorage !== 'undefined') {
-  Object.defineProperty(global, '_localStorage', {
-    value: new LocalStorage(jest),
-    writable: false,
-  });
-} else {
-  global.localStorage = new LocalStorage(jest);
-}
+export const setup = () => {
+  if (typeof global._localStorage !== 'undefined') {
+    Object.defineProperty(global, '_localStorage', {
+      value: new LocalStorage(jest),
+      writable: false,
+    });
+  } else {
+    global.localStorage = new LocalStorage(jest);
+  }
 
-if (typeof global._sessionStorage !== 'undefined') {
-  Object.defineProperty(global, '_sessionStorage', {
-    value: new LocalStorage(jest),
-    writable: false,
-  });
-} else {
-  global.sessionStorage = new LocalStorage(jest);
-}
+  if (typeof global._sessionStorage !== 'undefined') {
+    Object.defineProperty(global, '_sessionStorage', {
+      value: new LocalStorage(jest),
+      writable: false,
+    });
+  } else {
+    global.sessionStorage = new LocalStorage(jest);
+  }
+};