Fixed the problem of mapMutations, mapGetters, mapActions with vuex

Author: rafaesc

package.json

@@ -1,6 +1,6 @@
 {
   "name": "vue-docgen-api",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "description": "Toolbox to extract information from Vue component files for documentation generation purposes.",
   "bugs": {
     "url": "https://github.com/vue-styleguidist/vue-docgen-api/issues"
@@ -32,6 +32,7 @@
     "babel-preset-env": "^1.5.2",
     "babel-preset-es2015": "^6.24.1",
     "babel-preset-flow": "^6.23.0",
+    "babel-preset-stage-2": "^6.24.1",
     "doctrine": "^2.0.0",
     "enhanced-require": "^0.5.0-beta6",
     "hash-sum": "^1.0.2",

src/utils/getSandbox.js

@@ -1,4 +1,3 @@
-import getDocFile from './getDocFile';
 import stateDoc from './stateDoc';
 const babel = require('babel-core');
 const fs = require('fs');
@@ -10,7 +9,7 @@ const babelifyCode = code => {
 	const options = {
 		ast: false,
 		comments: false,
-		presets: ['babel-preset-es2015'],
+		presets: ['babel-preset-es2015', 'babel-preset-stage-2'],
 	};
 	return babel.transform(code, options);
 };
@@ -68,7 +67,15 @@ const evalComponentCode = (code) => {
 				exports: {
 				},
 			},
-			require:()=> {
+			require:(element)=> {
+				if (element === 'vuex') {
+					return {
+						mapState: function(){},
+						mapMutations: function(){},
+						mapGetters: function(){},
+						mapActions: function(){}
+					}
+				}
 				return function(){}
 			},
 			document: {},

tests/components/vuex/example.vue

@@ -0,0 +1,23 @@
+<template>
+</template>
+<script>
+    import {mapGetters} from 'vuex'
+    /**
+    * Partial mapping, object spread operator example
+    */
+    export default {
+        name: 'example',
+        computed: {
+            total () {
+                return this.$store.getters.allCustomers.length
+            },
+            ...mapGetters({
+                duplicate: 'hasDuplicateCustomer',
+                characters: 'alphaKeys'
+            })
+        },
+        mounted () {
+            console.log('Second component loaded')
+        }
+    }
+</script>

tests/parse.js

@@ -2,11 +2,12 @@ const path = require('path');
 var api = require('../dist/main');
 var grid = path.join(__dirname,  './components/grid/Grid.vue');
 var button = path.join(__dirname,  './components/button/Button.vue');
+var exampleVuex = path.join(__dirname,  './components/vuex/example.vue');
 const expect = require("chai").expect;
 let docGrid;
 let docButton;
 
-describe('tests lite grid', () => {
+describe('tests grid', () => {
 	docGrid = api.parse(grid);
 
 	it('should return an object', () => {
@@ -54,9 +55,9 @@ describe('tests lite grid', () => {
 	})
 })
 
-describe('tests lite button', () => {
+describe('tests button', () => {
 	docButton = api.parse(button);
-	console.log(JSON.stringify(docButton, null, 2));
+	// console.log(JSON.stringify(docButton, null, 2));
 
 	it('should return an object', () => {
 		expect(docButton).to.be.an('object')
@@ -102,3 +103,19 @@ describe('tests lite button', () => {
 		expect(Object.keys(docButton['props']).length).to.equal(4)
 	})
 })
+
+describe('test example vuex', () =>{
+	const docVuex = api.parse(exampleVuex);
+
+	it('should return an object', () => {
+		expect(docVuex).to.be.an('object')
+	})
+
+	it('The component name should be example', () => {
+		expect(docVuex.displayName).to.equal('example');
+	})
+
+	it('The component should have a description', () => {
+		expect(docVuex.description).to.equal('Partial mapping, object spread operator example');
+	})
+})