Как проверить изменение состояния между двумя компонентами после нажатия кнопки?

Question

У меня есть 2 компонента для этой задачи. Первый - это мой компонент-оболочка, а второй - компонент для кнопок.

Я нажимаю одну из кнопок моего компонента-кнопки, и она меняет состояние в моей оболочке.component.

Теперь, как мне перенести все это действие в среду Jest (Enzyme)?

it('should update state', () => {
      const onButtonClickMock = jest.fn();
      const wrapper = shallow(<MyComponent/>)
      const buttons = shallow(<Colors onClick={onButtonClickMock} colors={['gray', 'black', 'white']}/>);


      const d = buttons.find('div#buttons');
      const b = d.find('button.btn');
      b.at(1).simulate('click');//Pick second button from bunch and click

      wrapper.update();

      expect(onButtonClickMock).toHaveBeenCalledTimes(1);//This works
      expect(component2.state(['selection'])).toBe(1);

})