feat(transform): pass config options through to transformer

jestjs · Dec 7, 2020 · f143ce6 · f143ce6
1 parent 0f09ca4
commit f143ce6
Show file tree

Hide file tree

Showing 8 changed files with 62 additions and 37 deletions.
diff --git a/docs/CodeTransformation.md b/docs/CodeTransformation.md
@@ -35,17 +35,17 @@ interface Transformer<OptionType = unknown> {
   getCacheKey?: (
     sourceText: string,
     sourcePath: string,
-    options: TransformOptions,
+    options: TransformOptions<OptionType>,
   ) => string;
 
   process: (
     sourceText: string,
     sourcePath: string,
-    options: TransformOptions,
+    options: TransformOptions<OptionType>,
   ) => TransformedSource;
 }
 
-interface TransformOptions {
+interface TransformOptions<OptionType> {
   /**
    * If a transformer does module resolution and reads files, it should populate `cacheFS` so that
    * Jest avoids reading the same files again, improving performance. `cacheFS` stores entries of
@@ -61,6 +61,8 @@ interface TransformOptions {
   supportsExportNamespaceFrom: boolean;
   supportsStaticESM: boolean;
   supportsTopLevelAwait: boolean;
+  /** the options passed through Jest's config by the user */
+  transformerConfig: OptionType;
 }
 
 type TransformedSource =

diff --git a/packages/babel-jest/src/index.ts b/packages/babel-jest/src/index.ts
@@ -50,7 +50,7 @@ const createTransformer: CreateTransformer = userOptions => {
   function loadBabelConfig(
     cwd: Config.Path,
     filename: Config.Path,
-    transformOptions?: JestTransformOptions,
+    transformOptions: JestTransformOptions,
   ): PartialConfig {
     // `cwd` first to allow incoming options to override it
     const babelConfig = loadPartialConfig({
@@ -59,16 +59,16 @@ const createTransformer: CreateTransformer = userOptions => {
       caller: {
         ...options.caller,
         supportsDynamicImport:
-          transformOptions?.supportsDynamicImport ??
+          transformOptions.supportsDynamicImport ??
           options.caller.supportsDynamicImport,
         supportsExportNamespaceFrom:
-          transformOptions?.supportsExportNamespaceFrom ??
+          transformOptions.supportsExportNamespaceFrom ??
           options.caller.supportsExportNamespaceFrom,
         supportsStaticESM:
-          transformOptions?.supportsStaticESM ??
+          transformOptions.supportsStaticESM ??
           options.caller.supportsStaticESM,
         supportsTopLevelAwait:
-          transformOptions?.supportsTopLevelAwait ??
+          transformOptions.supportsTopLevelAwait ??
           options.caller.supportsTopLevelAwait,
       },
       filename,

diff --git a/packages/jest-repl/src/cli/repl.ts b/packages/jest-repl/src/cli/repl.ts
@@ -15,6 +15,7 @@ import type {Transformer} from '@jest/transform';
 import type {Config} from '@jest/types';
 
 let transformer: Transformer;
+let transformerConfig: unknown;
 
 const evalCommand: repl.REPLEval = (
   cmd: string,
@@ -37,6 +38,7 @@ const evalCommand: repl.REPLEval = (
           supportsExportNamespaceFrom: false,
           supportsStaticESM: false,
           supportsTopLevelAwait: false,
+          transformerConfig,
         },
       );
       cmd =
@@ -69,6 +71,7 @@ if (jestProjectConfig.transform) {
   for (let i = 0; i < jestProjectConfig.transform.length; i++) {
     if (new RegExp(jestProjectConfig.transform[i][0]).test('foobar.js')) {
       transformerPath = jestProjectConfig.transform[i][1];
+      transformerConfig = jestProjectConfig.transform[i][2];
       break;
     }
   }

diff --git a/packages/jest-transform/src/ScriptTransformer.ts b/packages/jest-transform/src/ScriptTransformer.ts
@@ -67,7 +67,10 @@ export default class ScriptTransformer {
   private readonly _cache: ProjectCache;
   private readonly _cacheFS: StringMap;
   private readonly _config: Config.ProjectConfig;
-  private readonly _transformCache: Map<Config.Path, Transformer>;
+  private readonly _transformCache: Map<
+    Config.Path,
+    {transformer: Transformer; transformerConfig: unknown}
+  >;
   private readonly _transformConfigCache: Map<Config.Path, unknown>;
 
   constructor(
@@ -102,7 +105,8 @@ export default class ScriptTransformer {
     options: ReducedTransformOptions,
   ): string {
     const configString = this._cache.configString;
-    const transformer = this._getTransformer(filename);
+    const {transformer, transformerConfig = {}} =
+      this._getTransformer(filename) || {};
 
     if (transformer && typeof transformer.getCacheKey === 'function') {
       return createHash('md5')
@@ -112,6 +116,7 @@ export default class ScriptTransformer {
             cacheFS: this._cacheFS,
             config: this._config,
             configString,
+            transformerConfig,
           }),
         )
         .update(CACHE_VERSION)
@@ -181,28 +186,30 @@ export default class ScriptTransformer {
       return null;
     }
 
-    const transformer = this._transformCache.get(transformPath);
-    if (transformer) {
-      return transformer;
+    const cached = this._transformCache.get(transformPath);
+    if (cached) {
+      return cached;
     }
 
-    let transform: Transformer = require(transformPath);
+    let transformer: Transformer = require(transformPath);
 
-    if (!transform) {
+    if (!transformer) {
       throw new TypeError('Jest: a transform must export something.');
     }
-    const transformerConfig = this._transformConfigCache.get(transformPath);
-    if (typeof transform.createTransformer === 'function') {
-      transform = transform.createTransformer(transformerConfig);
+    const transformerConfig =
+      this._transformConfigCache.get(transformPath) || {};
+    if (typeof transformer.createTransformer === 'function') {
+      transformer = transformer.createTransformer(transformerConfig);
     }
-    if (typeof transform.process !== 'function') {
+    if (typeof transformer.process !== 'function') {
       throw new TypeError(
         'Jest: a transform must export a `process` function.',
       );
     }
-    this._transformCache.set(transformPath, transform);
+    const res = {transformer, transformerConfig};
+    this._transformCache.set(transformPath, res);
 
-    return transform;
+    return res;
   }
 
   private _instrumentFile(
@@ -262,18 +269,19 @@ export default class ScriptTransformer {
     options: ReducedTransformOptions,
   ): TransformResult {
     const filename = tryRealpath(filepath);
-    const transform = this._getTransformer(filename);
+    const {transformer, transformerConfig = {}} =
+      this._getTransformer(filename) || {};
     const cacheFilePath = this._getFileCachePath(filename, content, options);
     let sourceMapPath: Config.Path | null = cacheFilePath + '.map';
     // Ignore cache if `config.cache` is set (--no-cache)
     let code = this._config.cache ? readCodeCacheFile(cacheFilePath) : null;
 
-    const shouldCallTransform = transform && this.shouldTransform(filename);
+    const shouldCallTransform = transformer && this.shouldTransform(filename);
 
     // That means that the transform has a custom instrumentation
     // logic and will handle it based on `config.collectCoverage` option
     const transformWillInstrument =
-      shouldCallTransform && transform && transform.canInstrument;
+      shouldCallTransform && transformer && transformer.canInstrument;
 
     if (code) {
       // This is broken: we return the code, and a path for the source map
@@ -292,12 +300,13 @@ export default class ScriptTransformer {
       map: null,
     };
 
-    if (transform && shouldCallTransform) {
-      const processed = transform.process(content, filename, {
+    if (transformer && shouldCallTransform) {
+      const processed = transformer.process(content, filename, {
         ...options,
         cacheFS: this._cacheFS,
         config: this._config,
         configString: this._cache.configString,
+        transformerConfig,
       });
 
       if (typeof processed === 'string') {
@@ -343,7 +352,7 @@ export default class ScriptTransformer {
        *
        */
       const shouldEmitSourceMaps =
-        (transform != null && map != null) || transform == null;
+        (transformer != null && map != null) || transformer == null;
 
       const instrumented = this._instrumentFile(
         filename,

diff --git a/.../src/__tests__/script_transformer.test.ts → ...m/src/__tests__/ScriptTransformer.test.ts b/.../src/__tests__/script_transformer.test.ts → ...m/src/__tests__/ScriptTransformer.test.ts
@@ -663,7 +663,10 @@ describe('ScriptTransformer', () => {
   });
 
   it('passes expected transform options to getCacheKey', () => {
-    config = {...config, transform: [['\\.js$', 'test_preprocessor', {}]]};
+    config = {
+      ...config,
+      transform: [['\\.js$', 'test_preprocessor', {configKey: 'configValue'}]],
+    };
     const scriptTransformer = new ScriptTransformer(config);
 
     scriptTransformer.transform(
@@ -779,8 +782,8 @@ describe('ScriptTransformer', () => {
 
     scriptTransformer.transform(fileName2, getCoverageOptions());
 
-    expect(testPreprocessor.getCacheKey.mock.calls[0][2].cacheFS).toBeDefined()
-    expect(testPreprocessor.process.mock.calls[0][2].cacheFS).toBeDefined()
+    expect(testPreprocessor.getCacheKey.mock.calls[0][2].cacheFS).toBeDefined();
+    expect(testPreprocessor.process.mock.calls[0][2].cacheFS).toBeDefined();
     expect(fs.readFileSync).toHaveBeenCalledTimes(1);
     expect(fs.readFileSync).toBeCalledWith(fileName1, 'utf8');
   });