Merge pull request #41 from smooth-code/fix-snap-getState

fix(snapshot): fix snap usage

README.md

@@ -334,7 +334,7 @@ You need to instruct your snapshot solution to save state of `loadable-component
 For example, to do this with [`react-snap`](https://github.com/stereobooster/react-snap) you can use following code:
 
 ```js
-import { getState } from 'loadable-components/snap'
+import { getState } from 'loadable-components'
 
 // Set up for react-snap.
 window.snapSaveState = () => getState()

src/snap/index.js → src/getState.js

@@ -1,9 +1,9 @@
 /* eslint-env browser */
 /* eslint-disable import/prefer-default-export */
-import { getAll as getAllComponents } from '../componentTracker'
-import { LOADABLE_STATE } from '../constants'
+import { getAll as getAllComponents } from './componentTracker'
+import { LOADABLE_STATE } from './constants'
 
-export function getState() {
+function getState() {
   const componentByIds = getAllComponents()
   const children = Object.keys(componentByIds).reduce((ids, id) => {
     const component = componentByIds[id]
@@ -12,3 +12,5 @@ export function getState() {
   }, [])
   return { [LOADABLE_STATE]: { children } }
 }
+
+export default getState

src/getState.test.js

@@ -0,0 +1,16 @@
+import { getState } from './'
+import loadable from './loadable'
+
+describe('#getState', () => {
+  it('should return only loaded components', () => {
+    const getComponent = jest.fn(() => import('./__fixtures__/Dummy'))
+    const Loadable = loadable(getComponent, {
+      modules: ['./__fixtures__/Dummy'],
+    })
+    expect(getState()).toEqual({ __LOADABLE_STATE__: { children: [] } })
+    Loadable.load()
+    expect(getState()).toEqual({
+      __LOADABLE_STATE__: { children: [{ id: './__fixtures__/Dummy' }] },
+    })
+  })
+})

src/index.js

@@ -1,6 +1,7 @@
 import * as tracker from './componentTracker'
 
 export { default as loadComponents } from './loadComponents'
+export { default as getState } from './getState'
 export { LOADABLE } from './constants'
 export { default } from './loadable'
 export const componentTracker = tracker