Tests added for withFirestore.

package.json

@@ -82,12 +82,14 @@
     "babel-register": "^6.24.0",
     "chai": "^3.5.0",
     "chai-as-promised": "^6.0.0",
+    "chai-enzyme": "^0.8.0",
     "child-process-promise": "^2.2.1",
     "codecov": "^2.1.0",
     "cross-env": "^5.1.0",
     "docdown": "^0.7.2",
     "documentation": "^4.0.0-beta15",
     "documentation-markdown-api-theme": "^1.0.2",
+    "enzyme": "^2.3.0",
     "eslint": "^3.19.0",
     "eslint-config-standard": "^10.0.0",
     "eslint-config-standard-react": "^4.3.0",
@@ -107,6 +109,7 @@
     "react-addons-test-utils": "^15.4.2",
     "react-dom": "^15.4.2",
     "redux": "3.6.0",
+    "redux-firestore": "^0.2.0",
     "rimraf": "^2.6.2",
     "sinon": "^2.1.0",
     "sinon-chai": "^2.9.0",

src/withFirestore.js

@@ -38,6 +38,7 @@ export const createWithFirestore = (storeKey = 'store') => WrappedComponent => {
           {...this.state}
           firestore={this.store.firestore}
           firebase={this.store.firebase}
+          dispatch={this.store.dispatch}
         />
       )
     }

tests/setup.js

@@ -1,13 +1,14 @@
 /* eslint-disable no-unused-vars */
 process.env.NODE_ENV = 'test'
 
-var chai = require('chai')
-var sinon = require('sinon')
-var chaiAsPromised = require('chai-as-promised')
-var sinonChai = require('sinon-chai')
-var jsdom = require('jsdom').jsdom
-var FirebaseServer = require('firebase-server')
-var uid = global.uid = 'Iq5b0qK2NtgggT6U3bU6iZRGyma2'
+const chai = require('chai')
+const sinon = require('sinon')
+const chaiAsPromised = require('chai-as-promised')
+const sinonChai = require('sinon-chai')
+const jsdom = require('jsdom').jsdom
+const chaiEnzyme = require('chai-enzyme')
+const FirebaseServer = require('firebase-server')
+const uid = global.uid = 'Iq5b0qK2NtgggT6U3bU6iZRGyma2'
 
 new FirebaseServer(5000, 'localhost.firebaseio.test', { // eslint-disable-line no-new
   users: {
@@ -20,6 +21,7 @@ new FirebaseServer(5000, 'localhost.firebaseio.test', { // eslint-disable-line n
 // Chai Plugins
 chai.use(chaiAsPromised)
 chai.use(sinonChai)
+chai.use(chaiEnzyme())
 
 // globals
 global.expect = chai.expect

tests/unit/withFirestore.spec.js

@@ -0,0 +1,64 @@
+import React, { createElement } from 'react'
+import TestUtils from 'react-addons-test-utils'
+import { createSink, nest, renderComponent } from 'recompose'
+import {mount, render, shallow} from 'enzyme'
+import { TestContainer, ProviderMock, storeWithFirestore } from '../utils/components'
+import withFirestore, { createWithFirestore } from '../../src/withFirestore'
+
+let store
+let TestComponent
+let wrapper
+
+describe('withFirestore', () => {
+  beforeEach(() => {
+    store = storeWithFirestore()
+    TestComponent = withFirestore(createSink)
+    wrapper = shallow(<TestComponent />, { context: { store } })
+  })
+
+  it('adds firestore as prop', () => {
+    expect(wrapper.prop('firestore')).to.exist
+    expect(wrapper.prop('firestore')).to.respondTo('add')
+  })
+
+  it('adds firebase as prop', () => {
+    expect(wrapper.prop('firebase')).to.exist
+    expect(wrapper.prop('firebase')).to.respondTo('push')
+  })
+
+  it('adds dispatch as prop', () => {
+    expect(wrapper.prop('dispatch')).to.exist
+    expect(wrapper.prop('dispatch')).to.be.a.function
+  })
+
+  describe('sets displayName static as', () => {
+    describe('withFirestore(${WrappedComponentName}) for', () => { // eslint-disable-line no-template-curly-in-string
+      it.skip('standard components', () => {
+        wrapper = shallow(<TestComponent />, { context: { store } })
+        expect(wrapper.instance.displayName).to.equal(`withFirestore(TestContainer)`)
+      })
+
+      it('string components', () => {
+        const str = 'Test'
+        const stringComp = withFirestore(str)
+        expect(stringComp.displayName).to.equal(`withFirestore(${str})`)
+      })
+    })
+
+    it.skip('"Component" for all other types', () => {
+      wrapper = shallow(withFirestore()(<div />))
+      expect(wrapper.displayName).to.equal('withFirestore(Component)')
+    })
+  })
+
+  it.skip('sets WrappedComponent static as component which was wrapped', () => {
+    wrapper = shallow(<TestComponent />, { context: { store } })
+    expect(wrapper.wrappedComponent).to.be.instanceOf(TestComponent)
+  })
+})
+
+describe('createwithFirestore', () => {
+  it('accepts a store key', () => {
+    createWithFirestore('store')
+  })
+})

tests/utils/components.js

@@ -0,0 +1,60 @@
+import React, { Children, Component, cloneElement } from 'react'
+import PropTypes from 'prop-types'
+import ReactDOM from 'react-dom'
+import TestUtils from 'react-addons-test-utils'
+import { createSink } from 'recompose'
+import { createStore, compose, combineReducers } from 'redux'
+import reactReduxFirebase from '../../src/enhancer'
+import { reduxFirestore } from 'redux-firestore'
+
+export class Passthrough extends Component {
+  render () {
+    return <div></div>
+  }
+}
+
+export class ProviderMock extends Component {
+  getChildContext () {
+    return { store: this.props.store }
+  }
+
+  state = { test: null, dynamic: '' }
+
+  render () {
+    return Children.only(
+      cloneElement(this.props.children, {
+        testProp: this.state.test,
+        dynamicProp: this.state.dynamic
+      }))
+  }
+}
+
+ProviderMock.childContextTypes = {
+  store: PropTypes.object.isRequired
+}
+
+ProviderMock.propTypes = {
+  store: PropTypes.object,
+  children: PropTypes.node
+}
+
+export class TestContainer extends Component {
+  render () {
+    return <Passthrough {...this.props} />
+  }
+}
+
+export const storeWithFirebase = () => {
+  const createStoreWithMiddleware = compose(
+    reactReduxFirebase(Firebase, { userProfile: 'users' })
+  )(createStore)
+  return createStoreWithMiddleware(combineReducers({ test: (state = {}) => state }))
+}
+
+export const storeWithFirestore = () => {
+  const createStoreWithMiddleware = compose(
+    reactReduxFirebase(Firebase, { userProfile: 'users' }),
+    reduxFirestore(Firebase)
+  )(createStore)
+  return createStoreWithMiddleware(combineReducers({ test: (state = {}) => state }))
+}