Merge pull request #38 from NickStefan/fullDOM

Allow rendering into the document.body

package.json

@@ -1,6 +1,6 @@
 {
   "name": "legit-tests",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "a chainable testing library for React",
   "main": "legit-tests.js",
   "scripts": {

src/middleware.js

@@ -1,9 +1,11 @@
 import Find from './middleware/find'
 import SetState from './middleware/setState'
 import Simulate from './middleware/simulate'
+import Clean from './middleware/clean'
 
 export default {
   Find,
   SetState,
-  Simulate
+  Simulate,
+  Clean
 }

src/middleware/clean.js

@@ -0,0 +1,7 @@
+export default function Clean(){
+  this.instance = null
+  this.elements = null
+  if (global.window){
+    global.window.document.body.innerHTML = ''
+  }
+}

src/tests.js

@@ -1,9 +1,10 @@
 import TestUtils from 'react-addons-test-utils'
 import ReactDOMServer from 'react-dom/server'
+import ReactDOM from 'react-dom'
 import React from 'react'
 global.React = React
 
-import { Find, SetState, Simulate } from './middleware'
+import { Find, SetState, Simulate, Clean } from './middleware'
 
 function Test(component, config) {
 
@@ -12,6 +13,10 @@ function Test(component, config) {
     const shallowRenderer = TestUtils.createRenderer()
     shallowRenderer.render(component)
     instance = shallowRenderer.getRenderOutput()
+  } else if (config && config.fullDOM && global.window) {
+    var div = global.window.document.createElement('div')
+    global.window.document.body.appendChild(div)
+    instance = ReactDOM.render(component, div)
   } else {
     instance = TestUtils.renderIntoDocument(component)
   }
@@ -69,7 +74,8 @@ function Test(component, config) {
   return testComponent.mixin({
     find: Find,
     setState: SetState,
-    simulate: Simulate
+    simulate: Simulate,
+    clean: Clean
   })
 }
 

tests/full-dom.jsx

@@ -0,0 +1,18 @@
+import Test from '../src/legit-tests'
+import { expect } from 'chai'
+
+describe('Render into document.body', () => {
+
+  it('should render and clean up component', () => {
+    Test(<section />, {fullDOM: true})
+    .test(function() {
+      expect(global.window.document.querySelector('section'))
+      .to.not.equal(null)
+    })
+    .clean()
+
+    // clean should clean up the document.body
+    expect(global.window.document.body.innerHTML).to.equal('')
+  })
+
+})