replace variable reference with constant value

jestjs · Nov 1, 2020 · 9e088f1 · 9e088f1
1 parent bdff07a
commit 9e088f1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/packages/babel-plugin-jest-hoist/src/__tests__/__snapshots__/hoistPlugin.test.ts.snap b/packages/babel-plugin-jest-hoist/src/__tests__/__snapshots__/hoistPlugin.test.ts.snap
@@ -15,7 +15,7 @@ _getJestObj().mock("./App", () => () =>
     void 0,
     false,
     {
-      fileName: _jsxFileName,
+      fileName: "/root/project/src/file.js",
       lineNumber: 1,
       columnNumber: 32
     },

diff --git a/packages/babel-plugin-jest-hoist/src/index.ts b/packages/babel-plugin-jest-hoist/src/index.ts
@@ -146,7 +146,9 @@ FUNCTIONS.mock = args => {
             const initNode = binding.path.node.init;
 
             if (initNode && binding.constant && scope.isPure(initNode, true)) {
-              // how to hoist???
+              // replace the reference with its constant value
+              id.replaceWith(initNode);
+
               isAllowedIdentifier = true;
             }
           }
@@ -286,7 +288,7 @@ export default (): PluginObj<{
   visitor: {
     ExpressionStatement(exprStmt) {
       const jestObjExpr = extractJestObjExprIfHoistable(
-        exprStmt.get<'expression'>('expression'),
+        exprStmt.get('expression'),
       );
       if (jestObjExpr) {
         jestObjExpr.replaceWith(