From 90759087fa5b5d2a991a433702a383a8cfe615a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=AE=9D=E7=A0=81?= <nongyoubao@alibaba-inc.com>
Date: Wed, 21 Dec 2016 23:32:29 +0800
Subject: [PATCH] fix: test case

---
 .babelrc     | 6 +++++-
 package.json | 4 +++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/.babelrc b/.babelrc
index 397caa9..9f79e09 100644
--- a/.babelrc
+++ b/.babelrc
@@ -1,4 +1,8 @@
 {
   "presets": ["es2015"],
-  "plugins": ["transform-async-to-generator"]
+  "env": {
+    "test": {
+      "plugins": ["transform-runtime", "transform-async-to-generator"]
+    }
+  }
 }
\ No newline at end of file
diff --git a/package.json b/package.json
index ca0fb53..e592dfb 100644
--- a/package.json
+++ b/package.json
@@ -5,7 +5,7 @@
   "main": "lib/index.js",
   "scripts": {
     "test": "jest",
-    "babel": "babel ./src --out-dir ./lib --presets es2015",
+    "babel": "babel ./src --out-dir ./lib",
     "prepublish": "npm run babel"
   },
   "files": [
@@ -34,7 +34,9 @@
     "babel-cli": "^6.18.0",
     "babel-jest": "^18.0.0",
     "babel-plugin-transform-async-to-generator": "^6.16.0",
+    "babel-plugin-transform-runtime": "^6.15.0",
     "babel-preset-es2015": "^6.18.0",
+    "babel-runtime": "^6.20.0",
     "co": "^4.6.0",
     "jest": "^18.0.0"
   },