feat(repo): spin up .ts daemon when running tests that use daemon

jest.preset.js

@@ -12,4 +12,6 @@ module.exports = {
   coverageReporters: ['html'],
   maxWorkers: 1,
   testEnvironment: 'node',
+  globalSetup: '../../scripts/unit-test-setup.js',
+  globalTeardown: '../../scripts/unit-test-teardown.js',
 };

packages/nx/src/daemon/client/client.ts

@@ -4,7 +4,7 @@ import { readFileSync, statSync } from 'fs';
 import { FileHandle, open } from 'fs/promises';
 import { ensureDirSync, ensureFileSync } from 'fs-extra';
 import { connect } from 'net';
-import { join } from 'path';
+import { extname, join } from 'path';
 import { performance } from 'perf_hooks';
 import { output } from '../../utils/output';
 import { getFullOsSocketPath, killSocketOrPath } from '../socket-utils';
@@ -412,6 +412,8 @@ export class DaemonClient {
 
       if (!(await this.isServerAvailable())) {
         await this.startInBackground();
+      } else {
+        console.log('Available daemon found');
       }
       this.setUpConnection();
       this._daemonStatus = DaemonStatus.CONNECTED;
@@ -478,18 +480,35 @@ export class DaemonClient {
     this._out = await open(DAEMON_OUTPUT_LOG_FILE, 'a');
     this._err = await open(DAEMON_OUTPUT_LOG_FILE, 'a');
 
-    const backgroundProcess = spawn(
-      process.execPath,
-      [join(__dirname, '../server/start.js')],
-      {
-        cwd: workspaceRoot,
-        stdio: ['ignore', this._out.fd, this._err.fd],
-        detached: true,
-        windowsHide: true,
-        shell: false,
-        env: { ...process.env, ...DAEMON_ENV_SETTINGS },
-      }
-    );
+    // This stuff is only hit during our unit tests,
+    // as the daemon server is typescript during their execution.
+    const isTypescript = extname(__filename) === '.ts';
+    const workerExt = isTypescript ? 'ts' : 'js';
+
+    const args = [join(__dirname, `../server/start.${workerExt}`)];
+
+    // If the daemon is typescript, we need to add the ts-node/register to support it.
+    if (isTypescript) {
+      args.unshift('-r', 'ts-node/register');
+    }
+
+    const backgroundProcess = spawn(process.execPath, args, {
+      cwd: workspaceRoot,
+      stdio: ['ignore', this._out.fd, this._err.fd],
+      detached: true,
+      windowsHide: true,
+      shell: false,
+      env: {
+        ...process.env,
+        ...DAEMON_ENV_SETTINGS,
+        ...(isTypescript
+          ? {
+              // Ensures that the worker uses the same tsconfig as the main process
+              TS_NODE_PROJECT: join(__dirname, '../../../tsconfig.lib.json'),
+            }
+          : {}),
+      },
+    });
     backgroundProcess.unref();
 
     /**

packages/nx/src/daemon/server/server.ts

@@ -132,11 +132,12 @@ async function handleMessage(socket, data: string) {
     );
   }
 
-  if (daemonIsOutdated()) {
+  const outdated = daemonIsOutdated();
+  if (outdated) {
     await respondWithErrorAndExit(
       socket,
-      `Lock files changed`,
-      new Error('LOCK-FILES-CHANGED')
+      `Daemon outdated`,
+      new Error(outdated)
     );
   }
 
@@ -274,11 +275,19 @@ function registerProcessTerminationListeners() {
 
 let existingLockHash: string | undefined;
 
-function daemonIsOutdated(): boolean {
-  return nxVersionChanged() || lockFileHashChanged();
+function daemonIsOutdated(): string | null {
+  if (nxVersionChanged()) {
+    return 'NX_VERSION_CHANGED';
+  } else if (lockFileHashChanged()) {
+    return 'LOCK_FILES_CHANGED';
+  }
+  return null;
 }
 
 function nxVersionChanged(): boolean {
+  if (process.env.NX_SKIP_NX_VERSION_CHECK === 'true') {
+    return false;
+  }
   return nxVersion !== getInstalledNxVersion();
 }
 
@@ -332,10 +341,11 @@ const handleWorkspaceChanges: FileWatcherCallback = async (
   try {
     resetInactivityTimeout(handleInactivityTimeout);
 
-    if (daemonIsOutdated()) {
+    const outdatedReason = daemonIsOutdated();
+    if (outdatedReason) {
       await handleServerProcessTermination({
         server,
-        reason: 'Lock file changed',
+        reason: outdatedReason,
       });
       return;
     }

scripts/unit-test-setup.js

@@ -0,0 +1,9 @@
+// Shuts down daemon before running any tests. This ensures that if a
+// test needs the daemon, it will start up fresh and use the `.ts` daemon.
+module.exports = () => {
+  process.env.NX_SKIP_NX_VERSION_CHECK = 'true';
+  require('child_process').execSync('pnpm exec nx daemon --stop', {
+    shell: true,
+    stdio: 'inherit',
+  });
+};

scripts/unit-test-teardown.js

@@ -0,0 +1,6 @@
+// Shuts down daemon after running any tests. This ensures that subsequent
+// commands use the installed Nx version's daemon.
+module.exports = () =>
+  require('child_process').execSync('pnpm exec nx daemon --stop', {
+    shell: true,
+  });

typedoc-theme/jest.config.ts

@@ -14,5 +14,7 @@ export default {
   resolver: '../scripts/patched-jest-resolver.js',
   moduleFileExtensions: ['ts', 'tsx', 'js', 'jsx'],
   coverageDirectory: '../coverage/typedoc-theme',
+  globalSetup: null,
+  globalTeardown: null,
   preset: '../jest.preset.js',
 };