Merge pull request #4 from santino/fix_hooks

Fix for React hooks

src/NavigationListener.jsx

@@ -1,5 +1,4 @@
 import React from 'react'
-import PropTypes from 'prop-types'
 import { action } from '@storybook/addon-actions'
 import { linkTo } from '@storybook/addon-links'
 import { routerShape } from 'found/PropTypes'
@@ -62,12 +61,11 @@ class NavigationListener extends React.Component {
   }
 
   render () {
-    return this.props.story()
+    return <>{this.props.children}</>
   }
 }
 
 NavigationListener.propTypes = {
-  story: PropTypes.func.isRequired,
   router: routerShape.isRequired
 }
 

src/NavigationListener.test.js

@@ -1,5 +1,5 @@
 import React from 'react'
-import { mount, shallow } from 'enzyme'
+import { mount } from 'enzyme'
 import { action } from '@storybook/addon-actions'
 import { linkTo } from '@storybook/addon-links'
 import NavigationListener from './NavigationListener'
@@ -39,18 +39,18 @@ describe('NavigationListener', () => {
     }
   }
   const wrap = () =>
-    mount(<NavigationListener {...routerProp} story={storyFn} />)
+    mount(<NavigationListener {...routerProp}>{storyFn()}</NavigationListener>)
 
   describe('React Component', () => {
     afterEach(() => {
       routerProp.router.addNavigationListener.mockClear()
     })
 
-    it('renders the passed story element as a children', () => {
+    it('renders children component (story) correctly', () => {
       const wrapper = wrap()
 
       expect(wrapper.find('NavigationListener')).toHaveLength(1)
-      expect(wrapper.children().html()).toEqual(shallow(storyFn()).html())
+      expect(wrapper.children().equals(storyFn())).toBe(true)
     })
 
     it('adds the internal onNavigation method as a NavigationListener', () => {

src/index.js

@@ -35,7 +35,7 @@ const storyRouterDecorator = (routes = [{}], initialLocation = '') => {
           path: rootPath,
           Component: NavigationListener,
           render: ({ Component, props }) => (
-            <Component {...props} story={story} />
+            <Component {...props}>{story()}</Component>
           ),
           children: routes.map(normaliseRouteChildren)
         }