From fc7d01e2ab8c96e61b90785daf9ec7d2f6455620 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mart=C3=ADn=20Bigio?= <martinb@fb.com>
Date: Fri, 9 Oct 2015 07:56:00 -0700
Subject: [PATCH] fix E2E tests

Reviewed By: @mkonicek

Differential Revision: D2526310

fb-gh-sync-id: 1c04f56073dc5b393a36b258c039fdfd85ab83a2
---
 babelRegisterOnly.js    | 16 ++++++++++++++++
 packager.js             |  2 +-
 react-packager/index.js |  2 +-
 3 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 babelRegisterOnly.js

diff --git a/babelRegisterOnly.js b/babelRegisterOnly.js
new file mode 100644
index 00000000..349ba638
--- /dev/null
+++ b/babelRegisterOnly.js
@@ -0,0 +1,16 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree. An additional grant
+ * of patent rights can be found in the PATENTS file in the same directory.
+ */
+'use strict';
+
+var _only = [];
+
+module.exports = function(onlyList) {
+  _only = _only.concat(onlyList);
+  require('babel-core/register')({only: _only});
+};
diff --git a/packager.js b/packager.js
index 686f53f9..ddda537f 100644
--- a/packager.js
+++ b/packager.js
@@ -8,7 +8,7 @@
  */
 'use strict';
 
-require('../babelRegisterOnly')([
+require('./babelRegisterOnly')([
   /packager\/[^\/]*/
 ]);
 
diff --git a/react-packager/index.js b/react-packager/index.js
index b435a57a..64ad1985 100644
--- a/react-packager/index.js
+++ b/react-packager/index.js
@@ -8,7 +8,7 @@
  */
 'use strict';
 
-require('../../babelRegisterOnly')([/react-packager\/src/]);
+require('../babelRegisterOnly')([/react-packager\/src/]);
 
 useGracefulFs();