UI Test Automation (#39777)

### Summary

This pull request includes the following updates:

1. Improvements and stabilization of the UI automation framework  
2. Setup of the UI automation pipeline  
3. Add UI test cases for FancyZones  
4. Add UI test cases for MouseUtils  
5. Improvements of Hosts Editor UI tests

---

### Related Links

- **Current Release checklist coverage**:  
 
https://github.com/microsoft/PowerToys/blob/feature/UITestAutomation/src/common/UITestAutomation/Doc/ui-automation-cover-list.md

- **UI Automation pipeline**:  
 
https://microsoft.visualstudio.com/Dart/_build?definitionId=161438&_a=summary

---------

Signed-off-by: Shawn Yuan <shuai.yuan.zju@gmail.com>
Signed-off-by: Shawn Yuan <shuaiyuan@microsoft.com>
Co-authored-by: Jerry Xu <n.xu@outlook.com>
Co-authored-by: Zhaopeng Wang <zhaopengwang@microsoft.com>
Co-authored-by: Xiaofeng Wang (from Dev Box) <xiaofengwang@microsoft.com>
Co-authored-by: Mengyuan <162882040+chenmy77@users.noreply.github.com>
Co-authored-by: yaqingmi <miyaqing01@gmail.com>
Co-authored-by: Clint Rutkas <clint@rutkas.com>
Co-authored-by: Yaqing Mi (from Dev Box) <yaqingmi@microsoft.com>
Co-authored-by: Kai Tao <69313318+vanzue@users.noreply.github.com>
Co-authored-by: zhaopeng wang <33367956+wang563681252@users.noreply.github.com>
Co-authored-by: Laszlo Nemeth <57342539+donlaci@users.noreply.github.com>
Co-authored-by: RokyZevon <12629919+RokyZevon@users.noreply.github.com>
Co-authored-by: Yu Leng <42196638+moooyo@users.noreply.github.com>
Co-authored-by: Yu Leng (from Dev Box) <yuleng@microsoft.com>
Co-authored-by: Davide Giacometti <25966642+davidegiacometti@users.noreply.github.com>
Co-authored-by: Gordon Lam <73506701+yeelam-gordon@users.noreply.github.com>
Co-authored-by: ruslanlap <106077551+ruslanlap@users.noreply.github.com>
Co-authored-by: Muhammad Danish <mdanishkhdev@gmail.com>
Co-authored-by: Bennett Blodinger <benwa@users.noreply.github.com>
Co-authored-by: Jaime Bernardo <jaime@janeasystems.com>
Co-authored-by: Ionuț Manța <ionut@janeasystems.com>
Co-authored-by: Hao Liu <liuhaobupt@163.com>
Co-authored-by: OlegHarchevkin <40352094+OlegKharchevkin@users.noreply.github.com>
Co-authored-by: dcog989 <89043002+dcog989@users.noreply.github.com>
Co-authored-by: PesBandi <127593627+PesBandi@users.noreply.github.com>
Co-authored-by: Stefan Markovic <57057282+stefansjfw@users.noreply.github.com>
Co-authored-by: vanzue <vanzue@outlook.com>
Co-authored-by: Typpi <20943337+Nick2bad4u@users.noreply.github.com>
Co-authored-by: Mike Griese <migrie@microsoft.com>
Co-authored-by: Carlos Zamora <carlos.zamora@microsoft.com>
Co-authored-by: Abhyudit <64366765+bitmap4@users.noreply.github.com>
Co-authored-by: Heiko <61519853+htcfreek@users.noreply.github.com>
Co-authored-by: Ved Nig <vednig12@outlook.com>
Co-authored-by: Niels Laute <niels.laute@live.nl>
Co-authored-by: Aung Khaing Khant <aungkhaingkhant.dev@gmail.com>
Co-authored-by: Aung Khaing Khant <aungkhaingkhant@advent-soft.com>
Co-authored-by: Dustin L. Howett <duhowett@microsoft.com>
Co-authored-by: leileizhang <leilzh@microsoft.com>
Co-authored-by: Dustin L. Howett <dustin@howett.net>
Co-authored-by: Shawn Yuan <128874481+shuaiyuanxx@users.noreply.github.com>
Co-authored-by: Shawn Yuan <shuai.yuan.zju@gmail.com>
Co-authored-by: cryolithic <cryolithic@gmail.com>
Co-authored-by: Lemonyte <49930425+lemonyte@users.noreply.github.com>
Co-authored-by: Gordon Lam (SH) <yeelam@microsoft.com>
Co-authored-by: Corey Hayward <72159232+CoreyHayward@users.noreply.github.com>
Co-authored-by: Jerry Xu <nxu@microsoft.com>
Co-authored-by: Shawn Yuan <shuaiyuan@microsoft.com>
Co-authored-by: Kayla Cinnamon <cinnamon@microsoft.com>
Co-authored-by: Jeremy Sinclair <4016293+snickler@users.noreply.github.com>

src/common/UITestAutomation/Element/CheckBox.cs

@@ -0,0 +1,53 @@
+// 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
+{
+    public class CheckBox : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.CheckBox";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="CheckBox"/> class.
+        /// </summary>
+        public CheckBox()
+        {
+            this.TargetControlType = CheckBox.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Select the item of the ComboBox.
+        /// </summary>
+        /// <param name="value">The text to select from the list view.</param>
+        public void Select(string value)
+        {
+            this.Find<NavigationViewItem>(value).Click();
+        }
+
+        /// <summary>
+        /// Gets a value indicating whether the CheckBox is checked.
+        /// </summary>
+        public bool IsChecked => this.Selected;
+
+        public CheckBox SetCheck(bool value = true, int msPreAction = 500, int msPostAction = 500)
+        {
+            if (this.IsChecked != value)
+            {
+                if (msPreAction > 0)
+                {
+                    Task.Delay(msPreAction).Wait();
+                }
+
+                // Toggle the switch
+                this.Click();
+                if (msPostAction > 0)
+                {
+                    Task.Delay(msPostAction).Wait();
+                }
+            }
+
+            return this;
+        }
+    }
+}

src/common/UITestAutomation/Element/ComboBox.cs

@@ -0,0 +1,28 @@
+// 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
+{
+    public class ComboBox : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.ComboBox";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="ComboBox"/> class.
+        /// </summary>
+        public ComboBox()
+        {
+            this.TargetControlType = ComboBox.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Select the item of the ComboBox.
+        /// </summary>
+        /// <param name="value">The text to select from the list view.</param>
+        public void Select(string value)
+        {
+            this.Find<NavigationViewItem>(value).Click();
+        }
+    }
+}

src/common/UITestAutomation/Element/Custom.cs

@@ -0,0 +1,74 @@
+// 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;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class Custom : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Custom";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Custom"/> class.
+        /// </summary>
+        public Custom()
+        {
+            this.TargetControlType = Custom.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Sends a combination of keys.
+        /// </summary>
+        /// <param name="keys">The keys to send.</param>
+        public void SendKeys(params Key[] keys)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                KeyboardHelper.SendKeys(keys);
+            });
+        }
+
+        /// <summary>
+        /// Drag element move offset.
+        /// </summary>
+        /// <param name="offsetX">The offsetX to move.</param>
+        /// <param name="offsetY">The offsetY to move.</param>
+        public void Drag(int offsetX, int offsetY)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement).MoveByOffset(10, 10).ClickAndHold(windowElement).MoveByOffset(offsetX, offsetY).Release();
+                actions.Build().Perform();
+            });
+        }
+
+        /// <summary>
+        /// Drag element move to other element.
+        /// </summary>
+        /// <param name="element">Move to this element.</param>
+        public void Drag(Element element)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement).ClickAndHold();
+                Assert.IsNotNull(element.WindowsElement, "element is null");
+                int dx = (element.WindowsElement.Rect.X - windowElement.Rect.X) / 10;
+                int dy = (element.WindowsElement.Rect.Y - windowElement.Rect.Y) / 10;
+                for (int i = 0; i < 10; i++)
+                {
+                    actions.MoveByOffset(dx, dy);
+                }
+
+                actions.Release();
+                actions.Build().Perform();
+            });
+        }
+    }
+}

src/common/UITestAutomation/Element/Element.cs

@@ -7,6 +7,7 @@ using System.Drawing;
 using System.Runtime.CompilerServices;
 using System.Xml.Linq;
 using ABI.Windows.Foundation;
+using Microsoft.PowerToys.UITest;
 using Microsoft.VisualStudio.TestPlatform.CommunicationUtilities;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using OpenQA.Selenium;
@@ -25,8 +26,20 @@ namespace Microsoft.PowerToys.UITest
     {
         private WindowsElement? windowsElement;
 
+        protected internal WindowsElement? WindowsElement
+        {
+            get => windowsElement;
+            set => windowsElement = value;
+        }
+
         private WindowsDriver<WindowsElement>? driver;
 
+        protected internal WindowsDriver<WindowsElement>? Driver
+        {
+            get => driver;
+            set => driver = value;
+        }
+
         protected string? TargetControlType { get; set; }
 
         internal bool IsMatchingTarget()
@@ -112,9 +125,12 @@ 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 virtual void Click(bool rightClick = false)
+        /// <param name="msPreAction">Delay in milliseconds before performing the click action. Default is 500 ms.</param>
+        /// <param name="msPostAction">Delay in milliseconds after performing the click action. Default is 500 ms.</param>
+        public virtual void Click(bool rightClick = false, int msPreAction = 500, int msPostAction = 500)
         {
-            PerformAction((actions, windowElement) =>
+            PerformAction(
+                (actions, windowElement) =>
             {
                 actions.MoveToElement(windowElement);
 
@@ -131,7 +147,9 @@ namespace Microsoft.PowerToys.UITest
                 }
 
                 actions.Build().Perform();
-            });
+            },
+                msPreAction,
+                msPostAction);
         }
 
         /// <summary>
@@ -152,51 +170,20 @@ namespace Microsoft.PowerToys.UITest
         }
 
         /// <summary>
-        /// Drag element move offset.
+        /// Release action
         /// </summary>
-        /// <param name="offsetX">The offsetX to move.</param>
-        /// <param name="offsetY">The offsetY to move.</param>
-        public void Drag(int offsetX, int offsetY)
+        public void ReleaseAction()
         {
-            PerformAction((actions, windowElement) =>
-            {
-                actions.MoveToElement(windowElement).MoveByOffset(10, 10).ClickAndHold(windowElement).MoveByOffset(offsetX, offsetY).Release();
-                actions.Build().Perform();
-            });
+            var releaseAction = new Actions(driver);
+            releaseAction.Release().Perform();
         }
 
         /// <summary>
-        /// Drag element move to other element.
+        /// Release key
         /// </summary>
-        /// <param name="element">Move to this element.</param>
-        public void Drag(Element element)
+        public void ReleaseKey(Key key)
         {
-            PerformAction((actions, windowElement) =>
-            {
-                actions.MoveToElement(windowElement).ClickAndHold();
-                Assert.IsNotNull(element.windowsElement, "element is null");
-                int dx = (element.windowsElement.Rect.X - windowElement.Rect.X) / 10;
-                int dy = (element.windowsElement.Rect.Y - windowElement.Rect.Y) / 10;
-                for (int i = 0; i < 10; i++)
-                {
-                    actions.MoveByOffset(dx, dy);
-                }
-
-                actions.Release();
-                actions.Build().Perform();
-            });
-        }
-
-        /// <summary>
-        /// Send Key of the element.
-        /// </summary>
-        /// <param name="key">The Key to Send.</param>
-        public void SendKeys(string key)
-        {
-            PerformAction((actions, windowElement) =>
-            {
-                windowElement.SendKeys(key);
-            });
+            KeyboardHelper.ReleaseKey(key);
         }
 
         /// <summary>
@@ -218,7 +205,7 @@ namespace Microsoft.PowerToys.UITest
         /// <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 T Find<T>(By by, int timeoutMS = 3000)
+        public T Find<T>(By by, int timeoutMS = 5000)
             where T : Element, new()
         {
             Assert.IsNotNull(this.windowsElement, $"WindowsElement is null in method Find<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
@@ -226,7 +213,7 @@ namespace Microsoft.PowerToys.UITest
             // 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}");
+            Assert.IsTrue(collection.Count > 0, $"UI-Element({typeof(T).Name}) not found using selector: {by}");
 
             return collection[0];
         }
@@ -239,7 +226,7 @@ namespace Microsoft.PowerToys.UITest
         /// <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)
+        public T Find<T>(string name, int timeoutMS = 5000)
             where T : Element, new()
         {
             return this.Find<T>(By.Name(name), timeoutMS);
@@ -252,7 +239,7 @@ namespace Microsoft.PowerToys.UITest
         /// <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)
+        public Element Find(By by, int timeoutMS = 5000)
         {
             return this.Find<Element>(by, timeoutMS);
         }
@@ -264,7 +251,7 @@ namespace Microsoft.PowerToys.UITest
         /// <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)
+        public Element Find(string name, int timeoutMS = 5000)
         {
             return this.Find<Element>(By.Name(name), timeoutMS);
         }
@@ -276,7 +263,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 = 5000)
             where T : Element, new()
         {
             Assert.IsNotNull(this.windowsElement, $"WindowsElement is null in method FindAll<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
@@ -308,7 +295,7 @@ namespace Microsoft.PowerToys.UITest
         /// <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)
+        public ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 5000)
             where T : Element, new()
         {
             return this.FindAll<T>(By.Name(name), timeoutMS);
@@ -321,7 +308,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<Element> FindAll(By by, int timeoutMS = 3000)
+        public ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 5000)
         {
             return this.FindAll<Element>(by, timeoutMS);
         }
@@ -333,11 +320,23 @@ namespace Microsoft.PowerToys.UITest
         /// <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)
+        public ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 5000)
         {
             return this.FindAll<Element>(By.Name(name), timeoutMS);
         }
 
+        /// <summary>
+        /// Send Key of the element.
+        /// </summary>
+        /// <param name="key">The Key to Send.</param>
+        protected void SendKeys(string key)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                windowElement.SendKeys(key);
+            });
+        }
+
         /// <summary>
         /// Simulates a manual operation on the element.
         /// </summary>
@@ -360,5 +359,15 @@ namespace Microsoft.PowerToys.UITest
                 Task.Delay(msPostAction).Wait();
             }
         }
+
+        /// <summary>
+        /// Save UI Element to a PNG file.
+        /// </summary>
+        /// <param name="path">the full path</param>
+        internal void SaveToPngFile(string path)
+        {
+            Assert.IsNotNull(this.windowsElement, $"WindowsElement is null in method SaveToPngFile with parameter: path = {path}");
+            this.windowsElement.GetScreenshot().SaveAsFile(path);
+        }
     }
 }

src/common/UITestAutomation/Element/Group.cs

@@ -0,0 +1,25 @@
+// 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;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class Group : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Group";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Group"/> class.
+        /// </summary>
+        public Group()
+        {
+            this.TargetControlType = Group.ExpectedControlType;
+        }
+    }
+}

src/common/UITestAutomation/Element/NavigationViewItem.cs

@@ -24,9 +24,12 @@ namespace Microsoft.PowerToys.UITest
         /// 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)
+        /// <param name="msPreAction">Pre action delay in milliseconds. Default value is 500</param>
+        /// <param name="msPostAction">Post action delay in milliseconds. Default value is 500</param>
+        public override void Click(bool rightClick = false, int msPreAction = 500, int msPostAction = 500)
         {
-            PerformAction((actions, windowElement) =>
+            PerformAction(
+                (actions, windowElement) =>
             {
                 actions.MoveToElement(windowElement, 10, 10);
 
@@ -40,7 +43,36 @@ namespace Microsoft.PowerToys.UITest
                 }
 
                 actions.Build().Perform();
-            });
+            },
+                msPreAction,
+                msPostAction);
+        }
+
+        /// <summary>
+        /// Click the center of the ListItem element.
+        /// </summary>
+        /// <param name="rightClick">If true, performs a right-click; otherwise, performs a left-click. Default value is false</param>
+        /// <param name="msPreAction">Pre action delay in milliseconds. Default value is 500</param>
+        /// <param name="msPostAction">Post action delay in milliseconds. Default value is 500</param>
+        public void ClickCenter(bool rightClick = false, int msPreAction = 500, int msPostAction = 500)
+        {
+            PerformAction(
+                (actions, windowElement) =>
+                {
+                    actions.MoveToElement(windowElement);
+                    if (rightClick)
+                    {
+                        actions.ContextClick();
+                    }
+                    else
+                    {
+                        actions.Click();
+                    }
+
+                    actions.Build().Perform();
+                },
+                msPreAction,
+                msPostAction);
         }
 
         /// <summary>

src/common/UITestAutomation/Element/Pane.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 Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium.Appium.Windows;
+using OpenQA.Selenium.Interactions;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class Pane : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Pane";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Pane"/> class.
+        /// </summary>
+        public Pane()
+        {
+            this.TargetControlType = Pane.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Drag element move offset.
+        /// </summary>
+        /// <param name="offsetX">The offsetX to move.</param>
+        /// <param name="offsetY">The offsetY to move.</param>
+        public void Drag(int offsetX, int offsetY)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement).MoveByOffset(10, 10).ClickAndHold(windowElement).MoveByOffset(offsetX, offsetY).Release();
+                actions.Build().Perform();
+            });
+        }
+
+        /// <summary>
+        /// Simulates holding when dragging to target position.
+        /// </summary>
+        /// <param name="offsetX">The offsetX to move.</param>
+        /// <param name="offsetY">The offsetY to move.</param>
+        public void DragAndHold(int offsetX, int offsetY)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement).MoveByOffset(10, 10).ClickAndHold(windowElement).MoveByOffset(offsetX, offsetY);
+                actions.Build().Perform();
+            });
+        }
+
+        public void ReleaseDrag()
+        {
+            var releaseAction = new Actions(this.Driver);
+            releaseAction.Release().Perform();
+        }
+
+        /// <summary>
+        /// Simulates holding a key, clicking and dragging a UI element to the specified screen coordinates.
+        /// </summary>
+        /// <param name="key">The keyboard key to press and hold during the drag operation.</param>
+        /// <param name="targetX">The target X-coordinate to drag the element to.</param>
+        /// <param name="targetY">The target Y-coordinate to drag the element to.</param>
+        public void KeyDownAndDrag(Key key, int targetX, int targetY)
+        {
+            HoldShiftToDrag(key, targetX, targetY);
+            ReleaseAction();
+            ReleaseKey(key);
+        }
+
+        /// <summary>
+        /// Simulates holding a key, clicking and dragging a UI element to the specified screen coordinates.
+        /// </summary>
+        /// <param name="key">The keyboard key to press and hold during the drag operation.</param>
+        /// <param name="targetX">The target X-coordinate to drag the element to.</param>
+        /// <param name="targetY">The target Y-coordinate to drag the element to.</param>
+        public void HoldShiftToDrag(Key key, int targetX, int targetY)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                KeyboardHelper.PressKey(key);
+
+                actions.MoveToElement(WindowsElement)
+                .ClickAndHold()
+                .Perform();
+
+                int dx = targetX - windowElement.Rect.X;
+                int dy = targetY - windowElement.Rect.Y;
+
+                int stepCount = 10;
+                int stepX = dx / stepCount;
+                int stepY = dy / stepCount;
+
+                for (int i = 0; i < stepCount; i++)
+                {
+                    var stepAction = new Actions(Driver);
+                    stepAction.MoveByOffset(stepX, stepY).Perform();
+                }
+            });
+        }
+    }
+}

src/common/UITestAutomation/Element/Slider.cs

@@ -0,0 +1,131 @@
+// 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 Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium.Appium.Windows;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class Slider : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Slider";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Slider"/> class.
+        /// </summary>
+        public Slider()
+        {
+            this.TargetControlType = Slider.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Gets the value of a Slider (WindowsElement)
+        /// </summary>
+        /// <returns>The integer value of the slider</returns>
+        public int GetValue()
+        {
+            return this.Text == string.Empty ? 0 : int.Parse(this.Text);
+        }
+
+        /// <summary>
+        /// Sends a combination of keys.
+        /// </summary>
+        /// <param name="keys">The keys to send.</param>
+        public void SendKeys(params Key[] keys)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                KeyboardHelper.SendKeys(keys);
+            });
+        }
+
+        /// <summary>
+        /// Sets the value of a Slider (WindowsElement) to the specified integer value.
+        /// Throws an exception if the value is out of the slider's valid range.
+        /// </summary>
+        /// <param name="targetValue">The target integer value to set</param>
+        public void SetValue(int targetValue)
+        {
+            // Read range and current value
+            int min = int.Parse(this.GetAttribute("RangeValue.Minimum"));
+            int max = int.Parse(this.GetAttribute("RangeValue.Maximum"));
+            int current = int.Parse(this.Text);
+
+            // Use Assert to check if the target value is within the valid range
+            Assert.IsTrue(
+                targetValue >= min && targetValue <= max,
+                $"Target value {targetValue} is out of range (min: {min}, max: {max}).");
+
+            // Compute difference
+            int diff = targetValue - current;
+            if (diff == 0)
+            {
+                return;
+            }
+
+            string key = diff > 0 ? OpenQA.Selenium.Keys.Right : OpenQA.Selenium.Keys.Left;
+            int steps = Math.Abs(diff);
+
+            for (int i = 0; i < steps; i++)
+            {
+                this.SendKeys(key);
+
+                // Thread.Sleep(2);
+            }
+
+            // Final check
+            int finalValue = int.Parse(this.Text);
+            Assert.AreEqual(
+                targetValue, finalValue, $"Slider value mismatch: expected {targetValue}, but got {finalValue}.");
+        }
+
+        /// <summary>
+        /// Sets the value of a Slider (WindowsElement) to the specified integer value.
+        /// Throws an exception if the value is out of the slider's valid range.
+        /// </summary>
+        /// <param name="targetValue">The target integer value to set</param>
+        public void QuickSetValue(int targetValue)
+        {
+            // Read range and current value
+            int min = int.Parse(this.GetAttribute("RangeValue.Minimum"));
+            int max = int.Parse(this.GetAttribute("RangeValue.Maximum"));
+            int current = int.Parse(this.Text);
+
+            // Use Assert to check if the target value is within the valid range
+            Assert.IsTrue(
+                targetValue >= min && targetValue <= max,
+                $"Target value {targetValue} is out of range (min: {min}, max: {max}).");
+
+            // Compute difference
+            int diff = targetValue - current;
+            if (diff == 0)
+            {
+                return;
+            }
+
+            string key = diff > 0 ? OpenQA.Selenium.Keys.Right : OpenQA.Selenium.Keys.Left;
+            int steps = Math.Abs(diff);
+
+            int maxKeysPerSend = 50;
+            int fullChunks = steps / maxKeysPerSend;
+            int remainder = steps % maxKeysPerSend;
+            for (int i = 0; i < fullChunks; i++)
+            {
+                SendKeys(new string(key[0], maxKeysPerSend));
+                Thread.Sleep(2);
+            }
+
+            if (remainder > 0)
+            {
+                SendKeys(new string(key[0], remainder));
+                Thread.Sleep(2);
+            }
+
+            // Final check
+            int finalValue = int.Parse(this.Text);
+            Assert.AreEqual(
+                targetValue, finalValue, $"Slider value mismatch: expected {targetValue}, but got {finalValue}.");
+        }
+    }
+}

src/common/UITestAutomation/Element/Tab.cs

@@ -0,0 +1,67 @@
+// 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 Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium.Appium.Windows;
+using OpenQA.Selenium.Interactions;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class Tab : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Tab";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Tab"/> class.
+        /// </summary>
+        public Tab()
+        {
+            this.TargetControlType = Tab.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Simulates holding a key, clicking and dragging a UI element to the specified screen coordinates.
+        /// </summary>
+        /// <param name="key">The keyboard key to press and hold during the drag operation.</param>
+        /// <param name="targetX">The target X-coordinate to drag the element to.</param>
+        /// <param name="targetY">The target Y-coordinate to drag the element to.</param>
+        public void KeyDownAndDrag(Key key, int targetX, int targetY)
+        {
+            HoldShiftToDrag(key, targetX, targetY);
+            ReleaseAction();
+            ReleaseKey(key);
+        }
+
+        /// <summary>
+        /// Simulates holding a key, clicking and dragging a UI element to the specified screen coordinates.
+        /// </summary>
+        /// <param name="key">The keyboard key to press and hold during the drag operation.</param>
+        /// <param name="targetX">The target X-coordinate to drag the element to.</param>
+        /// <param name="targetY">The target Y-coordinate to drag the element to.</param>
+        public void HoldShiftToDrag(Key key, int targetX, int targetY)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                KeyboardHelper.PressKey(key);
+
+                actions.MoveToElement(WindowsElement)
+                .ClickAndHold()
+                .Perform();
+
+                int dx = targetX - windowElement.Rect.X;
+                int dy = targetY - windowElement.Rect.Y;
+
+                int stepCount = 10;
+                int stepX = dx / stepCount;
+                int stepY = dy / stepCount;
+
+                for (int i = 0; i < stepCount; i++)
+                {
+                    var stepAction = new Actions(Driver);
+                    stepAction.MoveByOffset(stepX, stepY).Perform();
+                }
+            });
+        }
+    }
+}

src/common/UITestAutomation/Element/TextBox.cs

@@ -2,8 +2,6 @@
 // The Microsoft Corporation licenses this file to you under the MIT license.
 // See the LICENSE file in the project root for more information.
 
-using OpenQA.Selenium;
-
 namespace Microsoft.PowerToys.UITest
 {
     /// <summary>
@@ -35,9 +33,10 @@ namespace Microsoft.PowerToys.UITest
                 PerformAction((actions, windowElement) =>
                 {
                     // select all text and delete it
-                    windowElement.SendKeys(Keys.Control + "a");
-                    windowElement.SendKeys(Keys.Delete);
+                    windowElement.SendKeys(OpenQA.Selenium.Keys.Control + "a");
+                    windowElement.SendKeys(OpenQA.Selenium.Keys.Delete);
                 });
+                Task.Delay(500).Wait();
             }
 
             PerformAction((actions, windowElement) =>

src/common/UITestAutomation/Element/Thumb.cs

@@ -0,0 +1,36 @@
+// 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 Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium.Appium.Windows;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class Thumb : Element
+    {
+        private static readonly string ExpectedControlType = "ControlType.Thumb";
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="Thumb"/> class.
+        /// </summary>
+        public Thumb()
+        {
+            this.TargetControlType = Thumb.ExpectedControlType;
+        }
+
+        /// <summary>
+        /// Drag element move offset.
+        /// </summary>
+        /// <param name="offsetX">The offsetX to move.</param>
+        /// <param name="offsetY">The offsetY to move.</param>
+        public void Drag(int offsetX, int offsetY)
+        {
+            PerformAction((actions, windowElement) =>
+            {
+                actions.MoveToElement(windowElement).MoveByOffset(10, 10).ClickAndHold(windowElement).MoveByOffset(offsetX, offsetY).Release();
+                actions.Build().Perform();
+            });
+        }
+    }
+}

src/common/UITestAutomation/FindHelper.cs

@@ -20,7 +20,7 @@ namespace Microsoft.PowerToys.UITest
         public static ReadOnlyCollection<T>? FindAll<T, TW>(Func<IReadOnlyCollection<TW>> findElementsFunc, WindowsDriver<WindowsElement>? driver, int timeoutMS)
             where T : Element, new()
         {
-            var items = findElementsFunc();
+            var items = FindElementsWithRetry(findElementsFunc, timeoutMS);
             var res = items.Select(item =>
             {
                 var element = item as WindowsElement;
@@ -30,17 +30,30 @@ namespace Microsoft.PowerToys.UITest
             return new ReadOnlyCollection<T>(res);
         }
 
-        public static ReadOnlyCollection<T>? FindAll<T, TW>(Func<ReadOnlyCollection<TW>> findElementsFunc, WindowsDriver<WindowsElement>? driver, int timeoutMS)
-            where T : Element, new()
+        private static ReadOnlyCollection<TW> FindElementsWithRetry<TW>(Func<IReadOnlyCollection<TW>> findElementsFunc, int timeoutMS)
         {
-            var items = findElementsFunc();
-            var res = items.Select(item =>
-            {
-                var element = item as WindowsElement;
-                return NewElement<T>(element, driver, timeoutMS);
-            }).Where(item => item.IsMatchingTarget()).ToList();
+            var timeout = TimeSpan.FromMilliseconds(timeoutMS);
+            var retryIntervalMS = TimeSpan.FromMilliseconds(500);
+            DateTime startTime = DateTime.Now;
 
-            return new ReadOnlyCollection<T>(res);
+            while (DateTime.Now - startTime < timeout)
+            {
+                try
+                {
+                    var items = findElementsFunc();
+                    if (items.Count > 0)
+                    {
+                        return new ReadOnlyCollection<TW>((IList<TW>)items);
+                    }
+
+                    Task.Delay(retryIntervalMS).Wait();
+                }
+                catch (Exception)
+                {
+                }
+            }
+
+            return new ReadOnlyCollection<TW>(new List<TW>());
         }
 
         public static T NewElement<T>(WindowsElement? element, WindowsDriver<WindowsElement>? driver, int timeoutMS)
@@ -50,11 +63,6 @@ namespace Microsoft.PowerToys.UITest
             Assert.IsNotNull(element, $"New Element {typeof(T).Name} error: element is null.");
 
             T newElement = new T();
-            if (timeoutMS > 0)
-            {
-                // Only set timeout if it is positive value
-                driver.Manage().Timeouts().ImplicitWait = TimeSpan.FromMilliseconds(timeoutMS);
-            }
 
             newElement.SetSession(driver);
             newElement.SetWindowsElement(element);

src/common/UITestAutomation/KeyboardHelper.cs

@@ -0,0 +1,475 @@
+// 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;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Microsoft.PowerToys.UITest
+{
+    /// <summary>
+    /// Represents keyboard keys.
+    /// </summary>
+    public enum Key
+    {
+        Ctrl,
+        LCtrl,
+        RCtrl,
+        Alt,
+        Shift,
+        Tab,
+        Esc,
+        Enter,
+        Win,
+        A,
+        B,
+        C,
+        D,
+        E,
+        F,
+        G,
+        H,
+        I,
+        J,
+        K,
+        L,
+        M,
+        N,
+        O,
+        P,
+        Q,
+        R,
+        S,
+        T,
+        U,
+        V,
+        W,
+        X,
+        Y,
+        Z,
+        Num0,
+        Num1,
+        Num2,
+        Num3,
+        Num4,
+        Num5,
+        Num6,
+        Num7,
+        Num8,
+        Num9,
+        F1,
+        F2,
+        F3,
+        F4,
+        F5,
+        F6,
+        F7,
+        F8,
+        F9,
+        F10,
+        F11,
+        F12,
+        Space,
+        Backspace,
+        Delete,
+        Insert,
+        Home,
+        End,
+        PageUp,
+        PageDown,
+        Up,
+        Down,
+        Left,
+        Right,
+        Other,
+    }
+
+    /// <summary>
+    /// Provides methods for simulating keyboard input.
+    /// </summary>
+    internal static class KeyboardHelper
+    {
+        [DllImport("user32.dll")]
+#pragma warning disable SA1300 // Element should begin with upper-case letter
+        private static extern void keybd_event(byte bVk, byte bScan, uint dwFlags, UIntPtr dwExtraInfo);
+#pragma warning restore SA1300 // Element should begin with upper-case letter
+
+#pragma warning disable SA1310 // Field names should not contain underscore
+        private const byte VK_LWIN = 0x5B;
+        private const uint KEYEVENTF_KEYDOWN = 0x0000;
+        private const uint KEYEVENTF_KEYUP = 0x0002;
+#pragma warning restore SA1310 // Field names should not contain underscore
+
+        /// <summary>
+        /// Sends a combination of keys.
+        /// </summary>
+        /// <param name="keys">The keys to send.</param>
+        public static void SendKeys(params Key[] keys)
+        {
+            string keysToSend = string.Join(string.Empty, keys.Select(TranslateKey));
+            SendWinKeyCombination(keysToSend);
+        }
+
+        public static void PressKey(Key key)
+        {
+            PressVirtualKey(TranslateKeyHex(key));
+        }
+
+        public static void ReleaseKey(Key key)
+        {
+            ReleaseVirtualKey(TranslateKeyHex(key));
+        }
+
+        public static void SendKey(Key key)
+        {
+            PressVirtualKey(TranslateKeyHex(key));
+            ReleaseVirtualKey(TranslateKeyHex(key));
+        }
+
+        /// <summary>
+        /// Translates a key to its corresponding SendKeys representation.
+        /// </summary>
+        /// <param name="key">The key to translate.</param>
+        /// <returns>The SendKeys representation of the key.</returns>
+        private static string TranslateKey(Key key)
+        {
+            switch (key)
+            {
+                case Key.Ctrl:
+                    return "^";
+                case Key.LCtrl:
+                    return "^";
+                case Key.RCtrl:
+                    return "^";
+                case Key.Alt:
+                    return "%";
+                case Key.Shift:
+                    return "+";
+                case Key.Tab:
+                    return "{TAB}";
+                case Key.Esc:
+                    return "{ESC}";
+                case Key.Enter:
+                    return "{ENTER}";
+                case Key.Win:
+                    return "{WIN}";
+                case Key.Space:
+                    return " ";
+                case Key.Backspace:
+                    return "{BACKSPACE}";
+                case Key.Delete:
+                    return "{DELETE}";
+                case Key.Insert:
+                    return "{INSERT}";
+                case Key.Home:
+                    return "{HOME}";
+                case Key.End:
+                    return "{END}";
+                case Key.PageUp:
+                    return "{PGUP}";
+                case Key.PageDown:
+                    return "{PGDN}";
+                case Key.Up:
+                    return "{UP}";
+                case Key.Down:
+                    return "{DOWN}";
+                case Key.Left:
+                    return "{LEFT}";
+                case Key.Right:
+                    return "{RIGHT}";
+                case Key.F1:
+                    return "{F1}";
+                case Key.F2:
+                    return "{F2}";
+                case Key.F3:
+                    return "{F3}";
+                case Key.F4:
+                    return "{F4}";
+                case Key.F5:
+                    return "{F5}";
+                case Key.F6:
+                    return "{F6}";
+                case Key.F7:
+                    return "{F7}";
+                case Key.F8:
+                    return "{F8}";
+                case Key.F9:
+                    return "{F9}";
+                case Key.F10:
+                    return "{F10}";
+                case Key.F11:
+                    return "{F11}";
+                case Key.F12:
+                    return "{F12}";
+                case Key.A:
+                    return "a";
+                case Key.B:
+                    return "b";
+                case Key.C:
+                    return "c";
+                case Key.D:
+                    return "d";
+                case Key.E:
+                    return "e";
+                case Key.F:
+                    return "f";
+                case Key.G:
+                    return "g";
+                case Key.H:
+                    return "h";
+                case Key.I:
+                    return "i";
+                case Key.J:
+                    return "j";
+                case Key.K:
+                    return "k";
+                case Key.L:
+                    return "l";
+                case Key.M:
+                    return "m";
+                case Key.N:
+                    return "n";
+                case Key.O:
+                    return "o";
+                case Key.P:
+                    return "p";
+                case Key.Q:
+                    return "q";
+                case Key.R:
+                    return "r";
+                case Key.S:
+                    return "s";
+                case Key.T:
+                    return "t";
+                case Key.U:
+                    return "u";
+                case Key.V:
+                    return "v";
+                case Key.W:
+                    return "w";
+                case Key.X:
+                    return "x";
+                case Key.Y:
+                    return "y";
+                case Key.Z:
+                    return "z";
+                case Key.Num0:
+                    return "0";
+                case Key.Num1:
+                    return "1";
+                case Key.Num2:
+                    return "2";
+                case Key.Num3:
+                    return "3";
+                case Key.Num4:
+                    return "4";
+                case Key.Num5:
+                    return "5";
+                case Key.Num6:
+                    return "6";
+                case Key.Num7:
+                    return "7";
+                case Key.Num8:
+                    return "8";
+                case Key.Num9:
+                    return "9";
+                default:
+                    return string.Empty;
+            }
+        }
+
+        /// <summary>
+        /// map the virtual key codes to the corresponding keys.
+        /// </summary>
+        private static byte TranslateKeyHex(Key key)
+        {
+            switch (key)
+            {
+                case Key.Win:
+                    return 0x5B;  // Windows Key - 0x5B in hex
+                case Key.Ctrl:
+                    return 0x11;  // Ctrl Key - 0x11 in hex
+                case Key.Alt:
+                    return 0x12;  // Alt Key - 0x12 in hex
+                case Key.Shift:
+                    return 0x10;  // Shift Key - 0x10 in hex
+                case Key.LCtrl:
+                    return 0xA2;  // Left Ctrl Key - 0xA2 in hex
+                case Key.RCtrl:
+                    return 0xA3;  // Right Ctrl Key - 0xA3 in hex
+                case Key.A:
+                    return 0x41;  // A Key - 0x41 in hex
+                case Key.B:
+                    return 0x42;  // B Key - 0x42 in hex
+                case Key.C:
+                    return 0x43;  // C Key - 0x43 in hex
+                case Key.D:
+                    return 0x44;  // D Key - 0x44 in hex
+                case Key.E:
+                    return 0x45;  // E Key - 0x45 in hex
+                case Key.F:
+                    return 0x46;  // F Key - 0x46 in hex
+                case Key.G:
+                    return 0x47;  // G Key - 0x47 in hex
+                case Key.H:
+                    return 0x48;  // H Key - 0x48 in hex
+                case Key.I:
+                    return 0x49;  // I Key - 0x49 in hex
+                case Key.J:
+                    return 0x4A;  // J Key - 0x4A in hex
+                case Key.K:
+                    return 0x4B;  // K Key - 0x4B in hex
+                case Key.L:
+                    return 0x4C;  // L Key - 0x4C in hex
+                case Key.M:
+                    return 0x4D;  // M Key - 0x4D in hex
+                case Key.N:
+                    return 0x4E;  // N Key - 0x4E in hex
+                case Key.O:
+                    return 0x4F;  // O Key - 0x4F in hex
+                case Key.P:
+                    return 0x50;  // P Key - 0x50 in hex
+                case Key.Q:
+                    return 0x51;  // Q Key - 0x51 in hex
+                case Key.R:
+                    return 0x52;  // R Key - 0x52 in hex
+                case Key.S:
+                    return 0x53;  // S Key - 0x53 in hex
+                case Key.T:
+                    return 0x54;  // T Key - 0x54 in hex
+                case Key.U:
+                    return 0x55;  // U Key - 0x55 in hex
+                case Key.V:
+                    return 0x56;  // V Key - 0x56 in hex
+                case Key.W:
+                    return 0x57;  // W Key - 0x57 in hex
+                case Key.X:
+                    return 0x58;  // X Key - 0x58 in hex
+                case Key.Y:
+                    return 0x59;  // Y Key - 0x59 in hex
+                case Key.Z:
+                    return 0x5A;  // Z Key - 0x5A in hex
+                case Key.Num0:
+                    return 0x30;  // 0 Key - 0x30 in hex
+                case Key.Num1:
+                    return 0x31;  // 1 Key - 0x31 in hex
+                case Key.Num2:
+                    return 0x32;  // 2 Key - 0x32 in hex
+                case Key.Num3:
+                    return 0x33;  // 3 Key - 0x33 in hex
+                case Key.Num4:
+                    return 0x34;  // 4 Key - 0x34 in hex
+                case Key.Num5:
+                    return 0x35;  // 5 Key - 0x35 in hex
+                case Key.Num6:
+                    return 0x36;  // 6 Key - 0x36 in hex
+                case Key.Num7:
+                    return 0x37;  // 7 Key - 0x37 in hex
+                case Key.Num8:
+                    return 0x38;  // 8 Key - 0x38 in hex
+                case Key.Num9:
+                    return 0x39;  // 9 Key - 0x39 in hex
+                case Key.F1:
+                    return 0x70;  // F1 Key - 0x70 in hex
+                case Key.F2:
+                    return 0x71;  // F2 Key - 0x71 in hex
+                case Key.F3:
+                    return 0x72;  // F3 Key - 0x72 in hex
+                case Key.F4:
+                    return 0x73;  // F4 Key - 0x73 in hex
+                case Key.F5:
+                    return 0x74;  // F5 Key - 0x74 in hex
+                case Key.F6:
+                    return 0x75;  // F6 Key - 0x75 in hex
+                case Key.F7:
+                    return 0x76;  // F7 Key - 0x76 in hex
+                case Key.F8:
+                    return 0x77;  // F8 Key - 0x77 in hex
+                case Key.F9:
+                    return 0x78;  // F9 Key - 0x78 in hex
+                case Key.F10:
+                    return 0x79;  // F10 Key - 0x79 in hex
+                case Key.F11:
+                    return 0x7A;  // F11 Key - 0x7A in hex
+                case Key.F12:
+                    return 0x7B;  // F12 Key - 0x7B in hex
+                case Key.Up:
+                    return 0x26;  // Up Arrow Key - 0x26 in hex
+                case Key.Down:
+                    return 0x28;  // Down Arrow Key - 0x28 in hex
+                case Key.Left:
+                    return 0x25;  // Left Arrow Key - 0x25 in hex
+                case Key.Right:
+                    return 0x27;  // Right Arrow Key - 0x27 in hex
+                case Key.Home:
+                    return 0x24;  // Home Key - 0x24 in hex
+                case Key.End:
+                    return 0x23;  // End Key - 0x23 in hex
+                case Key.PageUp:
+                    return 0x21;  // Page Up Key - 0x21 in hex
+                case Key.PageDown:
+                    return 0x22;  // Page Down Key - 0x22 in hex
+                case Key.Space:
+                    return 0x20;  // Space Key - 0x20 in hex
+                case Key.Enter:
+                    return 0x0D;  // Enter Key - 0x0D in hex
+                case Key.Backspace:
+                    return 0x08;  // Backspace Key - 0x08 in hex
+                case Key.Tab:
+                    return 0x09;  // Tab Key - 0x09 in hex
+                case Key.Esc:
+                    return 0x1B;  // Escape Key - 0x1B in hex
+                case Key.Insert:
+                    return 0x2D;  // Insert Key - 0x2D in hex
+                case Key.Delete:
+                    return 0x2E;  // Delete Key - 0x2E in hex
+                default:
+                    throw new ArgumentException($"Key {key} is not supported, Please add your key at TranslateKeyHex for translation to hex.");
+            }
+        }
+
+        /// <summary>
+        /// Sends a combination of keys, including the Windows key, to the system.
+        /// </summary>
+        /// <param name="keys">The keys to send.</param>
+        private static void SendWinKeyCombination(string keys)
+        {
+            bool winKeyDown = false;
+
+            if (keys.Contains("{WIN}"))
+            {
+                keybd_event(VK_LWIN, 0, KEYEVENTF_KEYDOWN, UIntPtr.Zero);
+                winKeyDown = true;
+                keys = keys.Replace("{WIN}", string.Empty); // Remove {WIN} from the string
+            }
+
+            System.Windows.Forms.SendKeys.SendWait(keys);
+
+            // Release Windows key
+            if (winKeyDown)
+            {
+                keybd_event(VK_LWIN, 0, KEYEVENTF_KEYUP, UIntPtr.Zero);
+            }
+        }
+
+        /// <summary>
+        /// Just press the key.(no release)
+        /// </summary>
+        private static void PressVirtualKey(byte key)
+        {
+            keybd_event(key, 0, KEYEVENTF_KEYDOWN, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Release only the button (if pressed first)
+        /// </summary>
+        private static void ReleaseVirtualKey(byte key)
+        {
+            keybd_event(key, 0, KEYEVENTF_KEYUP, UIntPtr.Zero);
+        }
+    }
+}

src/common/UITestAutomation/ModuleConfigData.cs

@@ -29,6 +29,49 @@ namespace Microsoft.PowerToys.UITest
         PowerToysSettings,
         FancyZone,
         Hosts,
+        Runner,
+        Workspaces,
+    }
+
+    /// <summary>
+    /// Represents the window size for the UI test.
+    /// </summary>
+    public enum WindowSize
+    {
+        /// <summary>
+        /// Unspecified window size, won't make any size change
+        /// </summary>
+        UnSpecified,
+
+        /// <summary>
+        /// Small window size, 640 * 480
+        /// </summary>
+        Small,
+
+        /// <summary>
+        /// Small window size, 480 * 640
+        /// </summary>
+        Small_Vertical,
+
+        /// <summary>
+        /// Medium window size, 1024 * 768
+        /// </summary>
+        Medium,
+
+        /// <summary>
+        /// Medium window size, 768 * 1024
+        /// </summary>
+        Medium_Vertical,
+
+        /// <summary>
+        /// Large window size, 1920 * 1080
+        /// </summary>
+        Large,
+
+        /// <summary>
+        /// Large window size, 1080 * 1920
+        /// </summary>
+        Large_Vertical,
     }
 
     internal class ModuleConfigData
@@ -52,6 +95,8 @@ namespace Microsoft.PowerToys.UITest
                 [PowerToysModule.PowerToysSettings] = "PowerToys Settings",
                 [PowerToysModule.FancyZone] = "FancyZones Layout",
                 [PowerToysModule.Hosts] = "Hosts File Editor",
+                [PowerToysModule.Runner] = "PowerToys",
+                [PowerToysModule.Workspaces] = "Workspaces Editor",
             };
 
             // Exe start path for the module if it exists.
@@ -60,6 +105,8 @@ namespace Microsoft.PowerToys.UITest
                 [PowerToysModule.PowerToysSettings] = @"\..\..\..\WinUI3Apps\PowerToys.Settings.exe",
                 [PowerToysModule.FancyZone] = @"\..\..\..\PowerToys.FancyZonesEditor.exe",
                 [PowerToysModule.Hosts] = @"\..\..\..\WinUI3Apps\PowerToys.Hosts.exe",
+                [PowerToysModule.Runner] = @"\..\..\..\PowerToys.exe",
+                [PowerToysModule.Workspaces] = @"\..\..\..\PowerToys.WorkspacesEditor.exe",
             };
         }
 

src/common/UITestAutomation/MonitorInfoData.cs

@@ -0,0 +1,37 @@
+// 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.Collections.Generic;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public class MonitorInfoData
+    {
+        public MonitorInfoData()
+        {
+        }
+
+        public struct MonitorInfoDataWrapper
+        {
+            public string DeviceName { get; set; }
+
+            public string DeviceString { get; set; }
+
+            public string DeviceID { get; set; }
+
+            public string DeviceKey { get; set; }
+
+            public int PelsWidth { get; set; }
+
+            public int PelsHeight { get; set; }
+
+            public int DisplayFrequency { get; set; }
+        }
+
+        public struct ParamsWrapper
+        {
+            public List<MonitorInfoDataWrapper> Monitors { get; set; }
+        }
+    }
+}

src/common/UITestAutomation/MouseHelper.cs

@@ -0,0 +1,217 @@
+// 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;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public enum MouseActionType
+    {
+        LeftClick,
+        RightClick,
+        MiddleClick,
+        LeftDoubleClick,
+        RightDoubleClick,
+        LeftDown,
+        LeftUp,
+        RightDown,
+        RightUp,
+        MiddleDown,
+        MiddleUp,
+        ScrollUp,
+        ScrollDown,
+    }
+
+    internal static class MouseHelper
+    {
+        [StructLayout(LayoutKind.Sequential)]
+        public struct POINT
+        {
+            public int X;
+            public int Y;
+        }
+
+        [Flags]
+        internal enum MouseEvent
+        {
+            LeftDown = 0x0002,
+            LeftUp = 0x0004,
+            RightDown = 0x0008,
+            RightUp = 0x0010,
+            MiddleDown = 0x0020,
+            MiddleUp = 0x0040,
+            Wheel = 0x0800,
+        }
+
+        [DllImport("user32.dll")]
+        public static extern bool GetCursorPos(out POINT lpPoint);
+
+        [DllImport("user32.dll")]
+        public static extern bool SetCursorPos(int x, int y);
+
+        [DllImport("user32.dll")]
+#pragma warning disable SA1300 // Element should begin with upper-case letter
+        private static extern void mouse_event(uint dwFlags, uint dx, uint dy, uint dwData, UIntPtr dwExtraInfo);
+#pragma warning restore SA1300 // Element should begin with upper-case letter
+
+        /// <summary>
+        /// Gets the current position of the mouse cursor as a tuple.
+        /// </summary>
+        /// <returns>A tuple containing the X and Y coordinates of the cursor.</returns>
+        public static Tuple<int, int> GetMousePosition()
+        {
+            GetCursorPos(out POINT point);
+            return Tuple.Create(point.X, point.Y);
+        }
+
+        /// <summary>
+        /// Moves the mouse cursor to the specified screen coordinates.
+        /// </summary>
+        /// <param name="x">The new x-coordinate of the cursor.</param>
+        /// <param name="y">The new y-coordinate of the cursor.</param
+        public static void MoveMouseTo(int x, int y)
+        {
+            SetCursorPos(x, y);
+        }
+
+        /// <summary>
+        /// The delay in milliseconds between mouse down and up events to simulate a click.
+        /// </summary>
+        private const int ClickDelay = 100;
+
+        /// <summary>
+        /// The amount of scroll units to simulate a single mouse wheel tick.
+        /// </summary>
+        private const int ScrollAmount = 120;
+
+        /// <summary>
+        /// Simulates a left mouse click (press and release).
+        /// </summary>
+        public static void LeftClick()
+        {
+            LeftDown();
+            Thread.Sleep(ClickDelay);
+            LeftUp();
+        }
+
+        /// <summary>
+        /// Simulates a right mouse click (press and release).
+        /// </summary>
+        public static void RightClick()
+        {
+            RightDown();
+            Thread.Sleep(ClickDelay);
+            RightUp();
+        }
+
+        /// <summary>
+        /// Simulates a middle mouse click (press and release).
+        /// </summary>
+        public static void MiddleClick()
+        {
+            MiddleDown();
+            Thread.Sleep(ClickDelay);
+            MiddleUp();
+        }
+
+        /// <summary>
+        /// Simulates a left mouse double-click.
+        /// </summary>
+        public static void LeftDoubleClick()
+        {
+            LeftClick();
+            Thread.Sleep(ClickDelay);
+            LeftClick();
+        }
+
+        /// <summary>
+        /// Simulates a right mouse double-click.
+        /// </summary>
+        public static void RightDoubleClick()
+        {
+            RightClick();
+            Thread.Sleep(ClickDelay);
+            RightClick();
+        }
+
+        /// <summary>
+        /// Simulates pressing the left mouse button down.
+        /// </summary>
+        public static void LeftDown()
+        {
+            mouse_event((uint)MouseEvent.LeftDown, 0, 0, 0, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates pressing the right mouse button down.
+        /// </summary>
+        public static void RightDown()
+        {
+            mouse_event((uint)MouseEvent.RightDown, 0, 0, 0, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates pressing the middle mouse button down.
+        /// </summary>
+        public static void MiddleDown()
+        {
+            mouse_event((uint)MouseEvent.MiddleDown, 0, 0, 0, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates releasing the left mouse button.
+        /// </summary>
+        public static void LeftUp()
+        {
+            mouse_event((uint)MouseEvent.LeftUp, 0, 0, 0, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates releasing the right mouse button.
+        /// </summary>
+        public static void RightUp()
+        {
+            mouse_event((uint)MouseEvent.RightUp, 0, 0, 0, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates releasing the middle mouse button.
+        /// </summary>
+        public static void MiddleUp()
+        {
+            mouse_event((uint)MouseEvent.MiddleUp, 0, 0, 0, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates a mouse scroll wheel action by a specified amount.
+        /// Positive values scroll up, negative values scroll down.
+        /// </summary>
+        /// <param name="amount">The scroll amount. Typically 120 or -120 per tick.</param>
+        public static void ScrollWheel(int amount)
+        {
+            mouse_event((uint)MouseEvent.Wheel, 0, 0, (uint)amount, UIntPtr.Zero);
+        }
+
+        /// <summary>
+        /// Simulates scrolling the mouse wheel up by one tick.
+        /// </summary>
+        public static void ScrollUp()
+        {
+            ScrollWheel(ScrollAmount);
+        }
+
+        /// <summary>
+        /// Simulates scrolling the mouse wheel down by one tick.
+        /// </summary>
+        public static void ScrollDown()
+        {
+            ScrollWheel(-ScrollAmount);
+        }
+    }
+}

src/common/UITestAutomation/ScreenCapture.cs

@@ -0,0 +1,133 @@
+// 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;
+using System.Collections.Generic;
+using System.Drawing.Imaging;
+using System.Linq;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Threading.Tasks;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace Microsoft.PowerToys.UITest
+{
+    /// <summary>
+    /// Provides methods for capturing the screen with the mouse cursor.
+    /// </summary>
+    internal static class ScreenCapture
+    {
+        [DllImport("user32.dll")]
+        private static extern IntPtr GetDC(IntPtr hWnd);
+
+        [DllImport("gdi32.dll")]
+        private static extern int GetDeviceCaps(IntPtr hdc, int nIndex);
+
+        [DllImport("user32.dll")]
+        private static extern IntPtr ReleaseDC(IntPtr hWnd, IntPtr hDC);
+
+        [DllImport("user32.dll")]
+        private static extern bool GetCursorInfo(out CURSORINFO pci);
+
+        [DllImport("user32.dll")]
+        private static extern bool DrawIconEx(IntPtr hdc, int x, int y, IntPtr hIcon, int cx, int cy, int istepIfAniCur, IntPtr hbrFlickerFreeDraw, int diFlags);
+
+        /// <summary>
+        /// Represents a point with X and Y coordinates.
+        /// </summary>
+        [StructLayout(LayoutKind.Sequential)]
+        public struct POINT
+        {
+            public int X;
+            public int Y;
+        }
+
+        /// <summary>
+        /// Contains information about the cursor.
+        /// </summary>
+        [StructLayout(LayoutKind.Sequential)]
+        public struct CURSORINFO
+        {
+            /// <summary>
+            /// Gets or sets the size of the structure.
+            /// </summary>
+            public int CbSize;
+
+            /// <summary>
+            /// Gets or sets the cursor state.
+            /// </summary>
+            public int Flags;
+
+            /// <summary>
+            /// Gets or sets the handle to the cursor.
+            /// </summary>
+            public IntPtr HCursor;
+
+            /// <summary>
+            /// Gets or sets the screen position of the cursor.
+            /// </summary>
+            public POINT PTScreenPos;
+        }
+
+        private const int CURSORSHOWING = 0x00000001;
+        private const int DESKTOPHORZRES = 118;
+        private const int DESKTOPVERTRES = 117;
+        private const int DINORMAL = 0x0003;
+
+        /// <summary>
+        /// Captures the screen with the mouse cursor and saves it to the specified file path.
+        /// </summary>
+        /// <param name="filePath">The file path to save the captured image.</param>
+        private static void CaptureScreenWithMouse(string filePath)
+        {
+            IntPtr hdc = GetDC(IntPtr.Zero);
+            int screenWidth = GetDeviceCaps(hdc, DESKTOPHORZRES);
+            int screenHeight = GetDeviceCaps(hdc, DESKTOPVERTRES);
+            ReleaseDC(IntPtr.Zero, hdc);
+
+            Rectangle bounds = new Rectangle(0, 0, screenWidth, screenHeight);
+            using (Bitmap bitmap = new Bitmap(bounds.Width, bounds.Height))
+            {
+                using (System.Drawing.Graphics g = System.Drawing.Graphics.FromImage(bitmap))
+                {
+                    g.CopyFromScreen(bounds.Location, Point.Empty, bounds.Size);
+
+                    CURSORINFO cursorInfo;
+                    cursorInfo.CbSize = Marshal.SizeOf<CURSORINFO>();
+                    if (GetCursorInfo(out cursorInfo) && cursorInfo.Flags == CURSORSHOWING)
+                    {
+                        using (System.Drawing.Graphics gIcon = System.Drawing.Graphics.FromImage(bitmap))
+                        {
+                            IntPtr hdcDest = gIcon.GetHdc();
+                            DrawIconEx(hdcDest, cursorInfo.PTScreenPos.X, cursorInfo.PTScreenPos.Y, cursorInfo.HCursor, 0, 0, 0, IntPtr.Zero, DINORMAL);
+                            gIcon.ReleaseHdc(hdcDest);
+                        }
+                    }
+                }
+
+                bitmap.Save(filePath, ImageFormat.Png);
+            }
+        }
+
+        /// <summary>
+        /// Captures a screenshot and saves it to the specified directory.
+        /// </summary>
+        /// <param name="directory">The directory to save the screenshot.</param>
+        private static void CaptureScreenshot(string directory)
+        {
+            string filePath = Path.Combine(directory, $"screenshot_{DateTime.Now:yyyyMMdd_HHmmssfff}.png");
+            CaptureScreenWithMouse(filePath);
+        }
+
+        /// <summary>
+        /// Timer callback method to capture a screenshot.
+        /// </summary>
+        /// <param name="state">The state object passed to the callback method.</param>
+        public static void TimerCallback(object? state)
+        {
+            string directory = (string)state!;
+            CaptureScreenshot(directory);
+        }
+    }
+}

src/common/UITestAutomation/Session.cs

@@ -1,14 +1,17 @@
-// Copyright (c) Microsoft Corporation
+// 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.Collections.ObjectModel;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
 using System.Runtime.InteropServices;
-using System.Xml.Linq;
+using System.Text;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using OpenQA.Selenium.Appium;
 using OpenQA.Selenium.Appium.Windows;
 using OpenQA.Selenium.Interactions;
+using static Microsoft.PowerToys.UITest.WindowHelper;
 
 namespace Microsoft.PowerToys.UITest
 {
@@ -17,35 +20,70 @@ namespace Microsoft.PowerToys.UITest
     /// </summary>
     public class Session
     {
-        private WindowsDriver<WindowsElement> Root { get; set; }
+        public WindowsDriver<WindowsElement> Root { get; set; }
 
         private WindowsDriver<WindowsElement> WindowsDriver { get; set; }
 
-        [DllImport("user32.dll")]
-        private static extern bool SetForegroundWindow(nint hWnd);
+        private List<IntPtr> windowHandlers = new List<IntPtr>();
 
-        public Session(WindowsDriver<WindowsElement> root, WindowsDriver<WindowsElement> windowsDriver)
+        private Window? MainWindow { get; set; }
+
+        /// <summary>
+        /// Gets Main Window Handler
+        /// </summary>
+        public IntPtr MainWindowHandler { get; private set; }
+
+        /// <summary>
+        /// Gets Init Scope
+        /// </summary>
+        public PowerToysModule InitScope { get; private set; }
+
+        /// <summary>
+        /// Gets the RunAsAdmin flag.
+        /// If true, the session is running as admin.
+        /// If false, the session is not running as admin.
+        /// If null, no information is available.
+        /// </summary>
+        public bool? IsElevated { get; private set; }
+
+        public Session(WindowsDriver<WindowsElement> pRoot, WindowsDriver<WindowsElement> pDriver, PowerToysModule scope, WindowSize size)
         {
-            this.Root = root;
-            this.WindowsDriver = windowsDriver;
+            this.MainWindowHandler = IntPtr.Zero;
+            this.Root = pRoot;
+            this.WindowsDriver = pDriver;
+            this.InitScope = scope;
+
+            if (size != WindowSize.UnSpecified)
+            {
+                // Attach to the scope & reset MainWindowHandler
+                this.Attach(scope, size);
+            }
         }
 
         /// <summary>
-        /// Finds an element by selector.
+        /// Cleans up the Session Exe.
+        /// </summary>
+        public void Cleanup()
+        {
+            windowHandlers.Clear();
+        }
+
+        /// <summary>
+        /// Finds an Element or its derived class by selector.
         /// </summary>
         /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
         /// <param name="by">The selector to find the element.</param>
-        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        public T Find<T>(By by, int timeoutMS = 3000)
+        public T Find<T>(By by, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
             Assert.IsNotNull(this.WindowsDriver, $"WindowsElement is null in method Find<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
 
             // leverage findAll to filter out mismatched elements
-            var collection = this.FindAll<T>(by, timeoutMS);
+            var collection = this.FindAll<T>(by, timeoutMS, global);
 
-            Assert.IsTrue(collection.Count > 0, $"Element not found using selector: {by}");
+            Assert.IsTrue(collection.Count > 0, $"UI-Element({typeof(T).Name}) not found using selector: {by}");
 
             return collection[0];
         }
@@ -55,62 +93,159 @@ namespace Microsoft.PowerToys.UITest
         /// </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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        public T Find<T>(string name, int timeoutMS = 3000)
+        public T Find<T>(string name, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
-            return this.Find<T>(By.Name(name), timeoutMS);
+            return this.Find<T>(By.Name(name), timeoutMS, global);
         }
 
         /// <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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        public Element Find(By by, int timeoutMS = 3000)
+        public Element Find(By by, int timeoutMS = 5000, bool global = false)
         {
-            return this.Find<Element>(by, timeoutMS);
+            return this.Find<Element>(by, timeoutMS, global);
         }
 
         /// <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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        public Element Find(string name, int timeoutMS = 3000)
+        public Element Find(string name, int timeoutMS = 5000, bool global = false)
         {
-            return this.Find<Element>(By.Name(name), timeoutMS);
+            return this.Find<Element>(By.Name(name), timeoutMS, global);
         }
 
         /// <summary>
-        /// Finds all elements by selector.
+        /// Has only one Element or its derived class by selector.
+        /// </summary>
+        /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
+        /// <param name="by">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne<T>(By by, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(by, timeoutMS, global).Count == 1;
+        }
+
+        /// <summary>
+        /// Shortcut for this.HasOne<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne(By by, int timeoutMS = 5000, bool global = false)
+        {
+            return this.HasOne<Element>(by, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.HasOne<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 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne<T>(string name, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.HasOne<T>(By.Name(name), timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.HasOne<Element>(name, timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne(string name, int timeoutMS = 5000, bool global = false)
+        {
+            return this.HasOne<Element>(By.Name(name), timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Has one or more Element or its derived class by selector.
+        /// </summary>
+        /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
+        /// <param name="by">The selector to find the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has<T>(By by, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(by, timeoutMS, global).Count >= 1;
+        }
+
+        /// <summary>
+        /// Shortcut for this.Has<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to find the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has(By by, int timeoutMS = 5000, bool global = false)
+        {
+            return this.Has<Element>(by, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Has<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 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has<T>(string name, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.Has<T>(By.Name(name), timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Has<Element>(name, timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has(string name, int timeoutMS = 5000, bool global = false)
+        {
+            return this.Has<Element>(name, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Finds all Element or its derived class by selector.
         /// </summary>
         /// <typeparam name="T">The class of the elements, should be Element or its derived class.</typeparam>
         /// <param name="by">The selector to find the elements.</param>
-        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</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 = 5000, bool global = false)
             where T : Element, new()
         {
-            Assert.IsNotNull(this.WindowsDriver, $"WindowsElement is null in method FindAll<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
+            var driver = global ? this.Root : this.WindowsDriver;
+            Assert.IsNotNull(driver, $"WindowsElement is null in method FindAll<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
             var foundElements = FindHelper.FindAll<T, WindowsElement>(
                 () =>
                 {
                     if (by.GetIsAccessibilityId())
                     {
-                        var elements = this.WindowsDriver.FindElementsByAccessibilityId(by.GetAccessibilityId());
+                        var elements = driver.FindElementsByAccessibilityId(by.GetAccessibilityId());
                         return elements;
                     }
                     else
                     {
-                        var elements = this.WindowsDriver.FindElements(by.ToSeleniumBy());
+                        var elements = driver.FindElements(by.ToSeleniumBy());
                         return elements;
                     }
                 },
-                this.WindowsDriver,
+                driver,
                 timeoutMS);
 
             return foundElements ?? new ReadOnlyCollection<T>([]);
@@ -122,12 +257,12 @@ namespace Microsoft.PowerToys.UITest
         /// </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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        public ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 3000)
+        public ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
-            return this.FindAll<T>(By.Name(name), timeoutMS);
+            return this.FindAll<T>(By.Name(name), timeoutMS, global);
         }
 
         /// <summary>
@@ -135,11 +270,11 @@ namespace Microsoft.PowerToys.UITest
         /// 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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        public ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 3000)
+        public ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 5000, bool global = false)
         {
-            return this.FindAll<Element>(by, timeoutMS);
+            return this.FindAll<Element>(by, timeoutMS, global);
         }
 
         /// <summary>
@@ -147,55 +282,186 @@ namespace Microsoft.PowerToys.UITest
         /// 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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        public ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 3000)
+        public ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 5000, bool global = false)
         {
-            return this.FindAll<Element>(By.Name(name), timeoutMS);
+            return this.FindAll<Element>(By.Name(name), timeoutMS, global);
         }
 
         /// <summary>
-        /// Keyboard Action key.
+        /// Close the main window.
         /// </summary>
-        /// <param name="key1">The Keys1 to click.</param>
-        /// <param name="key2">The Keys2 to click.</param>
-        /// <param name="key3">The Keys3 to click.</param>
-        /// <param name="key4">The Keys4 to click.</param>
-        public void KeyboardAction(string key1, string key2 = "", string key3 = "", string key4 = "")
+        public void CloseMainWindow()
         {
-            PerformAction((actions, windowElement) =>
+            if (MainWindow != null)
             {
-                if (string.IsNullOrEmpty(key2))
-                {
-                    actions.SendKeys(key1);
-                }
-                else if (string.IsNullOrEmpty(key3))
-                {
-                    actions.SendKeys(key1).SendKeys(key2);
-                }
-                else if (string.IsNullOrEmpty(key4))
-                {
-                    actions.SendKeys(key1).SendKeys(key2).SendKeys(key3);
-                }
-                else
-                {
-                    actions.SendKeys(key1).SendKeys(key2).SendKeys(key3).SendKeys(key4);
-                }
+                MainWindow.Close();
+                MainWindow = null;
+            }
+        }
 
-                actions.Release();
-                actions.Build().Perform();
+        /// <summary>
+        /// Sends a combination of keys.
+        /// </summary>
+        /// <param name="keys">The keys to send.</param>
+        public void SendKeys(params Key[] keys)
+        {
+            PerformAction(() =>
+            {
+                KeyboardHelper.SendKeys(keys);
             });
         }
 
+        /// <summary>
+        /// release the key (after the hold key and drag is completed.)
+        /// </summary>
+        /// <param name="key">The key release.</param>
+        public void PressKey(Key key)
+        {
+            PerformAction(() =>
+            {
+                KeyboardHelper.PressKey(key);
+            });
+        }
+
+        /// <summary>
+        /// press and hold the specified key.
+        /// </summary>
+        /// <param name="key">The key to press and hold .</param>
+        public void ReleaseKey(Key key)
+        {
+            PerformAction(() =>
+            {
+                KeyboardHelper.ReleaseKey(key);
+            });
+        }
+
+        /// <summary>
+        /// press and hold the specified key.
+        /// </summary>
+        /// <param name="key">The key to press and release .</param>
+        public void SendKey(Key key, int msPreAction = 500, int msPostAction = 500)
+        {
+            PerformAction(
+                () =>
+                {
+                    KeyboardHelper.SendKey(key);
+                },
+                msPreAction,
+                msPostAction);
+        }
+
+        /// <summary>
+        /// Sends a sequence of keys.
+        /// </summary>
+        /// <param name="keys">An array of keys to send.</param>
+        public void SendKeySequence(params Key[] keys)
+        {
+            PerformAction(() =>
+            {
+                foreach (var key in keys)
+                {
+                    KeyboardHelper.SendKeys(key);
+                }
+            });
+        }
+
+        /// <summary>
+        /// Gets the current position of the mouse cursor as a tuple.
+        /// </summary>
+        /// <returns>A tuple containing the X and Y coordinates of the cursor.</returns>
+        public Tuple<int, int> GetMousePosition()
+        {
+            return MouseHelper.GetMousePosition();
+        }
+
+        /// <summary>
+        /// Moves the mouse cursor to the specified screen coordinates.
+        /// </summary>
+        /// <param name="x">The new x-coordinate of the cursor.</param>
+        /// <param name="y">The new y-coordinate of the cursor.</param
+        public void MoveMouseTo(int x, int y, int msPreAction = 500, int msPostAction = 500)
+        {
+            PerformAction(
+                () =>
+                {
+                    MouseHelper.MoveMouseTo(x, y);
+                },
+                msPreAction,
+                msPostAction);
+        }
+
+        /// <summary>
+        /// Performs a mouse action based on the specified action type.
+        /// </summary>
+        /// <param name="action">The mouse action to perform.</param>
+        /// <param name="msPreAction">Pre-action delay in milliseconds.</param>
+        /// <param name="msPostAction">Post-action delay in milliseconds.</param>
+        public void PerformMouseAction(MouseActionType action, int msPreAction = 500, int msPostAction = 500)
+        {
+            PerformAction(
+                () =>
+                {
+                    switch (action)
+                    {
+                        case MouseActionType.LeftClick:
+                            MouseHelper.LeftClick();
+                            break;
+                        case MouseActionType.RightClick:
+                            MouseHelper.RightClick();
+                            break;
+                        case MouseActionType.MiddleClick:
+                            MouseHelper.MiddleClick();
+                            break;
+                        case MouseActionType.LeftDoubleClick:
+                            MouseHelper.LeftDoubleClick();
+                            break;
+                        case MouseActionType.RightDoubleClick:
+                            MouseHelper.RightDoubleClick();
+                            break;
+                        case MouseActionType.LeftDown:
+                            MouseHelper.LeftDown();
+                            break;
+                        case MouseActionType.LeftUp:
+                            MouseHelper.LeftUp();
+                            break;
+                        case MouseActionType.RightDown:
+                            MouseHelper.RightDown();
+                            break;
+                        case MouseActionType.RightUp:
+                            MouseHelper.RightUp();
+                            break;
+                        case MouseActionType.MiddleDown:
+                            MouseHelper.MiddleDown();
+                            break;
+                        case MouseActionType.MiddleUp:
+                            MouseHelper.MiddleUp();
+                            break;
+                        case MouseActionType.ScrollUp:
+                            MouseHelper.ScrollUp();
+                            break;
+                        case MouseActionType.ScrollDown:
+                            MouseHelper.ScrollDown();
+                            break;
+                        default:
+                            throw new ArgumentException("Unsupported mouse action.", nameof(action));
+                    }
+                },
+                msPreAction,
+                msPostAction);
+        }
+
         /// <summary>
         /// Attaches to an existing PowerToys module.
         /// </summary>
         /// <param name="module">The PowerToys module to attach to.</param>
+        /// <param name="size">The window size to set. Default is no change to window size</param>
         /// <returns>The attached session.</returns>
-        public Session Attach(PowerToysModule module)
+        public Session Attach(PowerToysModule module, WindowSize size = WindowSize.UnSpecified)
         {
             string windowName = ModuleConfigData.Instance.GetModuleWindowName(module);
-            return this.Attach(windowName);
+            return this.Attach(windowName, size);
         }
 
         /// <summary>
@@ -203,35 +469,148 @@ namespace Microsoft.PowerToys.UITest
         /// The session should be attached when a new app is started.
         /// </summary>
         /// <param name="windowName">The window name to attach to.</param>
+        /// <param name="size">The window size to set. Default is no change to window size</param>
         /// <returns>The attached session.</returns>
-        public Session Attach(string windowName)
+        public Session Attach(string windowName, WindowSize size = WindowSize.UnSpecified)
         {
+            this.IsElevated = null;
+            this.MainWindowHandler = IntPtr.Zero;
+
             if (this.Root != null)
             {
-                var window = this.Root.FindElementByName(windowName);
-                Assert.IsNotNull(window, $"Failed to attach. Window '{windowName}' not found");
+                // search window handler by window title (admin and non-admin titles)
+                var timeout = TimeSpan.FromMinutes(2);
+                var retryInterval = TimeSpan.FromSeconds(5);
+                DateTime startTime = DateTime.Now;
+
+                List<(IntPtr HWnd, string Title)>? matchingWindows = null;
+
+                while (DateTime.Now - startTime < timeout)
+                {
+                    matchingWindows = WindowHelper.ApiHelper.FindDesktopWindowHandler(
+                    new[] { windowName, WindowHelper.AdministratorPrefix + windowName });
+
+                    if (matchingWindows.Count > 0 && matchingWindows[0].HWnd != IntPtr.Zero)
+                    {
+                        break;
+                    }
+
+                    Task.Delay(retryInterval).Wait();
+                }
+
+                if (matchingWindows == null || matchingWindows.Count == 0 || matchingWindows[0].HWnd == IntPtr.Zero)
+                {
+                    Assert.Fail($"Failed to attach. Window '{windowName}' not found after {timeout.TotalSeconds} seconds.");
+                }
+
+                // pick one from matching windows
+                this.MainWindowHandler = matchingWindows[0].HWnd;
+                this.IsElevated = matchingWindows[0].Title.StartsWith(WindowHelper.AdministratorPrefix);
+
+                ApiHelper.SetForegroundWindow(this.MainWindowHandler);
+
+                var hexWindowHandle = this.MainWindowHandler.ToInt64().ToString("x");
 
-                var windowHandle = new nint(int.Parse(window.GetAttribute("NativeWindowHandle")));
-                SetForegroundWindow(windowHandle);
-                var hexWindowHandle = windowHandle.ToString("x");
                 var appCapabilities = new AppiumOptions();
-
                 appCapabilities.AddAdditionalCapability("appTopLevelWindow", hexWindowHandle);
                 appCapabilities.AddAdditionalCapability("deviceName", "WindowsPC");
                 this.WindowsDriver = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), appCapabilities);
-                Assert.IsNotNull(this.WindowsDriver, "Attach WindowsDriver is null");
 
-                // Set implicit timeout to make element search retry every 500 ms
-                this.WindowsDriver.Manage().Timeouts().ImplicitWait = TimeSpan.FromSeconds(3);
+                this.windowHandlers.Add(this.MainWindowHandler);
+
+                if (size != WindowSize.UnSpecified)
+                {
+                    WindowHelper.SetWindowSize(this.MainWindowHandler, size);
+                }
+
+                // Set MainWindow
+                MainWindow = Find<Window>(matchingWindows[0].Title);
             }
             else
             {
                 Assert.IsNotNull(this.Root, $"Failed to attach to the window '{windowName}'. Root driver is null");
             }
 
+            Task.Delay(3000).Wait();
             return this;
         }
 
+        /// <summary>
+        /// Sets the main window size.
+        /// </summary>
+        /// <param name="size">WindowSize enum</param>
+        public void SetMainWindowSize(WindowSize size)
+        {
+            if (this.MainWindowHandler == IntPtr.Zero)
+            {
+                // Attach to the scope & reset MainWindowHandler
+                this.Attach(this.InitScope);
+            }
+
+            WindowHelper.SetWindowSize(this.MainWindowHandler, size);
+        }
+
+        /// <summary>
+        /// Gets the main window center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public (int CenterX, int CenterY) GetMainWindowCenter()
+        {
+            return WindowHelper.GetWindowCenter(this.MainWindowHandler);
+        }
+
+        /// <summary>
+        /// Gets the main window center coordinates.
+        /// </summary>
+        /// <returns>(int Left, int Top, int Right, int Bottom)</returns>
+        public (int Left, int Top, int Right, int Bottom) GetMainWindowRect()
+        {
+            return WindowHelper.GetWindowRect(this.MainWindowHandler);
+        }
+
+        /// <summary>
+        /// Launches the specified executable with optional arguments and simulates a delay before and after execution.
+        /// </summary>
+        /// <param name="executablePath">The full path to the executable to launch.</param>
+        /// <param name="arguments">Optional command-line arguments to pass to the executable.</param>
+        /// <param name="msPreAction">The number of milliseconds to wait before launching the executable. Default is 0 ms.</param>
+        /// <param name="msPostAction">The number of milliseconds to wait after launching the executable. Default is 2000 ms.</param>
+        public void StartExe(string executablePath, string arguments = "", int msPreAction = 0, int msPostAction = 2000)
+        {
+            PerformAction(
+                () =>
+                {
+                    StartExeInternal(executablePath, arguments);
+                },
+                msPreAction,
+                msPostAction);
+        }
+
+        private void StartExeInternal(string executablePath, string arguments = "")
+        {
+            var processInfo = new ProcessStartInfo
+            {
+                FileName = executablePath,
+                Arguments = arguments,
+                UseShellExecute = true,
+            };
+            Process.Start(processInfo);
+        }
+
+        /// <summary>
+        /// Terminates all running processes that match the specified process name.
+        /// Waits for each process to exit after sending the kill signal.
+        /// </summary>
+        /// <param name="processName">The name of the process to terminate (without extension, e.g., "notepad").</param>
+        public void KillAllProcessesByName(string processName)
+        {
+            foreach (var process in Process.GetProcessesByName(processName))
+            {
+                process.Kill();
+                process.WaitForExit();
+            }
+        }
+
         /// <summary>
         /// Simulates a manual operation on the element.
         /// </summary>
@@ -254,5 +633,26 @@ namespace Microsoft.PowerToys.UITest
                 Task.Delay(msPostAction).Wait();
             }
         }
+
+        /// <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 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 action, int msPreAction = 500, int msPostAction = 500)
+        {
+            if (msPreAction > 0)
+            {
+                Task.Delay(msPreAction).Wait();
+            }
+
+            action();
+
+            if (msPostAction > 0)
+            {
+                Task.Delay(msPostAction).Wait();
+            }
+        }
     }
 }

src/common/UITestAutomation/SessionHelper.cs

@@ -15,39 +15,58 @@ namespace Microsoft.PowerToys.UITest
     /// <summary>
     /// Nested class for test initialization.
     /// </summary>
-    internal class SessionHelper
+    public class SessionHelper
     {
         // Default session path is PowerToys settings dashboard
         private readonly string sessionPath = ModuleConfigData.Instance.GetModulePath(PowerToysModule.PowerToysSettings);
 
+        private readonly string runnerPath = ModuleConfigData.Instance.GetModulePath(PowerToysModule.Runner);
+
         private string? locationPath;
 
-        private WindowsDriver<WindowsElement> Root { get; set; }
+        private static WindowsDriver<WindowsElement>? root;
 
         private WindowsDriver<WindowsElement>? Driver { get; set; }
 
-        private Process? appDriver;
+        private static Process? appDriver;
+        private Process? runner;
+
+        private PowerToysModule scope;
 
         [UnconditionalSuppressMessage("SingleFile", "IL3000:Avoid accessing Assembly file path when publishing as a single file", Justification = "<Pending>")]
         public SessionHelper(PowerToysModule scope)
         {
+            this.scope = scope;
             this.sessionPath = ModuleConfigData.Instance.GetModulePath(scope);
             this.locationPath = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
 
-            var winAppDriverProcessInfo = new ProcessStartInfo
+            CheckWinAppDriverAndRoot();
+
+            var runnerProcessInfo = new ProcessStartInfo
             {
-                FileName = "C:\\Program Files (x86)\\Windows Application Driver\\WinAppDriver.exe",
+                FileName = locationPath + this.runnerPath,
                 Verb = "runas",
             };
 
-            this.appDriver = Process.Start(winAppDriverProcessInfo);
+            if (scope == PowerToysModule.PowerToysSettings)
+            {
+                this.ExitExe(runnerProcessInfo.FileName);
+                this.runner = Process.Start(runnerProcessInfo);
+            }
+        }
 
-            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 WinAppDriver And Root.
+        /// </summary>
+        public void CheckWinAppDriverAndRoot()
+        {
+            if (SessionHelper.root == null || SessionHelper.appDriver?.SessionId == null || SessionHelper.appDriver == null || SessionHelper.appDriver.HasExited)
+            {
+                this.StartWindowsAppDriverApp();
+                var desktopCapabilities = new AppiumOptions();
+                desktopCapabilities.AddAdditionalCapability("app", "Root");
+                SessionHelper.root = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), desktopCapabilities);
+            }
         }
 
         /// <summary>
@@ -56,7 +75,8 @@ namespace Microsoft.PowerToys.UITest
         /// <param name="scope">The PowerToys module to start.</param>
         public SessionHelper Init()
         {
-            this.StartExe(locationPath + this.sessionPath);
+            this.ExitExe(this.locationPath + this.sessionPath);
+            this.StartExe(this.locationPath + this.sessionPath);
 
             Assert.IsNotNull(this.Driver, $"Failed to initialize the test environment. Driver is null.");
 
@@ -71,8 +91,11 @@ namespace Microsoft.PowerToys.UITest
             ExitScopeExe();
             try
             {
-                appDriver?.Kill();
-                appDriver?.WaitForExit(); // Optional: Wait for the process to exit
+                if (this.scope == PowerToysModule.PowerToysSettings)
+                {
+                    runner?.Kill();
+                    runner?.WaitForExit(); // Optional: Wait for the process to exit
+                }
             }
             catch (Exception ex)
             {
@@ -81,31 +104,15 @@ namespace Microsoft.PowerToys.UITest
             }
         }
 
-        /// <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);
-            Console.WriteLine($"appPath: {appPath}");
-            this.Driver = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), opts);
-
-            // Set default timeout to 5 seconds
-            this.Driver.Manage().Timeouts().ImplicitWait = TimeSpan.FromSeconds(5);
-        }
-
         /// <summary>
         /// Exit a exe.
         /// </summary>
-        /// <param name="path">The path to the application executable.</param>
-        public void ExitExe(string path)
+        /// <param name="appPath">The path to the application executable.</param>
+        public void ExitExe(string appPath)
         {
             // Exit Exe
-            string exeName = Path.GetFileNameWithoutExtension(path);
+            string exeName = Path.GetFileNameWithoutExtension(appPath);
 
-            // PowerToys.FancyZonesEditor
             Process[] processes = Process.GetProcessesByName(exeName);
             foreach (Process process in processes)
             {
@@ -121,6 +128,48 @@ namespace Microsoft.PowerToys.UITest
             }
         }
 
+        /// <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 = NewWindowsDriver(opts);
+        }
+
+        /// <summary>
+        /// Starts a new exe and takes control of it.
+        /// </summary>
+        /// <param name="info">The path to the application executable.</param>
+        private WindowsDriver<WindowsElement> NewWindowsDriver(AppiumOptions info)
+        {
+            // Create driver with retry
+            var timeout = TimeSpan.FromMinutes(2);
+            var retryInterval = TimeSpan.FromSeconds(5);
+            DateTime startTime = DateTime.Now;
+
+            while (true)
+            {
+                try
+                {
+                    var res = new WindowsDriver<WindowsElement>(new Uri(ModuleConfigData.Instance.GetWindowsApplicationDriverUrl()), info);
+                    return res;
+                }
+                catch (Exception)
+                {
+                    if (DateTime.Now - startTime > timeout)
+                    {
+                        throw;
+                    }
+
+                    Task.Delay(retryInterval).Wait();
+                    CheckWinAppDriverAndRoot();
+                }
+            }
+        }
+
         /// <summary>
         /// Exit now exe.
         /// </summary>
@@ -134,16 +183,31 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         public void RestartScopeExe()
         {
-            ExitExe(sessionPath);
+            ExitScopeExe();
             StartExe(locationPath + sessionPath);
         }
 
-        public WindowsDriver<WindowsElement> GetRoot() => this.Root;
+        public WindowsDriver<WindowsElement> GetRoot()
+        {
+            return SessionHelper.root!;
+        }
 
         public WindowsDriver<WindowsElement> GetDriver()
         {
             Assert.IsNotNull(this.Driver, $"Failed to get driver. Driver is null.");
             return this.Driver;
         }
+
+        private void StartWindowsAppDriverApp()
+        {
+            var winAppDriverProcessInfo = new ProcessStartInfo
+            {
+                FileName = "C:\\Program Files (x86)\\Windows Application Driver\\WinAppDriver.exe",
+                Verb = "runas",
+            };
+
+            this.ExitExe(winAppDriverProcessInfo.FileName);
+            SessionHelper.appDriver = Process.Start(winAppDriverProcessInfo);
+        }
     }
 }

src/common/UITestAutomation/UITestAutomation.csproj

@@ -8,6 +8,9 @@
     <Nullable>enable</Nullable>
     <PublishAot>true</PublishAot>
     <InvariantGlobalization>true</InvariantGlobalization>
+    <TargetFramework>net9.0-windows10.0.22621.0</TargetFramework>
+    <UseWindowsForms>true</UseWindowsForms>
+    <PublishTrimmed>false</PublishTrimmed>
   </PropertyGroup>
 
   <ItemGroup>

src/common/UITestAutomation/UITestBase.cs

@@ -6,10 +6,16 @@ using System.Collections.ObjectModel;
 using System.Diagnostics;
 using System.Diagnostics.CodeAnalysis;
 using System.Reflection;
+using System.Runtime.InteropServices;
 using System.Xml.Linq;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium;
 using OpenQA.Selenium.Appium;
 using OpenQA.Selenium.Appium.Windows;
+using Windows.Devices.Display.Core;
+using Windows.Foundation.Metadata;
+using static Microsoft.PowerToys.UITest.UITestBase.NativeMethods;
+using static Microsoft.PowerToys.UITest.WindowHelper;
 
 namespace Microsoft.PowerToys.UITest
 {
@@ -17,19 +23,37 @@ namespace Microsoft.PowerToys.UITest
     /// Base class that should be inherited by all Test Classes.
     /// </summary>
     [TestClass]
-    public class UITestBase
+    public class UITestBase : IDisposable
     {
-        public Session Session { get; set; }
+        public required TestContext TestContext { get; set; }
 
-        private readonly SessionHelper sessionHelper;
+        public required Session Session { get; set; }
+
+        public bool IsInPipeline { get; }
+
+        public static MonitorInfoData.ParamsWrapper MonitorInfoData { get; set; } = new MonitorInfoData.ParamsWrapper() { Monitors = new List<MonitorInfoData.MonitorInfoDataWrapper>() };
 
         private readonly PowerToysModule scope;
+        private readonly WindowSize size;
+        private SessionHelper? sessionHelper;
+        private System.Threading.Timer? screenshotTimer;
+        private string? screenshotDirectory;
 
-        public UITestBase(PowerToysModule scope = PowerToysModule.PowerToysSettings)
+        public UITestBase(PowerToysModule scope = PowerToysModule.PowerToysSettings, WindowSize size = WindowSize.UnSpecified)
         {
+            this.IsInPipeline = !string.IsNullOrEmpty(Environment.GetEnvironmentVariable("platform"));
+            Console.WriteLine($"Running tests on platform: {Environment.GetEnvironmentVariable("platform")}");
+            if (IsInPipeline)
+            {
+                NativeMethods.ChangeDisplayResolution(1920, 1080);
+                NativeMethods.GetMonitorInfo();
+
+                // Escape Popups before starting
+                System.Windows.Forms.SendKeys.SendWait("{ESC}");
+            }
+
             this.scope = scope;
-            this.sessionHelper = new SessionHelper(scope).Init();
-            this.Session = new Session(this.sessionHelper.GetRoot(), this.sessionHelper.GetDriver());
+            this.size = size;
         }
 
         /// <summary>
@@ -38,6 +62,22 @@ namespace Microsoft.PowerToys.UITest
         [TestInitialize]
         public void TestInit()
         {
+            CloseOtherApplications();
+            if (IsInPipeline)
+            {
+                screenshotDirectory = Path.Combine(this.TestContext.TestResultsDirectory ?? string.Empty, "UITestScreenshots_" + Guid.NewGuid().ToString());
+                Directory.CreateDirectory(screenshotDirectory);
+
+                // Take screenshot every 1 second
+                screenshotTimer = new System.Threading.Timer(ScreenCapture.TimerCallback, screenshotDirectory, TimeSpan.Zero, TimeSpan.FromMilliseconds(1000));
+
+                // Escape Popups before starting
+                System.Windows.Forms.SendKeys.SendWait("{ESC}");
+            }
+
+            this.sessionHelper = new SessionHelper(scope).Init();
+            this.Session = new Session(this.sessionHelper.GetRoot(), this.sessionHelper.GetDriver(), scope, size);
+
             if (this.scope == PowerToysModule.PowerToysSettings)
             {
                 // close Debug warning dialog if any
@@ -50,12 +90,32 @@ namespace Microsoft.PowerToys.UITest
         }
 
         /// <summary>
-        /// UnInitializes the test.
+        /// Cleanups the test.
         /// </summary>
         [TestCleanup]
-        public void TestClean()
+        public void TestCleanup()
         {
-            this.sessionHelper.Cleanup();
+            if (IsInPipeline)
+            {
+                screenshotTimer?.Change(Timeout.Infinite, Timeout.Infinite);
+                Dispose();
+                if (TestContext.CurrentTestOutcome is UnitTestOutcome.Failed
+                    or UnitTestOutcome.Error
+                    or UnitTestOutcome.Unknown)
+                {
+                    Task.Delay(1000).Wait();
+                    AddScreenShotsToTestResultsDirectory();
+                }
+            }
+
+            this.Session.Cleanup();
+            this.sessionHelper!.Cleanup();
+        }
+
+        public void Dispose()
+        {
+            screenshotTimer?.Dispose();
+            GC.SuppressFinalize(this);
         }
 
         /// <summary>
@@ -64,47 +124,143 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
         /// <param name="by">The selector to find the element.</param>
-        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        protected T Find<T>(By by, int timeoutMS = 3000)
+        protected T Find<T>(By by, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
-            return this.Session.Find<T>(by, timeoutMS);
+            return this.Session.Find<T>(by, timeoutMS, global);
         }
 
         /// <summary>
-        /// Shortcut for this.Session.Find<Element>(By.Name(name), timeoutMS)
+        /// Shortcut for this.Session.Find<Element>(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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        protected T Find<T>(string name, int timeoutMS = 3000)
+        protected T Find<T>(string name, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
-            return this.Session.Find<T>(By.Name(name), timeoutMS);
+            return this.Session.Find<T>(By.Name(name), timeoutMS, global);
         }
 
         /// <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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        protected Element Find(By by, int timeoutMS = 3000)
+        protected Element Find(By by, int timeoutMS = 5000, bool global = false)
         {
-            return this.Session.Find(by, timeoutMS);
+            return this.Session.Find(by, timeoutMS, global);
         }
 
         /// <summary>
-        /// Shortcut for this.Session.Find<Element>(By.Name(name), timeoutMS)
+        /// Shortcut for this.Session.Find<Element>(name, timeoutMS)
         /// </summary>
         /// <param name="name">The name of the element.</param>
-        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>The found element.</returns>
-        protected Element Find(string name, int timeoutMS = 3000)
+        protected Element Find(string name, int timeoutMS = 5000, bool global = false)
         {
-            return this.Session.Find(name, timeoutMS);
+            return this.Session.Find(name, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Has only one Element or its derived class by selector.
+        /// </summary>
+        /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
+        /// <param name="by">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne<T>(By by, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(by, timeoutMS, global).Count == 1;
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.HasOne<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne(By by, int timeoutMS = 5000, bool global = false)
+        {
+            return this.Session.HasOne<Element>(by, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.HasOne<T>(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 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne<T>(string name, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.Session.HasOne<T>(By.Name(name), timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.HasOne<Element>(name, timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if only has one element, otherwise false.</returns>
+        public bool HasOne(string name, int timeoutMS = 5000, bool global = false)
+        {
+            return this.Session.HasOne<Element>(name, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.Has<T>(by, timeoutMS)
+        /// </summary>
+        /// <typeparam name="T">The class of the element, should be Element or its derived class.</typeparam>
+        /// <param name="by">The selector to find the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has<T>(By by, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.Session.FindAll<T>(by, timeoutMS, global).Count >= 1;
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.Has<Element>(by, timeoutMS)
+        /// </summary>
+        /// <param name="by">The selector to find the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has(By by, int timeoutMS = 5000, bool global = false)
+        {
+            return this.Session.Has<Element>(by, timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.Has<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 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has<T>(string name, int timeoutMS = 5000, bool global = false)
+            where T : Element, new()
+        {
+            return this.Session.Has<T>(By.Name(name), timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Shortcut for this.Session.Has<Element>(name, timeoutMS)
+        /// </summary>
+        /// <param name="name">The name of the element.</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
+        /// <returns>True if  has one or more element, otherwise false.</returns>
+        public bool Has(string name, int timeoutMS = 5000, bool global = false)
+        {
+            return this.Session.Has<Element>(name, timeoutMS, global);
         }
 
         /// <summary>
@@ -113,12 +269,12 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         /// <typeparam name="T">The class of the elements, should be Element or its derived class.</typeparam>
         /// <param name="by">The selector to find the elements.</param>
-        /// <param name="timeoutMS">The timeout in milliseconds (default is 3000).</param>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        protected ReadOnlyCollection<T> FindAll<T>(By by, int timeoutMS = 3000)
+        protected ReadOnlyCollection<T> FindAll<T>(By by, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
-            return this.Session.FindAll<T>(by, timeoutMS);
+            return this.Session.FindAll<T>(by, timeoutMS, global);
         }
 
         /// <summary>
@@ -127,12 +283,12 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         /// <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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        protected ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 3000)
+        protected ReadOnlyCollection<T> FindAll<T>(string name, int timeoutMS = 5000, bool global = false)
             where T : Element, new()
         {
-            return this.Session.FindAll<T>(By.Name(name), timeoutMS);
+            return this.Session.FindAll<T>(By.Name(name), timeoutMS, global);
         }
 
         /// <summary>
@@ -140,11 +296,11 @@ namespace Microsoft.PowerToys.UITest
         /// 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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        protected ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 3000)
+        protected ReadOnlyCollection<Element> FindAll(By by, int timeoutMS = 5000, bool global = false)
         {
-            return this.Session.FindAll<Element>(by, timeoutMS);
+            return this.Session.FindAll<Element>(by, timeoutMS, global);
         }
 
         /// <summary>
@@ -152,11 +308,137 @@ namespace Microsoft.PowerToys.UITest
         /// 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>
+        /// <param name="timeoutMS">The timeout in milliseconds (default is 5000).</param>
         /// <returns>A read-only collection of the found elements.</returns>
-        protected ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 3000)
+        protected ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 5000, bool global = false)
         {
-            return this.Session.FindAll<Element>(By.Name(name), timeoutMS);
+            return this.Session.FindAll<Element>(By.Name(name), timeoutMS, global);
+        }
+
+        /// <summary>
+        /// Scrolls the page
+        /// </summary>
+        /// <param name="scrollCount">The number of scroll attempts.</param>
+        /// <param name="direction">The direction to scroll.</param>
+        /// <param name="msPreAction">Pre-action delay in milliseconds.</param>
+        /// <param name="msPostAction">Post-action delay in milliseconds.</param>
+        public void Scroll(int scrollCount = 5, string direction = "Up", int msPreAction = 500, int msPostAction = 500)
+        {
+            MouseActionType mouseAction = direction == "Up" ? MouseActionType.ScrollUp : MouseActionType.ScrollDown;
+            for (int i = 0; i < scrollCount; i++)
+            {
+                Session.PerformMouseAction(mouseAction, msPreAction, msPostAction); // Ensure settings are visible
+            }
+        }
+
+        /// <summary>
+        /// Captures the last screenshot when the test fails.
+        /// </summary>
+        protected void CaptureLastScreenshot()
+        {
+            // Implement your screenshot capture logic here
+            // For example, save a screenshot to a file and return the file path
+            string screenshotPath = Path.Combine(this.TestContext.TestResultsDirectory ?? string.Empty, "last_screenshot.png");
+
+            this.Session.Root.GetScreenshot().SaveAsFile(screenshotPath, ScreenshotImageFormat.Png);
+
+            // Save screenshot to screenshotPath & upload to test attachment
+            this.TestContext.AddResultFile(screenshotPath);
+        }
+
+        /// <summary>
+        /// Retrieves the color of the pixel at the specified screen coordinates.
+        /// </summary>
+        /// <param name="x">The X coordinate on the screen.</param>
+        /// <param name="y">The Y coordinate on the screen.</param>
+        /// <returns>The color of the pixel at the specified coordinates.</returns>
+        public Color GetPixelColor(int x, int y)
+        {
+            return WindowHelper.GetPixelColor(x, y);
+        }
+
+        /// <summary>
+        /// Retrieves the color of the pixel at the specified screen coordinates as a string.
+        /// </summary>
+        /// <param name="x">The X coordinate on the screen.</param>
+        /// <param name="y">The Y coordinate on the screen.</param>
+        /// <returns>The color of the pixel at the specified coordinates.</returns>
+        public string GetPixelColorString(int x, int y)
+        {
+            return WindowHelper.GetPixelColorString(x, y);
+        }
+
+        /// <summary>
+        /// Gets the size of the display.
+        /// </summary>
+        /// <returns>
+        /// A tuple containing the width and height of the display.
+        /// </returns
+        public Tuple<int, int> GetDisplaySize()
+        {
+            return WindowHelper.GetDisplaySize();
+        }
+
+        /// <summary>
+        /// Sends a combination of keys.
+        /// </summary>
+        /// <param name="keys">The keys to send.</param>
+        public void SendKeys(params Key[] keys)
+        {
+            this.Session.SendKeys(keys);
+        }
+
+        /// <summary>
+        /// Sends a sequence of keys.
+        /// </summary>
+        /// <param name="keys">An array of keys to send.</param>
+        public void SendKeySequence(params Key[] keys)
+        {
+            this.Session.SendKeySequence(keys);
+        }
+
+        /// <summary>
+        /// Gets the current position of the mouse cursor as a tuple.
+        /// </summary>
+        /// <returns>A tuple containing the X and Y coordinates of the cursor.</returns>
+        public Tuple<int, int> GetMousePosition()
+        {
+            return this.Session.GetMousePosition();
+        }
+
+        /// <summary>
+        /// Gets the screen center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public (int CenterX, int CenterY) GetScreenCenter()
+        {
+            return WindowHelper.GetScreenCenter();
+        }
+
+        public bool IsWindowOpen(string windowName)
+        {
+            return WindowHelper.IsWindowOpen(windowName);
+        }
+
+        /// <summary>
+        /// Moves the mouse cursor to the specified screen coordinates.
+        /// </summary>
+        /// <param name="x">The new x-coordinate of the cursor.</param>
+        /// <param name="y">The new y-coordinate of the cursor.</param
+        public void MoveMouseTo(int x, int y)
+        {
+            this.Session.MoveMouseTo(x, y);
+        }
+
+        protected void AddScreenShotsToTestResultsDirectory()
+        {
+            if (screenshotDirectory != null)
+            {
+                foreach (string file in Directory.GetFiles(screenshotDirectory))
+                {
+                    this.TestContext.AddResultFile(file);
+                }
+            }
         }
 
         /// <summary>
@@ -164,8 +446,8 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         public void RestartScopeExe()
         {
-            this.sessionHelper.RestartScopeExe();
-            this.Session = new Session(this.sessionHelper.GetRoot(), this.sessionHelper.GetDriver());
+            this.sessionHelper!.RestartScopeExe();
+            this.Session = new Session(this.sessionHelper.GetRoot(), this.sessionHelper.GetDriver(), this.scope, this.size);
             return;
         }
 
@@ -174,8 +456,194 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         public void ExitScopeExe()
         {
-            this.sessionHelper.ExitScopeExe();
+            this.sessionHelper!.ExitScopeExe();
             return;
         }
+
+        private void CloseOtherApplications()
+        {
+            // Close other applications
+            var processNamesToClose = new List<string>
+            {
+                "PowerToys",
+                "PowerToys.Settings",
+                "PowerToys.FancyZonesEditor",
+            };
+            foreach (var processName in processNamesToClose)
+            {
+                foreach (var process in Process.GetProcessesByName(processName))
+                {
+                    process.Kill();
+                    process.WaitForExit();
+                }
+            }
+        }
+
+        public class NativeMethods
+        {
+            [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Ansi)]
+            public struct DISPLAY_DEVICE
+            {
+                public int cb;
+                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 32)]
+                public string DeviceName;
+                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 128)]
+                public string DeviceString;
+                public int StateFlags;
+                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 128)]
+                public string DeviceID;
+                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 128)]
+                public string DeviceKey;
+            }
+
+            [DllImport("user32.dll")]
+            private static extern int EnumDisplaySettings(IntPtr deviceName, int modeNum, ref DEVMODE devMode);
+
+            [DllImport("user32.dll")]
+            private static extern int EnumDisplaySettings(string deviceName, int modeNum, ref DEVMODE devMode);
+
+            [DllImport("user32.dll", CharSet = CharSet.Ansi)]
+            private static extern bool EnumDisplayDevices(IntPtr lpDevice, int iDevNum, ref DISPLAY_DEVICE lpDisplayDevice, int dwFlags);
+
+            [DllImport("user32.dll")]
+            private static extern int ChangeDisplaySettings(ref DEVMODE devMode, int flags);
+
+            [DllImport("user32.dll", CharSet = CharSet.Ansi)]
+            private static extern int ChangeDisplaySettingsEx(IntPtr lpszDeviceName, ref DEVMODE lpDevMode, IntPtr hwnd, uint dwflags, IntPtr lParam);
+
+            private const int DM_PELSWIDTH = 0x80000;
+            private const int DM_PELSHEIGHT = 0x100000;
+
+            public const int ENUM_CURRENT_SETTINGS = -1;
+            public const int CDS_TEST = 0x00000002;
+            public const int CDS_UPDATEREGISTRY = 0x01;
+            public const int DISP_CHANGE_SUCCESSFUL = 0;
+            public const int DISP_CHANGE_RESTART = 1;
+            public const int DISP_CHANGE_FAILED = -1;
+
+            [StructLayout(LayoutKind.Sequential)]
+            public struct DEVMODE
+            {
+                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 32)]
+                public string DmDeviceName;
+                public short DmSpecVersion;
+                public short DmDriverVersion;
+                public short DmSize;
+                public short DmDriverExtra;
+                public int DmFields;
+                public int DmPositionX;
+                public int DmPositionY;
+                public int DmDisplayOrientation;
+                public int DmDisplayFixedOutput;
+                public short DmColor;
+                public short DmDuplex;
+                public short DmYResolution;
+                public short DmTTOption;
+                public short DmCollate;
+                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 32)]
+                public string DmFormName;
+                public short DmLogPixels;
+                public int DmBitsPerPel;
+                public int DmPelsWidth;
+                public int DmPelsHeight;
+                public int DmDisplayFlags;
+                public int DmDisplayFrequency;
+                public int DmICMMethod;
+                public int DmICMIntent;
+                public int DmMediaType;
+                public int DmDitherType;
+                public int DmReserved1;
+                public int DmReserved2;
+                public int DmPanningWidth;
+                public int DmPanningHeight;
+            }
+
+            public static void GetMonitorInfo()
+            {
+                int deviceIndex = 0;
+                DISPLAY_DEVICE d = default(DISPLAY_DEVICE);
+                d.cb = Marshal.SizeOf(d);
+
+                Console.WriteLine("monitor list :");
+                while (EnumDisplayDevices(IntPtr.Zero, deviceIndex, ref d, 0))
+                {
+                    Console.WriteLine($"monitor {deviceIndex + 1}:");
+                    Console.WriteLine($"  name: {d.DeviceName}");
+                    Console.WriteLine($"  string: {d.DeviceString}");
+                    Console.WriteLine($"  ID: {d.DeviceID}");
+                    Console.WriteLine($"  key: {d.DeviceKey}");
+                    Console.WriteLine();
+
+                    DEVMODE dm = default(DEVMODE);
+                    dm.DmSize = (short)Marshal.SizeOf<DEVMODE>();
+                    int modeNum = 0;
+                    while (EnumDisplaySettings(d.DeviceName, modeNum, ref dm) > 0)
+                    {
+                        MonitorInfoData.Monitors.Add(new MonitorInfoData.MonitorInfoDataWrapper()
+                        {
+                            DeviceName = d.DeviceName,
+                            DeviceString = d.DeviceString,
+                            DeviceID = d.DeviceID,
+                            DeviceKey = d.DeviceKey,
+                            PelsWidth = dm.DmPelsWidth,
+                            PelsHeight = dm.DmPelsHeight,
+                            DisplayFrequency = dm.DmDisplayFrequency,
+                        });
+                        Console.WriteLine($"  mode {modeNum}: {dm.DmPelsWidth}x{dm.DmPelsHeight} @ {dm.DmDisplayFrequency}Hz");
+                        modeNum++;
+                    }
+
+                    deviceIndex++;
+                    d.cb = Marshal.SizeOf(d); // Reset the size for the next device
+                }
+            }
+
+            public static void ChangeDisplayResolution(int PelsWidth, int PelsHeight)
+            {
+                Screen screen = Screen.PrimaryScreen!;
+                if (screen.Bounds.Width == PelsWidth && screen.Bounds.Height == PelsHeight)
+                {
+                    return;
+                }
+
+                DEVMODE devMode = default(DEVMODE);
+                devMode.DmDeviceName = new string(new char[32]);
+                devMode.DmFormName = new string(new char[32]);
+                devMode.DmSize = (short)Marshal.SizeOf<DEVMODE>();
+
+                int modeNum = 0;
+                while (EnumDisplaySettings(IntPtr.Zero, modeNum, ref devMode) > 0)
+                {
+                    Console.WriteLine($"Mode {modeNum}: {devMode.DmPelsWidth}x{devMode.DmPelsHeight} @ {devMode.DmDisplayFrequency}Hz");
+                    modeNum++;
+                }
+
+                devMode.DmPelsWidth = PelsWidth;
+                devMode.DmPelsHeight = PelsHeight;
+
+                int result = NativeMethods.ChangeDisplaySettings(ref devMode, NativeMethods.CDS_TEST);
+
+                if (result == DISP_CHANGE_SUCCESSFUL)
+                {
+                    result = ChangeDisplaySettings(ref devMode, CDS_UPDATEREGISTRY);
+                    if (result == DISP_CHANGE_SUCCESSFUL)
+                    {
+                        Console.WriteLine($"Changing display resolution to {devMode.DmPelsWidth}x{devMode.DmPelsHeight}");
+                    }
+                    else
+                    {
+                        Console.WriteLine($"Failed to change display resolution. Error code: {result}");
+                    }
+                }
+                else if (result == DISP_CHANGE_RESTART)
+                {
+                    Console.WriteLine($"Changing display resolution to {devMode.DmPelsWidth}x{devMode.DmPelsHeight} requires a restart");
+                }
+                else
+                {
+                    Console.WriteLine($"Failed to change display resolution. Error code: {result}");
+                }
+            }
+        }
     }
 }

src/common/UITestAutomation/VisualAssert.cs

@@ -0,0 +1,160 @@
+// 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.Drawing;
+using System.IO;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace Microsoft.PowerToys.UITest
+{
+    public static class VisualAssert
+    {
+        /// <summary>
+        /// Asserts current visual state of the element is equal with base line image.
+        /// To use this VisualAssert, you need to set Window Theme to Light-Mode to avoid Theme color difference in baseline image.
+        /// Such limitation could be removed either Auto-generate baseline image for both Light & Dark mode
+        /// </summary>
+        /// <param name="testContext">TestContext object</param>
+        /// <param name="element">Element object</param>
+        /// <param name="scenarioSubname">additional scenario name if two or more scenarios in one test</param>
+        [RequiresUnreferencedCode("This method uses reflection which may not be compatible with trimming.")]
+        public static void AreEqual(TestContext? testContext, Element element, string scenarioSubname = "")
+        {
+            var pipelinePlatform = Environment.GetEnvironmentVariable("platform");
+
+            // Perform visual validation only in the pipeline
+            if (string.IsNullOrEmpty(pipelinePlatform))
+            {
+                Console.WriteLine("Skip visual validation in the local run.");
+                return;
+            }
+
+            if (element == null)
+            {
+                Assert.Fail("Element object is null or invalid");
+            }
+
+            var stackTrace = new StackTrace();
+            var callerFrame = stackTrace.GetFrame(1);
+            var callerMethod = callerFrame?.GetMethod();
+
+            var callerName = callerMethod?.Name;
+            var callerClassName = callerMethod?.DeclaringType?.Name;
+
+            if (string.IsNullOrEmpty(callerName) || string.IsNullOrEmpty(callerClassName))
+            {
+                Assert.Fail("Unable to determine the caller method and class name.");
+            }
+
+            if (string.IsNullOrWhiteSpace(scenarioSubname))
+            {
+                scenarioSubname = string.Join("_", callerClassName, callerName, pipelinePlatform);
+            }
+            else
+            {
+                scenarioSubname = string.Join("_", callerClassName, callerName, scenarioSubname.Trim(), pipelinePlatform);
+            }
+
+            var baselineImageResourceName = callerMethod!.DeclaringType!.Assembly.GetManifestResourceNames().Where(name => name.Contains(scenarioSubname)).FirstOrDefault();
+
+            var tempTestImagePath = GetTempFilePath(scenarioSubname, "test", ".png");
+
+            element.SaveToPngFile(tempTestImagePath);
+
+            if (string.IsNullOrEmpty(baselineImageResourceName)
+                || !Path.GetFileNameWithoutExtension(baselineImageResourceName).EndsWith(scenarioSubname))
+            {
+                testContext?.AddResultFile(tempTestImagePath);
+                Assert.Fail($"Baseline image for scenario {scenarioSubname} can't be found, test image saved in file://{tempTestImagePath.Replace('\\', '/')}");
+            }
+
+            var tempBaselineImagePath = GetTempFilePath(scenarioSubname, "baseline", Path.GetExtension(baselineImageResourceName));
+
+            bool isSame = false;
+
+            using (var stream = callerMethod!.DeclaringType!.Assembly.GetManifestResourceStream(baselineImageResourceName))
+            {
+                if (stream == null)
+                {
+                    Assert.Fail($"Resource stream '{baselineImageResourceName}' is null.");
+                }
+
+                using (var baselineImage = new Bitmap(stream))
+                {
+                    using (var testImage = new Bitmap(tempTestImagePath))
+                    {
+                        isSame = VisualAssert.AreEqual(baselineImage, testImage);
+
+                        if (!isSame)
+                        {
+                            // Copy baseline image to temp folder as well
+                            baselineImage.Save(tempBaselineImagePath);
+                        }
+                    }
+                }
+            }
+
+            if (!isSame)
+            {
+                if (testContext != null)
+                {
+                    testContext.AddResultFile(tempBaselineImagePath);
+                    testContext.AddResultFile(tempTestImagePath);
+                }
+
+                Assert.Fail($"Fail to validate visual result for scenario {scenarioSubname}, baseline image can be found file://{tempBaselineImagePath.Replace('\\', '/')}, and test image can be found file://{tempTestImagePath.Replace('\\', '/')}");
+            }
+        }
+
+        /// <summary>
+        /// Get temp file path
+        /// </summary>
+        /// <param name="scenario">scenario name</param>
+        /// <param name="imageType">baseline or test image</param>
+        /// <param name="extension">image file extension</param>
+        /// <returns>full temp file path</returns>
+        private static string GetTempFilePath(string scenario, string imageType, string extension)
+        {
+            var tempFileFullName = $"{scenario}_{imageType}{extension}";
+
+            // Remove invalid filename character if any
+            Path.GetInvalidFileNameChars().ToList().ForEach(c => tempFileFullName = tempFileFullName.Replace(c, '-'));
+
+            return Path.Combine(Path.GetTempPath(), tempFileFullName);
+        }
+
+        /// <summary>
+        /// Test if two images are equal bit-by-bit
+        /// </summary>
+        /// <param name="baselineImage">baseline image</param>
+        /// <param name="testImage">test image</param>
+        /// <returns>true if are equal,otherwise false</returns>
+        private static bool AreEqual(Bitmap baselineImage, Bitmap testImage)
+        {
+            if (baselineImage.Width != testImage.Width || baselineImage.Height != testImage.Height)
+            {
+                return false;
+            }
+
+            // WinAppDriver sometimes adds a border to the screenshot (around 2 pix width), and it is not always consistent.
+            // So we exclude the border when comparing the images, and usually it is the edge of the windows, won't affect the comparison.
+            int excludeBorderWidth = 5, excludeBorderHeight = 5;
+
+            for (int x = excludeBorderWidth; x < baselineImage.Width - excludeBorderWidth; x++)
+            {
+                for (int y = excludeBorderHeight; y < baselineImage.Height - excludeBorderHeight; y++)
+                {
+                    if (!VisualHelper.PixIsSame(baselineImage.GetPixel(x, y), testImage.GetPixel(x, y)))
+                    {
+                        return false;
+                    }
+                }
+            }
+
+            return true;
+        }
+    }
+}

src/common/UITestAutomation/VisualHelper.cs

@@ -0,0 +1,33 @@
+// 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;
+using System.Collections.Generic;
+using System.Drawing;
+using System.Drawing.Imaging;
+using System.Linq;
+using System.Runtime.InteropServices;
+using System.Security.Cryptography;
+using System.Text;
+using System.Threading.Tasks;
+
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace Microsoft.PowerToys.UITest
+{
+    internal static class VisualHelper
+    {
+        /// <summary>
+        /// Compare two pixels with a fuzz factor
+        /// </summary>
+        /// <param name="c1">base color</param>
+        /// <param name="c2">test color</param>
+        /// <param name="fuzz">fuzz factor, default is 10</param>
+        /// <returns>true if same, otherwise is false</returns>
+        public static bool PixIsSame(Color c1, Color c2, int fuzz = 10)
+        {
+            return Math.Abs(c1.A - c2.A) <= fuzz && Math.Abs(c1.R - c2.R) <= fuzz && Math.Abs(c1.G - c2.G) <= fuzz && Math.Abs(c1.B - c2.B) <= fuzz;
+        }
+    }
+}

src/common/UITestAutomation/WindowHelper.cs

@@ -0,0 +1,331 @@
+// 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;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Microsoft.PowerToys.UITest
+{
+    internal static class WindowHelper
+    {
+        internal const string AdministratorPrefix = "Administrator: ";
+
+        /// <summary>
+        /// Sets the main window size.
+        /// </summary>
+        /// <param name="size">WindowSize enum</param>
+        public static void SetWindowSize(IntPtr windowHandler, WindowSize size)
+        {
+            if (size == WindowSize.UnSpecified)
+            {
+                return;
+            }
+
+            int width = 0, height = 0;
+
+            switch (size)
+            {
+                case WindowSize.Small:
+                    width = 640;
+                    height = 480;
+                    break;
+                case WindowSize.Small_Vertical:
+                    width = 480;
+                    height = 640;
+                    break;
+                case WindowSize.Medium:
+                    width = 1024;
+                    height = 768;
+                    break;
+                case WindowSize.Medium_Vertical:
+                    width = 768;
+                    height = 1024;
+                    break;
+                case WindowSize.Large:
+                    width = 1920;
+                    height = 1080;
+                    break;
+                case WindowSize.Large_Vertical:
+                    width = 1080;
+                    height = 1920;
+                    break;
+            }
+
+            if (width > 0 && height > 0)
+            {
+                WindowHelper.SetMainWindowSize(windowHandler, width, height);
+            }
+        }
+
+        /// <summary>
+        /// Gets the main window center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public static (int CenterX, int CenterY) GetWindowCenter(IntPtr windowHandler)
+        {
+            if (windowHandler == IntPtr.Zero)
+            {
+                return (0, 0);
+            }
+            else
+            {
+                var rect = ApiHelper.GetWindowCenter(windowHandler);
+                return (rect.CenterX, rect.CenterY);
+            }
+        }
+
+        /// <summary>
+        /// Gets the main window center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public static (int Left, int Top, int Right, int Bottom) GetWindowRect(IntPtr windowHandler)
+        {
+            if (windowHandler == IntPtr.Zero)
+            {
+                return (0, 0, 0, 0);
+            }
+            else
+            {
+                var rect = ApiHelper.GetWindowRect(windowHandler);
+                return (rect.Left, rect.Top, rect.Right, rect.Bottom);
+            }
+        }
+
+        /// <summary>
+        /// Gets the screen center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public static (int CenterX, int CenterY) GetScreenCenter()
+        {
+            return ApiHelper.GetScreenCenter();
+        }
+
+        /// <summary>
+        /// Sets the main window size based on Width and Height.
+        /// </summary>
+        /// <param name="width">the width in pixel</param>
+        /// <param name="height">the height in pixel</param>
+        public static void SetMainWindowSize(IntPtr windowHandler, int width, int height)
+        {
+            if (windowHandler == IntPtr.Zero
+                || width <= 0
+                || height <= 0)
+            {
+                return;
+            }
+
+            ApiHelper.SetWindowPos(windowHandler, IntPtr.Zero, 0, 0, width, height, ApiHelper.SetWindowPosNoZorder | ApiHelper.SetWindowPosShowWindow);
+
+            // Wait for 1000ms after resize
+            Task.Delay(1000).Wait();
+        }
+
+        /// <summary>
+        /// Retrieves the color of the pixel at the specified screen coordinates.
+        /// </summary>
+        /// <param name="x">The X coordinate on the screen.</param>
+        /// <param name="y">The Y coordinate on the screen.</param>
+        /// <returns>The color of the pixel at the specified coordinates.</returns>
+        public static Color GetPixelColor(int x, int y)
+        {
+            IntPtr hdc = ApiHelper.GetDC(IntPtr.Zero);
+            uint pixel = ApiHelper.GetPixel(hdc, x, y);
+            _ = ApiHelper.ReleaseDC(IntPtr.Zero, hdc);
+
+            int r = (int)(pixel & 0x000000FF);
+            int g = (int)((pixel & 0x0000FF00) >> 8);
+            int b = (int)((pixel & 0x00FF0000) >> 16);
+
+            return Color.FromArgb(r, g, b);
+        }
+
+        /// <summary>
+        /// Retrieves the color of the pixel at the specified screen coordinates as a string.
+        /// </summary>
+        /// <param name="x">The X coordinate on the screen.</param>
+        /// <param name="y">The Y coordinate on the screen.</param>
+        /// <returns>The color of the pixel at the specified coordinates.</returns>
+        public static string GetPixelColorString(int x, int y)
+        {
+            Color color = WindowHelper.GetPixelColor(x, y);
+            return $"#{color.R:X2}{color.G:X2}{color.B:X2}";
+        }
+
+        /// <summary>
+        /// Gets the size of the display.
+        /// </summary>
+        /// <returns>
+        /// A tuple containing the width and height of the display.
+        /// </returns
+        public static Tuple<int, int> GetDisplaySize()
+        {
+            IntPtr hdc = ApiHelper.GetDC(IntPtr.Zero);
+            int screenWidth = ApiHelper.GetDeviceCaps(hdc, ApiHelper.DESKTOPHORZRES);
+            int screenHeight = ApiHelper.GetDeviceCaps(hdc, ApiHelper.DESKTOPVERTRES);
+            _ = ApiHelper.ReleaseDC(IntPtr.Zero, hdc);
+
+            return Tuple.Create(screenWidth, screenHeight);
+        }
+
+        public static bool IsWindowOpen(string windowName)
+        {
+            var matchingWindows = ApiHelper.FindDesktopWindowHandler([windowName, AdministratorPrefix + windowName]);
+            return matchingWindows.Count > 0;
+        }
+
+        internal static class ApiHelper
+        {
+            [DllImport("user32.dll")]
+            public static extern bool SetForegroundWindow(IntPtr hWnd);
+
+            public const uint SetWindowPosNoMove = 0x0002;
+            public const uint SetWindowPosNoZorder = 0x0004;
+            public const uint SetWindowPosShowWindow = 0x0040;
+
+            [DllImport("user32.dll", SetLastError = true)]
+            public static extern bool SetWindowPos(IntPtr hWnd, IntPtr hWndInsertAfter, int x, int y, int cx, int cy, uint uFlags);
+
+            // Delegate for the EnumWindows callback function
+            private delegate bool EnumWindowsProc(IntPtr hWnd, IntPtr lParam);
+
+            // P/Invoke declaration for EnumWindows
+            [DllImport("user32.dll")]
+            [return: MarshalAs(UnmanagedType.Bool)]
+            private static extern bool EnumWindows(EnumWindowsProc lpEnumFunc, IntPtr lParam);
+
+            // P/Invoke declaration for GetWindowTextLength
+            [DllImport("user32.dll", SetLastError = true, CharSet = CharSet.Auto)]
+            private static extern int GetWindowTextLength(IntPtr hWnd);
+
+            // P/Invoke declaration for GetWindowText
+            [DllImport("user32.dll", SetLastError = true, CharSet = CharSet.Auto)]
+            private static extern int GetWindowText(IntPtr hWnd, StringBuilder lpString, int nMaxCount);
+
+            [DllImport("user32.dll")]
+            public static extern IntPtr GetDC(IntPtr hWnd);
+
+            [DllImport("gdi32.dll")]
+            public static extern uint GetPixel(IntPtr hdc, int x, int y);
+
+            [DllImport("gdi32.dll")]
+            public static extern int GetDeviceCaps(IntPtr hdc, int nIndex);
+
+            public const int DESKTOPHORZRES = 118;
+            public const int DESKTOPVERTRES = 117;
+
+            [DllImport("user32.dll")]
+            public static extern int ReleaseDC(IntPtr hWnd, IntPtr hDC);
+
+            // Define the Win32 RECT structure
+            [StructLayout(LayoutKind.Sequential)]
+            public struct RECT
+            {
+                public int Left;    // X coordinate of the left edge of the window
+                public int Top;     // Y coordinate of the top edge of the window
+                public int Right;   // X coordinate of the right edge of the window
+                public int Bottom;  // Y coordinate of the bottom edge of the window
+            }
+
+            // Import GetWindowRect API to retrieve window's screen coordinates
+            [DllImport("user32.dll")]
+            public static extern bool GetWindowRect(IntPtr hWnd, out RECT lpRect);
+
+            public static List<(IntPtr HWnd, string Title)> FindDesktopWindowHandler(string[] matchingWindowsTitles)
+            {
+                var windows = new List<(IntPtr HWnd, string Title)>();
+
+                _ = EnumWindows(
+                    (hWnd, lParam) =>
+                    {
+                        int length = GetWindowTextLength(hWnd);
+                        if (length > 0)
+                        {
+                            var builder = new StringBuilder(length + 1);
+                            _ = GetWindowText(hWnd, builder, builder.Capacity);
+
+                            var title = builder.ToString();
+                            if (matchingWindowsTitles.Contains(title))
+                            {
+                                windows.Add((hWnd, title));
+                            }
+                        }
+
+                        return true; // Continue enumeration
+                    },
+                    IntPtr.Zero);
+
+                return windows;
+            }
+
+            /// <summary>
+            /// Get the center point coordinates of a specified window (in screen coordinates)
+            /// </summary>
+            /// <param name="hWnd">The window handle</param>
+            /// <returns>The center point (x, y)</returns>
+            public static (int CenterX, int CenterY) GetWindowCenter(IntPtr hWnd)
+            {
+                if (hWnd == IntPtr.Zero)
+                {
+                    throw new ArgumentException("Invalid window handle");
+                }
+
+                if (GetWindowRect(hWnd, out RECT rect))
+                {
+                    int width = rect.Right - rect.Left;
+                    int height = rect.Bottom - rect.Top;
+
+                    int centerX = rect.Left + (width / 2);
+                    int centerY = rect.Top + (height / 2);
+
+                    return (centerX, centerY);
+                }
+                else
+                {
+                    throw new InvalidOperationException("Failed to retrieve window coordinates");
+                }
+            }
+
+            public static (int Left, int Top, int Right, int Bottom) GetWindowRect(IntPtr hWnd)
+            {
+                if (hWnd == IntPtr.Zero)
+                {
+                    throw new ArgumentException("Invalid window handle");
+                }
+
+                if (GetWindowRect(hWnd, out RECT rect))
+                {
+                    return (rect.Left, rect.Top, rect.Right, rect.Bottom);
+                }
+                else
+                {
+                    throw new InvalidOperationException("Failed to retrieve window coordinates");
+                }
+            }
+
+            [DllImport("user32.dll")]
+            public static extern int GetSystemMetrics(int nIndex);
+
+            public enum SystemMetric
+            {
+                ScreenWidth = 0,            // Width of the primary screen in pixels (SM_CXSCREEN)
+                ScreenHeight = 1,           // Height of the primary screen in pixels (SM_CYSCREEN)
+                VirtualScreenWidth = 78,    // Width of the virtual screen that includes all monitors (SM_CXVIRTUALSCREEN)
+                VirtualScreenHeight = 79,   // Height of the virtual screen that includes all monitors (SM_CYVIRTUALSCREEN)
+                MonitorCount = 80,          // Number of display monitors (SM_CMONITORS, available on Windows XP+)
+            }
+
+            public static (int CenterX, int CenterY) GetScreenCenter()
+            {
+                int width = GetSystemMetrics((int)SystemMetric.ScreenWidth);
+                int height = GetSystemMetrics((int)SystemMetric.ScreenHeight);
+
+                return (width / 2, height / 2);
+            }
+        }
+    }
+}