Pass jest runtime to setup scripts (closes jestjs#108)

johannhof · May 6, 2015 · ea97506 · ea97506
1 parent 68f3cbd
commit ea97506
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/HasteModuleLoader/HasteModuleLoader.js b/src/HasteModuleLoader/HasteModuleLoader.js
@@ -938,6 +938,10 @@ Loader.prototype.requireModuleOrMock = function(currPath, moduleName) {
   }
 };
 
+Loader.prototype.getJestRuntime = function(dir){
+    return this._builtInModules['jest-runtime'](dir).exports;
+};
+
 /**
  * Clears all cached module objects. This allows one to reset the state of
  * all modules in the system. It will reset (read: clear) the export objects

diff --git a/src/TestRunner.js b/src/TestRunner.js
@@ -362,7 +362,8 @@ TestRunner.prototype.runTest = function(testFilePath) {
           global: env.global,
           require: moduleLoader.constructBoundRequire(
             config.setupEnvScriptFile
-          )
+          ),
+          jest: moduleLoader.getJestRuntime(config.setupEnvScriptFile)
         }
       );
     }

diff --git a/src/jasmineTestRunner/jasmineTestRunner.js b/src/jasmineTestRunner/jasmineTestRunner.js
@@ -176,7 +176,8 @@ function jasmineTestRunner(config, environment, moduleLoader, testPath) {
           __filename: config.setupTestFrameworkScriptFile,
           require: moduleLoader.constructBoundRequire(
             config.setupTestFrameworkScriptFile
-          )
+          ),
+          jest: moduleLoader.getJestRuntime(config.setupTestFrameworkScriptFile)
         }
       );
     }