From 4a44746975a4d8710af1bae52a340189dd41815e Mon Sep 17 00:00:00 2001
From: Helly Zhang <49181894+hellyzh@users.noreply.github.com>
Date: Thu, 19 Dec 2024 16:40:52 +0800
Subject: [PATCH] test: update to open a sample view (#12943)

* test: update to open a sample view

* test: update to create outlook add in
---
 .../src/ui-test/treeview/treeview-invalidname.test.ts | 11 +++++++----
 .../src/ui-test/treeview/treeview-samples.test.ts     | 11 +++++++----
 packages/tests/src/utils/vscodeOperation.ts           |  4 ++--
 3 files changed, 16 insertions(+), 10 deletions(-)

diff --git a/packages/tests/src/ui-test/treeview/treeview-invalidname.test.ts b/packages/tests/src/ui-test/treeview/treeview-invalidname.test.ts
index 651918a23f..6d7e00dcce 100644
--- a/packages/tests/src/ui-test/treeview/treeview-invalidname.test.ts
+++ b/packages/tests/src/ui-test/treeview/treeview-invalidname.test.ts
@@ -18,6 +18,7 @@ import {
 } from "../../utils/vscodeOperation";
 import { it } from "../../utils/it";
 import * as os from "os";
+import { RetryHandler } from "../../utils/retryHandler";
 
 describe("New project Tests", function () {
   this.timeout(Timeout.testCase);
@@ -47,10 +48,12 @@ describe("New project Tests", function () {
     },
     async function () {
       const driver = VSBrowser.instance.driver;
-      await execCommandIfExist(
-        CommandPaletteCommands.CreateProjectCommand,
-        Timeout.webView
-      );
+      await RetryHandler.retry(async () => {
+        await execCommandIfExist(
+          CommandPaletteCommands.CreateProjectCommand,
+          Timeout.webView
+        );
+      });
       const input = await InputBox.create();
       // if exist click it
       await input.selectQuickPick(CreateProjectQuestion.Tab);
diff --git a/packages/tests/src/ui-test/treeview/treeview-samples.test.ts b/packages/tests/src/ui-test/treeview/treeview-samples.test.ts
index 028b601b1c..a1965ca2f7 100644
--- a/packages/tests/src/ui-test/treeview/treeview-samples.test.ts
+++ b/packages/tests/src/ui-test/treeview/treeview-samples.test.ts
@@ -26,6 +26,7 @@ import { cleanUpLocalProject } from "../../utils/cleanHelper";
 import { it } from "../../utils/it";
 import { assert } from "chai";
 import { join } from "path";
+import { RetryHandler } from "../../utils/retryHandler";
 
 describe("Create sample project and open sample view to download sample Tests", function () {
   this.timeout(Timeout.testCase);
@@ -59,10 +60,12 @@ describe("Create sample project and open sample view to download sample Tests",
           console.log("close editor tab");
           await new EditorView().closeAllEditors();
           console.log("open new sample tab from tree view");
-          await execCommandIfExist(
-            CommandPaletteCommands.SamplesCommand,
-            Timeout.webView
-          );
+          await RetryHandler.retry(async () => {
+            await execCommandIfExist(
+              CommandPaletteCommands.SamplesCommand,
+              Timeout.webView
+            );
+          });
           const webView = new WebView();
           console.log("find title");
           await driver.sleep(Timeout.webView);
diff --git a/packages/tests/src/utils/vscodeOperation.ts b/packages/tests/src/utils/vscodeOperation.ts
index 4d738b327a..74478a9a55 100644
--- a/packages/tests/src/utils/vscodeOperation.ts
+++ b/packages/tests/src/utils/vscodeOperation.ts
@@ -801,7 +801,7 @@ export async function createNewProject(
     }
     case "addin": {
       await input.selectQuickPick(CreateProjectQuestion.OfficeAddin);
-      await input.selectQuickPick("Taskpane");
+      await input.selectQuickPick("Task pane");
       await driver.sleep(Timeout.input);
       break;
     }
@@ -809,7 +809,7 @@ export async function createNewProject(
       const importPath: string =
         testRootFolder + "\\..\\src\\ui-test\\treeview\\office-xml-addin";
       await input.selectQuickPick(CreateProjectQuestion.OfficeAddin);
-      await input.selectQuickPick("Import an Existing Outlook Add-in");
+      await input.selectQuickPick("Import an Existing Outlook Add-ins");
 
       console.log("choose import path: ", importPath);
       await input.selectQuickPick("Browse...");