var React = require.requireActual('react')
var assign = require.requireActual('object-assign')

function stubContext(Component, context) {
  var _component = Component
  var _context = context
  var _contextTypes = {}

  Object.keys(_context).forEach((key) => {
    _contextTypes[key] = React.PropTypes.any
  })

  var ContextWrapper = React.createClass({
    childContextTypes: _contextTypes,
    getChildContext: function() { return _context },
    render: function() { return React.Children.only(this.props.children) }
  })

  var ContextWrappedComponent = React.createClass({
    render() {
      return <ContextWrapper><Component {...this.props} /></ContextWrapper>
    }
  })

  return ContextWrappedComponent
}

module.exports = stubContext