fix(shallow): fix shallow test util

Author: Alexander Fedyashov

test/utils/shallow.js

@@ -2,17 +2,20 @@ import { ShallowWrapper } from 'enzyme'
 
 const meetsName = (wrapper, name) => name === wrapper.type().name
 
-const dive = wrapper => {
+const dive = (wrapper) => {
+  /* eslint-disable no-param-reassign */
   if (meetsName(wrapper, 'refHOC')) wrapper = wrapper.dive()
   if (meetsName(wrapper, 'Ref')) wrapper = wrapper.dive()
+  /* eslint-enable */
 
   return wrapper
 }
 
-class Wrapper extends ShallowWrapper {
-  shallow(options) {
-    return this.single('shallow', n => dive(new Wrapper(n, null, options)))
-  }
+// Heads up! This monkey patch overrides behaviour of shallow() method of Enzyme
+const shallowMethod = ShallowWrapper.prototype.shallow
+
+ShallowWrapper.prototype.shallow = function (...args) {
+  return dive(shallowMethod.apply(this, ...args))
 }
 
-export default (...args) => dive(new Wrapper(...args))
+export default (...args) => dive(new ShallowWrapper(...args))