fix: adjusts to tests cleanup (#229)

* cleanup adjusts

* add mocha cleanup

Author: mshima

Diff for: .mocharc.cjs

@@ -1,3 +1,5 @@
 module.exports = {
   extension: ['spec.ts'],
+  parallel: true,
+  require: ['src/mocha-cleanup.hooks.js'],
 };

Diff for: package-lock.json

@@ -12,8 +12,13 @@
   "author": "The Yeoman Team",
   "type": "module",
   "exports": {
-    "types": "./dist/index.d.ts",
-    "import": "./dist/index.js"
+    "./mocha-cleanup": {
+      "import": "./dist/mocha-cleanup.hooks.js"
+    },
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
   },
   "types": "./dist/index.d.ts",
   "files": [
@@ -43,7 +48,8 @@
     "mem-fs-editor": "^11.1.1",
     "sinon": "^18.0.0",
     "temp-dir": "^3.0.0",
-    "type-fest": "^4.3.1"
+    "type-fest": "^4.3.1",
+    "when-exit": "^2.1.3"
   },
   "devDependencies": {
     "@types/inquirer": "^9.0.3",

Diff for: package.json

@@ -0,0 +1,7 @@
+import { context } from './index.js';
+
+export const mochaHooks = {
+  afterAll() {
+    context.startNewContext();
+  },
+};

Diff for: src/mocha-cleanup.hooks.js

@@ -1,27 +1,41 @@
 import process from 'node:process';
+import onExit from 'when-exit';
 import type RunContext from './run-context.js';
 import type RunResult from './run-result.js';
 
 class TestContext {
+  beforeCwd?: string;
+  autoRestore = true;
+  autoCleanup?: boolean;
   runResult?: RunResult;
   private runContext?: RunContext<any>;
 
-  startNewContext(runContext?: RunContext<any>) {
-    this.runContext?.cleanupTemporaryDir();
+  startNewContext(runContext?, autoCleanup = true) {
+    if (this.beforeCwd !== process.cwd()) {
+      if (this.autoCleanup) {
+        this.runContext?.cleanupTemporaryDir();
+      } else if (this.autoRestore) {
+         
+        this.runContext?.restore();
+      }
+    }
+
+    if (this.beforeCwd && this.beforeCwd !== process.cwd()) {
+      console.log('Test failed to restore context', this.beforeCwd, process.cwd());
+    }
+
+    this.autoCleanup = autoCleanup;
+    this.beforeCwd = runContext ? process.cwd() : undefined;
     this.runContext = runContext;
     this.runResult = undefined;
   }
 }
 
 const testContext = new TestContext();
 
-const cleanupTemporaryDir = () => {
+onExit(() => {
   testContext.startNewContext();
-};
-
-process.on('exit', cleanupTemporaryDir);
-process.on('SIGINT', cleanupTemporaryDir);
-process.on('SIGTERM', cleanupTemporaryDir);
+});
 
 export default testContext;
 

Diff for: src/test-context.ts

@@ -171,7 +171,7 @@ describe('run-result', () => {
     beforeEach(() => {
       cwd = path.join(process.cwd(), 'fixtures', 'tmp');
       if (!fs.existsSync(cwd)) {
-        fs.mkdirSync(cwd);
+        fs.mkdirSync(cwd, { recursive: true });
       }
 
       runResult = new RunResult({

Diff for: test/run-result.spec.ts

@@ -14,6 +14,7 @@
     /* Emit */
     "outDir": "./dist" /* Specify an output folder for all emitted files. */,
     "declaration": true,
+    "skipLibCheck": true,
 
     /* Interop Constraints */
     "allowSyntheticDefaultImports": true /* Allow 'import x from y' when a module doesn't have a default export. */,