Flatten styles in test snapshots

jest/__tests__/serializer-test.js

@@ -4,12 +4,9 @@
 import { mount, render, shallow } from 'enzyme';
 import React from 'react';
 import renderer from 'react-test-renderer';
-import serializer from '../serializer';
-import { StyleSheet, Text, View } from '../../dist';
+import { StyleSheet, Text, View } from '../../src';
 import toJson from 'enzyme-to-json';
 
-expect.addSnapshotSerializer(serializer);
-
 /**
  * Fixtures
  */

jest/createSerializer.js

@@ -0,0 +1,56 @@
+const React = require('react');
+
+function createSerializer(styleSheet) {
+  function flattenNodeStyles(node) {
+    if (node && node.props) {
+      // check for React elements in any props
+      const nextProps = Object.keys(node.props).reduce((acc, curr) => {
+        const value = node.props[curr];
+        if (React.isValidElement(value)) {
+          acc[curr] = flattenNodeStyles(value);
+        }
+        return acc;
+      }, {});
+
+      // flatten styles and avoid empty objects in snapshots
+      if (node.props.style) {
+        const style = styleSheet.flatten(node.props.style);
+        if (Object.keys(style).length > 0) {
+          nextProps.style = style;
+        } else {
+          delete nextProps.style;
+        }
+      }
+
+      const args = [node, nextProps];
+
+      // recurse over children too
+      const children = node.children || node.props.children;
+      if (children) {
+        if (Array.isArray(children)) {
+          children.forEach(child => {
+            args.push(flattenNodeStyles(child));
+          });
+        } else {
+          args.push(flattenNodeStyles(children));
+        }
+      }
+
+      return React.cloneElement.apply(React.cloneElement, args);
+    }
+
+    return node;
+  }
+
+  function test(value) {
+    return !!value && value.$$typeof === Symbol.for('react.test.json');
+  }
+
+  function print(value, serializer) {
+    return serializer(flattenNodeStyles(value));
+  }
+
+  return { test, print };
+}
+
+module.exports = createSerializer;

jest/serializer.js

@@ -1,61 +1,6 @@
-const React = require('react');
+const createSerializer = require('./createSerializer');
 const { StyleSheet } = require('../dist');
 
-function createSerializer(styleSheet) {
-  function flattenNodeStyles(node) {
-    if (node && node.props) {
-      // check for React elements in any props
-      const nextProps = Object.keys(node.props).reduce((acc, curr) => {
-        const value = node.props[curr];
-        if (React.isValidElement(value)) {
-          acc[curr] = flattenNodeStyles(value);
-        }
-        return acc;
-      }, {});
-
-      // flatten styles and avoid empty objects in snapshots
-      if (node.props.style) {
-        const style = styleSheet.flatten(node.props.style);
-        if (Object.keys(style).length > 0) {
-          nextProps.style = style;
-        } else {
-          delete nextProps.style;
-        }
-      }
-
-      const args = [node, nextProps];
-
-      // recurse over children too
-      const children = node.children || node.props.children;
-      if (children) {
-        if (Array.isArray(children)) {
-          children.forEach(child => {
-            args.push(flattenNodeStyles(child));
-          });
-        } else {
-          args.push(flattenNodeStyles(children));
-        }
-      }
-
-      return React.cloneElement.apply(React.cloneElement, args);
-    }
-
-    return node;
-  }
-
-  function test(value) {
-    return !!value && value.$$typeof === Symbol.for('react.test.json');
-  }
-
-  function print(value, serializer) {
-    return serializer(flattenNodeStyles(value));
-  }
-
-  return { test, print };
-}
-
 const serializer = createSerializer(StyleSheet);
-createSerializer.test = serializer.test;
-createSerializer.print = serializer.print;
 
-module.exports = createSerializer;
+module.exports = serializer;