test: update

Author: wenytang-ms

test/ui/command.test.ts

@@ -109,6 +109,17 @@ describe("Command Tests", function() {
         }
     });
 
+    afterEach(async function() {
+        if (this.currentTest?.state === 'failed') {
+            const safeName = this.currentTest.title.replace(/[^a-zA-Z0-9_-]/g, '_');
+            try {
+                await VSBrowser.instance.takeScreenshot(safeName);
+                console.log(`Screenshot saved for failed test: ${safeName}`);
+            } catch (e) {
+                console.warn(`Failed to take screenshot for ${safeName}: ${e}`);
+            }
+        }
+    });
 
     it("Test javaProjectExplorer.focus", async function() {
         await new Workbench().executeCommand("javaProjectExplorer.focus");