Проверка асинхронного вложенного компонента

Скажем, у меня есть следующий компонент оболочки:

'use strict'

import React, {PropTypes, PureComponent} from 'react'
import {update} from '../../actions/actions'
import LoadFromServerButton from '../LoadFromServerButton'
import {connect} from 'react-redux'

export class FooDisplay extends PureComponent {
  render () {
    return (
      <p>
        <span className='foo'>
          {this.props.foo}
        </span>
        <LoadFromServerButton updateFunc={this.props.update} />
      </p>
    )
  }
}

export const mapStateToProps = (state) => {
  return {foo: state.foo.foo}
}

FooDisplay.propTypes = {
  foo: PropTypes.string
}

export const mapDispatchToProps = (dispatch) => {
  return {
    update: (foo) => dispatch(update(foo))
  }
}

export default connect(mapStateToProps, mapDispatchToProps)(FooDisplay)

И следующий внутренний компонент:

'use strict'

import React, {PropTypes, PureComponent} from 'react'
import {get} from '../../actions/actions'
import ActiveButton from '../ActiveButton'
import {connect} from 'react-redux'

export class LoadFromServerButton extends PureComponent {
  doUpdate () {
    return this.props.get().then(this.props.updateFunc)
  }

  render () {
    return (
      <ActiveButton action={this.doUpdate.bind(this)} actionArguments={[this.props.foo]} text='fetch serverside address' />
    )
  }
}

export const mapStateToProps = (state) => {
  return {foo: state.foo.foo}
}

export const mapDispatchToProps = (dispatch) => {
  return {
    get: () => dispatch(get())
  }
}

LoadAddressFromServerButton.propTypes = {
  updateFunc: PropTypes.func.isRequired
}

export default connect(mapStateToProps, mapDispatchToProps)(LoadFromServerButton)

ActiveButton - очень тонкая обертка вокруг кнопки с кликом и деструктуризацией аргументов.

Теперь давайте скажем, что мое действие get записывается следующим образом:

export const get = () => dispatch => http('/dummy_route')
      .spread((response, body) => dispatch(actOnThing(update, body)))

Теперь, если я напишу тест так:

/* global window, test, expect, beforeAll, afterAll, describe */

'use strict'

import React from 'react'
import FooDisplay from './index'
import {mount} from 'enzyme'
import {Provider} from 'react-redux'
import configureStore from '../../store/configureStore'
import nock, {uriString} from '../../config/nock'
import _ from 'lodash'


const env = _.cloneDeep(process.env)
describe('the component behaves correctly when integrating with store and reducers/http', () => {
  beforeAll(() => {
    nock.disableNetConnect()
    process.env.API_URL = uriString
  })

  afterAll(() => {
    process.env = _.cloneDeep(env)
    nock.enableNetConnect()
    nock.cleanAll()
  })

  test('when deep rendering, the load event populates the input correctly', () => {
    const store = configureStore({
      address: {
        address: 'foo'
      }
    })
    const display = mount(<Provider store={store}><FooDisplay /></Provider>,
        {attachTo: document.getElementById('root')})
    expect(display.find('p').find('.address').text()).toEqual('foo')
    const button = display.find('LoadFromServerButton')
    expect(button.text()).toEqual('fetch serverside address')
    nock.get('/dummy_address').reply(200, {address: 'new address'})
    button.simulate('click')
  })
})

Это приводит к:

Unhandled rejection Error: Error: connect ECONNREFUSED 127.0.0.1:8080

Немного подумав, это связано с тем, что тест не возвращает обещание, поскольку нажатие кнопки приводит к тому, что обещание срабатывает из-под колпака, поэтому afterAll запускается немедленно, очищает nock и реальное http соединение идет по проводам.

Как я могу проверить этот случай? Кажется, у меня нет простого способа вернуть правильное обещание ... Как проверить обновления DOM, полученные в результате этих обновлений?