deepShallow.js

import { shallow } from 'enzyme';

const deepShallow = (endCondition) => (element) => {
  let wrapper = shallow(element);
  while (wrapper.node && !endCondition(wrapper)) {
    wrapper = wrapper.shallow();
  }

  return wrapper;
}

export const deepShallowTo = (ExpectedComp) => deepShallow(
  (wrapper) => wrapper.type() === ExpectedComp
);