Generate individual *.test.js files from webpack compilation

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

Author: Copilot

src/test/index.ts

@@ -3,15 +3,29 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
-// @ts-nocheck
-
 // This file is providing the test runner to use when running extension tests.
 import * as path from 'path';
 import * as vscode from 'vscode';
+import glob from 'glob';
 import Mocha from 'mocha';
 import { mockWebviewEnvironment } from './mocks/mockWebviewEnvironment';
 import { EXTENSION_ID } from '../constants';
 
+function addTests(mocha: Mocha, root: string): Promise<void> {
+	return new Promise((resolve, reject) => {
+		glob('**/**.test.js', { cwd: root }, (error, files) => {
+			if (error) {
+				return reject(error);
+			}
+
+			for (const testFile of files) {
+				mocha.addFile(path.join(root, testFile));
+			}
+			resolve();
+		});
+	});
+}
+
 async function runAllExtensionTests(testsRoot: string, clb: (error: Error | null, failures?: number) => void): Promise<any> {
 	// Ensure the dev-mode extension is activated
 	await vscode.extensions.getExtension(EXTENSION_ID)!.activate();
@@ -22,23 +36,10 @@ async function runAllExtensionTests(testsRoot: string, clb: (error: Error | null
 		ui: 'bdd',
 		color: true
 	});
+	mocha.addFile(path.resolve(testsRoot, 'globalHooks.js'));
 
-	// Load globalHooks if it exists
-	try {
-		mocha.addFile(path.resolve(testsRoot, 'globalHooks.js'));
-	} catch (e) {
-		// globalHooks might not exist in webpack bundle, ignore
-	}
-
-	// Import all test files using webpack's require.context
-	try {
-		// Load tests from src/test directory only
-		// Webview tests are compiled separately with the webview configuration
-		const importAll = (r: __WebpackModuleApi.RequireContext) => r.keys().forEach(r);
-		importAll(require.context('./', true, /\.test$/));
-	} catch (e) {
-		console.log('Error loading tests:', e);
-	}
+	await addTests(mocha, testsRoot);
+	await addTests(mocha, path.resolve(testsRoot, '../../../webviews/'));
 
 	if (process.env.TEST_JUNIT_XML_PATH) {
 		mocha.reporter('mocha-multi-reporters', {

webpack.config.js

@@ -156,14 +156,12 @@ async function getWebviewConfig(mode, env, entry) {
  */
 async function getExtensionConfig(target, mode, env) {
 	const basePath = path.join(__dirname, 'src');
+	const glob = require('glob');
 
 	/**
 	 * @type WebpackConfig['plugins'] | any
 	 */
 	const plugins = [
-		new webpack.optimize.LimitChunkCountPlugin({
-			maxChunks: 1
-		}),
 		new ForkTsCheckerPlugin({
 			async: false,
 			formatter: 'basic',
@@ -190,10 +188,28 @@ async function getExtensionConfig(target, mode, env) {
 	const entry = {
 		extension: './src/extension.ts',
 	};
+
+	// Add test entry points
 	if (target === 'webworker') {
 		entry['test/index'] = './src/test/browser/index.ts';
 	} else if (target === 'node') {
+		// Add main test runner
 		entry['test/index'] = './src/test/index.ts';
+		
+	// Add individual test files as separate entry points
+		const testFiles = glob.sync('src/test/**/*.test.ts', { cwd: __dirname });
+		testFiles.forEach(testFile => {
+			// Convert src/test/github/utils.test.ts -> test/github/utils.test
+			const entryName = testFile.replace('src/', '').replace('.ts', '');
+			entry[entryName] = `./${testFile}`;
+		});
+	}
+
+	// Don't limit chunks for node target when we have individual test files
+	if (target !== 'node' || !('test/index' in entry && Object.keys(entry).some(key => key.endsWith('.test')))) {
+		plugins.unshift(new webpack.optimize.LimitChunkCountPlugin({
+			maxChunks: 1
+		}));
 	}
 
 	return {