class MyComponent extends Component {
  // Note you have to *return* the promise from your lifecycle method
  componentWillMount() {
   return new Promise(function(resolve) { 
       setTimeout(resolve, 100);
   });
  }
}

it('waits for the mount', async () => {
  const wrapper = await willMount(<MyComponent ... />);
});