Revert pipeline settings

Co-authored-by: Xiaofeng Wang (from Dev Box) <xiaofengwang@microsoft.com>

.pipelines/v2/templates/job-build-project.yml

@@ -50,6 +50,9 @@ parameters:
   - name: runTests
     type: boolean
     default: true
+  - name: buildTests
+    type: boolean
+    default: true
   - name: useVSPreview
     type: boolean
     default: false
@@ -110,7 +113,7 @@ jobs:
     JobOutputArtifactName: build-$(BuildPlatform)-$(BuildConfiguration)${{ parameters.artifactStem }}
     NUGET_RESTORE_MSBUILD_ARGS: /p:Platform=$(BuildPlatform) # Required for nuget to work due to self contained
     NODE_OPTIONS: --max_old_space_size=16384
-    ${{ if eq(parameters.runTests, true) }}:
+    ${{ if or(eq(parameters.runTests, true), eq(parameters.buildTests, true)) }}:
       MSBuildMainBuildTargets: Build;Test
     ${{ else }}:
       MSBuildMainBuildTargets: Build
@@ -514,7 +517,7 @@ jobs:
     displayName: Stage GPO files
 
   # Running the tests may result in future jobs consuming artifacts out of this build
-  - ${{ if eq(parameters.runTests, true) }}:
+  - ${{ if or(eq(parameters.runTests, true), eq(parameters.buildTests, true)) }}:
     - task: CopyFiles@2
       displayName: Stage entire build output
       inputs:

.pipelines/v2/templates/job-test-project.yml

@@ -11,10 +11,14 @@ parameters:
   - name: useLatestWebView2
     type: boolean
     default: false
+  - name: isUIAutomationPipeline
+    type: boolean
+    default: false
 
 jobs:
 - job: Test${{ parameters.platform }}${{ parameters.configuration }}
   displayName: Test ${{ parameters.platform }} ${{ parameters.configuration }}
+  timeoutInMinutes: 300
   variables:
     BuildPlatform: ${{ parameters.platform }}
     BuildConfiguration: ${{ parameters.configuration }}
@@ -101,8 +105,16 @@ jobs:
       vsTestVersion: 'toolsInstaller'
       uiTests: true
       rerunFailedTests: true
-      testAssemblyVer2: |
-        **\UITests-FancyZones.dll
-        **\UITests-FancyZonesEditor.dll
-        !**\obj\**
-        !**\ref\**
+      ${{ if eq(parameters.isUIAutomationPipeline, true) }}:
+        testAssemblyVer2: |
+          **\*UITest*.dll
+          !**\obj\**
+          !**\ref\**
+      ${{ else }}:
+        testAssemblyVer2: |
+          **\UITests-FancyZones.dll
+          **\UITests-FancyZonesEditor.dll
+          !**\obj\**
+          !**\ref\**
+    env:
+      platform: '$(BuildPlatform)'

.pipelines/v2/templates/pipeline-ui-tests-automation.yml

@@ -0,0 +1,60 @@
+variables:
+  - name: runCodesignValidationInjectionBG
+    value: false
+  - name: EnablePipelineCache
+    value: true
+  - ${{ if eq(parameters.enableMsBuildCaching, true) }}:
+    - name: EnablePipelineCache
+      value: true
+
+parameters:
+  - name: buildPlatforms
+    type: object
+    default:
+      - x64
+      - arm64
+  - name: enableMsBuildCaching
+    type: boolean
+    default: false
+  - name: useVSPreview
+    type: boolean
+    default: false
+  - name: useLatestWebView2
+    type: boolean
+    default: false
+
+stages:
+  - ${{ each platform in parameters.buildPlatforms }}:
+    - stage: Build_${{ platform }}
+      displayName: Build ${{ platform }}
+      dependsOn: []
+      jobs:
+        - template: job-build-project.yml
+          parameters:
+            pool:
+              ${{ if eq(variables['System.CollectionId'], 'cb55739e-4afe-46a3-970f-1b49d8ee7564') }}:
+                name: SHINE-INT-L
+              ${{ else }}:
+                name: SHINE-OSS-L
+              ${{ if eq(parameters.useVSPreview, true) }}:
+                demands: ImageOverride -equals SHINE-VS17-Preview
+            buildPlatforms:
+              - ${{ platform }}
+            buildConfigurations: [Release]
+            enablePackageCaching: true
+            enableMsBuildCaching: ${{ parameters.enableMsBuildCaching }}
+            runTests: false
+            buildTests: true
+            useVSPreview: ${{ parameters.useVSPreview }}
+
+    - stage: Test_${{ platform }}
+      displayName: Test ${{ platform }}
+      dependsOn:
+        - Build_${{platform}}
+      jobs:
+        - template: job-test-project.yml
+          parameters:
+            platform: ${{ platform }}
+            configuration: Release
+            useLatestWebView2: ${{ parameters.useLatestWebView2 }}
+            isUIAutomationPipeline: true

.pipelines/verifyDepsJsonLibraryVersions.ps1

@@ -91,5 +91,4 @@ if ($totalFailures -gt 0) {
 }
 
 Write-Host -ForegroundColor Green "All " $referencedFileVersionsPerDll.keys.Count " libraries are mentioned with the same version across the dependencies.`r`n"
-exit 0
-
+exit 0

PowerToys.sln

@@ -706,6 +706,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "RegistryPreview.FuzzTests",
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Microsoft.CmdPal.Ext.System", "src\modules\cmdpal\Exts\Microsoft.CmdPal.Ext.System\Microsoft.CmdPal.Ext.System.csproj", "{64B88F02-CD88-4ED8-9624-989A800230F9}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "MouseUtils.UITests", "src\modules\MouseUtils\MouseUtils.UITests\MouseUtils.UITests.csproj", "{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|ARM64 = Debug|ARM64
@@ -2574,14 +2576,18 @@ Global
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Debug|ARM64.Build.0 = Debug|ARM64
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Debug|x64.ActiveCfg = Debug|x64
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Debug|x64.Build.0 = Debug|x64
-		{64B88F02-CD88-4ED8-9624-989A800230F9}.Debug|x86.ActiveCfg = Debug|x64
-		{64B88F02-CD88-4ED8-9624-989A800230F9}.Debug|x86.Build.0 = Debug|x64
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Release|ARM64.ActiveCfg = Release|ARM64
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Release|ARM64.Build.0 = Release|ARM64
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Release|x64.ActiveCfg = Release|x64
 		{64B88F02-CD88-4ED8-9624-989A800230F9}.Release|x64.Build.0 = Release|x64
-		{64B88F02-CD88-4ED8-9624-989A800230F9}.Release|x86.ActiveCfg = Release|x64
-		{64B88F02-CD88-4ED8-9624-989A800230F9}.Release|x86.Build.0 = Release|x64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Debug|ARM64.ActiveCfg = Debug|ARM64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Debug|ARM64.Build.0 = Debug|ARM64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Debug|x64.ActiveCfg = Debug|x64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Debug|x64.Build.0 = Debug|x64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Release|ARM64.ActiveCfg = Release|ARM64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Release|ARM64.Build.0 = Release|ARM64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Release|x64.ActiveCfg = Release|x64
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}.Release|x64.Build.0 = Release|x64
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
@@ -2721,7 +2727,7 @@ Global
 		{25C91A4E-BA4E-467A-85CD-8B62545BF674} = {A50C70A6-2DA0-4027-B90E-B1A40755A8A5}
 		{6AB6A2D6-F859-4A82-9184-0BD29C9F07D1} = {A50C70A6-2DA0-4027-B90E-B1A40755A8A5}
 		{212AD910-8488-4036-BE20-326931B75FB2} = {4AFC9975-2456-4C70-94A4-84073C1CED93}
-		{7AC943C9-52E8-44CF-9083-744D8049667B} = {4574FDD0-F61D-4376-98BF-E5A1262C11EC}
+		{7AC943C9-52E8-44CF-9083-744D8049667B} = {322566EF-20DC-43A6-B9F8-616AF942579A}
 		{54A93AF7-60C7-4F6C-99D2-FBB1F75F853A} = {7AC943C9-52E8-44CF-9083-744D8049667B}
 		{92C39820-9F84-4529-BC7D-22AAE514D63B} = {7AC943C9-52E8-44CF-9083-744D8049667B}
 		{515554D1-D004-4F7F-A107-2211FC0F6B2C} = {7AC943C9-52E8-44CF-9083-744D8049667B}
@@ -2852,6 +2858,7 @@ Global
 		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A0} = {F05E590D-AD46-42BE-9C25-6A63ADD2E3EA}
 		{5702B3CC-8575-48D5-83D8-15BB42269CD3} = {929C1324-22E8-4412-A9A8-80E85F3985A5}
 		{64B88F02-CD88-4ED8-9624-989A800230F9} = {ECB8E0D1-7603-4E5C-AB10-D1E545E6F8E2}
+		{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1} = {322566EF-20DC-43A6-B9F8-616AF942579A}
 	EndGlobalSection
 	GlobalSection(ExtensibilityGlobals) = postSolution
 		SolutionGuid = {C3A2F9D1-7930-4EF4-A6FC-7EE0A99821D0}

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,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 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;
+        }
+    }
+}

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;
@@ -112,9 +113,10 @@ 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)
+        public virtual void Click(bool rightClick = false, int msPreAction = 500, int msPostAction = 500)
         {
-            PerformAction((actions, windowElement) =>
+            PerformAction(
+                (actions, windowElement) =>
             {
                 actions.MoveToElement(windowElement);
 
@@ -131,7 +133,9 @@ namespace Microsoft.PowerToys.UITest
                 }
 
                 actions.Build().Perform();
-            });
+            },
+                msPreAction,
+                msPostAction);
         }
 
         /// <summary>
@@ -199,6 +203,42 @@ namespace Microsoft.PowerToys.UITest
             });
         }
 
+        /// <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)
+        {
+            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();
+                }
+
+                var releaseAction = new Actions(driver);
+                releaseAction.Release().Perform();
+
+                KeyboardHelper.ReleaseKey(key);
+            });
+        }
+
         /// <summary>
         /// Gets the attribute value of the UI element.
         /// </summary>
@@ -218,7 +258,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 +266,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 +279,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 +292,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 +304,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 +316,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 +348,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 +361,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,7 +373,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<Element> FindAll(string name, int timeoutMS = 3000)
+        public ReadOnlyCollection<Element> FindAll(string name, int timeoutMS = 5000)
         {
             return this.FindAll<Element>(By.Name(name), timeoutMS);
         }
@@ -360,5 +400,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 SaveToFile 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,35 @@ namespace Microsoft.PowerToys.UITest
                 }
 
                 actions.Build().Perform();
-            });
+            },
+                msPreAction,
+                msPostAction);
+        }
+
+        /// <summary>
+        /// Click the ListItem element.
+        /// </summary>
+        /// <param name="rightClick">If true, performs a right-click; otherwise, performs a left-click. Default value is false</param>
+        /// <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) =>
+                {
+                    if (rightClick)
+                    {
+                        actions.ContextClick();
+                    }
+                    else
+                    {
+                        actions.Click();
+                    }
+
+                    actions.Build().Perform();
+                },
+                msPreAction,
+                msPostAction);
         }
 
         /// <summary>

src/common/UITestAutomation/Element/Slider.cs

@@ -0,0 +1,119 @@
+// 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>
+        /// 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/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,8 +33,8 @@ 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);
                 });
             }
 

src/common/UITestAutomation/FindHelper.cs

@@ -33,7 +33,7 @@ namespace Microsoft.PowerToys.UITest
         public static ReadOnlyCollection<T>? FindAll<T, TW>(Func<ReadOnlyCollection<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;
@@ -43,6 +43,27 @@ namespace Microsoft.PowerToys.UITest
             return new ReadOnlyCollection<T>(res);
         }
 
+        private static ReadOnlyCollection<TW> FindElementsWithRetry<TW>(Func<ReadOnlyCollection<TW>> findElementsFunc, int timeoutMS)
+        {
+            int retryIntervalMS = 500;
+            timeoutMS = 1;
+            int elapsedTime = 0;
+
+            while (elapsedTime < timeoutMS)
+            {
+                var items = findElementsFunc();
+                if (items.Count > 0)
+                {
+                    return items;
+                }
+
+                Task.Delay(retryIntervalMS).Wait();
+                elapsedTime += retryIntervalMS;
+            }
+
+            return new ReadOnlyCollection<TW>(new List<TW>());
+        }
+
         public static T NewElement<T>(WindowsElement? element, WindowsDriver<WindowsElement>? driver, int timeoutMS)
              where T : Element, new()
         {
@@ -50,11 +71,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,351 @@
+// 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: // Right Ctrl Key - 0xA3 in hex
+                    return 0xA3;
+                case Key.A:
+                    return 0x41;  // A Key - 0x41 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,48 @@ namespace Microsoft.PowerToys.UITest
         PowerToysSettings,
         FancyZone,
         Hosts,
+        Runner,
+    }
+
+    /// <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 +94,7 @@ namespace Microsoft.PowerToys.UITest
                 [PowerToysModule.PowerToysSettings] = "PowerToys Settings",
                 [PowerToysModule.FancyZone] = "FancyZones Layout",
                 [PowerToysModule.Hosts] = "Hosts File Editor",
+                [PowerToysModule.Runner] = "PowerToys",
             };
 
             // Exe start path for the module if it exists.
@@ -60,6 +103,7 @@ 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",
             };
         }
 

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

@@ -3,8 +3,9 @@
 // See the LICENSE file in the project root for more information.
 
 using System.Collections.ObjectModel;
+using System.Diagnostics;
 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;
@@ -17,27 +18,78 @@ 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 const string AdministratorPrefix = "Administrator: ";
 
-        public Session(WindowsDriver<WindowsElement> root, WindowsDriver<WindowsElement> windowsDriver)
+        private List<IntPtr> windowHandlers = new List<IntPtr>();
+
+        private Window? MainWindow { get; set; }
+
+        /// <summary>
+        /// Gets Main Window Handler
+        /// </summary>
+        public IntPtr MainWindowHandler { 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> root, WindowsDriver<WindowsElement> windowsDriver, PowerToysModule scope, WindowSize size)
         {
+            this.MainWindowHandler = IntPtr.Zero;
             this.Root = root;
             this.WindowsDriver = windowsDriver;
+
+            // 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()
+        {
+            /*
+            foreach (var windowHandle in this.windowHandlers)
+            {
+                if (windowHandle == IntPtr.Zero)
+                {
+                    continue;
+                }
+
+                try
+                {
+                    var process = Process.GetProcessById((int)windowHandle);
+                    if (process != null && !process.HasExited)
+                    {
+                        process.Kill();
+                        process.WaitForExit();
+                    }
+                }
+                catch
+                {
+                }
+            }
+            */
+
+            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)
             where T : Element, new()
         {
             Assert.IsNotNull(this.WindowsDriver, $"WindowsElement is null in method Find<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
@@ -45,7 +97,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];
         }
@@ -55,9 +107,9 @@ 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)
             where T : Element, new()
         {
             return this.Find<T>(By.Name(name), timeoutMS);
@@ -67,9 +119,9 @@ namespace Microsoft.PowerToys.UITest
         /// 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)
         {
             return this.Find<Element>(by, timeoutMS);
         }
@@ -78,21 +130,117 @@ namespace Microsoft.PowerToys.UITest
         /// 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)
         {
             return this.Find<Element>(By.Name(name), timeoutMS);
         }
 
         /// <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)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(by, timeoutMS).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)
+        {
+            return this.HasOne<Element>(by, timeoutMS);
+        }
+
+        /// <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)
+            where T : Element, new()
+        {
+            return this.HasOne<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <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)
+        {
+            return this.HasOne<Element>(By.Name(name), timeoutMS);
+        }
+
+        /// <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)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(by, timeoutMS).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)
+        {
+            return this.Has<Element>(by, timeoutMS);
+        }
+
+        /// <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)
+            where T : Element, new()
+        {
+            return this.Has<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <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)
+        {
+            return this.Has<Element>(name, timeoutMS);
+        }
+
+        /// <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)
             where T : Element, new()
         {
             Assert.IsNotNull(this.WindowsDriver, $"WindowsElement is null in method FindAll<{typeof(T).Name}> with parameters: by = {by}, timeoutMS = {timeoutMS}");
@@ -122,9 +270,9 @@ 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)
             where T : Element, new()
         {
             return this.FindAll<T>(By.Name(name), timeoutMS);
@@ -135,9 +283,9 @@ 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)
         {
             return this.FindAll<Element>(by, timeoutMS);
         }
@@ -147,55 +295,327 @@ 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)
         {
             return this.FindAll<Element>(By.Name(name), timeoutMS);
         }
 
         /// <summary>
-        /// Keyboard Action key.
+        /// Sets the main window size.
         /// </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 = "")
+        /// <param name="size">WindowSize enum</param>
+        public void SetMainWindowSize(WindowSize size)
         {
-            PerformAction((actions, windowElement) =>
+            if (size == WindowSize.UnSpecified)
             {
-                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);
-                }
+                return;
+            }
 
-                actions.Release();
-                actions.Build().Perform();
+            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)
+            {
+                this.SetMainWindowSize(width, height);
+            }
+        }
+
+        /// <summary>
+        /// Gets the main window center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public (int CenterX, int CenterY) GetWindowCenter()
+        {
+            if (this.MainWindowHandler == IntPtr.Zero)
+            {
+                return (0, 0);
+            }
+            else
+            {
+                var rect = ApiHelper.GetWindowCenter(this.MainWindowHandler);
+                return (rect.CenterX, rect.CenterY);
+            }
+        }
+
+        /// <summary>
+        /// Gets the screen center coordinates.
+        /// </summary>
+        /// <returns>(x, y)</returns>
+        public (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 void SetMainWindowSize(int width, int height)
+        {
+            if (this.MainWindowHandler == IntPtr.Zero
+                || width <= 0
+                || height <= 0)
+            {
+                return;
+            }
+
+            ApiHelper.SetWindowPos(this.MainWindowHandler, IntPtr.Zero, 0, 0, width, height, ApiHelper.SetWindowPosNoZorder | ApiHelper.SetWindowPosShowWindow);
+
+            // Wait for 1000ms after resize
+            Task.Delay(1000).Wait();
+        }
+
+        /// <summary>
+        /// Close the main window.
+        /// </summary>
+        public void CloseMainWindow()
+        {
+            if (MainWindow != null)
+            {
+                MainWindow.Close();
+                MainWindow = null;
+            }
+        }
+
+        /// <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)
+        {
+            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 string GetPixelColorString(int x, int y)
+        {
+            Color color = this.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 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);
+        }
+
+        /// <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));
+                        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,26 +623,44 @@ 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 matchingWindows = ApiHelper.FindDesktopWindowHandler([windowName, AdministratorPrefix + windowName]);
+                if (matchingWindows.Count == 0 || matchingWindows[0].HWnd == IntPtr.Zero)
+                {
+                    Assert.Fail($"Failed to attach. Window '{windowName}' not found");
+                }
+
+                // pick one from matching windows
+                this.MainWindowHandler = matchingWindows[0].HWnd;
+                this.IsElevated = matchingWindows[0].Title.StartsWith(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)
+                {
+                    this.SetMainWindowSize(size);
+                }
+
+                // Set MainWindow
+                MainWindow = Find<Window>(matchingWindows[0].Title);
             }
             else
             {
@@ -232,6 +670,176 @@ namespace Microsoft.PowerToys.UITest
             return this;
         }
 
+        public bool IsWindowOpen(string windowName)
+        {
+            var matchingWindows = ApiHelper.FindDesktopWindowHandler([windowName, AdministratorPrefix + windowName]);
+            return matchingWindows.Count > 0;
+        }
+
+        private 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");
+                }
+            }
+
+            [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);
+            }
+        }
+
+        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);
+        }
+
+        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 +862,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

@@ -20,6 +20,8 @@ namespace Microsoft.PowerToys.UITest
         // 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; }
@@ -27,10 +29,14 @@ namespace Microsoft.PowerToys.UITest
         private WindowsDriver<WindowsElement>? Driver { get; set; }
 
         private 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);
 
@@ -40,14 +46,23 @@ namespace Microsoft.PowerToys.UITest
                 Verb = "runas",
             };
 
+            this.ExitExe(winAppDriverProcessInfo.FileName);
             this.appDriver = Process.Start(winAppDriverProcessInfo);
 
+            var runnerProcessInfo = new ProcessStartInfo
+            {
+                FileName = locationPath + this.runnerPath,
+                Verb = "runas",
+            };
+
+            if (scope == PowerToysModule.PowerToysSettings)
+            {
+                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>
@@ -56,7 +71,11 @@ namespace Microsoft.PowerToys.UITest
         /// <param name="scope">The PowerToys module to start.</param>
         public SessionHelper Init()
         {
-            this.StartExe(locationPath + this.sessionPath);
+            // Exit setting exe to fix CommandPalette error, remove after fixing the issue
+            this.ExitExe(this.locationPath + ModuleConfigData.Instance.GetModulePath(PowerToysModule.PowerToysSettings));
+
+            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.");
 
@@ -73,6 +92,11 @@ namespace Microsoft.PowerToys.UITest
             {
                 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 +105,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 +129,21 @@ 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);
+            opts.AddAdditionalCapability("ms:waitForAppLaunch", "5");
+            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 now exe.
         /// </summary>

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,15 @@ 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;
 
 namespace Microsoft.PowerToys.UITest
 {
@@ -17,19 +22,36 @@ 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; }
 
+        private readonly bool isInPipeline;
         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)
+        // private System.Threading.Timer? screenshotTimer;
+        // private string? screenshotDirectory;
+        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.ChangeDispalyResolution();
+                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 +60,18 @@ namespace Microsoft.PowerToys.UITest
         [TestInitialize]
         public void TestInit()
         {
+            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));
+            }
+
+            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 +84,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,22 +118,22 @@ 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)
             where T : Element, new()
         {
             return this.Session.Find<T>(by, timeoutMS);
         }
 
         /// <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)
             where T : Element, new()
         {
             return this.Session.Find<T>(By.Name(name), timeoutMS);
@@ -89,33 +143,129 @@ namespace Microsoft.PowerToys.UITest
         /// 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)
         {
             return this.Session.Find(by, timeoutMS);
         }
 
         /// <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)
         {
             return this.Session.Find(name, timeoutMS);
         }
 
+        /// <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)
+            where T : Element, new()
+        {
+            return this.FindAll<T>(by, timeoutMS).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)
+        {
+            return this.Session.HasOne<Element>(by, timeoutMS);
+        }
+
+        /// <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)
+            where T : Element, new()
+        {
+            return this.Session.HasOne<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <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)
+        {
+            return this.Session.HasOne<Element>(name, timeoutMS);
+        }
+
+        /// <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)
+            where T : Element, new()
+        {
+            return this.Session.FindAll<T>(by, timeoutMS).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)
+        {
+            return this.Session.Has<Element>(by, timeoutMS);
+        }
+
+        /// <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)
+            where T : Element, new()
+        {
+            return this.Session.Has<T>(By.Name(name), timeoutMS);
+        }
+
+        /// <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)
+        {
+            return this.Session.Has<Element>(name, timeoutMS);
+        }
+
         /// <summary>
         /// Finds all elements by selector.
         /// Shortcut for this.Session.FindAll<T>(by, timeoutMS)
         /// </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)
             where T : Element, new()
         {
             return this.Session.FindAll<T>(by, timeoutMS);
@@ -127,9 +277,9 @@ 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)
             where T : Element, new()
         {
             return this.Session.FindAll<T>(By.Name(name), timeoutMS);
@@ -140,9 +290,9 @@ 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)
         {
             return this.Session.FindAll<Element>(by, timeoutMS);
         }
@@ -152,20 +302,105 @@ 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)
         {
             return this.Session.FindAll<Element>(By.Name(name), timeoutMS);
         }
 
+        /// <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 this.Session.GetPixelColor(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 this.Session.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>
+        /// 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>
         /// Restart scope exe.
         /// </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 +409,165 @@ namespace Microsoft.PowerToys.UITest
         /// </summary>
         public void ExitScopeExe()
         {
-            this.sessionHelper.ExitScopeExe();
+            this.sessionHelper!.ExitScopeExe();
             return;
         }
+
+        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)
+                    {
+                        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 ChangeDispalyResolution()
+            {
+                Screen screen = Screen.PrimaryScreen!;
+                if (screen.Bounds.Width == 1920 && screen.Bounds.Height == 1080)
+                {
+                    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 = 1920;
+                devMode.DmPelsHeight = 1080;
+
+                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 limiation 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 not 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,170 @@
+// 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
+    {
+        #pragma warning disable SA1307
+        [StructLayout(LayoutKind.Sequential)]
+        private struct IMMERSIVE_COLOR_PREFERENCE
+        {
+            public uint dwColorSetIndex;
+            public uint crStartColor;
+            public uint crAccentColor;
+        }
+        #pragma warning restore SA1307
+
+        [DllImport("uxtheme.dll", EntryPoint = "#120")]
+        private static extern IntPtr GetUserColorPreference(ref IMMERSIVE_COLOR_PREFERENCE pcpPreference, bool fForceReload);
+
+        /// <summary>
+        /// Gets the system accent color.
+        /// </summary>
+        /// <returns>The system accent color as a Color object.</returns>
+        private static Color GetSystemAccentColor()
+        {
+            IMMERSIVE_COLOR_PREFERENCE colorPreference = default(IMMERSIVE_COLOR_PREFERENCE);
+            GetUserColorPreference(ref colorPreference, true);
+            return ToColor(colorPreference.crStartColor);
+        }
+
+        /// <summary>
+        /// Converts a color value to a Color object.
+        /// </summary>
+        /// <param name="c">The color value.</param>
+        /// <returns>The Color object.</returns>
+        private static Color ToColor(uint c)
+        {
+            int r = (int)(c & 0xFF) % 256;
+            int g = (int)((c >> 8) & 0xFF) % 256;
+            int b = (int)(c >> 16) % 256;
+            return Color.FromArgb(r, g, b);
+        }
+
+        /// <summary>
+        /// Gets HSL values from a Color object.
+        /// </summary>
+        /// <param name="color">The Color object.</param>
+        /// <returns>A tuple containing the HSL values.</returns>
+        private static (double H, double S, double L) GetHSL(Color color)
+        {
+            double rNorm = color.R / 255.0;
+            double gNorm = color.G / 255.0;
+            double bNorm = color.B / 255.0;
+
+            double max = Math.Max(rNorm, Math.Max(gNorm, bNorm));
+            double min = Math.Min(rNorm, Math.Min(gNorm, bNorm));
+            double h = 0, s = 0, l = (max + min) / 2;
+
+            if (max != min)
+            {
+                double delta = max - min;
+                s = l > 0.5 ? delta / (2 - max - min) : delta / (max + min);
+
+                if (max == rNorm)
+                {
+                    h = ((gNorm - bNorm) / delta) + (gNorm < bNorm ? 6 : 0);
+                }
+                else if (max == gNorm)
+                {
+                    h = ((bNorm - rNorm) / delta) + 2;
+                }
+                else if (max == bNorm)
+                {
+                    h = ((rNorm - gNorm) / delta) + 4;
+                }
+
+                h /= 6;
+            }
+
+            return (h * 360, s * 100, l * 100);
+        }
+
+        /// <summary>
+        /// Makes a specific color in an image transparent.
+        /// </summary>
+        /// <param name="imagePath">The path to the image file.</param>
+        /// <param name="outputPath">The path to save the output image file.</param>
+        /// <param name="targetColor">The target color to make transparent.</param>
+        /// <param name="fuzz">The fuzz factor for color comparison, default is 2.</param>
+        private static void MakeColorTransparent(string imagePath, string outputPath, Color targetColor, int fuzz = 2)
+        {
+            var hsl = GetHSL(targetColor);
+
+            // Assert.IsNotNull(null, $"Target Color - H: {hsl.H}, S: {hsl.S}, L: {hsl.L}");
+            using (Bitmap originalBitmap = new Bitmap(imagePath))
+            {
+                using (Bitmap bitmap = new Bitmap(originalBitmap))
+                {
+                    for (int y = 0; y < bitmap.Height; y++)
+                    {
+                        for (int x = 0; x < bitmap.Width; x++)
+                        {
+                            Color pixelColor = bitmap.GetPixel(x, y);
+                            if (HueIsSame(pixelColor, targetColor, fuzz))
+                            {
+                                bitmap.SetPixel(x, y, Color.Transparent);
+                            }
+                        }
+                    }
+
+                    bitmap.Save(outputPath, ImageFormat.Png);
+                }
+            }
+        }
+
+        /// <summary>
+        /// Erases the user preference color from an image. Will overwrite this image.
+        /// </summary>
+        /// <param name="imagePath">The path to the image file.</param>
+        /// <param name="fuzz">The fuzz factor for color comparison, default is 2.</param>
+        public static void EraseUserPreferenceColor(string imagePath, int fuzz = 2)
+        {
+            Color systemColor = GetSystemAccentColor();
+            string tempPath = Path.GetTempFileName();
+            MakeColorTransparent(imagePath, tempPath, systemColor, fuzz);
+            File.Delete(imagePath);
+            File.Move(tempPath, imagePath);
+        }
+
+        /// <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;
+        }
+
+        /// <summary>
+        /// Compares the hue of two colors with a fuzz factor.
+        /// </summary>
+        /// <param name="c1">The first color.</param>
+        /// <param name="c2">The second color.</param>
+        /// <param name="fuzz">The fuzz factor, default is 2.</param>
+        /// <returns>True if the hues are the same, otherwise false.</returns>
+        public static bool HueIsSame(Color c1, Color c2, int fuzz = 2)
+        {
+            var h1 = GetHSL(c1).H;
+            var h2 = GetHSL(c2).H;
+            return Math.Abs(h1 - h2) <= fuzz;
+        }
+    }
+}

src/modules/Hosts/Hosts.UITests/HostModuleTests.cs

@@ -13,7 +13,7 @@ namespace Hosts.UITests
     public class HostModuleTests : UITestBase
     {
         public HostModuleTests()
-            : base(PowerToysModule.Hosts)
+            : base(PowerToysModule.Hosts, WindowSize.Small_Vertical)
         {
         }
 
@@ -31,14 +31,16 @@ namespace Hosts.UITests
         /// </item>
         /// </list>
         /// </summary>
-        [TestMethod]
+        [TestMethod("Hosts.Basic.EmptyViewShouldWork")]
         public void TestEmptyView()
         {
             this.CloseWarningDialog();
             this.RemoveAllEntries();
 
             // 'Add an entry' button (only show-up when list is empty) should be visible
-            Assert.IsTrue(this.FindAll<HyperlinkButton>("Add an entry").Count == 1, "'Add an entry' button should be visible in the empty view");
+            Assert.IsTrue(this.HasOne<HyperlinkButton>("Add an entry"), "'Add an entry' button should be visible in the empty view");
+
+            VisualAssert.AreEqual(this.TestContext, this.Find("Entries"), "EmptyView");
 
             // Click 'Add an entry' from empty-view for adding Host override rule
             this.Find<HyperlinkButton>("Add an entry").Click();
@@ -46,8 +48,10 @@ namespace Hosts.UITests
             this.AddEntry("192.168.0.1", "localhost", false, false);
 
             // Should have one row now and not more empty view
-            Assert.IsTrue(this.FindAll<Button>("Delete").Count == 1, "Should have one row now");
-            Assert.IsTrue(this.FindAll<HyperlinkButton>("Add an entry").Count == 0, "'Add an entry' button should be invisible if not empty view");
+            Assert.IsTrue(this.Has<Button>("Delete"), "Should have one row now");
+            Assert.IsFalse(this.Has<HyperlinkButton>("Add an entry"), "'Add an entry' button should be invisible if not empty view");
+
+            VisualAssert.AreEqual(this.TestContext, this.Find("Entries"), "NonEmptyView");
         }
 
         /// <summary>
@@ -58,17 +62,19 @@ namespace Hosts.UITests
         /// </item>
         /// </list>
         /// </summary>
-        [TestMethod]
+        [TestMethod("Hosts.Basic.AddEntryButtonShouldWork")]
         public void TestAddingEntry()
         {
             this.CloseWarningDialog();
             this.RemoveAllEntries();
 
-            Assert.IsTrue(this.FindAll<Button>("Delete").Count == 0, "Should have no row after removing all");
+            Assert.IsFalse(this.Has<Button>("Delete"), "Should have no row after removing all");
 
             this.AddEntry("192.168.0.1", "localhost", true);
 
-            Assert.IsTrue(this.FindAll<Button>("Delete").Count == 1, "Should have one row now");
+            Assert.IsTrue(this.Has<Button>("Delete"), "Should have one row now");
+
+            VisualAssert.AreEqual(this.TestContext, this.Find("Entries"));
         }
 
         /// <summary>
@@ -82,7 +88,7 @@ namespace Hosts.UITests
         /// </item>
         /// </list>
         /// </summary>
-        [TestMethod]
+        [TestMethod("Hosts.Basic.CanNotAddMoreThenNighHosts")]
         public void TestTooManyHosts()
         {
             this.CloseWarningDialog();
@@ -116,18 +122,46 @@ namespace Hosts.UITests
         /// </item>
         /// </list>
         /// </summary>
-        [TestMethod]
+        [TestMethod("Hosts.Basic.ErrorMessgeShowupIfNotRunAsAdmin")]
         public void TestErrorMessageWithNonAdminPermission()
         {
-            this.CloseWarningDialog();
-            this.RemoveAllEntries();
+            if (this.Session.IsElevated == false)
+            {
+                this.CloseWarningDialog();
+                this.RemoveAllEntries();
 
-            // Add new URL override and a warning tip should be shown
-            this.AddEntry("192.168.0.1", "localhost", true);
+                // Add new URL override and a warning tip should be shown
+                this.AddEntry("192.168.0.1", "localhost", true);
 
-            Assert.IsTrue(
-                this.FindAll<TextBlock>("The hosts file cannot be saved because the program isn't running as administrator.").Count == 1,
-                "Should display host-file saving error if not run as administrator");
+                Assert.IsTrue(
+                    this.FindAll<TextBlock>("The hosts file cannot be saved because the program isn't running as administrator.").Count == 1,
+                    "Should display host-file saving error if not run as administrator");
+            }
+        }
+
+        /// <summary>
+        /// Test No Error-message in the Hosts-File-Editor
+        /// <list type="bullet">
+        /// <item>
+        /// <description>Validating error message should be shown if not run as admin.</description>
+        /// </item>
+        /// </list>
+        /// </summary>
+        [TestMethod("Hosts.Basic.NoErrorMessgeShowupIfRunAsAdmin")]
+        public void TestNoErrorMessageWithNonAdminPermission()
+        {
+            if (this.Session.IsElevated == true)
+            {
+                this.CloseWarningDialog();
+                this.RemoveAllEntries();
+
+                // Add new URL override and a warning tip should be shown
+                this.AddEntry("192.168.0.1", "localhost", true);
+
+                Assert.IsFalse(
+                    this.FindAll<TextBlock>("The hosts file cannot be saved because the program isn't running as administrator.").Count > 0,
+                    "Should display host-file saving error if not run as administrator");
+            }
         }
 
         /// <summary>
@@ -144,7 +178,7 @@ namespace Hosts.UITests
         /// </item>
         /// </list>
         /// </summary>
-        [TestMethod]
+        [TestMethod("Hosts.Basic.FiltersControlShouldWork")]
         public void TestFilterControl()
         {
             this.CloseWarningDialog();
@@ -212,7 +246,7 @@ namespace Hosts.UITests
 
             // Close-filter-panel
             this.Find<Button>("Filters").Click();
-            Assert.IsFalse(this.FindAll<Button>("Clear filters").Count == 1, "Filter panel should be closed afer click Filter Button");
+            Assert.IsFalse(this.FindAll<Button>("Clear filters").Count == 1, "Filter panel should be closed after clicking Filter Button");
         }
 
         private void AddEntry(string ip, string host, bool active = true, bool clickAddEntryButton = true)

src/modules/Hosts/Hosts.UITests/Hosts.UITests.csproj

@@ -17,6 +17,14 @@
     <PropertyGroup>
       <OutputPath>$(SolutionDir)$(Platform)\$(Configuration)\tests\Hosts.UITests\</OutputPath>
     </PropertyGroup>
+    <ItemGroup>
+      <EmbeddedResource Include="Baseline\HostModuleTests_TestAddingEntry_arm64.png" />
+      <EmbeddedResource Include="Baseline\HostModuleTests_TestEmptyView_EmptyView_arm64.png" />
+      <EmbeddedResource Include="Baseline\HostModuleTests_TestEmptyView_NonEmptyView_arm64.png" />
+      <EmbeddedResource Include="Baseline\HostModuleTests_TestAddingEntry_x64.png" />
+      <EmbeddedResource Include="Baseline\HostModuleTests_TestEmptyView_EmptyView_x64.png" />
+      <EmbeddedResource Include="Baseline\HostModuleTests_TestEmptyView_NonEmptyView_x64.png" />
+    </ItemGroup>
 
     <ItemGroup>
       <PackageReference Include="MSTest" />

src/modules/Hosts/Hosts.UITests/HostsSettingTests.cs

@@ -12,6 +12,11 @@ namespace Hosts.UITests
     [TestClass]
     public class HostsSettingTests : UITestBase
     {
+        public HostsSettingTests()
+        : base(PowerToysModule.PowerToysSettings, WindowSize.Medium)
+        {
+        }
+
         /// <summary>
         /// Test Warning Dialog at startup
         /// <list type="bullet">
@@ -29,7 +34,7 @@ namespace Hosts.UITests
         /// </item>
         /// </list>
         /// </summary>
-        [TestMethod]
+        [TestMethod("Hosts.Settings.ShowWarningDialogIfRunAsAdmin")]
         public void TestWarningDialog()
         {
             this.LaunchFromSetting(showWarning: true);
@@ -54,7 +59,7 @@ namespace Hosts.UITests
             // wait for 500 ms to make sure Hosts File Editor is launched
             Task.Delay(500).Wait();
 
-            this.Session.Attach(PowerToysModule.Hosts);
+            this.Session.Attach(PowerToysModule.Hosts, WindowSize.Small_Vertical);
 
             // Should show warning dialog
             Assert.IsTrue(this.FindAll("Warning").Count > 0, "Should show warning dialog");
@@ -68,7 +73,7 @@ namespace Hosts.UITests
             Assert.IsFalse(this.IsHostsFileEditorClosed(), "Hosts File Editor should NOT be closed after click Accept button in Warning Dialog");
 
             // Close Hosts File Editor window
-            this.Session.Find<Window>("Hosts File Editor").Close();
+            this.Session.CloseMainWindow();
 
             // Restore back to PowerToysSettings Session
             this.Session.Attach(PowerToysModule.PowerToysSettings);
@@ -82,7 +87,7 @@ namespace Hosts.UITests
             Assert.IsFalse(this.IsHostsFileEditorClosed(), "Hosts File Editor should NOT be closed");
 
             // Close Hosts File Editor window
-            this.Session.Find<Window>("Hosts File Editor").Close();
+            this.Session.CloseMainWindow();
 
             // Restore back to PowerToysSettings Session
             this.Session.Attach(PowerToysModule.PowerToysSettings);

src/modules/MouseUtils/MouseUtils.UITests/FindMyMouseTests.cs

@@ -0,0 +1,558 @@
+// 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.Globalization;
+using System.Threading.Tasks;
+using System.Xml.Linq;
+using Microsoft.PowerToys.UITest;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Windows.Devices.Printers;
+
+namespace MouseUtils.UITests
+{
+    [TestClass]
+    public class FindMyMouseTests : UITestBase
+    {
+        /// <summary>
+        /// Test Warning Dialog at startup
+        /// <list type="bullet">
+        /// <item>
+        /// <description>Validating Warning-Dialog will be shown if 'Show a warning at startup' toggle is On.</description>
+        /// </item>
+        /// <item>
+        /// <description>Validating Warning-Dialog will NOT be shown if 'Show a warning at startup' toggle is Off.</description>
+        /// </item>
+        /// <item>
+        /// <description>Validating click 'Quit' button in Warning-Dialog, the Hosts File Editor window would be closed.</description>
+        /// </item>
+        /// <item>
+        /// <description>Validating click 'Accept' button in Warning-Dialog, the Hosts File Editor window would NOT be closed.</description>
+        /// </item>
+        /// </list>
+        /// </summary>
+        [TestMethod]
+        public void TestEnableFindMyMouse()
+        {
+            LaunchFromSetting();
+
+            var settings = new FindMyMouseSettings();
+            settings.OverlayOpacity = "100";
+            settings.Radius = "50";
+            settings.InitialZoom = "1";
+            settings.AnimationDuration = "0";
+            settings.BackgroundColor = "000000";
+            settings.SpotlightColor = "FFFFFF";
+
+            var foundCustom = this.Find<Custom>("Find My Mouse");
+            Assert.IsNotNull(foundCustom);
+
+            if (CheckAnimationEnable(ref foundCustom))
+            {
+                foundCustom = this.Find<Custom>("Find My Mouse");
+            }
+
+            if (foundCustom != null)
+            {
+                foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(true);
+
+                // foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+                SetFindMyMouseActivationMethod(ref foundCustom, "Press Left Control twice");
+                Assert.IsNotNull(foundCustom, "Find My Mouse group not found.");
+                SetFindMyMouseAppearanceBehavior(ref foundCustom, ref settings);
+
+                var excludedApps = foundCustom.Find<TextBlock>("Excluded apps");
+                if (excludedApps != null)
+                {
+                    excludedApps.Click();
+                    excludedApps.Click();
+                }
+                else
+                {
+                    Assert.Fail("Activation method group not found.");
+                }
+            }
+            else
+            {
+                Assert.Fail("Find My Mouse group not found.");
+            }
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press Left Ctrl twice and verify the overlay appears.
+            VerifySpotlightSettings(ref settings);
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press any other key and verify the overlay disappears.
+            Session.SendKeys(Key.A);
+            VerifySpotlightDisappears(ref settings);
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press Left Ctrl twice and verify the overlay appears.
+            VerifySpotlightSettings(ref settings);
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press a mouse button and verify the overlay disappears.
+            Task.Delay(1000).Wait();
+
+            // MouseSimulator.LeftClick();
+            Session.PerformMouseAction(MouseActionType.LeftClick, 500, 1000);
+
+            VerifySpotlightDisappears(ref settings);
+        }
+
+        [TestMethod]
+        public void TestEnableFindMyMouse2()
+        {
+            LaunchFromSetting();
+
+            var settings = new FindMyMouseSettings();
+            settings.OverlayOpacity = "100";
+            settings.Radius = "80";
+            settings.InitialZoom = "1";
+            settings.AnimationDuration = "0";
+            settings.BackgroundColor = "FFFFFF";
+            settings.SpotlightColor = "000000";
+
+            var foundCustom = this.Find<Custom>("Find My Mouse");
+            Assert.IsNotNull(foundCustom);
+
+            if (CheckAnimationEnable(ref foundCustom))
+            {
+                foundCustom = this.Find<Custom>("Find My Mouse");
+            }
+
+            if (foundCustom != null)
+            {
+                foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(true);
+
+                // foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+                SetFindMyMouseActivationMethod(ref foundCustom, "Press Left Control twice");
+                Assert.IsNotNull(foundCustom, "Find My Mouse group not found.");
+                SetFindMyMouseAppearanceBehavior(ref foundCustom, ref settings);
+
+                var excludedApps = foundCustom.Find<TextBlock>("Excluded apps");
+                if (excludedApps != null)
+                {
+                    excludedApps.Click();
+                    excludedApps.Click();
+                }
+                else
+                {
+                    Assert.Fail("Excluded apps group not found.");
+                }
+            }
+            else
+            {
+                Assert.Fail("Find My Mouse group not found.");
+            }
+
+            // [Test Case]Test the different settings and verify they apply, Background color
+            // [Test Case]Test the different settings and verify they apply, Spotlight color
+            // [Test Case]Test the different settings and verify they apply, Spotlight radius
+            VerifySpotlightSettings(ref settings);
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press any other key and verify the overlay disappears.
+            Session.SendKeys(Key.A);
+            VerifySpotlightDisappears(ref settings);
+        }
+
+        [TestMethod]
+        public void TestDisableFindMyMouse()
+        {
+            LaunchFromSetting();
+
+            var settings = new FindMyMouseSettings();
+            settings.OverlayOpacity = "100";
+            settings.Radius = "50";
+            settings.InitialZoom = "1";
+            settings.AnimationDuration = "0";
+            settings.BackgroundColor = "000000";
+            settings.SpotlightColor = "FFFFFF";
+            var foundCustom = this.Find<Custom>("Find My Mouse");
+
+            Assert.IsNotNull(foundCustom);
+
+            if (CheckAnimationEnable(ref foundCustom))
+            {
+                foundCustom = this.Find<Custom>("Find My Mouse");
+            }
+
+            if (foundCustom != null)
+            {
+                foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(true);
+
+                // foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+                SetFindMyMouseActivationMethod(ref foundCustom, "Press Left Control twice");
+                Assert.IsNotNull(foundCustom);
+                SetFindMyMouseAppearanceBehavior(ref foundCustom, ref settings);
+
+                var excludedApps = foundCustom.Find<TextBlock>("Excluded apps");
+                if (excludedApps != null)
+                {
+                    excludedApps.Click();
+                    excludedApps.Click();
+                }
+                else
+                {
+                    Assert.Fail("Activation method group not found.");
+                }
+            }
+            else
+            {
+                Assert.Fail("Find My Mouse group not found.");
+            }
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press Left Ctrl twice and verify the overlay appears.
+            // VerifySpotlightSettings(ref settings);
+            ActivateSpotlight(ref settings);
+            VerifySpotlightAppears(ref settings);
+
+            // [Test Case] Disable FindMyMouse. Verify the overlay no longer appears when you press Left Ctrl twice
+            foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+            Task.Delay(1000).Wait();
+            ActivateSpotlight(ref settings);
+
+            VerifySpotlightDisappears(ref settings);
+
+            // [Test Case] Press Left Ctrl twice and verify the overlay appears
+            foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(true);
+            ActivateSpotlight(ref settings);
+            VerifySpotlightAppears(ref settings);
+
+            Session.PerformMouseAction(MouseActionType.LeftClick);
+        }
+
+        [TestMethod]
+        public void TestDisableFindMyMouse2()
+        {
+            LaunchFromSetting();
+
+            var settings = new FindMyMouseSettings();
+            settings.OverlayOpacity = "100";
+            settings.Radius = "50";
+            settings.InitialZoom = "1";
+            settings.AnimationDuration = "0";
+            settings.BackgroundColor = "000000";
+            settings.SpotlightColor = "FFFFFF";
+            var foundCustom = this.Find<Custom>("Find My Mouse");
+            if (foundCustom != null)
+            {
+                foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(true);
+
+                // foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+                SetFindMyMouseActivationMethod(ref foundCustom, "Press Left Control twice");
+                Assert.IsNotNull(foundCustom, "Find My Mouse group not found.");
+
+                // SetFindMyMouseAppearanceBehavior(ref foundCustom, ref settings);
+                var excludedApps = foundCustom.Find<TextBlock>("Excluded apps");
+                if (excludedApps != null)
+                {
+                    excludedApps.Click();
+                    excludedApps.Click();
+                }
+                else
+                {
+                    Assert.Fail("Activation method group not found.");
+                }
+            }
+            else
+            {
+                Assert.Fail("Find My Mouse group not found.");
+            }
+
+            // [Test Case]Enable FindMyMouse. Then, without moving your mouse: Press Left Ctrl twice and verify the overlay appears.
+            // VerifySpotlightSettings(ref settings);
+
+            // [Test Case] Disable FindMyMouse. Verify the overlay no longer appears when you press Left Ctrl twice
+            foundCustom.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+            Task.Delay(2000).Wait();
+            Session.SendKey(Key.LCtrl, 0, 0);
+            Task.Delay(100).Wait();
+            Session.SendKey(Key.LCtrl, 0, 0);
+
+            VerifySpotlightDisappears(ref settings);
+        }
+
+        private void VerifySpotlightDisappears(ref FindMyMouseSettings settings)
+        {
+            Task.Delay(2000).Wait();
+
+            var location = Session.GetMousePosition();
+            int radius = int.Parse(settings.Radius, CultureInfo.InvariantCulture);
+            var colorSpotlight = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreNotEqual("#" + settings.SpotlightColor, colorSpotlight);
+
+            var colorBackground = Session.GetPixelColorString(location.Item1 + radius + 50, location.Item2 + radius + 50);
+            Assert.AreNotEqual("#" + settings.BackgroundColor, colorBackground);
+
+            var colorBackground2 = Session.GetPixelColorString(location.Item1 + radius + 100, location.Item2 + radius + 100);
+            Assert.AreNotEqual("#" + settings.BackgroundColor, colorBackground2);
+        }
+
+        private void VerifySpotlightAppears(ref FindMyMouseSettings settings)
+        {
+            Task.Delay(2000).Wait();
+
+            var location = Session.GetMousePosition();
+            int radius = int.Parse(settings.Radius, CultureInfo.InvariantCulture);
+            var colorSpotlight = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreEqual("#" + settings.SpotlightColor, colorSpotlight);
+
+            var colorSpotlight2 = Session.GetPixelColorString(location.Item1 + radius - 1, location.Item2);
+
+            // Session.MoveMouseTo(location.Item1 + radius - 10, location.Item2);
+            Assert.AreEqual("#" + settings.SpotlightColor, colorSpotlight2);
+            Task.Delay(100).Wait();
+
+            var colorBackground = Session.GetPixelColorString(location.Item1 + radius + 50, location.Item2 + radius + 50);
+            Assert.AreEqual("#" + settings.BackgroundColor, colorBackground);
+
+            var colorBackground2 = Session.GetPixelColorString(location.Item1 + radius + 100, location.Item2 + radius + 100);
+            Assert.AreEqual("#" + settings.BackgroundColor, colorBackground2);
+        }
+
+        private void ActivateSpotlight(ref FindMyMouseSettings settings)
+        {
+            var xy = Session.GetMousePosition();
+            Session.MoveMouseTo(xy.Item1 - 200, xy.Item2 - 100);
+            Task.Delay(1000).Wait();
+
+            Session.PerformMouseAction(MouseActionType.LeftClick);
+            Task.Delay(5000).Wait();
+            if (settings.SelectedActivationMethod == FindMyMouseSettings.ActivationMethod.PressLeftControlTwice)
+            {
+                Session.SendKey(Key.LCtrl, 0, 0);
+                Task.Delay(100).Wait();
+                Session.SendKey(Key.LCtrl, 0, 0);
+            }
+            else if (settings.SelectedActivationMethod == FindMyMouseSettings.ActivationMethod.PressRightControlTwice)
+            {
+                Session.SendKey(Key.RCtrl, 0, 0);
+                Task.Delay(100).Wait();
+                Session.SendKey(Key.RCtrl, 0, 0);
+            }
+            else if (settings.SelectedActivationMethod == FindMyMouseSettings.ActivationMethod.ShakeMouse)
+            {
+                // Simulate shake mouse;
+            }
+            else if (settings.SelectedActivationMethod == FindMyMouseSettings.ActivationMethod.CustomShortcut)
+            {
+                // Simulate custom shortcut
+            }
+        }
+
+        private void VerifySpotlightSettings(ref FindMyMouseSettings settings, bool equal = true)
+        {
+            ActivateSpotlight(ref settings);
+
+            VerifySpotlightAppears(ref settings);
+        }
+
+        private void SetFindMyMouseActivationMethod(ref Custom? foundCustom, string method)
+        {
+            Assert.IsNotNull(foundCustom);
+            var groupActivation = foundCustom.Find<TextBlock>("Activation method");
+            if (groupActivation != null)
+            {
+                groupActivation.Click();
+                string findMyMouseComboBoxKey = "Activation method";
+                var foundElements = foundCustom.FindAll<ComboBox>(findMyMouseComboBoxKey);
+                if (foundElements.Count != 0)
+                {
+                    var myMouseComboBox = foundCustom.Find<ComboBox>(findMyMouseComboBoxKey);
+                    Assert.IsNotNull(myMouseComboBox);
+                    myMouseComboBox.Click();
+                    var selectedItem = myMouseComboBox.Find<NavigationViewItem>(method);
+                    Assert.IsNotNull(selectedItem);
+                    selectedItem.Click();
+                }
+                else
+                {
+                    Assert.IsTrue(false, "ComboBox is not found in the setting page.");
+                }
+            }
+            else
+            {
+                Assert.Fail("Activation method group not found.");
+            }
+        }
+
+        private void SetFindMyMouseAppearanceBehavior(ref Custom foundCustom, ref FindMyMouseSettings settings)
+        {
+            Assert.IsNotNull(foundCustom);
+            var groupAppearanceBehavior = foundCustom.Find<TextBlock>("Appearance & behavior");
+            if (groupAppearanceBehavior != null)
+            {
+                // groupAppearanceBehavior.Click();
+                if (foundCustom.FindAll<Slider>("Overlay opacity (%)").Count == 0)
+                {
+                    groupAppearanceBehavior.Click();
+                }
+
+                // Set the BackGround color
+                var backgroundColor = foundCustom.Find<Group>("Background color");
+                Assert.IsNotNull(backgroundColor);
+
+                var button = backgroundColor.Find<Button>(By.XPath(".//Button"));
+                Assert.IsNotNull(button);
+                button.Click();
+
+                var popupWindow = this.Find<Window>("Popup");
+                Assert.IsNotNull(popupWindow);
+                Task.Delay(1000).Wait();
+                var colorModelComboBox = this.Find<ComboBox>("Color model");
+                Assert.IsNotNull(colorModelComboBox);
+                colorModelComboBox.Click();
+                var selectedItem = colorModelComboBox.Find<NavigationViewItem>("RGB");
+                selectedItem.Click();
+                Task.Delay(100).Wait();
+                var rgbHexEdit = this.Find<TextBox>("RGB hex");
+                Assert.IsNotNull(rgbHexEdit);
+                Task.Delay(100).Wait();
+                rgbHexEdit.SetText(settings.BackgroundColor);
+                Task.Delay(100).Wait();
+                button.Click();
+
+                // Set the Spotlight color
+                var spotlightColor = foundCustom.Find<Group>("Spotlight color");
+                Assert.IsNotNull(spotlightColor);
+
+                var spotlightColorButton = spotlightColor.Find<Button>(By.XPath(".//Button"));
+                Assert.IsNotNull(spotlightColorButton);
+                spotlightColorButton.Click();
+
+                var spotlightColorPopupWindow = Session.Find<Window>("Popup");
+                Assert.IsNotNull(spotlightColorPopupWindow);
+                var spotlightColorModelComboBox = this.Find<ComboBox>("Color model");
+                Assert.IsNotNull(spotlightColorModelComboBox);
+                spotlightColorModelComboBox.Click();
+                var selectedItem2 = spotlightColorModelComboBox.Find<NavigationViewItem>("RGB");
+                Assert.IsNotNull(selectedItem2);
+                selectedItem2.Click();
+                Task.Delay(100).Wait();
+                var rgbHexEdit2 = this.Find<TextBox>("RGB hex");
+                Assert.IsNotNull(rgbHexEdit2);
+                Task.Delay(100).Wait();
+                rgbHexEdit2.SetText(settings.SpotlightColor);
+                Task.Delay(100).Wait();
+                spotlightColorButton.Click(false, 500, 1500);
+
+                // Set the overlay opacity to overlayOpacity%
+                var overlayOpacitySlider = foundCustom.Find<Slider>("Overlay opacity (%)");
+                Assert.IsNotNull(overlayOpacitySlider);
+                Assert.IsNotNull(settings.OverlayOpacity);
+                int overlayOpacityValue = int.Parse(settings.OverlayOpacity, CultureInfo.InvariantCulture);
+                overlayOpacitySlider.QuickSetValue(overlayOpacityValue);
+                Assert.AreEqual(settings.OverlayOpacity, overlayOpacitySlider.Text);
+                Task.Delay(1000).Wait();
+
+                // Set the Fade Initial zoom to 0
+                var spotlightInitialZoomSlider = foundCustom.Find<Slider>("Spotlight initial zoom");
+                Assert.IsNotNull(spotlightInitialZoomSlider);
+                Task.Delay(1000).Wait();
+                spotlightInitialZoomSlider.QuickSetValue(int.Parse(settings.InitialZoom, CultureInfo.InvariantCulture));
+                Assert.AreEqual(settings.InitialZoom, spotlightInitialZoomSlider.Text);
+                Task.Delay(1000).Wait();
+
+                //// Change the edit value
+                var spotlightRadiusEdit = foundCustom.Find<TextBox>("Spotlight radius (px) Minimum5");
+                Assert.IsNotNull(spotlightRadiusEdit);
+                Task.Delay(1000).Wait();
+                spotlightRadiusEdit.SetText(settings.Radius);
+                Assert.AreEqual(settings.Radius, spotlightRadiusEdit.Text);
+                Task.Delay(1000).Wait();
+
+                // Set the duration to 0 ms
+                var spotlightAnimationDuration = foundCustom.Find<TextBox>("Animation duration (ms) Minimum0");
+                Assert.IsNotNull(spotlightAnimationDuration);
+                Task.Delay(1000).Wait();
+                spotlightAnimationDuration.SetText(settings.AnimationDuration);
+                Assert.AreEqual(settings.AnimationDuration, spotlightAnimationDuration.Text);
+                Task.Delay(1000).Wait();
+
+                // groupAppearanceBehavior.Click();
+            }
+            else
+            {
+                Assert.Fail("Appearance & behavior group not found.");
+            }
+        }
+
+        private bool FindGroup(string groupName)
+        {
+            try
+            {
+                var foundElements = this.FindAll<Element>(groupName);
+                foreach (var element in foundElements)
+                {
+                    string className = element.ClassName;
+                    string name = element.Name;
+                    string text = element.Text;
+                    string helptext = element.HelpText;
+                    string controlType = element.ControlType;
+                }
+
+                if (foundElements.Count == 0)
+                {
+                    return false;
+                }
+            }
+            catch (Exception ex)
+            {
+                // Validate if group is not found by checking exception.Message
+                return ex.Message.Contains("No element found");
+            }
+
+            return true;
+        }
+
+        private bool CheckAnimationEnable(ref Custom? foundCustom)
+        {
+            Assert.IsNotNull(foundCustom, "Find My Mouse group not found.");
+            var foundElements = foundCustom.FindAll<TextBlock>("Animations are disabled in your system settings.");
+
+            // Assert.IsNull(animationDisabledWarning);
+            if (foundElements.Count != 0)
+            {
+                var openSettingsLink = foundCustom.Find<Element>("Open settings");
+                Assert.IsNotNull(openSettingsLink);
+                openSettingsLink.Click(false, 500, 3000);
+
+                string settingsWindow = "Settings";
+                this.Session.Attach(settingsWindow);
+                var animationEffects = this.Find<ToggleSwitch>("Animation effects");
+                Assert.IsNotNull(animationEffects);
+                animationEffects.Toggle(true);
+
+                Task.Delay(2000).Wait();
+                Session.SendKeys(Key.Alt, Key.F4);
+                this.Session.Attach(PowerToysModule.PowerToysSettings);
+                this.LaunchFromSetting(reload: true);
+            }
+            else
+            {
+                return false;
+            }
+
+            return true;
+        }
+
+        private void LaunchFromSetting(bool reload = false, bool launchAsAdmin = false)
+        {
+            // this.Session.Attach(PowerToysModule.PowerToysSettings);
+            Session.SetMainWindowSize(WindowSize.Large);
+
+            // Goto Hosts File Editor setting page
+            if (this.FindAll<NavigationViewItem>("Mouse utilities", 10000).Count == 0)
+            {
+                // Expand Advanced list-group if needed
+                this.Find<NavigationViewItem>("Input / Output").Click();
+            }
+
+            if (reload)
+            {
+                this.Find<NavigationViewItem>("Keyboard Manager").Click();
+            }
+
+            Task.Delay(1000).Wait();
+            this.Find<NavigationViewItem>("Mouse utilities").Click();
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/MouseHighlighterTests.cs

@@ -0,0 +1,382 @@
+// 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.Globalization;
+using System.Threading.Tasks;
+using System.Xml.Linq;
+using Microsoft.PowerToys.UITest;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Windows.Devices.Printers;
+
+namespace MouseUtils.UITests
+{
+    [TestClass]
+    public class MouseHighlighterTests : UITestBase
+    {
+        [TestMethod]
+        public void TestEnableMouseHighlighter()
+        {
+            LaunchFromSetting();
+            var foundCustom0 = this.Find<Custom>("Find My Mouse");
+            if (foundCustom0 != null)
+            {
+                foundCustom0.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+            }
+            else
+            {
+                Assert.Fail("Find My Mouse custom not found.");
+            }
+
+            var settings = new MouseHighlighterSettings();
+            settings.PrimaryButtonHighlightColor = "FFFF0000";
+            settings.SecondaryButtonHighlightColor = "FF00FF00";
+            settings.AlwaysHighlightColor = "004cFF71";
+            settings.Radius = "50";
+            settings.FadeDelay = "0";
+            settings.FadeDuration = "90";
+
+            var foundCustom = this.Find<Custom>("Mouse Highlighter");
+            if (foundCustom != null)
+            {
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Highlighter").Toggle(true);
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Highlighter").Toggle(false);
+
+                var xy = Session.GetMousePosition();
+                Session.MoveMouseTo(xy.Item1, xy.Item2 - 100);
+
+                Session.PerformMouseAction(MouseActionType.ScrollDown);
+                Session.PerformMouseAction(MouseActionType.ScrollDown);
+                Session.PerformMouseAction(MouseActionType.ScrollDown);
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Highlighter").Toggle(true);
+
+                // Change the shortcut key for MouseHighlighter
+                // [TestCase]Change activation shortcut and test it
+                var activationShortcutButton = foundCustom.Find<Button>("Activation shortcut");
+                Assert.IsNotNull(activationShortcutButton);
+
+                activationShortcutButton.Click();
+                Task.Delay(500).Wait();
+                var activationShortcutWindow = Session.Find<Window>("Activation shortcut");
+                Assert.IsNotNull(activationShortcutWindow);
+
+                // Invalid shortcut key
+                Session.SendKeySequence(Key.H);
+
+                // IOUtil.SimulateKeyPress(0x41);
+                var invalidShortcutText = activationShortcutWindow.Find<TextBlock>("Invalid shortcut");
+                Assert.IsNotNull(invalidShortcutText);
+
+                // IOUtil.SimulateShortcut(0x5B, 0x10, 0x45);
+                Session.SendKeys(Key.Win, Key.Shift, Key.H);
+
+                // Assert.IsNull(activationShortcutWindow.Find<TextBlock>("Invalid shortcut"));
+                var saveButton = activationShortcutWindow.Find<Button>("Save");
+                Assert.IsNotNull(saveButton);
+                saveButton.Click(false, 500, 1000);
+
+                SetMouseHighlighterAppearanceBehavior(ref foundCustom, ref settings);
+
+                var xy0 = Session.GetMousePosition();
+                Session.MoveMouseTo(xy0.Item1 - 100, xy0.Item2);
+                Session.PerformMouseAction(MouseActionType.LeftClick);
+
+                // Check the mouse highlighter is enabled
+                Session.SendKeys(Key.Win, Key.Shift, Key.H);
+
+                // IOUtil.SimulateShortcut(0x5B, 0x10, 0x45);
+                Task.Delay(1000).Wait();
+
+                // MouseSimulator.LeftClick();
+                // [Test Case] Press the activation shortcut and press left and right click somewhere, verifying the highlights are applied.
+                // [Test Case] Press the activation shortcut again and verify no highlights appear when the mouse buttons are clicked.
+                VerifyMouseHighlighterAppears(ref settings, "leftClick");
+                VerifyMouseHighlighterAppears(ref settings, "rightClick");
+
+                // Disable mouse highlighter
+                Session.SendKeys(Key.Win, Key.Shift, Key.H);
+                Task.Delay(1000).Wait();
+
+                VerifyMouseHighlighterNotAppears(ref settings, "leftClick");
+                VerifyMouseHighlighterNotAppears(ref settings, "rightClick");
+
+                // [Test Case] Disable Mouse Highlighter and verify that the module is not activated when you press the activation shortcut.
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Highlighter").Toggle(false);
+                xy = Session.GetMousePosition();
+                Session.MoveMouseTo(xy.Item1 - 100, xy.Item2);
+
+                Session.SendKeys(Key.Win, Key.Shift, Key.H);
+                Task.Delay(1000).Wait();
+
+                VerifyMouseHighlighterNotAppears(ref settings, "leftClick");
+                VerifyMouseHighlighterNotAppears(ref settings, "rightClick");
+
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Highlighter").Toggle(true);
+                xy = Session.GetMousePosition();
+                Session.MoveMouseTo(xy.Item1 - 100, xy.Item2);
+
+                Session.SendKeys(Key.Win, Key.Shift, Key.H);
+                Task.Delay(1000).Wait();
+                VerifyMouseHighlighterDrag(ref settings, "leftClick");
+                VerifyMouseHighlighterDrag(ref settings, "rightClick");
+            }
+            else
+            {
+                Assert.Fail("Mouse Highlighter Custom not found.");
+            }
+
+            Task.Delay(500).Wait();
+        }
+
+        private void VerifyMouseHighlighterDrag(ref MouseHighlighterSettings settings, string action = "leftClick")
+        {
+            Task.Delay(500).Wait();
+            string expectedColor = string.Empty;
+            if (action == "leftClick")
+            {
+                // MouseSimulator.LeftDown();
+                Session.PerformMouseAction(MouseActionType.LeftDown);
+                expectedColor = settings.PrimaryButtonHighlightColor.Substring(2);
+            }
+            else if (action == "rightClick")
+            {
+                // MouseSimulator.RightDown();
+                Session.PerformMouseAction(MouseActionType.RightDown);
+                expectedColor = settings.SecondaryButtonHighlightColor.Substring(2);
+            }
+            else
+            {
+                Assert.Fail("Invalid action specified.");
+            }
+
+            expectedColor = "#" + expectedColor;
+
+            var location = Session.GetMousePosition();
+            int radius = int.Parse(settings.Radius, CultureInfo.InvariantCulture);
+            var colorLeftClick = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreEqual(expectedColor, colorLeftClick);
+
+            var colorLeftClick2 = Session.GetPixelColorString(location.Item1 + radius - 1, location.Item2);
+
+            Assert.AreEqual(expectedColor, colorLeftClick2);
+
+            var colorBackground = Session.GetPixelColorString(location.Item1 + radius + 50, location.Item2 + radius + 50);
+            Assert.AreNotEqual(expectedColor, colorBackground);
+
+            // Drag the mouse
+            Session.MoveMouseTo(location.Item1 - 200, location.Item2);
+            Task.Delay(2000).Wait();
+
+            location = Session.GetMousePosition();
+            colorLeftClick = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreEqual(expectedColor, colorLeftClick);
+
+            colorLeftClick2 = Session.GetPixelColorString(location.Item1 + radius - 1, location.Item2);
+
+            Assert.AreEqual(expectedColor, colorLeftClick2);
+
+            colorBackground = Session.GetPixelColorString(location.Item1 + radius + 50, location.Item2 + radius + 50);
+            Assert.AreNotEqual(expectedColor, colorBackground);
+
+            if (action == "leftClick")
+            {
+                // MouseSimulator.LeftUp();
+                Session.PerformMouseAction(MouseActionType.LeftUp);
+            }
+            else if (action == "rightClick")
+            {
+                // MouseSimulator.RightUp();
+                Session.PerformMouseAction(MouseActionType.RightUp);
+            }
+
+            int duration = int.Parse(settings.FadeDuration, CultureInfo.InvariantCulture);
+            Task.Delay(duration + 100).Wait();
+            colorLeftClick = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreNotEqual("#" + settings.PrimaryButtonHighlightColor, colorLeftClick);
+        }
+
+        private void VerifyMouseHighlighterNotAppears(ref MouseHighlighterSettings settings, string action = "leftClick")
+        {
+            Task.Delay(500).Wait();
+            string expectedColor = string.Empty;
+            if (action == "leftClick")
+            {
+                // MouseSimulator.LeftDown();
+                Session.PerformMouseAction(MouseActionType.LeftDown);
+                expectedColor = settings.PrimaryButtonHighlightColor.Substring(2);
+            }
+            else if (action == "rightClick")
+            {
+                // MouseSimulator.RightDown();
+                Session.PerformMouseAction(MouseActionType.RightDown);
+                expectedColor = settings.SecondaryButtonHighlightColor.Substring(2);
+            }
+            else
+            {
+                Assert.Fail("Invalid action specified.");
+            }
+
+            expectedColor = "#" + expectedColor;
+            var location = Session.GetMousePosition();
+            int radius = int.Parse(settings.Radius, CultureInfo.InvariantCulture);
+            var colorLeftClick = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreNotEqual(expectedColor, colorLeftClick);
+            var colorLeftClick2 = Session.GetPixelColorString(location.Item1 + radius - 1, location.Item2);
+            Assert.AreNotEqual(expectedColor, colorLeftClick2);
+            if (action == "leftClick")
+            {
+                Session.PerformMouseAction(MouseActionType.LeftUp);
+            }
+            else if (action == "rightClick")
+            {
+                Session.PerformMouseAction(MouseActionType.RightUp);
+            }
+        }
+
+        private void VerifyMouseHighlighterAppears(ref MouseHighlighterSettings settings, string action = "leftClick")
+        {
+            Task.Delay(500).Wait();
+            string expectedColor = string.Empty;
+            if (action == "leftClick")
+            {
+                // MouseSimulator.LeftDown();
+                Session.PerformMouseAction(MouseActionType.LeftDown);
+                expectedColor = settings.PrimaryButtonHighlightColor.Substring(2);
+            }
+            else if (action == "rightClick")
+            {
+                // MouseSimulator.RightDown();
+                Session.PerformMouseAction(MouseActionType.RightDown);
+                expectedColor = settings.SecondaryButtonHighlightColor.Substring(2);
+            }
+            else
+            {
+                Assert.Fail("Invalid action specified.");
+            }
+
+            expectedColor = "#" + expectedColor;
+
+            var location = Session.GetMousePosition();
+            int radius = int.Parse(settings.Radius, CultureInfo.InvariantCulture);
+            var colorLeftClick = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreEqual(expectedColor, colorLeftClick);
+
+            var colorLeftClick2 = Session.GetPixelColorString(location.Item1 + radius - 1, location.Item2);
+
+            Assert.AreEqual(expectedColor, colorLeftClick2);
+            Task.Delay(500).Wait();
+
+            var colorBackground = Session.GetPixelColorString(location.Item1 + radius + 50, location.Item2 + radius + 50);
+            Assert.AreNotEqual(expectedColor, colorBackground);
+            if (action == "leftClick")
+            {
+                // MouseSimulator.LeftUp();
+                Session.PerformMouseAction(MouseActionType.LeftUp);
+            }
+            else if (action == "rightClick")
+            {
+                // MouseSimulator.RightUp();
+                Session.PerformMouseAction(MouseActionType.RightUp);
+            }
+
+            int duration = int.Parse(settings.FadeDuration, CultureInfo.InvariantCulture);
+            Task.Delay(duration + 100).Wait();
+            colorLeftClick = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreNotEqual("#" + settings.PrimaryButtonHighlightColor, colorLeftClick);
+        }
+
+        private void SetColor(ref Custom foundCustom, string colorName = "Primary button highlight color", string colorValue = "000000", string opacity = "0")
+        {
+            Assert.IsNotNull(foundCustom);
+            var groupAppearanceBehavior = foundCustom.Find<TextBlock>("Appearance & behavior");
+            if (groupAppearanceBehavior != null)
+            {
+                if (foundCustom.FindAll<TextBox>("Fade duration (ms) Minimum0").Count == 0)
+                {
+                    groupAppearanceBehavior.Click();
+                }
+
+                // Set primary button highlight color
+                var primaryButtonHighlightColor = foundCustom.Find<Group>(colorName);
+                Assert.IsNotNull(primaryButtonHighlightColor);
+
+                var button = primaryButtonHighlightColor.Find<Button>(By.XPath(".//Button"));
+                Assert.IsNotNull(button);
+                button.Click(false, 500, 700);
+
+                var popupWindow = Session.Find<Window>("Popup");
+                Assert.IsNotNull(popupWindow);
+                var colorModelComboBox = this.Find<ComboBox>("Color model");
+                Assert.IsNotNull(colorModelComboBox);
+                colorModelComboBox.Click(false, 500, 700);
+                var selectedItem = colorModelComboBox.Find<NavigationViewItem>("RGB");
+                selectedItem.Click();
+                var rgbHexEdit = this.Find<TextBox>("RGB hex");
+                Assert.IsNotNull(rgbHexEdit);
+                rgbHexEdit.SetText(colorValue);
+
+                button.Click();
+            }
+        }
+
+        private void SetMouseHighlighterAppearanceBehavior(ref Custom foundCustom, ref MouseHighlighterSettings settings)
+        {
+            Assert.IsNotNull(foundCustom);
+            var groupAppearanceBehavior = foundCustom.Find<TextBlock>("Appearance & behavior");
+            if (groupAppearanceBehavior != null)
+            {
+                // groupAppearanceBehavior.Click();
+                if (foundCustom.FindAll<TextBox>(settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.FadeDurationEdit)).Count == 0)
+                {
+                    groupAppearanceBehavior.Click();
+                }
+
+                // Set primary button highlight color
+                SetColor(ref foundCustom, settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.PrimaryButtonHighlightColorGroup), settings.PrimaryButtonHighlightColor);
+
+                // Set secondary button highlight color
+                SetColor(ref foundCustom, settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.SecondaryButtonHighlightColorGroup), settings.SecondaryButtonHighlightColor);
+
+                // Set the duration to duration ms
+                var fadeDurationEdit = foundCustom.Find<TextBox>(settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.FadeDurationEdit));
+                Assert.IsNotNull(fadeDurationEdit);
+                fadeDurationEdit.SetText(settings.FadeDuration);
+                Assert.AreEqual(settings.FadeDuration, fadeDurationEdit.Text);
+
+                // Set Fade delay(ms)
+                var fadeDelayEdit = foundCustom.Find<TextBox>(settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.FadeDelayEdit));
+                Assert.IsNotNull(fadeDelayEdit);
+                fadeDelayEdit.SetText(settings.FadeDelay);
+                Assert.AreEqual(settings.FadeDelay, fadeDelayEdit.Text);
+
+                // Set the fade radius (px)
+                var fadeRadiusEdit = foundCustom.Find<TextBox>(settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.RadiusEdit));
+                Assert.IsNotNull(fadeRadiusEdit);
+                fadeRadiusEdit.SetText(settings.Radius);
+                Assert.AreEqual(settings.Radius, fadeRadiusEdit.Text);
+
+                // Set always highlight color
+                SetColor(ref foundCustom, settings.GetElementName(MouseHighlighterSettings.SettingsUIElements.AlwaysHighlightColorGroup), settings.AlwaysHighlightColor);
+            }
+            else
+            {
+                Assert.Fail("Appearance & behavior group not found.");
+            }
+        }
+
+        private void LaunchFromSetting(bool showWarning = false, bool launchAsAdmin = false)
+        {
+            Session.SetMainWindowSize(WindowSize.Large);
+
+            // Goto Hosts File Editor setting page
+            if (this.FindAll<NavigationViewItem>("Mouse utilities").Count == 0)
+            {
+                // Expand Advanced list-group if needed
+                this.Find<NavigationViewItem>("Input / Output").Click();
+            }
+
+            this.Find<NavigationViewItem>("Mouse utilities").Click();
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/MouseJumpTests.cs

@@ -0,0 +1,144 @@
+// 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.Threading.Tasks;
+using Microsoft.PowerToys.UITest;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+
+namespace MouseUtils.UITests
+{
+    [TestClass]
+    public class MouseJumpTests : UITestBase
+    {
+        [TestMethod]
+        public void TestEnableMouseJump()
+        {
+            LaunchFromSetting();
+            var foundCustom0 = this.Find<Custom>("Find My Mouse");
+            if (foundCustom0 != null)
+            {
+                foundCustom0.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(true);
+                foundCustom0.Find<ToggleSwitch>("Enable Find My Mouse").Toggle(false);
+            }
+            else
+            {
+                Assert.Fail("Find My Mouse custom not found.");
+            }
+
+            for (int i = 0; i < 10; i++)
+            {
+                Session.PerformMouseAction(MouseActionType.ScrollDown);
+            }
+
+            var foundCustom = this.Find<Custom>("Mouse Jump");
+            if (foundCustom != null)
+            {
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Jump").Toggle(true);
+
+                var xy = Session.GetMousePosition();
+                Session.MoveMouseTo(xy.Item1, xy.Item2 - 100);
+
+                // Change the shortcut key for MouseHighlighter
+                // [TestCase]Change activation shortcut and test it
+                var activationShortcutButton = foundCustom.Find<Button>("Activation shortcut");
+                Assert.IsNotNull(activationShortcutButton);
+
+                activationShortcutButton.Click(false, 500, 1000);
+                var activationShortcutWindow = Session.Find<Window>("Activation shortcut");
+                Assert.IsNotNull(activationShortcutWindow);
+
+                // Invalid shortcut key
+                Session.SendKeySequence(Key.H);
+
+                // IOUtil.SimulateKeyPress(0x41);
+                var invalidShortcutText = activationShortcutWindow.Find<TextBlock>("Invalid shortcut");
+                Assert.IsNotNull(invalidShortcutText);
+
+                // IOUtil.SimulateShortcut(0x5B, 0x10, 0x45);
+                Session.SendKeys(Key.Win, Key.Shift, Key.Z);
+
+                // Assert.IsNull(activationShortcutWindow.Find<TextBlock>("Invalid shortcut"));
+                var saveButton = activationShortcutWindow.Find<Button>("Save");
+                Assert.IsNotNull(saveButton);
+                saveButton.Click(false, 500, 1500);
+
+                var screenCenter = this.Session.GetScreenCenter();
+                Session.MoveMouseTo(screenCenter.CenterX, screenCenter.CenterY, 500, 1000);
+                Session.MoveMouseTo(screenCenter.CenterX, screenCenter.CenterY - 300, 500, 1000);
+
+                Session.SendKeys(Key.Win, Key.Shift, Key.Z);
+                VerifyWindowAppears();
+
+                Task.Delay(1000).Wait();
+                foundCustom.Find<ToggleSwitch>("Enable Mouse Jump").Toggle(false);
+                Session.MoveMouseTo(screenCenter.CenterX, screenCenter.CenterY - 300, 500, 1000);
+                Session.SendKeys(Key.Win, Key.Shift, Key.Z);
+                Task.Delay(500).Wait();
+                VerifyWindowNotAppears();
+            }
+            else
+            {
+                Assert.Fail("Mouse Highlighter Custom not found.");
+            }
+
+            Task.Delay(500).Wait();
+        }
+
+        private void VerifyWindowAppears()
+        {
+            string windowName = "MouseJump";
+            Session.Attach(windowName);
+            var center = this.Session.GetWindowCenter();
+            Session.MoveMouseTo(center.CenterX, center.CenterY);
+            Session.PerformMouseAction(MouseActionType.LeftClick, 1000, 1000);
+            var screenCenter = this.Session.GetScreenCenter();
+
+            // Get Mouse position
+            var xy = Session.GetMousePosition();
+
+            double distance = CalculateDistance(xy.Item1, xy.Item2, screenCenter.CenterX, screenCenter.CenterY);
+            Assert.IsTrue(distance <= 10, "Mouse Jump window should be opened and mouse should be moved to the center of the screen.");
+        }
+
+        private void VerifyWindowNotAppears()
+        {
+            string windowName = "MouseJump";
+            bool open = Session.IsWindowOpen(windowName);
+            Assert.IsFalse(open, "Mouse Jump window should not be opened.");
+        }
+
+        /// <summary>
+        /// Calculate the Euclidean distance between two 2D points
+        /// </summary>
+        /// <param name="x1">X coordinate of first point</param>
+        /// <param name="y1">Y coordinate of first point</param>
+        /// <param name="x2">X coordinate of second point</param>
+        /// <param name="y2">Y coordinate of second point</param>
+        /// <returns>Distance (double)</returns>
+        public double CalculateDistance(int x1, int y1, int x2, int y2)
+        {
+            int dx = x2 - x1;
+            int dy = y2 - y1;
+            return Math.Sqrt((dx * dx) + (dy * dy));
+        }
+
+        private void LaunchFromSetting(bool showWarning = false, bool launchAsAdmin = false)
+        {
+            Session.SetMainWindowSize(WindowSize.Large);
+            Task.Delay(1000).Wait();
+
+            // Goto Hosts File Editor setting page
+            if (this.FindAll<NavigationViewItem>("Mouse utilities").Count == 0)
+            {
+                // Expand Advanced list-group if needed
+                this.Find<NavigationViewItem>("Input / Output").ClickCenter();
+            }
+
+            // Click on the Mouse utilities
+            Task.Delay(2000).Wait();
+            this.Find<NavigationViewItem>("Mouse utilities").Click();
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/MousePointerCrosshairsTests.cs

@@ -0,0 +1,310 @@
+// 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.Globalization;
+using System.Reflection;
+using System.Threading.Tasks;
+using System.Xml.Linq;
+using Microsoft.PowerToys.UITest;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Windows.Devices.Printers;
+
+namespace MouseUtils.UITests
+{
+    [TestClass]
+    public class MousePointerCrosshairsTests : UITestBase
+    {
+        [TestMethod]
+        public void TestEnableMousePointerCrosshairs()
+        {
+            LaunchFromSetting();
+
+            var settings = new MousePointerCrosshairsSettings();
+            settings.CrosshairsColor = "FF0000";
+            settings.CrosshairsBorderColor = "FF0000";
+            settings.Opacity = "100";
+            settings.CenterRadius = "0";
+            settings.Thickness = "20";
+            settings.BorderSize = "0";
+            settings.IsFixLength = false;
+            settings.FixedLength = "1";
+
+            var foundCustom = FindMouseUtilElement(MouseUtilsSettings.MouseUtils.FindMyMouse);
+            MouseUtilsSettings.SetMouseUtilEnabled(foundCustom, MouseUtilsSettings.MouseUtils.FindMyMouse, false);
+
+            foundCustom = FindMouseUtilElement(MouseUtilsSettings.MouseUtils.MouseHighlighter);
+            MouseUtilsSettings.SetMouseUtilEnabled(foundCustom, MouseUtilsSettings.MouseUtils.MouseHighlighter, true);
+            MouseUtilsSettings.SetMouseUtilEnabled(foundCustom, MouseUtilsSettings.MouseUtils.MouseHighlighter, false);
+
+            for (int i = 0; i < 10; i++)
+            {
+                Session.PerformMouseAction(MouseActionType.ScrollDown);
+            }
+
+            foundCustom = FindMouseUtilElement(MouseUtilsSettings.MouseUtils.MousePointerCrosshairs);
+
+            // this.FindGroup("Enable Mouse Pointer Crosshairs");
+            MouseUtilsSettings.SetMouseUtilEnabled(foundCustom, MouseUtilsSettings.MouseUtils.MousePointerCrosshairs, false);
+            Task.Delay(500).Wait();
+            MouseUtilsSettings.SetMouseUtilEnabled(foundCustom, MouseUtilsSettings.MouseUtils.MousePointerCrosshairs, true);
+
+            Assert.IsNotNull(foundCustom);
+
+            // [Test Case] Change activation shortcut and test it.
+            var activationShortcutButton = foundCustom.Find<Button>("Activation shortcut");
+            Assert.IsNotNull(activationShortcutButton);
+
+            activationShortcutButton.Click(false, 500, 1000);
+            var activationShortcutWindow = Session.Find<Window>("Activation shortcut");
+            Assert.IsNotNull(activationShortcutWindow);
+
+            // Invalid shortcut key
+            Session.SendKeySequence(Key.H);
+
+            var invalidShortcutText = activationShortcutWindow.Find<TextBlock>("Invalid shortcut");
+            Assert.IsNotNull(invalidShortcutText);
+
+            Session.SendKeys(Key.Win, Key.Alt, Key.A);
+
+            // Assert.IsNull(activationShortcutWindow.Find<TextBlock>("Invalid shortcut"));
+            var saveButton = activationShortcutWindow.Find<Button>("Save");
+            Assert.IsNotNull(saveButton);
+            saveButton.Click(false, 500, 1000);
+
+            SetMousePointerCrosshairsAppearanceBehavior(ref foundCustom, ref settings);
+            Task.Delay(500).Wait();
+
+            // [Test Case]  Press the activation shortcut and verify the crosshairs appear, and that they follow the mouse around.
+            var xy0 = Session.GetMousePosition();
+            Session.MoveMouseTo(xy0.Item1 - 100, xy0.Item2);
+            Session.PerformMouseAction(MouseActionType.LeftClick);
+            Session.SendKeys(Key.Win, Key.Alt, Key.A);
+            Task.Delay(1000).Wait();
+
+            xy0 = Session.GetMousePosition();
+            Session.MoveMouseTo(xy0.Item1 - 100, xy0.Item2 - 100);
+            Session.PerformMouseAction(MouseActionType.LeftClick);
+            VerifyMousePointerCrosshairsAppears(ref settings);
+            Task.Delay(500).Wait();
+
+            xy0 = Session.GetMousePosition();
+            Session.MoveMouseTo(xy0.Item1 - 50, xy0.Item2 - 50);
+            VerifyMousePointerCrosshairsAppears(ref settings);
+
+            // [Test Case] Press the activation shortcut again and verify the crosshairs disappear.
+            Session.SendKeys(Key.Win, Key.Alt, Key.A);
+            Task.Delay(1000).Wait();
+
+            xy0 = Session.GetMousePosition();
+            Session.MoveMouseTo(xy0.Item1 - 10, xy0.Item2);
+            Session.PerformMouseAction(MouseActionType.LeftClick);
+            VerifyMousePointerCrosshairsNotAppears(ref settings);
+            Task.Delay(500).Wait();
+
+            // [Test Case] Disable Mouse Pointer Crosshairs and verify that the module is not activated when you press the activation shortcut.
+            MouseUtilsSettings.SetMouseUtilEnabled(foundCustom, MouseUtilsSettings.MouseUtils.MousePointerCrosshairs, false);
+            xy0 = Session.GetMousePosition();
+            Session.MoveMouseTo(xy0.Item1 - 100, xy0.Item2);
+            Session.PerformMouseAction(MouseActionType.LeftClick);
+            Session.SendKeys(Key.Win, Key.Alt, Key.A);
+            Task.Delay(1000).Wait();
+
+            VerifyMousePointerCrosshairsNotAppears(ref settings);
+        }
+
+        private void VerifyMousePointerCrosshairsNotAppears(ref MousePointerCrosshairsSettings settings)
+        {
+            Task.Delay(500).Wait();
+            string expectedColor = string.Empty;
+            expectedColor = "#" + settings.CrosshairsColor;
+            var location = Session.GetMousePosition();
+
+            int radius = int.Parse(settings.CenterRadius, CultureInfo.InvariantCulture);
+
+            var color = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreNotEqual(expectedColor, color);
+        }
+
+        private void VerifyMousePointerCrosshairsAppears(ref MousePointerCrosshairsSettings settings)
+        {
+            Task.Delay(1000).Wait();
+            string expectedColor = string.Empty;
+            expectedColor = "#" + settings.CrosshairsColor;
+            var location = Session.GetMousePosition();
+
+            int radius = int.Parse(settings.CenterRadius, CultureInfo.InvariantCulture);
+
+            var color = Session.GetPixelColorString(location.Item1, location.Item2);
+            Assert.AreEqual(expectedColor, color, "Center color check failed");
+
+            var colorX = Session.GetPixelColorString(location.Item1 + 50, location.Item2);
+            Assert.AreEqual(expectedColor, colorX, "Center x + 50 color check failed");
+
+            colorX = Session.GetPixelColorString(location.Item1 - 50, location.Item2);
+            Assert.AreEqual(expectedColor, colorX, "Center x - 50 color check failed");
+
+            var colorY = Session.GetPixelColorString(location.Item1, location.Item2 + 50);
+            Assert.AreEqual(expectedColor, colorY, "Center y + 50 color check failed");
+
+            colorY = Session.GetPixelColorString(location.Item1, location.Item2 - 50);
+            Assert.AreEqual(expectedColor, colorY, "Center y + 50 color check failed");
+        }
+
+        private void SetColor(ref Custom foundCustom, string colorName, string colorValue = "000000")
+        {
+            Assert.IsNotNull(foundCustom);
+            var groupAppearanceBehavior = foundCustom.Find<TextBlock>("Appearance & behavior");
+            if (groupAppearanceBehavior != null)
+            {
+                // Set primary button highlight color
+                var primaryButtonHighlightColor = foundCustom.Find<Group>(colorName);
+                Assert.IsNotNull(primaryButtonHighlightColor);
+
+                var button = primaryButtonHighlightColor.Find<Button>(By.XPath(".//Button"));
+                Assert.IsNotNull(button);
+                button.Click(false);
+                var popupWindow = Session.Find<Window>("Popup");
+                Assert.IsNotNull(popupWindow);
+                var colorModelComboBox = this.Find<ComboBox>("Color model");
+                Assert.IsNotNull(colorModelComboBox);
+                colorModelComboBox.Click();
+                var selectedItem = colorModelComboBox.Find<NavigationViewItem>("RGB");
+                selectedItem.Click();
+                var rgbHexEdit = this.Find<TextBox>("RGB hex");
+                Assert.IsNotNull(rgbHexEdit);
+                rgbHexEdit.SetText(colorValue);
+
+                button.Click();
+            }
+        }
+
+        private void SetMousePointerCrosshairsAppearanceBehavior(ref Custom foundCustom, ref MousePointerCrosshairsSettings settings)
+        {
+            Assert.IsNotNull(foundCustom);
+            var groupAppearanceBehavior = foundCustom.Find<TextBlock>("Appearance & behavior");
+            if (groupAppearanceBehavior != null)
+            {
+                // groupAppearanceBehavior.Click();
+                if (foundCustom.FindAll<TextBox>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.ThicknessEdit)).Count == 0)
+                {
+                    groupAppearanceBehavior.Click();
+                    Session.PerformMouseAction(MouseActionType.ScrollDown);
+                    Session.PerformMouseAction(MouseActionType.ScrollDown);
+                    Session.PerformMouseAction(MouseActionType.ScrollDown);
+                }
+
+                // Set the crosshairs color
+                SetColor(ref foundCustom, settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.CrosshairsColorGroup), settings.CrosshairsColor);
+
+                // Set the crosshairs border color
+                SetColor(ref foundCustom, settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.CrosshairsBorderColorGroup), settings.CrosshairsBorderColor);
+
+                // Set the duration to duration ms
+                var opacitySlider = foundCustom.Find<Slider>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.OpacitySlider));
+                Assert.IsNotNull(opacitySlider);
+                Assert.IsNotNull(settings.Opacity);
+                int opacityValue = int.Parse(settings.Opacity, CultureInfo.InvariantCulture);
+                opacitySlider.QuickSetValue(opacityValue);
+                Assert.AreEqual(settings.Opacity, opacitySlider.Text);
+                Task.Delay(500).Wait();
+
+                // Set the center radius (px)
+                var centerRadiusEdit = foundCustom.Find<TextBox>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.CenterRadiusEdit));
+                Assert.IsNotNull(centerRadiusEdit);
+                centerRadiusEdit.SetText(settings.CenterRadius);
+                Assert.AreEqual(settings.CenterRadius, centerRadiusEdit.Text);
+
+                // Set the thickness (px)
+                var thicknessEdit = foundCustom.Find<TextBox>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.ThicknessEdit));
+                Assert.IsNotNull(thicknessEdit);
+                thicknessEdit.SetText(settings.Thickness);
+                Assert.AreEqual(settings.Thickness, thicknessEdit.Text);
+
+                // Set the border size (px)
+                var borderSizeEdit = foundCustom.Find<TextBox>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.BorderSizeEdit));
+                Assert.IsNotNull(borderSizeEdit);
+                borderSizeEdit.SetText(settings.BorderSize);
+                Assert.AreEqual(settings.BorderSize, borderSizeEdit.Text);
+
+                // Set the fixed length (px)
+                var isFixedLength = foundCustom.Find<ToggleSwitch>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.IsFixLengthToggle));
+                Assert.IsNotNull(isFixedLength);
+                isFixedLength.Toggle(settings.IsFixLength);
+                Assert.AreEqual(settings.IsFixLength, isFixedLength.IsOn);
+                if (settings.IsFixLength)
+                {
+                    var fixedLengthEdit = foundCustom.Find<TextBox>(settings.GetElementName(MousePointerCrosshairsSettings.SettingsUIElements.FixedLengthEdit));
+                    Assert.IsNotNull(fixedLengthEdit);
+                    fixedLengthEdit.SetText(settings.FixedLength);
+                    Assert.AreEqual(settings.FixedLength, fixedLengthEdit.Text);
+                }
+            }
+            else
+            {
+                Assert.Fail("Appearance & behavior group not found.");
+            }
+        }
+
+        private bool FindGroup(string groupName)
+        {
+            try
+            {
+                var foundElements = this.FindAll<Element>(groupName);
+                foreach (var element in foundElements)
+                {
+                    string className = element.ClassName;
+                    string name = element.Name;
+                    string text = element.Text;
+                    string helptext = element.HelpText;
+                    string controlType = element.ControlType;
+                }
+
+                if (foundElements.Count == 0)
+                {
+                    return false;
+                }
+            }
+            catch (Exception ex)
+            {
+                // Validate if group is not found by checking exception.Message
+                return ex.Message.Contains("No element found");
+            }
+
+            return true;
+        }
+
+        public Custom? FindMouseUtilElement(MouseUtilsSettings.MouseUtils element)
+        {
+            var elementName = MouseUtilsSettings.GetMouseUtilUIName(element);
+            var foundCustom = this.Find<Custom>(elementName);
+            for (int i = 0; i < 20; i++)
+            {
+                if (foundCustom != null)
+                {
+                    break;
+                }
+
+                Session.PerformMouseAction(MouseActionType.ScrollDown);
+                foundCustom = this.Find<Custom>(elementName);
+            }
+
+            return foundCustom;
+        }
+
+        private void LaunchFromSetting(bool showWarning = false, bool launchAsAdmin = false)
+        {
+            Session.SetMainWindowSize(WindowSize.Large);
+
+            // Goto Hosts File Editor setting page
+            if (this.FindAll<NavigationViewItem>("Mouse utilities").Count == 0)
+            {
+                // Expand Advanced list-group if needed
+                this.Find<NavigationViewItem>("Input / Output").Click();
+            }
+
+            this.Find<NavigationViewItem>("Mouse utilities").Click();
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/MouseUtils.UITests.csproj

@@ -0,0 +1,25 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <!-- Look at Directory.Build.props in root for common stuff as well -->
+  <Import Project="..\..\..\Common.Dotnet.CsWinRT.props" />
+
+  <PropertyGroup>
+    <ProjectGuid>{4E0AE3A4-2EE0-44D7-A2D0-8769977254A1}</ProjectGuid>
+      <RootNamespace>PowerToys.MouseUtils.UITests</RootNamespace>
+      <AssemblyName>PowerToys.MouseUtils.UITests</AssemblyName>
+      <IsPackable>false</IsPackable>
+      <IsTestProject>true</IsTestProject>
+      <Nullable>enable</Nullable>
+      <OutputType>Library</OutputType>
+
+      <!-- This is a UI test, so don't run as part of MSBuild -->
+      <RunVSTest>false</RunVSTest>
+    </PropertyGroup>
+    <PropertyGroup>
+      <OutputPath>$(SolutionDir)$(Platform)\$(Configuration)\tests\MouseUtils.UITests\</OutputPath>
+    </PropertyGroup>
+
+    <ItemGroup>
+      <PackageReference Include="MSTest" />
+      <ProjectReference Include="..\..\..\common\UITestAutomation\UITestAutomation.csproj" />
+    </ItemGroup>
+</Project>

src/modules/MouseUtils/MouseUtils.UITests/Release-Test-Checklist-Migration-Progress.md

@@ -0,0 +1,26 @@
+## This is for tracking UI-Tests migration progress for Hosts File Editor Module
+Refer to [release check list] (https://github.com/microsoft/PowerToys/blob/releaseChecklist/doc/releases/tests-checklist-template.md#hosts-file-editor) for all manual tests.
+
+### Existing Manual Test-cases run by previous PowerToys owner
+For existing manual test-cases, we will convert them to UI-Tests and run them in CI and Release pipeline
+
+ * Launch Host File Editor:
+   - [x] Verify the application exits if "Quit" is clicked on the initial warning. (**HostsSettingTests.TestWarningDialog**)
+   - [x] Launch Host File Editor again and click "Accept". The module should not close. (**HostModuleTests.TestEmptyView**)
+   - [ ] Launch Host File Editor again and click "Accept". The module should not close. Open the hosts file (`%WinDir%\System32\Drivers\Etc`) in a text editor that auto-refreshes so you can see the changes applied by the editor in real time. (VSCode is an editor like this, for example)
+   - [ ] Enable and disable lines and verify they are applied to the file.
+   - [ ] Add a new entry and verify it's applied.
+   - [ ] Add manually an entry with more than 9 hosts in hosts file (Windows limitation) and verify it is split correctly on loading and the info bar is shown.
+   - [x] Try to filter for lines and verify you can find them. (**HostModuleTests.TestFilterControl**)
+   - [ ] Click the "Open hosts file" button and verify it opens in your default editor. (likely Notepad)
+ * Test the different settings and verify they are applied:
+   - [ ] Launch as Administrator.
+   - [x] Show a warning at startup. (**HostsSettingTests.TestWarningDialog**)
+   - [ ] Additional lines position.
+
+### Additional UI-Tests cases
+  - [x] Add manually an entry with more than 9 hosts and Add button should be disabled. (**HostModuleTests.TestTooManyHosts**)
+  - [x] Add manually an entry with less or equal 9 hosts and Add button should be enabled. (**HostModuleTests.TestTooManyHosts**)
+  - [x] Should show empty view if no entries. (**HostModuleTests.TestEmptyView**)
+  - [x] Add a new entry with valid or invalid input (**HostModuleTests.TestAddHost**)
+  - [x] Show save host file error if not run as Administrator. (**HostModuleTests.TestErrorMessageWithNonAdminPermission**)

src/modules/MouseUtils/MouseUtils.UITests/util/FindMyMouseSettings.cs

@@ -0,0 +1,58 @@
+// 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 MouseUtils.UITests
+{
+    public class FindMyMouseSettings
+    {
+        // Appearance settings
+        public string OverlayOpacity { get; set; }
+
+        public string Radius { get; set; }
+
+        public string InitialZoom { get; set; }
+
+        public string AnimationDuration { get; set; }
+
+        // Color settings
+        public string BackgroundColor { get; set; }
+
+        public string SpotlightColor { get; set; }
+
+        // Activation method settings
+        public enum ActivationMethod
+        {
+            PressLeftControlTwice,
+            PressRightControlTwice,
+            ShakeMouse,
+            CustomShortcut,
+        }
+
+        public ActivationMethod SelectedActivationMethod { get; set; }
+
+        // Optional constructor to initialize properties
+        public FindMyMouseSettings(
+            string overlayOpacity = "",
+            string radius = "",
+            string initialZoom = "",
+            string animationDuration = "",
+            string backgroundColor = "",
+            string spotlightColor = "")
+        {
+            OverlayOpacity = overlayOpacity;
+            Radius = radius;
+            InitialZoom = initialZoom;
+            AnimationDuration = animationDuration;
+            BackgroundColor = backgroundColor;
+            SpotlightColor = spotlightColor;
+            SelectedActivationMethod = ActivationMethod.PressLeftControlTwice; // Default value
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/util/MouseHighlighterSettings.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;
+
+namespace MouseUtils.UITests
+{
+    public class MouseHighlighterSettings
+    {
+        // Appearance settings
+        public string Radius { get; set; }
+
+        public string FadeDelay { get; set; }
+
+        public string FadeDuration { get; set; }
+
+        // Color settings
+        public string PrimaryButtonHighlightColor { get; set; }
+
+        public string SecondaryButtonHighlightColor { get; set; }
+
+        public string AlwaysHighlightColor { get; set; }
+
+        // Settings UI Elements
+        public enum SettingsUIElements
+        {
+            PrimaryButtonHighlightColorGroup,
+            SecondaryButtonHighlightColorGroup,
+            AlwaysHighlightColorGroup,
+            RadiusEdit,
+            FadeDelayEdit,
+            FadeDurationEdit,
+        }
+
+        private Dictionary<SettingsUIElements, string> ElementNameMap { get; }
+
+        // Optional constructor to initialize properties
+        public MouseHighlighterSettings(
+            string radius = "",
+            string fadeDelay = "",
+            string fadeDuration = "",
+            string primaryButtonHighlightColor = "",
+            string secondaryButtonHighlightColor = "",
+            string alwaysHighlightColor = "")
+        {
+            this.Radius = radius;
+            this.FadeDelay = fadeDelay;
+            this.FadeDuration = fadeDuration;
+            this.PrimaryButtonHighlightColor = primaryButtonHighlightColor;
+            this.SecondaryButtonHighlightColor = secondaryButtonHighlightColor;
+            this.AlwaysHighlightColor = alwaysHighlightColor;
+
+            ElementNameMap = new Dictionary<SettingsUIElements, string>
+            {
+                [SettingsUIElements.PrimaryButtonHighlightColorGroup] = @"Primary button highlight color",
+                [SettingsUIElements.SecondaryButtonHighlightColorGroup] = @"Secondary button highlight color",
+                [SettingsUIElements.AlwaysHighlightColorGroup] = @"Always highlight color",
+                [SettingsUIElements.RadiusEdit] = @"Radius (px) Minimum5",
+                [SettingsUIElements.FadeDelayEdit] = @"Fade delay (ms) Minimum0",
+                [SettingsUIElements.FadeDurationEdit] = @"Fade duration (ms) Minimum0",
+            };
+        }
+
+        public string GetElementName(SettingsUIElements element)
+        {
+            return ElementNameMap[element];
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/util/MousePointerCrosshairsSettings.cs

@@ -0,0 +1,86 @@
+// 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 Windows.UI.Core.AnimationMetrics;
+
+namespace MouseUtils.UITests
+{
+    public class MousePointerCrosshairsSettings
+    {
+        // Appearance settings
+        public string Opacity { get; set; }
+
+        public string CenterRadius { get; set; }
+
+        public string Thickness { get; set; }
+
+        public string BorderSize { get; set; }
+
+        public bool IsFixLength { get; set; }
+
+        public string FixedLength { get; set; }
+
+        // Color settings
+        public string CrosshairsColor { get; set; }
+
+        public string CrosshairsBorderColor { get; set; }
+
+        // Settings UI Elements
+        public enum SettingsUIElements
+        {
+            CrosshairsColorGroup,
+            CrosshairsBorderColorGroup,
+            OpacitySlider,
+            CenterRadiusEdit,
+            ThicknessEdit,
+            BorderSizeEdit,
+            FixedLengthEdit,
+            IsFixLengthToggle,
+        }
+
+        private Dictionary<SettingsUIElements, string> ElementNameMap { get; }
+
+        // Optional constructor to initialize properties
+        public MousePointerCrosshairsSettings(
+            string opacity = "",
+            string centerRadius = "",
+            string thickness = "",
+            string borderSize = "",
+            bool isFixLength = false,
+            string fixedLength = "",
+            string crosshairsColor = "",
+            string crosshairsBorderColor = "")
+        {
+            this.Opacity = opacity;
+            this.CenterRadius = centerRadius;
+            this.Thickness = thickness;
+            this.BorderSize = borderSize;
+            this.IsFixLength = isFixLength;
+            this.FixedLength = fixedLength;
+            this.CrosshairsColor = crosshairsColor;
+            this.CrosshairsBorderColor = crosshairsBorderColor;
+            ElementNameMap = new Dictionary<SettingsUIElements, string>
+            {
+                [SettingsUIElements.CrosshairsColorGroup] = @"Crosshairs color",
+                [SettingsUIElements.CrosshairsBorderColorGroup] = @"Crosshairs border color",
+                [SettingsUIElements.OpacitySlider] = @"Crosshairs opacity (%)",
+                [SettingsUIElements.CenterRadiusEdit] = @"Crosshairs center radius (px) Minimum0 Maximum500",
+                [SettingsUIElements.ThicknessEdit] = @"Crosshairs thickness (px) Minimum1 Maximum50",
+                [SettingsUIElements.BorderSizeEdit] = @"Crosshairs border size (px) Minimum0 Maximum50",
+                [SettingsUIElements.FixedLengthEdit] = @"Crosshairs fixed length (px) Minimum1",
+                [SettingsUIElements.IsFixLengthToggle] = @"Fix crosshairs length",
+            };
+        }
+
+        public string GetElementName(SettingsUIElements element)
+        {
+            return ElementNameMap[element];
+        }
+    }
+}

src/modules/MouseUtils/MouseUtils.UITests/util/MouseUtilsSettings.cs

@@ -0,0 +1,59 @@
+// Copyright (c) Microsoft Corporation
+// The Microsoft Corporation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+using System.Collections.Generic;
+using Microsoft.PowerToys.UITest;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using OpenQA.Selenium.Appium.Windows;
+
+namespace MouseUtils.UITests
+{
+    public class MouseUtilsSettings
+    {
+        // Mouse Utils Modules
+        public enum MouseUtils
+        {
+            MouseHighlighter,
+            FindMyMouse,
+            MousePointerCrosshairs,
+            MouseJump,
+        }
+
+        private static readonly Dictionary<MouseUtils, string> MouseUtilUINameMap = new()
+        {
+            [MouseUtils.MouseHighlighter] = @"Mouse Highlighter",
+            [MouseUtils.FindMyMouse] = @"Find My Mouse",
+            [MouseUtils.MousePointerCrosshairs] = @"Mouse Pointer Crosshairs",
+            [MouseUtils.MouseJump] = @"Mouse Jump",
+        };
+
+        private static readonly Dictionary<MouseUtils, string> MouseUtilUIToggleMap = new()
+        {
+            [MouseUtils.MouseHighlighter] = @"Enable Mouse Highlighter",
+            [MouseUtils.FindMyMouse] = @"Enable Find My Mouse",
+            [MouseUtils.MousePointerCrosshairs] = @"Enable Mouse Pointer Crosshairs",
+            [MouseUtils.MouseJump] = @"Enable Mouse Jump",
+        };
+
+        public static string GetMouseUtilUIName(MouseUtils element)
+        {
+            return MouseUtilUINameMap[element];
+        }
+
+        public static void SetMouseUtilEnabled(Custom? custom, MouseUtils element, bool isEnable = true)
+        {
+            if (custom != null)
+            {
+                string toggleName = MouseUtilUIToggleMap[element];
+                var toggle = custom.Find<ToggleSwitch>(toggleName);
+
+                toggle.Toggle(isEnable);
+            }
+            else
+            {
+                Assert.Fail(element + " custom not found.");
+            }
+        }
+    }
+}

src/modules/fancyzones/UITests-FancyZones/OneZoneSwitchTests.cs

@@ -0,0 +1,221 @@
+// 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.Data.Common;
+using System.Diagnostics;
+using System.DirectoryServices;
+using System.Globalization;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
+using System.Windows.Forms;
+using System.Xml.Linq;
+using FancyZonesEditor.Models;
+using FancyZonesEditorCommon.Data;
+using Microsoft.FancyZones.UITests.Utils;
+using Microsoft.FancyZonesEditor.UITests.Utils;
+using Microsoft.FancyZonesEditor.UnitTests.Utils;
+using Microsoft.PowerToys.UITest;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using static Microsoft.FancyZonesEditor.UnitTests.Utils.FancyZonesEditorHelper;
+
+namespace UITests_FancyZones
+{
+    [TestClass]
+    public class OneZoneSwitchTests : UITestBase
+    {
+        private static readonly string WindowName = "Windows (C:) - File Explorer"; // set launch explorer window name
+        private static readonly string PowertoysWindowName = "PowerToys Settings"; // set powertoys settings window name
+        private static readonly int SubZones = 2;
+        private static readonly IOTestHelper AppZoneHistory = new FancyZonesEditorFiles().AppZoneHistoryIOHelper;
+
+        public OneZoneSwitchTests()
+            : base(PowerToysModule.PowerToysSettings, WindowSize.Medium)
+        {
+        }
+
+        [TestInitialize]
+        public void TestInitialize()
+        {
+            // set a custom layout with 2 subzones
+            CustomLayouts customLayouts = new CustomLayouts();
+            CustomLayouts.CustomLayoutListWrapper customLayoutListWrapper = CustomLayoutsList;
+            Files.CustomLayoutsIOHelper.WriteData(customLayouts.Serialize(customLayoutListWrapper));
+
+            // clear the app zone history
+            AppZoneHistory.DeleteFile();
+
+            // Goto Hosts File Editor setting page
+            if (this.FindAll<NavigationViewItem>("FancyZones").Count == 0)
+            {
+                // Expand Advanced list-group if needed
+                this.Find<NavigationViewItem>("Windowing & Layouts").Click();
+            }
+
+            this.Find<NavigationViewItem>("FancyZones").Click();
+            this.Find<ToggleSwitch>("Enable FancyZones").Toggle(true);
+            this.Session.SetMainWindowSize(WindowSize.Large_Vertical);
+
+            int tries = 5;
+            Pull(tries, "down"); // Pull the setting page up to make sure the setting is visible
+            bool switchWindowEnable = TestContext.TestName == "TestSwitchShortCutDisable" ? false : true;
+
+            this.Find<ToggleSwitch>("Switch between windows in the current zone").Toggle(switchWindowEnable);
+            Task.Delay(500).Wait(); // Wait for the setting to be applied
+            Pull(tries, "up"); // Pull the setting page down to make sure the setting is visible
+            this.Find<Microsoft.PowerToys.UITest.Button>("Launch layout editor").Click();
+
+            Task.Delay(1000).Wait();
+            this.Session.Attach(PowerToysModule.FancyZone);
+            this.Find<Element>(By.Name("Custom Column")).Click();
+            this.Find<Microsoft.PowerToys.UITest.Button>("Close").Click();
+            this.RestartScopeExe();
+        }
+
+        [TestMethod]
+        public void TestSwitchWindow()
+        {
+            SnapToOneZone();
+
+            string? activeWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreEqual(PowertoysWindowName, activeWindowTitle);
+
+            // switch to the previous window by shortcut win+page down
+            SendKeys(Key.Win, Key.PageDown);
+
+            activeWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreEqual(WindowName, activeWindowTitle);
+
+            // Clean settings
+            Clean();
+        }
+
+        [TestMethod]
+        public void TestSwitchAfterDesktopChange()
+        {
+            SnapToOneZone();
+
+            string? windowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreEqual(PowertoysWindowName, windowTitle);
+
+            // Add virtual desktop
+            SendKeys(Key.Ctrl, Key.Win, Key.D);
+            string? switchWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle(); // Fixed variable name to start with lower-case letter and removed unnecessary assignment warning by using the variable meaningfully.
+
+            // return back
+            SendKeys(Key.Ctrl, Key.Win, Key.Left);
+            Task.Delay(500).Wait(); // Optional: Wait for a moment to ensure window switch
+            string? returnWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreEqual(PowertoysWindowName, returnWindowTitle);
+
+            // check shortcut
+            SendKeys(Key.Win, Key.PageDown);
+            string? activeWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreEqual(WindowName, activeWindowTitle);
+
+            // close the virtual desktop
+            SendKeys(Key.Ctrl, Key.Win, Key.Right);
+            Task.Delay(500).Wait(); // Optional: Wait for a moment to ensure window switch
+            SendKeys(Key.Ctrl, Key.Win, Key.F4);
+            Task.Delay(500).Wait(); // Optional: Wait for a moment to ensure window switch
+
+            // Clean settings
+            Clean();
+        }
+
+        [TestMethod]
+        public void TestSwitchShortCutDisable()
+        {
+            SnapToOneZone();
+
+            string? activeWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreEqual(PowertoysWindowName, activeWindowTitle);
+
+            // switch to the previous window by shortcut win+page down
+            SendKeys(Key.Win, Key.PageDown);
+            Task.Delay(500).Wait(); // Optional: Wait for a moment to ensure window switch
+
+            activeWindowTitle = ZoneSwitchHelper.GetActiveWindowTitle();
+            Assert.AreNotEqual(WindowName, activeWindowTitle);
+
+            // Clean Setting
+            Clean();
+        }
+
+        private void SnapToOneZone()
+        {
+            // Set drag position of target zone
+            int screenWidth = Screen.PrimaryScreen?.Bounds.Width ?? 1920;  // default 1920
+            int screenHeight = Screen.PrimaryScreen?.Bounds.Height ?? 1080;
+
+            int targetX = screenWidth / SubZones / 3;
+            int targetY = screenWidth / SubZones / 2;
+
+            // assert the AppZoneHistory layout is set
+            Session.KillAllProcessesByName("explorer");
+            Session.StartExe("explorer.exe", "C:\\");
+
+            // Start Windows Explorer process
+            Session.Attach(WindowName, WindowSize.UnSpecified); // display window1
+            var tabView = Find<Element>(By.AccessibilityId("TabView"));
+            tabView.DoubleClick(); // maximize the window
+            tabView.KeyDownAndDrag(Key.Shift, targetX, targetY);
+
+            // Attach the PowerToys settings window to the front
+            Session.Attach(PowertoysWindowName, WindowSize.UnSpecified);
+            string name = "Non Client Input Sink Window";
+            Element settingsView = Find<Element>(By.Name(name));
+            settingsView.DoubleClick(); // maximize the window
+            settingsView.KeyDownAndDrag(Key.Shift, targetX, targetY);
+
+            // check the AppZoneHistory layout is set and in the same zone
+            string appZoneHistoryJson = AppZoneHistory.GetData();
+            Console.WriteLine($"{ZoneSwitchHelper.GetZoneIndexSetByAppName(WindowName, appZoneHistoryJson)},{ZoneSwitchHelper.GetZoneIndexSetByAppName(WindowName, appZoneHistoryJson)}");
+            Assert.AreEqual(
+                ZoneSwitchHelper.GetZoneIndexSetByAppName(WindowName, appZoneHistoryJson),
+                ZoneSwitchHelper.GetZoneIndexSetByAppName(PowertoysWindowName, appZoneHistoryJson));
+        }
+
+        private static readonly CustomLayouts.CustomLayoutListWrapper CustomLayoutsList = new CustomLayouts.CustomLayoutListWrapper
+        {
+            CustomLayouts = new List<CustomLayouts.CustomLayoutWrapper>
+            {
+                new CustomLayouts.CustomLayoutWrapper
+                {
+                    Uuid = "{63F09977-D327-4DAC-98F4-0C886CAE9517}",
+                    Type = CustomLayout.Grid.TypeToString(),
+                    Name = "Custom Column",
+                    Info = new CustomLayouts().ToJsonElement(new CustomLayouts.GridInfoWrapper
+                    {
+                        Rows = 1,
+                        Columns = SubZones,
+                        RowsPercentage = new List<int> { 10000 },
+                        ColumnsPercentage = new List<int> { 5000, 5000 },
+                        CellChildMap = new int[][] { [0, 1] },
+                        SensitivityRadius = 20,
+                        ShowSpacing = true,
+                        Spacing = 10,
+                    }),
+                },
+            },
+        };
+
+        private void Pull(int tries = 5, string direction = "up")
+        {
+            Key keyToSend = direction == "up" ? Key.Up : Key.Down;
+            for (int i = 0; i < tries; i++)
+            {
+               SendKeys(keyToSend);
+            }
+        }
+
+        private void Clean()
+        {
+            Session.KillAllProcessesByName("explorer");
+        }
+    }
+}

src/modules/fancyzones/UITests-FancyZones/UITests-FancyZones.csproj

@@ -25,4 +25,11 @@
   <ItemGroup>
     <Folder Include="Properties\" />
   </ItemGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="..\editor\FancyZonesEditor\FancyZonesEditor.csproj" />
+	<ProjectReference Include="..\UITests-FancyZonesEditor\UITests-FancyZonesEditor.csproj" />
+	<ProjectReference Include="..\..\..\common\UITestAutomation\UITestAutomation.csproj" />
+  </ItemGroup>
+
 </Project>

src/modules/fancyzones/UITests-FancyZones/Utils/ZoneSwitchHelper.cs

@@ -0,0 +1,84 @@
+// 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.Diagnostics;
+using System.Linq;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Text.Json;
+using System.Threading.Tasks;
+
+namespace Microsoft.FancyZones.UITests.Utils
+{
+    public class ZoneSwitchHelper
+    {
+        public static void LaunchExplorer(string path)
+        {
+            var explorerProcessInfo = new ProcessStartInfo
+            {
+                FileName = "explorer.exe",
+                Arguments = path,
+            };
+
+            Process.Start(explorerProcessInfo);
+            Task.Delay(2000).Wait(); // Wait for the Explorer window to fully launch
+        }
+
+        public static string? GetZoneIndexSetByAppName(string exeName, string json)
+        {
+            if (string.IsNullOrEmpty(exeName) || string.IsNullOrEmpty(json))
+            {
+                return null;
+            }
+
+            try
+            {
+                using var doc = JsonDocument.Parse(json);
+                var historyArray = doc.RootElement.GetProperty("app-zone-history");
+
+                foreach (var item in historyArray.EnumerateArray())
+                {
+                    if (item.TryGetProperty("app-path", out var appPathElement) &&
+                        appPathElement.GetString() is string path &&
+                        path.EndsWith(exeName, StringComparison.OrdinalIgnoreCase))
+                    {
+                        var history = item.GetProperty("history");
+                        if (history.GetArrayLength() > 0)
+                        {
+                            return history[0].GetProperty("zone-index-set")[0].GetString();
+                        }
+                    }
+                }
+            }
+            catch (JsonException ex)
+            {
+                throw new InvalidOperationException("JSON parse error: " + ex.Message, ex);
+            }
+
+            return null;
+        }
+
+        [DllImport("user32.dll")]
+        private static extern IntPtr GetForegroundWindow();
+
+        [DllImport("user32.dll", SetLastError = true, CharSet = CharSet.Unicode)]
+        private static extern int GetWindowText(IntPtr hWnd, StringBuilder lpString, int nMaxCount);
+
+        public static string? GetActiveWindowTitle()
+        {
+            const int nChars = 256;
+            StringBuilder buff = new StringBuilder(nChars);
+            IntPtr handle = GetForegroundWindow();
+
+            if (GetWindowText(handle, buff, nChars) > 0)
+            {
+                return buff.ToString();
+            }
+
+            return null;
+        }
+    }
+}

src/modules/fancyzones/UITests-FancyZonesEditor/ApplyLayoutTests.cs

@@ -135,6 +135,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             FancyZonesEditorHelper.Files.ParamsIOHelper.WriteData(editorParameters.Serialize(Parameters));
 
@@ -195,12 +196,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void ApplyCustomLayout()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/CopyLayoutTests.cs

@@ -76,6 +76,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             ParamsWrapper parameters = new ParamsWrapper
             {
@@ -172,12 +173,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void CopyTemplate_FromEditLayoutWindow()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/CreateLayoutTests.cs

@@ -23,6 +23,8 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
+
             // prepare test editor parameters with 2 monitors before launching the editor
             EditorParameters editorParameters = new EditorParameters();
             EditorParameters.ParamsWrapper parameters = new EditorParameters.ParamsWrapper
@@ -132,12 +134,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void CreateWithDefaultName()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/CustomLayoutsTests.cs

@@ -104,6 +104,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             CustomLayouts customLayouts = new CustomLayouts();
             FancyZonesEditorHelper.Files.CustomLayoutsIOHelper.WriteData(customLayouts.Serialize(Layouts));
 
@@ -208,12 +209,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void Name_Initialize()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/DefaultLayoutsTest.cs

@@ -121,6 +121,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             var defaultLayouts = new DefaultLayouts();
             FancyZonesEditorHelper.Files.DefaultLayoutsIOHelper.WriteData(defaultLayouts.Serialize(Layouts));
 
@@ -237,12 +238,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void Initialize()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/DeleteLayoutTests.cs

@@ -142,6 +142,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             FancyZonesEditorHelper.Files.ParamsIOHelper.WriteData(editorParameters.Serialize(Parameters));
 
@@ -215,19 +216,13 @@ namespace Microsoft.FancyZonesEditor.UITests
             Session.Find<Element>(CustomLayouts.CustomLayouts[0].Name).Click();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void DeleteNotAppliedLayout()
         {
             var deletedLayout = CustomLayouts.CustomLayouts[1].Name;
             Session.Find<Element>(deletedLayout).Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.EditLayoutButton)).Click();
             Session.Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.DeleteLayoutButton)).Click();
-            Session.KeyboardAction(Keys.Tab, Keys.Enter);
+            Session.SendKeySequence(Key.Tab, Key.Enter);
 
             // verify the layout is removed
             Assert.IsTrue(Session.FindAll<Element>(deletedLayout).Count == 0);
@@ -245,7 +240,8 @@ namespace Microsoft.FancyZonesEditor.UITests
             var deletedLayout = CustomLayouts.CustomLayouts[0].Name;
             Session.Find<Element>(deletedLayout).Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.EditLayoutButton)).Click();
             Session.Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.DeleteLayoutButton)).Click();
-            Session.KeyboardAction(Keys.Tab, Keys.Enter);
+
+            Session.SendKeySequence(Key.Tab, Key.Enter);
 
             // verify the layout is removed
             Assert.IsTrue(Session.FindAll<Element>(deletedLayout).Count == 0);
@@ -270,7 +266,7 @@ namespace Microsoft.FancyZonesEditor.UITests
             var deletedLayout = CustomLayouts.CustomLayouts[1].Name;
             Session.Find<Element>(deletedLayout).Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.EditLayoutButton)).Click();
             Session.Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.DeleteLayoutButton)).Click();
-            Session.KeyboardAction(Keys.Tab, Keys.Tab, Keys.Enter);
+            Session.SendKeySequence(Key.Tab, Key.Tab, Key.Enter);
 
             // verify the layout is not removed
             Assert.IsNotNull(Session.Find<Element>(deletedLayout));
@@ -287,7 +283,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         {
             var deletedLayout = CustomLayouts.CustomLayouts[1].Name;
             FancyZonesEditorHelper.ClickContextMenuItem(Session, deletedLayout, FancyZonesEditorHelper.ElementName.Delete);
-            Session.KeyboardAction(Keys.Tab, Keys.Enter);
+            Session.SendKeySequence(Key.Tab, Key.Enter);
 
             // verify the layout is removed
             Assert.IsTrue(Session.FindAll<Element>(deletedLayout).Count == 0);
@@ -304,7 +300,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         {
             var deletedLayout = CustomLayouts.CustomLayouts[1].Name;
             FancyZonesEditorHelper.ClickContextMenuItem(Session, deletedLayout, FancyZonesEditorHelper.ElementName.Delete);
-            Session.KeyboardAction(Keys.Tab, Keys.Enter);
+            Session.SendKeySequence(Key.Tab, Key.Enter);
 
             // verify the default layout is reset to the "default" default
             Session.Find<Element>(TestConstants.TemplateLayoutNames[LayoutType.PriorityGrid]).Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.EditLayoutButton)).Click();
@@ -323,7 +319,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         {
             var deletedLayout = CustomLayouts.CustomLayouts[1].Name;
             FancyZonesEditorHelper.ClickContextMenuItem(Session, deletedLayout, FancyZonesEditorHelper.ElementName.Delete);
-            Session.KeyboardAction(Keys.Tab, Keys.Enter);
+            Session.SendKeySequence(Key.Tab, Key.Enter);
 
             // verify the hotkey is available
             Session.Find<Element>(CustomLayouts.CustomLayouts[0].Name).Find<Button>(PowerToys.UITest.By.AccessibilityId(AccessibilityId.EditLayoutButton)).Click();

src/modules/fancyzones/UITests-FancyZonesEditor/EditLayoutTests.cs

@@ -105,6 +105,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             ParamsWrapper parameters = new ParamsWrapper
             {
@@ -209,12 +210,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void OpenEditMode()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/FirstLunchTest.cs

@@ -32,6 +32,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             ParamsWrapper parameters = new ParamsWrapper
             {
@@ -141,12 +142,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void FirstLaunch()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/LayoutHotkeysTests.cs

@@ -106,6 +106,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             ParamsWrapper parameters = new ParamsWrapper
             {
@@ -206,12 +207,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void Initialize()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/TemplateLayoutsTests.cs

@@ -80,6 +80,7 @@ namespace Microsoft.FancyZonesEditor.UITests
         [TestInitialize]
         public void TestInitialize()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             ParamsWrapper parameters = new ParamsWrapper
             {
@@ -191,12 +192,6 @@ namespace Microsoft.FancyZonesEditor.UITests
             this.RestartScopeExe();
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void ZoneNumber_Cancel()
         {

src/modules/fancyzones/UITests-FancyZonesEditor/UIInitializeTest.cs

@@ -27,12 +27,6 @@ namespace Microsoft.FancyZonesEditor.UITests
         {
         }
 
-        [TestCleanup]
-        public void TestCleanup()
-        {
-            FancyZonesEditorHelper.Files.Restore();
-        }
-
         [TestMethod]
         public void EditorParams_VerifySelectedMonitor()
         {
@@ -737,6 +731,7 @@ namespace Microsoft.FancyZonesEditor.UITests
 
         private void InitFileData()
         {
+            FancyZonesEditorHelper.Files.Restore();
             EditorParameters editorParameters = new EditorParameters();
             ParamsWrapper parameters = new ParamsWrapper
             {

src/modules/fancyzones/UITests-FancyZonesEditor/Utils/AppZoneHistory.cs

@@ -0,0 +1,78 @@
+// 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;
+using System.Text.Json;
+using static FancyZonesEditorCommon.Data.AppZoneHistory.AppZoneHistoryWrapper;
+using static FancyZonesEditorCommon.Data.CustomLayouts;
+
+namespace FancyZonesEditorCommon.Data
+{
+    public class AppZoneHistory : EditorData<AppZoneHistory.AppZoneHistoryListWrapper>
+    {
+        public string File
+        {
+            get
+            {
+                return GetDataFolder() + "\\Microsoft\\PowerToys\\FancyZones\\app-zone-history.json";
+            }
+        }
+
+        public struct AppZoneHistoryWrapper
+        {
+            public struct ZoneHistoryWrapper
+            {
+                public int[][] ZoneIndexSet { get; set; }
+
+                public DeviceIdWrapper Device { get; set; }
+
+                public string ZonesetUuid { get; set; }
+            }
+
+            public struct DeviceIdWrapper
+            {
+                public string Monitor { get; set; }
+
+                public string MonitorInstance { get; set; }
+
+                public int MonitorNumber { get; set; }
+
+                public string SerialNumber { get; set; }
+
+                public string VirtualDesktop { get; set; }
+            }
+
+            public string AppPath { get; set; }
+
+            public List<ZoneHistoryWrapper> History { get; set; }
+        }
+
+        public struct AppZoneHistoryListWrapper
+        {
+            public List<AppZoneHistoryWrapper> AppZoneHistory { get; set; }
+        }
+
+        public JsonElement ToJsonElement(ZoneHistoryWrapper info)
+        {
+            string json = JsonSerializer.Serialize(info, this.JsonOptions);
+            return JsonSerializer.Deserialize<JsonElement>(json);
+        }
+
+        public JsonElement ToJsonElement(DeviceIdWrapper info)
+        {
+            string json = JsonSerializer.Serialize(info, this.JsonOptions);
+            return JsonSerializer.Deserialize<JsonElement>(json);
+        }
+
+        public ZoneHistoryWrapper ZoneHistoryFromJsonElement(string json)
+        {
+            return JsonSerializer.Deserialize<ZoneHistoryWrapper>(json, this.JsonOptions);
+        }
+
+        public DeviceIdWrapper GridFromJsonElement(string json)
+        {
+            return JsonSerializer.Deserialize<DeviceIdWrapper>(json, this.JsonOptions);
+        }
+    }
+}

src/modules/fancyzones/UITests-FancyZonesEditor/Utils/FancyZonesEditorFiles.cs

@@ -20,6 +20,8 @@ namespace Microsoft.FancyZonesEditor.UITests.Utils
 
         public IOTestHelper LayoutTemplatesIOHelper { get; }
 
+        public IOTestHelper AppZoneHistoryIOHelper { get; }
+
         public FancyZonesEditorFiles()
         {
             ParamsIOHelper = new IOTestHelper(new EditorParameters().File);
@@ -28,6 +30,7 @@ namespace Microsoft.FancyZonesEditor.UITests.Utils
             DefaultLayoutsIOHelper = new IOTestHelper(new DefaultLayouts().File);
             LayoutHotkeysIOHelper = new IOTestHelper(new LayoutHotkeys().File);
             LayoutTemplatesIOHelper = new IOTestHelper(new LayoutTemplates().File);
+            AppZoneHistoryIOHelper = new IOTestHelper(new AppZoneHistory().File);
         }
 
         public void Restore()
@@ -38,6 +41,7 @@ namespace Microsoft.FancyZonesEditor.UITests.Utils
             DefaultLayoutsIOHelper.RestoreData();
             LayoutHotkeysIOHelper.RestoreData();
             LayoutTemplatesIOHelper.RestoreData();
+            AppZoneHistoryIOHelper.RestoreData();
         }
     }
 }

src/modules/fancyzones/UITests-FancyZonesEditor/Utils/IOTestHelper.cs

@@ -70,6 +70,12 @@ namespace Microsoft.FancyZonesEditor.UITests.Utils
             }
         }
 
+        // For get app zone history data
+        public string GetData()
+        {
+            return ReadFile(_file);
+        }
+
         private string ReadFile(string fileName)
         {
             var attempts = 0;