jest: support custom testEnvironment

packages/knip/fixtures/plugins/jest2/jest.config.js

@@ -1,3 +1,5 @@
 module.exports = {
   displayName: 'dev',
+    testEnvironment:
+    '<rootDir>/jest.environment.js',
 };

packages/knip/fixtures/plugins/jest2/jest.environment.js

@@ -0,0 +1 @@
+module.exports = {};

packages/knip/src/plugins/jest/index.ts

@@ -58,7 +58,14 @@ const resolveDependencies = async (config: JestInitialOptions, options: PluginOp
 
   const runner = config.runner ? [config.runner] : [];
   const runtime = config.runtime && config.runtime !== 'jest-circus' ? [config.runtime] : [];
-  const environments = config.testEnvironment === 'jsdom' ? ['jest-environment-jsdom'] : [];
+
+  let environments: Array<string> = [];
+  if (config.testEnvironment === 'jsdom') {
+    environments = ['jest-environment-jsdom'];
+  } else if (config.testEnvironment) {
+    environments = [config.testEnvironment];
+  }
+
   const resolvers = config.resolver ? [config.resolver] : [];
   const reporters = config.reporters
     ? config.reporters

packages/knip/test/plugins/jest2.test.ts

@@ -19,13 +19,17 @@ test('Find dependencies with the Jest plugin', async () => {
   // <rootDir> to reference the root directory.
   assert(!issues.files.has(join(cwd, 'project1/setupFiles/setup.js')));
 
+  // Correctly identifies a local environment file which uses
+  // <rootDir> to reference the root directory.
+  assert(!issues.files.has(join(cwd, 'jest.environment.js')));
+
   assert.deepEqual(counters, {
     ...baseCounters,
     files: 0,
     devDependencies: 1,
     unlisted: 0,
     unresolved: 0,
-    processed: 4,
-    total: 4,
+    processed: 5,
+    total: 5,
   });
 });