add tests

__fixtures__/invalidProps.vue

@@ -0,0 +1,15 @@
+<template>
+  <div></div>
+</template>
+
+<script>
+export default {
+  data () {
+    return {
+      props: {
+        a: String
+      }
+    }
+  }
+}
+</script>

__fixtures__/validProps.vue

@@ -0,0 +1,11 @@
+<template>
+  <div></div>
+</template>
+
+<script>
+export default {
+  props: {
+    a: String
+  }
+}
+</script>

package.json

@@ -18,7 +18,7 @@
     "node": ">=8"
   },
   "scripts": {
-    "test": "yarn lint && jest",
+    "test": "yarn lint && jest --verbose",
     "build": "yarn lint && bili",
     "postbuild": "find dist -type d -name __test__ -exec rm -r {} +",
     "prepublishOnly": "yarn build",

src/__test__/helpers.test.ts

@@ -0,0 +1,36 @@
+import traverse from '@babel/traverse'
+import { getAST } from './utils'
+import { isVueComponent, isPropsOption } from '../helpers'
+
+describe('helpers', () => {
+  const ast1 = getAST('validProps.vue')
+  const ast2 = getAST('invalidProps.vue')
+
+  test('The default export should be a Vue component', () => {
+    traverse(ast1, {
+      ExportDefaultDeclaration(path: any) {
+        expect(isVueComponent(path.node)).toBe(true)
+      }
+    })
+  })
+
+  test('The props property is a Vue component option', () => {
+    traverse(ast1, {
+      ObjectProperty(path: any) {
+        if (path.node.key.name === 'props') {
+          expect(isPropsOption(path)).toBe(true)
+        }
+      }
+    })
+  })
+
+  test('The props property is not a Vue component option', () => {
+    traverse(ast2, {
+      ObjectProperty(path: any) {
+        if (path.node.key.name === 'props') {
+          expect(isPropsOption(path)).toBe(false)
+        }
+      }
+    })
+  })
+})

src/__test__/index.test.ts

@@ -1,13 +1,8 @@
 import vuese, { ParserOptions } from '../index'
-import sfcToEstree from '../sfcToEstree'
-import * as path from 'path'
-import * as fs from 'fs'
-
-const p = path.resolve(__dirname, '../../__fixtures__/common.vue')
-const source = fs.readFileSync(p, 'utf-8')
+import { getAST } from './utils'
 
 test('get the correct props value', () => {
-  const ast = sfcToEstree(source)
+  const ast = getAST('common.vue')
   const options: ParserOptions = {
     onProp: origValue => {
       expect(origValue).toEqual({ a: String })

src/__test__/sfcToEstree.test.ts

@@ -1,12 +1,7 @@
-import sfcToEstree from '../sfcToEstree'
-import * as path from 'path'
-import * as fs from 'fs'
 import * as bt from '@babel/types'
-
-const p = path.resolve(__dirname, '../../__fixtures__/common.vue')
-const source = fs.readFileSync(p, 'utf-8')
+import { getAST } from './utils'
 
 test('The type of AST should be File', () => {
-  const ast = sfcToEstree(source)
+  const ast = getAST('common.vue')
   expect(bt.isFile(ast)).toBe(true)
 })

src/__test__/utils.ts

@@ -0,0 +1,11 @@
+import * as path from 'path'
+import * as fs from 'fs'
+import sfcToEstree from '../sfcToEstree'
+
+export function getAST(fileName: string): object {
+  const p = path.resolve(__dirname, `../../__fixtures__/${fileName}`)
+  const source = fs.readFileSync(p, 'utf-8')
+  return sfcToEstree(source)
+}
+
+test('placeholder', () => {})

src/helpers.ts

@@ -0,0 +1,36 @@
+import generate from '@babel/generator'
+import * as bt from '@babel/types'
+
+/**
+ * If a node satisfies the following conditions, then we will use this node as a Vue component.
+ * 1. It is a default export
+ * 2. others...
+ */
+export function isVueComponent(node: object): boolean {
+  return bt.isExportDefaultDeclaration(node)
+}
+
+export function getValueFromGenerate(node: any) {
+  let code: string = 'return'
+  const { code: genCode } = generate(node)
+  code += genCode
+  const fn = new Function(code)
+  try {
+    return fn()
+  } catch (e) {
+    console.error(e)
+  }
+}
+
+export function isPropsOption(path: any): boolean {
+  if (
+    bt.isObjectProperty(path.node) &&
+    path.parentPath &&
+    path.parentPath.parentPath &&
+    isVueComponent(path.parentPath.parentPath.node)
+  ) {
+    const keyPath = path.get('key')
+    return keyPath.node.name === 'props'
+  }
+  return false
+}

src/index.ts

@@ -1,10 +1,9 @@
 import traverse from '@babel/traverse'
-import { getValueFromGenerate } from './helpers/index'
+import { getValueFromGenerate, isPropsOption } from './helpers'
 
 const mainTraveres = {
   ObjectProperty(path: any) {
-    const keyPath = path.get('key')
-    if (keyPath.node.name === 'props') {
+    if (isPropsOption(path)) {
       const valuePath = path.get('value')
       const propsValue = getValueFromGenerate(valuePath.node)
       const { onProp } = (this as any).options
@@ -19,7 +18,7 @@ export interface ParserOptions {
   }
 }
 
-export default function(ast: object, options: ParserOptions = {}) {
+export default function(ast: any, options: ParserOptions = {}) {
   traverse(ast, {
     ExportDefaultDeclaration(path: any) {
       path.traverse(mainTraveres, {