test: use afterEach for mock cleanup to prevent leaks

Author: ryzrr

src/utils/__tests__/generators.test.mjs

@@ -1,5 +1,5 @@
 import assert from 'node:assert/strict';
-import { describe, it, mock } from 'node:test';
+import { describe, it, mock, afterEach } from 'node:test';
 
 import {
   groupNodesByModule,
@@ -144,6 +144,8 @@ describe('buildApiDocURL', () => {
 });
 
 describe('createLazyGenerator', () => {
+  afterEach(() => mock.restoreAll());
+
   it('spreads metadata properties onto the returned object', () => {
     const metadata = { name: 'ast', version: '1.0.0', dependsOn: undefined };
     const gen = createLazyGenerator(metadata);
@@ -163,7 +165,6 @@ describe('createLazyGenerator', () => {
     const gen = createLazyGenerator({ name: 'ast' });
     const result = await gen.generate('hello');
     assert.equal(result, 'processed:hello');
-    mock.restoreAll();
   });
 
   it('exposes a processChunk function that delegates to the lazily loaded module', async () => {
@@ -177,6 +178,5 @@ describe('createLazyGenerator', () => {
     const gen = createLazyGenerator({ name: 'ast' });
     const result = await gen.processChunk(['a', 'b', 'c'], [0, 2]);
     assert.deepStrictEqual(result, ['a', 'c']);
-    mock.restoreAll();
   });
 });

src/utils/__tests__/url.test.mjs

@@ -1,16 +1,13 @@
 import assert from 'node:assert/strict';
-import { describe, it, mock } from 'node:test';
+import { describe, it, mock, afterEach } from 'node:test';
 
-// Mock node:fs/promises so we don't touch the real filesystem
 let fileContent = 'hello from file';
 mock.module('node:fs/promises', {
   namedExports: {
     readFile: async () => fileContent,
   },
 });
 
-// Mock node:url passthrough (pathToFileURL is used inside importFromURL)
-// We load after mocking so the module picks up our stubs
 const { toParsedURL, loadFromURL } = await import('../url.mjs');
 
 describe('toParsedURL', () => {
@@ -31,6 +28,8 @@ describe('toParsedURL', () => {
 });
 
 describe('loadFromURL', () => {
+  afterEach(() => mock.restoreAll());
+
   it('should read content from the filesystem for a plain file path', async () => {
     fileContent = 'file content';
     const result = await loadFromURL('/some/path/file.txt');
@@ -50,6 +49,5 @@ describe('loadFromURL', () => {
 
     const result = await loadFromURL('https://nodejs.org/data.txt');
     assert.strictEqual(result, 'fetched content');
-    mock.restoreAll();
   });
 });