fix: shallow snapshots with decorators [no issue] (#170)

ornikar · May 6, 2019 · e98bf4e · e98bf4e
1 parent 2d467f7
commit e98bf4e
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/@ornikar/jest-config-react/__mocks__/@storybook/react.js b/@ornikar/jest-config-react/__mocks__/@storybook/react.js
@@ -85,16 +85,13 @@ exports.storiesOf = groupName => {
               expect(shallowToJson(child.dive())).toMatchSnapshot();
             });
           } else {
-            let belowWrapper = true;
-            const toJsonMap = json => {
-              if (!belowWrapper) return { ...json, props: {} };
-              if (json.type === 'JestStoryWrapper') {
-                belowWrapper = false;
-              }
-              return json;
-            };
             expect(
-              shallowToJson(wrapper, { map: toJsonMap })
+              shallowToJson(
+                wrapper
+                  .find(JestStoryWrapper)
+                  .children()
+                  .shallow()
+              )
             ).toMatchSnapshot();
           }
         });