import React, {Component, PropTypes} from "react";
import ReactDOM from "react-dom";

const wrapTestableComponent = (ComponentToWrap) => {

    class TestableComponent extends Component {

        componentDidMount() {
            ReactDOM.findDOMNode(this.wrappedRef).setAttribute("test_id",this.props.test_id);
        }

        render() {
            return <ComponentToWrap  {...this.props} ref={(wrapped) => { this.wrappedRef = wrapped; }} />;
        }
    }

    ComponentToWrap.propTypes = {
        test_id: PropTypes.string.isRequired,
    }

    return TestableComponent
}
export default wrapTestableComponent;