Async mode

package.json

@@ -27,7 +27,7 @@
   },
   "homepage": "https://github.com/lukechilds/react-jsdom",
   "dependencies": {
-    "react-dom": "^15.6.1",
+    "react-dom": "15.x",
     "window": "^4.1.1"
   },
   "devDependencies": {

src/index.js

@@ -22,13 +22,39 @@ const ReactJSDOM = {
 
 		ReactDOM.render(component, container);
 
+		return {
+			element: container.children.length > 1 ? container.children : container.children[0],
+			restore: () => {
 				Object.keys(origGlobals).forEach(prop => {
 					global[prop] = origGlobals[prop];
 				});
-
-		return container.children.length > 1 ? container.children : container.children[0];
 			}
-
+		};
+	}
 };
 
+Object.defineProperty(ReactJSDOM, 'async', {
+	enumerable: true,
+  configurable: false,
+  get: () => {
+  	let resolve;
+  	let element;
+  	let restore;
+  	return {
+  		done: () => {
+				resolve({
+					element,
+					restore
+				});
+			},
+			render: (component, window) => new Promise((res, rej) => {
+				resolve = res;
+				const r = ReactJSDOM.render(component, window);
+				element = r.element;
+				restore = r.restore;
+			});
+		};
+  }
+});
+
 module.exports = ReactJSDOM;

test/react-jsdom.js

@@ -1,7 +1,7 @@
 import test from 'ava';
 import Window from 'window';
 import React from 'react';
-import ReactJSDOM from 'this';
+import ReactJSDOM from '../src';
 
 test('ReactJSDOM is a object', t => {
 	t.is(typeof ReactJSDOM, 'object');
@@ -25,7 +25,7 @@ test('ReactJSDOM renders a React Component', t => {
 	t.is(elem.textContent, 'hi');
 });
 
-test('ReactJSDOM allows window instance to be passed in', t => {
+test('ReactJSDOM allows window instance to be passed in', async t => {
 	const window = new Window();
 	const elem = ReactJSDOM.render(
 		React.createElement('div', {}, 'hi'),
@@ -35,3 +35,37 @@ test('ReactJSDOM allows window instance to be passed in', t => {
 	t.is(elem.nodeName, 'DIV');
 	t.is(elem.textContent, 'hi');
 });
+
+test('Works with asynchronous components', async t => {
+	const originalDocument = global.document;
+	const { done, render } = ReactJSDOM.async;
+
+	class Test extends React.Component {
+		constructor(props) {
+			super(props);
+			this.state = {
+				children: 'test'
+			};
+		}
+		componentDidMount() {
+			setTimeout(() => {
+				this.setState({
+					children: 'updated'
+				});
+			}, 500);
+		}
+		componentDidUpdate() {
+			t.is(this.state.children, 'updated');
+			done();
+		}
+		render() {
+			return React.createElement('div', {}, this.state.children);
+		}
+	}
+
+	const { element, restore } = await render(React.createElement(Test));
+
+	t.is(element.nodeName, 'DIV');
+	restore();
+	t.is(global.document, originalDocument);
+});