reset stylesheet for all tests

Author: quantizor

src/index.js

@@ -1,8 +1,8 @@
-const toHaveStyleRule = require('./toHaveStyleRule')
-const styleSheetSerializer = require('./styleSheetSerializer')
-const { resetStyleSheet } = require('./utils')
+const toHaveStyleRule = require('./toHaveStyleRule');
+const styleSheetSerializer = require('./styleSheetSerializer');
+const { resetStyleSheet } = require('./utils');
 
-resetStyleSheet()
+global.beforeEach(resetStyleSheet);
 
-expect.addSnapshotSerializer(styleSheetSerializer)
-expect.extend({ toHaveStyleRule })
+expect.addSnapshotSerializer(styleSheetSerializer);
+expect.extend({ toHaveStyleRule });

test/setup.js

@@ -1,4 +1,6 @@
 import { configure } from 'enzyme';
 import Adapter from 'enzyme-adapter-react-16';
 
+import '../src';
+
 configure({ adapter: new Adapter() });

test/styleSheetSerializer.spec.js

@@ -3,8 +3,6 @@ import { mount, shallow } from 'enzyme';
 import React from 'react';
 import renderer from 'react-test-renderer';
 import styled, { ThemeContext, ThemeProvider } from 'styled-components';
-import '../src';
-import { resetStyleSheet } from '../src/utils';
 
 const toMatchSnapshot = (name, component) => {
   expect(renderer.create(component).toJSON()).toMatchSnapshot('react-test-renderer');
@@ -20,10 +18,6 @@ const shallowWithTheme = (tree, theme) => {
   return shallow(tree);
 };
 
-beforeEach(() => {
-  resetStyleSheet();
-});
-
 it('null', () => {
   expect(null).toMatchSnapshot();
 });

test/utils.spec.js

@@ -1,35 +1,32 @@
-const { getHashes } = require('../src/utils');
+const {
+  __PRIVATE__: { masterSheet },
+} = require('styled-components');
 
-jest.mock('styled-components', () => ({
-  ServerStyleSheet: true,
-  __PRIVATE__: {
-    masterSheet: {
-      names: new Map([['sc-1', new Set(['a'])], ['sc-2', new Set(['b', 'c'])], ['sc-3', new Set(['d', 'e'])]]),
-      toString() {
-        return `
-            <style data-styled="active">
-              .sc-1 {}
-              data-styled.g1[id="sc-1"]{content:"sc-1,"}
-              .a { color: red; }
-            </style>
-            <style data-styled="active">
-              .sc-2 {}
-              data-styled.g2[id="sc-2"]{content:"sc-2,"}
-              .b { color: green; }
-              .c { color: blue; }
-            </style>
-            <style data-styled="active">
-              .sc-3 {}
-              data-styled.g3[id="sc-3"]{content:"sc-3,"}
-              .d { color: pink; }
-              .e { color: indianred; }
-            </style>
-          `;
-      },
-    },
-  },
-}));
+const { getHashes } = require('../src/utils');
 
 it('extracts hashes', () => {
+  masterSheet.names = new Map([['sc-1', new Set(['a'])], ['sc-2', new Set(['b', 'c'])], ['sc-3', new Set(['d', 'e'])]]);
+  masterSheet.toString = function() {
+    return `
+      <style data-styled="active">
+        .sc-1 {}
+        data-styled.g1[id="sc-1"]{content:"sc-1,"}
+        .a { color: red; }
+      </style>
+      <style data-styled="active">
+        .sc-2 {}
+        data-styled.g2[id="sc-2"]{content:"sc-2,"}
+        .b { color: green; }
+        .c { color: blue; }
+      </style>
+      <style data-styled="active">
+        .sc-3 {}
+        data-styled.g3[id="sc-3"]{content:"sc-3,"}
+        .d { color: pink; }
+        .e { color: indianred; }
+      </style>
+    `;
+  };
+
   expect(getHashes()).toEqual(['sc-1', 'a', 'sc-2', 'b', 'c', 'sc-3', 'd', 'e']);
 });