stubContext.js

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

function stubContext(Component, context, props) {
  var _component = Component
  var _context = context
  var _props = props

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

  var ContextWrappedComponent = React.createClass({
    childContextTypes: _contextTypes,

    getChildContext: function() {
      return _context
    },

    render: function() {
      return <Component {...props} />
    }
  })

  ContextWrappedComponent.withProps = function(props) {
    return stubContext(_component, _context, assign({}, _props, props))
  }

  return ContextWrappedComponent
}

module.exports = stubContext