[Hosts] Converting manual release-check-list tests to UI-Test Automation and Adding more UI-Test-cases (#37657)

* Add more UI-Test, refactor UITestAutomation

* Convert manual test-case to automation

UI-Tests:

Validating Empty-view is shown if no entries in the list.
Validating Empty-view is NOT shown if 1 or more entries in the list.
Validating Add-an-entry HyperlinkButton in Empty-view works correctly.
Validating Adding-entry Button works correctly.
Validating the Add button should be Disabled if more than 9 hosts in one entry.
Validating the Add button should be Enabled if less or equal 9 hosts in one entry.
Validating error message should be shown if not run as admin.
Validating Warning-Dialog will be shown if 'Show a warning at startup' toggle is On.
Validating Warning-Dialog will NOT be shown if 'Show a warning at startup' toggle is Off.
Validating click 'Quit' button in Warning-Dialog, the Hosts File Editor window would be closed.
Validating click 'Accept' button in Warning-Dialog, the Hosts File Editor window would NOT be closed.

---------

Co-authored-by: Jerry Xu <nxu@microsoft.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

src/common/UITestAutomation/Element/Button.cs

@@ -9,5 +9,14 @@ namespace Microsoft.PowerToys.UITest
     /// </summary>
     public class Button : Element
     {
+        private static readonly string ExpectedControlType = "ControlType.Button";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Button"/> class.
+        /// </summary>
+        public Button()
+        {
+            this.TargetControlType = Button.ExpectedControlType;
+        }
     }
 }

src/common/UITestAutomation/Element/Element.cs

@@ -22,6 +22,14 @@ namespace Microsoft.PowerToys.UITest
 
         private WindowsDriver<WindowsElement>? driver;
 
+        protected string? TargetControlType { get; set; }
+
+        internal bool IsMatchingTarget()
+        {
+            var ct = this.ControlType;
+            return string.IsNullOrEmpty(this.TargetControlType) || this.TargetControlType == this.ControlType;
+        }
+
         internal void SetWindowsElement(WindowsElement windowsElement) => this.windowsElement = windowsElement;
 
         internal void SetSession(WindowsDriver<WindowsElement> driver) => this.driver = driver;
@@ -91,7 +99,7 @@ namespace Microsoft.PowerToys.UITest
         /// Click the UI element.
         /// </summary>
         /// <param name="rightClick">If true, performs a right-click; otherwise, performs a left-click. Default value is false</param>
-        public void Click(bool rightClick = false)
+        public virtual void Click(bool rightClick = false)
         {
             PerformAction((actions, windowElement) =>
             {
@@ -116,7 +124,7 @@ namespace Microsoft.PowerToys.UITest
         /// <summary>
         /// Double Click the UI element.
         /// </summary>
-        public void DoubleClick()
+        public virtual void DoubleClick()
         {
             PerformAction((actions, windowElement) =>
             {
@@ -139,7 +147,6 @@ namespace Microsoft.PowerToys.UITest
         {
             Assert.IsNotNull(this.windowsElement, $"WindowsElement is null in method GetAttribute with parameter: attributeName = {attributeName}");
             var attributeValue = this.windowsElement.GetAttribute(attributeName);
-            Assert.IsNotNull(attributeValue, $"Attribute '{attributeName}' is null.");
             return attributeValue;
         }
 
@@ -154,17 +161,51 @@ namespace Microsoft.PowerToys.UITest
             where T : Element, new()
         {
             Assert.IsNotNull(this.windowsElement, $"WindowsElement is null in method Find<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
-            var foundElement = FindHelper.Find<T, AppiumWebElement>(
-                () =>
-                {
-                    var element = this.windowsElement.FindElement(by.ToSeleniumBy());
-                    Assert.IsNotNull(element, $"Element not found using selector: {by}");
-                    return element;
-                },
-                this.driver,
-                timeoutMS);
 
-            return foundElement;
+            // leverage findAll to filter out mismatched elements
+            var collection = this.FindAll<T>(by, timeoutMS);
+
+            Assert.IsTrue(collection.Count > 0, $"Element not found using selector: {by}");
+
+            return collection[0];
+        }
+
+        /// <summary>
+        /// Finds an element by the selector.
+        /// Shortcut for this.Find<T>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <typeparam name="T">The class type of the element to find.</typeparam>
+        /// <param name="name">The name for finding the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds.</param>
+        /// <returns>The found element.</returns>
+        public T Find<T>(string name, int timeoutMS = 3000)
+            where T : Element, new()
+        {
+            return this.Find<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <summary>
+        /// Finds an element by the selector.
+        /// Shortcut for this.Find<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to use for finding the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds.</param>
+        /// <returns>The found element.</returns>
+        public Element Find(By by, int timeoutMS = 3000)
+        {
+            return this.Find<Element>(by, timeoutMS);
+        }
+
+        /// <summary>
+        /// Finds an element by the selector.
+        /// Shortcut for this.Find<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <param name="name">The name for finding the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds.</param>
+        /// <returns>The found element.</returns>
+        public Element Find(string name, int timeoutMS = 3000)
+        {
+            return this.Find<Element>(By.Name(name), timeoutMS);
         }
 
         /// <summary>
@@ -174,7 +215,7 @@ namespace Microsoft.PowerToys.UITest
         /// <param name="by">The selector to use for finding the elements.</param>
         /// <param name="timeoutMS">The timeout in milliseconds.</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        public ReadOnlyCollection<T>? FindAll<T>(By by, int timeoutMS = 3000)
+        public ReadOnlyCollection<T> FindAll<T>(By by, int timeoutMS = 3000)
             where T : Element, new()
         {
             Assert.IsNotNull(this.windowsElement, $"WindowsElement is null in method FindAll<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
@@ -182,22 +223,59 @@ namespace Microsoft.PowerToys.UITest
                 () =>
                 {
                     var elements = this.windowsElement.FindElements(by.ToSeleniumBy());
-                    Assert.IsTrue(elements.Count > 0, $"Elements not found using selector: {by}");
                     return elements;
                 },
                 this.driver,
                 timeoutMS);
 
-            return foundElements;
+            return foundElements ?? new ReadOnlyCollection<T>([]);
+        }
+
+        /// <summary>
+        /// Finds all elements by the selector.
+        /// Shortcut for this.FindAll<T>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <typeparam name="T">The class type of the elements to find.</typeparam>
+        /// <param name="name">The name for finding the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds.</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        public ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 3000)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <summary>
+        /// Finds all elements by the selector.
+        /// Shortcut for this.FindAll<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to use for finding the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds.</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        public ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 3000)
+        {
+            return this.FindAll<Element>(by, timeoutMS);
+        }
+
+        /// <summary>
+        /// Finds all elements by the selector.
+        /// Shortcut for this.FindAll<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <param name="name">The name for finding the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds.</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        public ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 3000)
+        {
+            return this.FindAll<Element>(By.Name(name), timeoutMS);
         }
 
         /// <summary>
         /// Simulates a manual operation on the element.
         /// </summary>
         /// <param name="action">The action to perform on the element.</param>
-        /// <param name="msPreAction">The number of milliseconds to wait before the action. Default value is 100 ms</param>
-        /// <param name="msPostAction">The number of milliseconds to wait after the action. Default value is 100 ms</param>
-        protected void PerformAction(Action<Actions, WindowsElement> action, int msPreAction = 100, int msPostAction = 100)
+        /// <param name="msPreAction">The number of milliseconds to wait before the action. Default value is 500 ms</param>
+        /// <param name="msPostAction">The number of milliseconds to wait after the action. Default value is 500 ms</param>
+        protected void PerformAction(Action<Actions, WindowsElement> action, int msPreAction = 500, int msPostAction = 500)
         {
             if (msPreAction > 0)
             {

src/common/UITestAutomation/Element/HyperlinkButton.cs

@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation
+// The Microsoft Corporation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+namespace Microsoft.PowerToys.UITest
+{
+    /// <summary>
+    /// Represents a HyperLinkButton in the UI test environment.
+    /// HyperLinkButton represents a button control that functions as a hyperlink.
+    /// </summary>
+    public class HyperlinkButton : Button
+    {
+        private static readonly string ExpectedControlType = "ControlType.HyperLink";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="HyperlinkButton"/> class.
+        /// </summary>
+        public HyperlinkButton()
+        {
+            this.TargetControlType = HyperlinkButton.ExpectedControlType;
+        }
+    }
+}

src/common/UITestAutomation/Element/NavigationViewItem.cs

@@ -0,0 +1,59 @@
+// Copyright (c) Microsoft Corporation
+// The Microsoft Corporation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+namespace Microsoft.PowerToys.UITest
+{
+    /// <summary>
+    /// Represents a NavigationViewItem in the UI test environment.
+    /// NavigationViewItem represents the container for an item in a NavigationView control.
+    /// </summary>
+    public class NavigationViewItem : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.ListItem";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="NavigationViewItem"/> class.
+        /// </summary>
+        public NavigationViewItem()
+        {
+            this.TargetControlType = NavigationViewItem.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Click the ListItem element.
+        /// </summary>
+        /// <param name="rightClick">If true, performs a right-click; otherwise, performs a left-click. Default value is false</param>
+        public override void Click(bool rightClick = false)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement, 10, 10);
+
+                if (rightClick)
+                {
+                    actions.ContextClick();
+                }
+                else
+                {
+                    actions.Click();
+                }
+
+                actions.Build().Perform();
+            });
+        }
+
+        /// <summary>
+        /// Double Click the ListItem element.
+        /// </summary>
+        public override void DoubleClick()
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement, 10, 10);
+                actions.DoubleClick();
+                actions.Build().Perform();
+            });
+        }
+    }
+}

src/common/UITestAutomation/Element/TextBlock.cs

@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation
+// The Microsoft Corporation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+namespace Microsoft.PowerToys.UITest
+{
+    /// <summary>
+    /// Represents a TextBlock in the UI test environment.
+    /// TextBlock provides a lightweight control for displaying small amounts of flow content.
+    /// </summary>
+    public class TextBlock : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Text";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="TextBlock"/> class.
+        /// </summary>
+        public TextBlock()
+        {
+            this.TargetControlType = TextBlock.ExpectedControlType;
+        }
+    }
+}

src/common/UITestAutomation/Element/TextBox.cs

@@ -7,10 +7,21 @@ using OpenQA.Selenium;
 namespace Microsoft.PowerToys.UITest
 {
     /// <summary>
-    /// Represents a textbox in the UI test environment.
+    /// Represents a TextBox in the UI test environment.
+    /// TextBox represents a control that can be used to display and edit plain text (single or multi-line).
     /// </summary>
     public class TextBox : Element
     {
+        private static readonly string ExpectedControlType = "ControlType.Edit";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="TextBox"/> class.
+        /// </summary>
+        public TextBox()
+        {
+            this.TargetControlType = TextBox.ExpectedControlType;
+        }
+
         /// <summary>
         /// Sets the text of the textbox.
         /// </summary>

src/common/UITestAutomation/Element/Window.cs

@@ -18,7 +18,7 @@ namespace Microsoft.PowerToys.UITest
         {
             if (byClickButton)
             {
-                Find<Button>(By.Name("Maximize")).Click();
+                Find<Button>("Maximize").Click();
             }
             else
             {
@@ -37,7 +37,7 @@ namespace Microsoft.PowerToys.UITest
         {
             if (byClickButton)
             {
-                Find<Button>(By.Name("Restore")).Click();
+                Find<Button>("Restore").Click();
             }
             else
             {
@@ -56,7 +56,7 @@ namespace Microsoft.PowerToys.UITest
         {
             if (byClickButton)
             {
-                Find<Button>(By.Name("Minimize")).Click();
+                Find<Button>("Minimize").Click();
             }
             else
             {
@@ -74,7 +74,7 @@ namespace Microsoft.PowerToys.UITest
         {
             if (byClickButton)
             {
-                Find<Button>(By.Name("Close")).Click();
+                Find<Button>("Close").Click();
             }
             else
             {

src/common/UITestAutomation/FindHelper.cs

@@ -17,13 +17,6 @@ namespace Microsoft.PowerToys.UITest
     /// </summary>
     internal static class FindHelper
     {
-        public static T Find<T, TW>(Func<TW> findElementFunc, WindowsDriver<WindowsElement>? driver, int timeoutMS)
-            where T : Element, new()
-        {
-            var item = findElementFunc() as WindowsElement;
-            return NewElement<T>(item, driver, timeoutMS);
-        }
-
         public static ReadOnlyCollection<T>? FindAll<T, TW>(Func<ReadOnlyCollection<TW>> findElementsFunc, WindowsDriver<WindowsElement>? driver, int timeoutMS)
             where T : Element, new()
         {
@@ -32,7 +25,7 @@ namespace Microsoft.PowerToys.UITest
             {
                 var element = item as WindowsElement;
                 return NewElement<T>(element, driver, timeoutMS);
-            }).ToList();
+            }).Where(item => item.IsMatchingTarget()).ToList();
 
             return new ReadOnlyCollection<T>(res);
         }

src/common/UITestAutomation/Session.cs

@@ -4,6 +4,7 @@
 
 using System.Collections.ObjectModel;
 using System.Runtime.InteropServices;
+using System.Xml.Linq;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using OpenQA.Selenium.Appium;
 using OpenQA.Selenium.Appium.Windows;
@@ -39,17 +40,48 @@ namespace Microsoft.PowerToys.UITest
             where T : Element, new()
         {
             Assert.IsNotNull(this.WindowsDriver, $"WindowsElement is null in method Find<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
-            var foundElement = FindHelper.Find<T, WindowsElement>(
-                () =>
-                {
-                    var element = this.WindowsDriver.FindElement(by.ToSeleniumBy());
-                    Assert.IsNotNull(element, $"Element not found using selector: {by}");
-                    return element;
-                },
-                this.WindowsDriver,
-                timeoutMS);
 
-            return foundElement;
+            // leverage findAll to filter out mismatched elements
+            var collection = this.FindAll<T>(by, timeoutMS);
+
+            Assert.IsTrue(collection.Count > 0, $"Element not found using selector: {by}");
+
+            return collection[0];
+        }
+
+        /// <summary>
+        /// Shortcut for this.Find<T>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>The found element.</returns>
+        public T Find<T>(string name, int timeoutMS = 3000)
+            where T : Element, new()
+        {
+            return this.Find<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Find<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to find the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>The found element.</returns>
+        public Element Find(By by, int timeoutMS = 3000)
+        {
+            return this.Find<Element>(by, timeoutMS);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Find<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>The found element.</returns>
+        public Element Find(string name, int timeoutMS = 3000)
+        {
+            return this.Find<Element>(By.Name(name), timeoutMS);
         }
 
         /// <summary>
@@ -72,7 +104,45 @@ namespace Microsoft.PowerToys.UITest
                 this.WindowsDriver,
                 timeoutMS);
 
-            return foundElements ?? new ReadOnlyCollection<T>(new List<T>());
+            return foundElements ?? new ReadOnlyCollection<T>([]);
+        }
+
+        /// <summary>
+        /// Finds all elements by selector.
+        /// Shortcut for this.FindAll<T>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <typeparam name="T">The class of the elements, should be Element or its derived class.</typeparam>
+        /// <param name="name">The name to find the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        public ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 3000)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <summary>
+        /// Finds all elements by selector.
+        /// Shortcut for this.FindAll<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to find the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        public ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 3000)
+        {
+            return this.FindAll<Element>(by, timeoutMS);
+        }
+
+        /// <summary>
+        /// Finds all elements by selector.
+        /// Shortcut for this.FindAll<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <param name="name">The name to find the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        public ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 3000)
+        {
+            return this.FindAll<Element>(By.Name(name), timeoutMS);
         }
 
         /// <summary>

src/common/UITestAutomation/SessionHelper.cs

@@ -0,0 +1,101 @@
+// Copyright (c) Microsoft Corporation
+// The Microsoft Corporation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+using System.Diagnostics;
+using System.Diagnostics.CodeAnalysis;
+using System.Reflection;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium.Appium;
+using OpenQA.Selenium.Appium.Windows;
+
+namespace Microsoft.PowerToys.UITest
+{
+    /// <summary>
+    /// Nested class for test initialization.
+    /// </summary>
+    internal class SessionHelper
+    {
+        // Default session path is PowerToys settings dashboard
+        private readonly string sessionPath = ModuleConfigData.Instance.GetModulePath(PowerToysModule.PowerToysSettings);
+
+        private WindowsDriver<WindowsElement> Root { get; set; }
+
+        private WindowsDriver<WindowsElement>? Driver { get; set; }
+
+        private Process? appDriver;
+
+        public SessionHelper(PowerToysModule scope)
+        {
+            this.sessionPath = ModuleConfigData.Instance.GetModulePath(scope);
+
+            var winAppDriverProcessInfo = new ProcessStartInfo
+            {
+                FileName = "C:\\Program Files (x86)\\Windows Application Driver\\WinAppDriver.exe",
+                Verb = "runas",
+            };
+
+            this.appDriver = Process.Start(winAppDriverProcessInfo);
+
+            var desktopCapabilities = new AppiumOptions();
+            desktopCapabilities.AddAdditionalCapability("app", "Root");
+            this.Root = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), desktopCapabilities);
+
+            // Set default timeout to 5 seconds
+            this.Root.Manage().Timeouts().ImplicitWait = TimeSpan.FromSeconds(5);
+        }
+
+        /// <summary>
+        /// Initializes the test environment.
+        /// </summary>
+        /// <param name="scope">The PowerToys module to start.</param>
+        [UnconditionalSuppressMessage("SingleFile", "IL3000:Avoid accessing Assembly file path when publishing as a single file", Justification = "<Pending>")]
+        public SessionHelper Init()
+        {
+            string? path = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
+            this.StartExe(path + this.sessionPath);
+
+            Assert.IsNotNull(this.Driver, $"Failed to initialize the test environment. Driver is null.");
+
+            // Set default timeout to 5 seconds
+            this.Driver.Manage().Timeouts().ImplicitWait = TimeSpan.FromSeconds(5);
+
+            return this;
+        }
+
+        /// <summary>
+        /// Cleans up the test environment.
+        /// </summary>
+        public void Cleanup()
+        {
+            try
+            {
+                appDriver?.Kill();
+            }
+            catch (Exception ex)
+            {
+                // Handle exceptions if needed
+                Debug.WriteLine($"Exception during Cleanup: {ex.Message}");
+            }
+        }
+
+        /// <summary>
+        /// Starts a new exe and takes control of it.
+        /// </summary>
+        /// <param name="appPath">The path to the application executable.</param>
+        public void StartExe(string appPath)
+        {
+            var opts = new AppiumOptions();
+            opts.AddAdditionalCapability("app", appPath);
+            this.Driver = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), opts);
+        }
+
+        public WindowsDriver<WindowsElement> GetRoot() => this.Root;
+
+        public WindowsDriver<WindowsElement> GetDriver()
+        {
+            Assert.IsNotNull(this.Driver, $"Failed to get driver. Driver is null.");
+            return this.Driver;
+        }
+    }
+}

src/common/UITestAutomation/UITestBase.cs

@@ -15,22 +15,42 @@ namespace Microsoft.PowerToys.UITest
     /// <summary>
     /// Base class that should be inherited by all Test Classes.
     /// </summary>
+    [TestClass]
     public class UITestBase
     {
         public Session Session { get; set; }
 
-        private readonly TestInit testInit = new TestInit();
+        private readonly SessionHelper sessionHelper;
+
+        private readonly PowerToysModule scope;
 
         public UITestBase(PowerToysModule scope = PowerToysModule.PowerToysSettings)
         {
-            this.testInit.SetScope(scope);
-            this.testInit.Init();
-            this.Session = new Session(this.testInit.GetRoot(), this.testInit.GetDriver());
+            this.scope = scope;
+            this.sessionHelper = new SessionHelper(scope).Init();
+            this.Session = new Session(this.sessionHelper.GetRoot(), this.sessionHelper.GetDriver());
         }
 
         ~UITestBase()
         {
-            this.testInit.Cleanup();
+            this.sessionHelper.Cleanup();
+        }
+
+        /// <summary>
+        /// Initializes the test.
+        /// </summary>
+        [TestInitialize]
+        public void TestInit()
+        {
+            if (this.scope == PowerToysModule.PowerToysSettings)
+            {
+                // close Debug warning dialog if any
+                // Such debug warning dialog seems only appear in PowerToys Settings
+                if (this.FindAll("DEBUG").Count > 0)
+                {
+                    this.Find("DEBUG").Find<Button>("Close").Click();
+                }
+            }
         }
 
         /// <summary>
@@ -47,6 +67,41 @@ namespace Microsoft.PowerToys.UITest
             return this.Session.Find<T>(by, timeoutMS);
         }
 
+        /// <summary>
+        /// Shortcut for this.Session.Find<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>The found element.</returns>
+        protected T Find<T>(string name, int timeoutMS = 3000)
+            where T : Element, new()
+        {
+            return this.Session.Find<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.Find<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to find the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>The found element.</returns>
+        protected Element Find(By by, int timeoutMS = 3000)
+        {
+            return this.Session.Find(by, timeoutMS);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.Find<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>The found element.</returns>
+        protected Element Find(string name, int timeoutMS = 3000)
+        {
+            return this.Session.Find(name, timeoutMS);
+        }
+
         /// <summary>
         /// Finds all elements by selector.
         /// Shortcut for this.Session.FindAll<T>(by, timeoutMS)
@@ -62,93 +117,41 @@ namespace Microsoft.PowerToys.UITest
         }
 
         /// <summary>
-        /// Nested class for test initialization.
+        /// Finds all elements by selector.
+        /// Shortcut for this.Session.FindAll<Element>(By.Name(name), timeoutMS)
         /// </summary>
-        private sealed class TestInit
+        /// <typeparam name="T">The class of the elements, should be Element or its derived class.</typeparam>
+        /// <param name="name">The name of the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        protected ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 3000)
+            where T : Element, new()
         {
-            private WindowsDriver<WindowsElement> Root { get; set; }
+            return this.Session.FindAll<T>(By.Name(name), timeoutMS);
+        }
 
-            private WindowsDriver<WindowsElement>? Driver { get; set; }
+        /// <summary>
+        /// Finds all elements by selector.
+        /// Shortcut for this.Session.FindAll<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to find the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        protected ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 3000)
+        {
+            return this.Session.FindAll<Element>(by, timeoutMS);
+        }
 
-            private static Process? appDriver;
-
-            // Default session path is PowerToys settings dashboard
-            private static string sessionPath = ModuleConfigData.Instance.GetModulePath(PowerToysModule.PowerToysSettings);
-
-            public TestInit()
-            {
-                appDriver = Process.Start(new ProcessStartInfo
-                {
-                    FileName = "C:\\Program Files (x86)\\Windows Application Driver\\WinAppDriver.exe",
-                    Verb = "runas",
-                });
-
-                var desktopCapabilities = new AppiumOptions();
-                desktopCapabilities.AddAdditionalCapability("app", "Root");
-                this.Root = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), desktopCapabilities);
-
-                // Set default timeout to 5 seconds
-                this.Root.Manage().Timeouts().ImplicitWait = TimeSpan.FromSeconds(5);
-            }
-
-            /// <summary>
-            /// Initializes the test environment.
-            /// </summary>
-            [UnconditionalSuppressMessage("SingleFile", "IL3000:Avoid accessing Assembly file path when publishing as a single file", Justification = "<Pending>")]
-            public void Init()
-            {
-                string? path = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
-                this.StartExe(path + sessionPath);
-
-                Assert.IsNotNull(this.Driver, $"Failed to initialize the test environment. Driver is null.");
-
-                // Set default timeout to 5 seconds
-                this.Driver.Manage().Timeouts().ImplicitWait = TimeSpan.FromSeconds(5);
-            }
-
-            /// <summary>
-            /// Cleans up the test environment.
-            /// </summary>
-            public void Cleanup()
-            {
-                try
-                {
-                    appDriver?.Kill();
-                }
-                catch (Exception ex)
-                {
-                    // Handle exceptions if needed
-                    Debug.WriteLine($"Exception during Cleanup: {ex.Message}");
-                }
-            }
-
-            /// <summary>
-            /// Starts a new exe and takes control of it.
-            /// </summary>
-            /// <param name="appPath">The path to the application executable.</param>
-            public void StartExe(string appPath)
-            {
-                var opts = new AppiumOptions();
-                opts.AddAdditionalCapability("app", appPath);
-                this.Driver = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), opts);
-            }
-
-            /// <summary>
-            /// Sets scope to the Test Class.
-            /// </summary>
-            /// <param name="scope">The PowerToys module to start.</param>
-            public void SetScope(PowerToysModule scope)
-            {
-                sessionPath = ModuleConfigData.Instance.GetModulePath(scope);
-            }
-
-            public WindowsDriver<WindowsElement> GetRoot() => this.Root;
-
-            public WindowsDriver<WindowsElement> GetDriver()
-            {
-                Assert.IsNotNull(this.Driver, $"Failed to get driver. Driver is null.");
-                return this.Driver;
-            }
+        /// <summary>
+        /// Finds all elements by selector.
+        /// Shortcut for this.Session.FindAll<Element>(By.Name(name), timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the elements.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <returns>A read-only collection of the found elements.</returns>
+        protected ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 3000)
+        {
+            return this.Session.FindAll<Element>(By.Name(name), timeoutMS);
         }
     }
 }