support for unittest on module find err

Author: eleanorjboyd

src/client/testing/testController/common/utils.ts

@@ -201,12 +201,10 @@ export function buildErrorNodeOptions(uri: Uri, message: string, testType: strin
     let errorMessage = message;
 
     // Check for missing module errors and provide specific messaging
-    if (testType === 'pytest') {
-        const missingModule = extractMissingModuleName(message);
-        if (missingModule) {
-            labelText = `Missing Module: ${missingModule}`;
-            errorMessage = `The module '${missingModule}' is not installed in the selected Python environment. Please install it to enable test discovery.`;
-        }
+    const missingModule = extractMissingModuleName(message);
+    if (missingModule) {
+        labelText = `Missing Module: ${missingModule}`;
+        errorMessage = `The module '${missingModule}' is not installed in the selected Python environment. Please install it to enable test discovery.`;
     }
 
     return {

src/test/testing/testController/common/buildErrorNodeOptions.unit.test.ts

@@ -63,12 +63,23 @@ suite('buildErrorNodeOptions - missing module detection', () => {
         expect(result.error).to.equal('Some other error occurred');
     });
 
-    test('Should use generic error for unittest errors', () => {
-        const errorMessage = "ModuleNotFoundError: No module named 'pytest'";
+    test('Should detect missing module for unittest errors', () => {
+        const errorMessage = "ModuleNotFoundError: No module named 'pandas'";
+
+        const result = buildErrorNodeOptions(workspaceUri, errorMessage, 'unittest');
+
+        expect(result.label).to.equal('Missing Module: pandas [workspace]');
+        expect(result.error).to.equal(
+            "The module 'pandas' is not installed in the selected Python environment. Please install it to enable test discovery.",
+        );
+    });
+
+    test('Should use generic error for unittest non-module errors', () => {
+        const errorMessage = 'Some other error occurred';
 
         const result = buildErrorNodeOptions(workspaceUri, errorMessage, 'unittest');
 
         expect(result.label).to.equal('Unittest Discovery Error [workspace]');
-        expect(result.error).to.equal("ModuleNotFoundError: No module named 'pytest'");
+        expect(result.error).to.equal('Some other error occurred');
     });
 });