From 58da50aace5cd3e5a322ac0eada423de4cfc79e5 Mon Sep 17 00:00:00 2001
From: Henry Zhu <hi@henryzoo.com>
Date: Mon, 16 Oct 2017 11:09:11 -0400
Subject: [PATCH] simplify register test (#6391)

---
 packages/babel-register/test/__data__/es2015.js |  7 +------
 packages/babel-register/test/index.js           | 11 +++++++++--
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/packages/babel-register/test/__data__/es2015.js b/packages/babel-register/test/__data__/es2015.js
index fdc8c1bed4..69571eb70b 100644
--- a/packages/babel-register/test/__data__/es2015.js
+++ b/packages/babel-register/test/__data__/es2015.js
@@ -1,6 +1 @@
-import a from "assert";
-
-export default () => {
-  a.foo();
-  return "b";
-};
+import "assert";
diff --git a/packages/babel-register/test/index.js b/packages/babel-register/test/index.js
index 212daa58ba..2ba7ace225 100644
--- a/packages/babel-register/test/index.js
+++ b/packages/babel-register/test/index.js
@@ -1,5 +1,4 @@
 import chai from "chai";
-import path from "path";
 
 const DATA_ES2015 = require.resolve("./__data__/es2015");
 
@@ -12,7 +11,15 @@ describe("babel-register", function() {
     babelRegister.default(
       Object.assign(
         {
-          presets: [path.join(__dirname, "../../babel-preset-es2015")],
+          plugins: [
+            {
+              visitor: {
+                ImportDeclaration(path) {
+                  path.remove();
+                },
+              },
+            },
+          ],
           babelrc: false,
         },
         config,