Test passing in document

package.json

@@ -40,6 +40,7 @@
     "ava": "^0.17.0",
     "browser-env": "^2.0.19",
     "coveralls": "^2.11.15",
+    "jsdom": "^9.9.1",
     "nyc": "^10.0.0",
     "xo": "^0.17.1"
   }

test/unit.js

@@ -1,12 +1,34 @@
 import test from 'ava';
+import jsdom from 'jsdom';
 import domLoaded from '../';
 
-test('domLoaded runs the callback', async t => {
+test('callback fires with global window', async t => {
 	t.plan(1);
 	domLoaded(() => t.pass());
 });
 
-test('domLoaded Promise resolves', async t => {
+test('Promise resolves with global window', async t => {
 	t.plan(1);
 	domLoaded().then(() => t.pass());
 });
+
+test('callback fires with local document', async t => {
+	t.plan(1);
+	const config = {
+		html: '',
+		onload: window => domLoaded(() => t.pass(), window.document)
+	};
+	jsdom.env(config);
+});
+
+test.cb('Promise resolves with local document', t => {
+	t.plan(1);
+	const config = {
+		html: '',
+		onload: window => domLoaded(window.document).then(() => {
+			t.pass();
+			t.end();
+		})
+	};
+	jsdom.env(config);
+});