feat: custom haste

docs/Configuration.md

@@ -514,6 +514,8 @@ type HasteConfig = {
   platforms?: Array<string>;
   /** Whether to throw on error on module collision. */
   throwOnModuleCollision?: boolean;
+  /** Custom HasteMap module */
+  hasteMapModulePath?: string;
 };
 ```
 

e2e/__tests__/customHaste.test.ts

@@ -0,0 +1,30 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+import * as path from 'path';
+import runJest from '../runJest';
+
+describe('Custom Haste Integration', () => {
+  test('valid test with fake module resolutions', () => {
+    const config = {
+      haste: {
+        hasteMapModulePath: path.resolve(
+          __dirname,
+          '..',
+          'custom-haste-map/hasteMap.js',
+        ),
+      },
+    };
+
+    const {exitCode} = runJest('custom-haste-map', [
+      '--config',
+      JSON.stringify(config),
+      'hasteExample.test.js',
+    ]);
+    expect(exitCode).toBe(0);
+  });
+});

e2e/custom-haste-map/__tests__/hasteExample.test.js

@@ -0,0 +1,18 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ */
+
+'use strict';
+
+const add = require('fakeModuleName');
+
+describe('Custom Haste', () => {
+  test('adds ok', () => {
+    expect(true).toBe(true);
+    expect(add(1, 2)).toBe(3);
+  });
+});

e2e/custom-haste-map/__tests__/hasteExampleHelper.js

@@ -0,0 +1,15 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ */
+
+'use strict';
+
+function add(a, b) {
+  return a + b;
+}
+
+module.exports = add;

e2e/custom-haste-map/hasteMap.js

@@ -0,0 +1,139 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+'use strict';
+
+const path = require('path');
+const fakeFile = {
+  file: path.resolve(__dirname, '__tests__/hasteExampleHelper.js'),
+  moduleName: 'fakeModuleName',
+  sha1: 'fakeSha1',
+};
+
+const fakeJSON = 'fakeJSON';
+
+const testPath = path.resolve(__dirname, '__tests__/hasteExample.test.js');
+
+const allFiles = [fakeFile.file, testPath];
+
+class HasteFS {
+  getModuleName(file) {
+    if (file === fakeFile.file) {
+      return fakeFile.moduleName;
+    }
+    return null;
+  }
+
+  getSize(file) {
+    return null;
+  }
+
+  getDependencies(file) {
+    if (file === testPath) {
+      return fakeFile.file;
+    }
+    return [];
+  }
+
+  getSha1(file) {
+    if (file === fakeFile.file) {
+      return fakeFile.sha1;
+    }
+    return null;
+  }
+
+  exists(file) {
+    return allFiles.includes(file);
+  }
+
+  getAllFiles() {
+    return allFiles;
+  }
+
+  getFileIterator() {
+    return allFiles;
+  }
+
+  getAbsoluteFileIterator() {
+    return allFiles;
+  }
+
+  matchFiles(pattern) {
+    if (!(pattern instanceof RegExp)) {
+      pattern = new RegExp(pattern);
+    }
+    const files = [];
+    for (const file of this.getAbsoluteFileIterator()) {
+      if (pattern.test(file)) {
+        files.push(file);
+      }
+    }
+    return files;
+  }
+
+  matchFilesWithGlob(globs, root) {
+    return [];
+  }
+}
+
+class ModuleMap {
+  getModule(name, platform, supportsNativePlatform, type) {
+    if (name === fakeFile.moduleName) {
+      return fakeFile.file;
+    }
+    return null;
+  }
+
+  getPackage() {
+    return null;
+  }
+
+  getMockModule() {
+    return undefined;
+  }
+
+  getRawModuleMap() {
+    return {};
+  }
+
+  toJSON() {
+    return fakeJSON;
+  }
+}
+
+class HasteMap {
+  constructor(options) {
+    this._cachePath = HasteMap.getCacheFilePath(
+      options.cacheDirectory,
+      options.name,
+    );
+  }
+
+  async build() {
+    return {
+      hasteFS: new HasteFS(),
+      moduleMap: new ModuleMap(),
+    };
+  }
+
+  static getCacheFilePath(tmpdir, name) {
+    return path.join(tmpdir, name);
+  }
+
+  getCacheFilePath() {
+    return this._cachePath;
+  }
+
+  static getModuleMapFromJSON(json) {
+    if (json === fakeJSON) {
+      return new ModuleMap();
+    }
+    throw new Error('Failed to parse serialized module map');
+  }
+}
+
+module.exports = HasteMap;

e2e/custom-haste-map/package.json

@@ -0,0 +1,7 @@
+{
+  "jest": {
+    "haste": {
+      "hasteMapModulePath": "<rootDir>/hasteMap.js"
+    }
+  }
+}

packages/jest-config/src/ValidConfig.ts

@@ -58,6 +58,7 @@ const initialOptions: Config.InitialOptions = {
     enableSymlinks: false,
     forceNodeFilesystemAPI: false,
     hasteImplModulePath: '<rootDir>/haste_impl.js',
+    hasteMapModulePath: '',
     platforms: ['ios', 'android'],
     throwOnModuleCollision: false,
   },

packages/jest-haste-map/src/ModuleMap.ts

@@ -16,6 +16,7 @@ import type {
   ModuleMetaData,
   RawModuleMap,
 } from './types';
+import type {IModuleMap} from '.';
 
 const EMPTY_OBJ: Record<string, ModuleMetaData> = {};
 const EMPTY_MAP = new Map();
@@ -29,7 +30,7 @@ export type SerializableModuleMap = {
   rootDir: Config.Path;
 };
 
-export default class ModuleMap {
+export default class ModuleMap implements IModuleMap<SerializableModuleMap> {
   static DuplicateHasteCandidatesError: typeof DuplicateHasteCandidatesError;
   private readonly _raw: RawModuleMap;
   private json: SerializableModuleMap | undefined;

packages/jest-haste-map/src/index.ts

@@ -18,7 +18,7 @@ import {escapePathForRegex} from 'jest-regex-util';
 import serializer from 'jest-serializer';
 import {Worker} from 'jest-worker';
 import HasteFS from './HasteFS';
-import HasteModuleMap from './ModuleMap';
+import HasteModuleMap, {SerializableModuleMap} from './ModuleMap';
 import H from './constants';
 import nodeCrawl = require('./crawlers/node');
 import watchmanCrawl = require('./crawlers/watchman');
@@ -32,12 +32,14 @@ import type {
   EventsQueue,
   FileData,
   FileMetaData,
+  HTypeValue,
   HasteRegExp,
   InternalHasteMap,
   HasteMap as InternalHasteMapObject,
   MockData,
   ModuleMapData,
   ModuleMetaData,
+  RawModuleMap,
   WorkerMetadata,
 } from './types';
 import FSEventsWatcher = require('./watchers/FSEventsWatcher');
@@ -60,6 +62,7 @@ type Options = {
   extensions: Array<string>;
   forceNodeFilesystemAPI?: boolean;
   hasteImplModulePath?: string;
+  hasteMapModulePath?: string;
   ignorePattern?: HasteRegExp;
   maxWorkers: number;
   mocksPattern?: string;
@@ -132,6 +135,36 @@ function invariant(condition: unknown, message?: string): asserts condition {
   }
 }
 
+export interface IModuleMap<S = SerializableModuleMap> {
+  getModule(
+    name: string,
+    platform?: string | null,
+    supportsNativePlatform?: boolean | null,
+    type?: HTypeValue | null,
+  ): Config.Path | null;
+
+  getPackage(
+    name: string,
+    platform: string | null | undefined,
+    _supportsNativePlatform: boolean | null,
+  ): Config.Path | null;
+
+  getMockModule(name: string): Config.Path | undefined;
+
+  getRawModuleMap(): RawModuleMap;
+
+  toJSON(): S;
+}
+
+export type HasteMapStatic<S = SerializableModuleMap> = {
+  getCacheFilePath(
+    tmpdir: Config.Path,
+    name: string,
+    ...extra: Array<string>
+  ): string;
+  getModuleMapFromJSON(json: S): IModuleMap<S>;
+};
+
 /**
  * HasteMap is a JavaScript implementation of Facebook's haste module system.
  *
@@ -219,7 +252,22 @@ export default class HasteMap extends EventEmitter {
   private _watchers: Array<Watcher>;
   private _worker: WorkerInterface | null;
 
-  constructor(options: Options) {
+  static getStatic(config: Config.ProjectConfig): HasteMapStatic {
+    if (config.haste.hasteMapModulePath) {
+      return require(config.haste.hasteMapModulePath);
+    }
+    return this;
+  }
+
+  static create(options: Options): HasteMap {
+    if (options.hasteMapModulePath) {
+      const CustomHasteMap = require(options.hasteMapModulePath);
+      return new CustomHasteMap(options);
+    }
+    return new HasteMap(options);
+  }
+
+  private constructor(options: Options) {
     super();
     this._options = {
       cacheDirectory: options.cacheDirectory || tmpdir(),
@@ -324,6 +372,10 @@ export default class HasteMap extends EventEmitter {
     );
   }
 
+  static getModuleMapFromJSON(json: SerializableModuleMap): HasteModuleMap {
+    return HasteModuleMap.fromJSON(json);
+  }
+
   getCacheFilePath(): string {
     return this._cachePath;
   }

packages/jest-haste-map/src/types.ts

@@ -9,6 +9,7 @@ import type {Stats} from 'graceful-fs';
 import type {Config} from '@jest/types';
 import type HasteFS from './HasteFS';
 import type ModuleMap from './ModuleMap';
+import type {IModuleMap} from '.';
 
 export type IgnoreMatcher = (item: string) => boolean;
 
@@ -71,6 +72,12 @@ export type InternalHasteMap = {
   mocks: MockData;
 };
 
+export type IHasteMap = {
+  hasteFS: HasteFS;
+  moduleMap: IModuleMap;
+  __hasteMapForTest?: InternalHasteMap | null;
+};
+
 export type HasteMap = {
   hasteFS: HasteFS;
   moduleMap: ModuleMap;

packages/jest-resolve/src/index.ts

@@ -11,7 +11,7 @@ import * as path from 'path';
 import chalk = require('chalk');
 import slash = require('slash');
 import type {Config} from '@jest/types';
-import type {ModuleMap} from 'jest-haste-map';
+import type {IModuleMap} from 'jest-haste-map';
 import {tryRealpath} from 'jest-util';
 import ModuleNotFoundError from './ModuleNotFoundError';
 import defaultResolver, {clearDefaultResolverCache} from './defaultResolver';
@@ -50,13 +50,13 @@ const nodePaths = NODE_PATH
 
 export default class Resolver {
   private readonly _options: ResolverConfig;
-  private readonly _moduleMap: ModuleMap;
+  private readonly _moduleMap: IModuleMap;
   private readonly _moduleIDCache: Map<string, string>;
   private readonly _moduleNameCache: Map<string, Config.Path>;
   private readonly _modulePathCache: Map<string, Array<Config.Path>>;
   private readonly _supportsNativePlatform: boolean;
 
-  constructor(moduleMap: ModuleMap, options: ResolverConfig) {
+  constructor(moduleMap: IModuleMap, options: ResolverConfig) {
     this._options = {
       defaultPlatform: options.defaultPlatform,
       extensions: options.extensions,