add tests

Author: corvinsz

tests/MaterialDesignThemes.UITests/Samples/DialogHost/WithMultipleTextBoxes.xaml

@@ -0,0 +1,28 @@
+<UserControl x:Class="MaterialDesignThemes.UITests.Samples.DialogHost.WithMultipleTextBoxes"
+             xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:local="clr-namespace:MaterialDesignThemes.UITests.Samples.DialogHost"
+             xmlns:materialDesign="http://materialdesigninxaml.net/winfx/xaml/themes"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             d:DesignHeight="450"
+             d:DesignWidth="800"
+             mc:Ignorable="d">
+  <materialDesign:DialogHost x:Name="SampleDialogHost" Loaded="DialogHost_Loaded">
+    <materialDesign:DialogHost.DialogContent>
+      <StackPanel Margin="16"
+                  HorizontalAlignment="Center"
+                  VerticalAlignment="Center">
+        <TextBox x:Name="TextBoxOne"
+                 MinWidth="200"
+                 Text="One" />
+        <TextBox x:Name="TextBoxTwo"
+                 MinWidth="200"
+                 Text="Two" />
+        <TextBox x:Name="TextBoxThree"
+                 MinWidth="200"
+                 Text="Three" />
+      </StackPanel>
+    </materialDesign:DialogHost.DialogContent>
+  </materialDesign:DialogHost>
+</UserControl>

tests/MaterialDesignThemes.UITests/Samples/DialogHost/WithMultipleTextBoxes.xaml.cs

@@ -0,0 +1,16 @@
+namespace MaterialDesignThemes.UITests.Samples.DialogHost;
+
+/// <summary>
+/// Interaction logic for WithMultipleTextBoxes.xaml
+/// </summary>
+public partial class WithMultipleTextBoxes : UserControl
+{
+    public WithMultipleTextBoxes()
+    {
+        InitializeComponent();
+    }
+    private void DialogHost_Loaded(object sender, RoutedEventArgs e)
+    {
+        SampleDialogHost.IsOpen = true;
+    }
+}

tests/MaterialDesignThemes.UITests/WPF/DialogHosts/DialogHostTests.cs

@@ -314,7 +314,7 @@ public async Task CornerRadius_AppliedToContentCoverBorder_WhenSetOnEmbeddedDial
         await Wait.For(async () =>
         {
             var contentCoverBorder = await dialogHost.GetElement<Border>("ContentCoverBorder");
-                
+
             await Assert.That((await contentCoverBorder.GetCornerRadius()).TopLeft).IsEqualTo(1);
             await Assert.That((await contentCoverBorder.GetCornerRadius()).TopRight).IsEqualTo(2);
             await Assert.That((await contentCoverBorder.GetCornerRadius()).BottomRight).IsEqualTo(3);
@@ -500,7 +500,7 @@ public async Task DialogHost_WithComboBox_CanSelectItem()
         var comboBox = await dialogHost.GetElement<ComboBox>("TargetedPlatformComboBox");
         await Task.Delay(500, TestContext.Current!.CancellationToken);
         await comboBox.LeftClick();
-        
+
         var item = await Wait.For(() => comboBox.GetElement<ComboBoxItem>("TargetItem"));
         await Task.Delay(TimeSpan.FromSeconds(1));
         await item.LeftClick();
@@ -514,4 +514,40 @@ await Wait.For(async () =>
 
         recorder.Success();
     }
+
+    [Test]
+    [Description("Issue 3434")]
+    [Arguments(WindowState.Minimized, WindowState.Maximized)]
+    [Arguments(WindowState.Minimized, WindowState.Normal)]
+    [Arguments(WindowState.Maximized, WindowState.Normal)]
+	public async Task DialogHost_WhenWindowStateChanges_FocusedElementStaysFocused(WindowState firstWindowState, WindowState secondWindowState)
+	{
+		await using var recorder = new TestRecorder(App);
+
+		var dialogHost = (await LoadUserControl<WithMultipleTextBoxes>()).As<DialogHost>();
+		await Task.Delay(400, TestContext.Current!.CancellationToken);
+
+        // Select the second TextBox
+        var tbTwo = await dialogHost.GetElement<TextBox>("TextBoxTwo");
+        await tbTwo.MoveKeyboardFocus();
+		await Assert.That(await tbTwo.GetIsFocused()).IsTrue();
+        
+		// First state
+		await dialogHost.RemoteExecute(SetStateOfParentWindow, firstWindowState);
+		await Task.Delay(400, TestContext.Current!.CancellationToken);
+        // Second state
+        await dialogHost.RemoteExecute(SetStateOfParentWindow, secondWindowState);
+		await Task.Delay(400, TestContext.Current!.CancellationToken);
+
+		// After changing state of the window the previously focused element should be focused again
+		await Assert.That(await tbTwo.GetIsFocused()).IsTrue();
+		recorder.Success();
+
+		static object SetStateOfParentWindow(DialogHost dialogHost, WindowState state)
+        {
+            var window = Window.GetWindow(dialogHost);
+            window.WindowState = state;
+            return null!;
+        }
+	}
 }