From 9b3be2d44e4c60214c3d5f7b99fe33b62ebb75f4 Mon Sep 17 00:00:00 2001
From: rabbitism <popmessiah@hotmail.com>
Date: Wed, 13 Nov 2024 00:31:15 +0800
Subject: [PATCH 1/4] feat: add DialogFocusHint attached property.

---
 demo/Ursa.Demo/Dialogs/CustomDemoDialog.axaml | 29 +++++++++++++++----
 src/Ursa/AssemblyInfo.cs                      |  3 +-
 .../OverlayShared/OverlayDialogHost.Dialog.cs | 10 +++++--
 .../OverlayShared/OverlayDialogHost.Drawer.cs |  9 +++++-
 src/Ursa/Helpers/FocusHelper.cs               | 13 +++++++++
 5 files changed, 54 insertions(+), 10 deletions(-)
 create mode 100644 src/Ursa/Helpers/FocusHelper.cs

diff --git a/demo/Ursa.Demo/Dialogs/CustomDemoDialog.axaml b/demo/Ursa.Demo/Dialogs/CustomDemoDialog.axaml
index 3b38ce34..dc61a442 100644
--- a/demo/Ursa.Demo/Dialogs/CustomDemoDialog.axaml
+++ b/demo/Ursa.Demo/Dialogs/CustomDemoDialog.axaml
@@ -17,7 +17,10 @@
             <GradientStop Offset="0.9" Color="{DynamicResource SemiLightBlue1Color}" />
         </LinearGradientBrush>
     </UserControl.Background>
-    <Grid Margin="24" RowDefinitions="Auto, *, Auto" MinWidth="400">
+    <Grid
+        MinWidth="400"
+        Margin="24"
+        RowDefinitions="Auto, *, Auto">
         <TextBlock
             Grid.Row="0"
             Margin="8"
@@ -45,7 +48,10 @@
                 Grid.Column="1"
                 Margin="32,8,0,8"
                 Label="Owner">
-                <TextBox u:FormItem.Label="Owner" Text="{Binding Owner}" />
+                <TextBox
+                    u:FocusHelper.DialogFocusHint="True"
+                    u:FormItem.Label="Owner"
+                    Text="{Binding Owner}" />
             </u:FormItem>
             <u:FormItem
                 Grid.Row="1"
@@ -69,9 +75,20 @@
             HorizontalAlignment="Right"
             Orientation="Horizontal"
             Spacing="8">
-            <Button Command="{Binding DialogCommand}" Content="Dialog" Theme="{DynamicResource SolidButton}" />
-            <Button Command="{Binding OKCommand}" Content="OK" Theme="{DynamicResource SolidButton}" Classes="Tertiary"/>
-            <Button Command="{Binding CancelCommand}" Content="Cancel" Theme="{DynamicResource SolidButton}" Classes="Tertiary"/>
+            <Button
+                Command="{Binding DialogCommand}"
+                Content="Dialog"
+                Theme="{DynamicResource SolidButton}" />
+            <Button
+                Classes="Tertiary"
+                Command="{Binding OKCommand}"
+                Content="OK"
+                Theme="{DynamicResource SolidButton}" />
+            <Button
+                Classes="Tertiary"
+                Command="{Binding CancelCommand}"
+                Content="Cancel"
+                Theme="{DynamicResource SolidButton}" />
             <ComboBox>
                 <ComboBoxItem>A</ComboBoxItem>
                 <ComboBoxItem>B</ComboBoxItem>
@@ -79,5 +96,5 @@
             </ComboBox>
         </StackPanel>
     </Grid>
-    
+
 </UserControl>
diff --git a/src/Ursa/AssemblyInfo.cs b/src/Ursa/AssemblyInfo.cs
index 5fb4320d..a45e3de2 100644
--- a/src/Ursa/AssemblyInfo.cs
+++ b/src/Ursa/AssemblyInfo.cs
@@ -3,4 +3,5 @@
 [assembly:XmlnsPrefix("https://irihi.tech/ursa", "u")]
 [assembly:XmlnsDefinition("https://irihi.tech/ursa", "Ursa")]
 [assembly:XmlnsDefinition("https://irihi.tech/ursa", "Ursa.Controls")]
-[assembly:XmlnsDefinition("https://irihi.tech/ursa", "Ursa.Controls.Shapes")]
\ No newline at end of file
+[assembly:XmlnsDefinition("https://irihi.tech/ursa", "Ursa.Controls.Shapes")]
+[assembly:XmlnsDefinition("https://irihi.tech/ursa", "Ursa.Helpers")]
\ No newline at end of file
diff --git a/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs b/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs
index e2f2be3f..86d9c6ec 100644
--- a/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs
+++ b/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs
@@ -4,6 +4,7 @@
 using Irihi.Avalonia.Shared.Helpers;
 using Irihi.Avalonia.Shared.Shapes;
 using Ursa.Controls.OverlayShared;
+using Ursa.Helpers;
 
 namespace Ursa.Controls;
 
@@ -113,12 +114,17 @@ internal void AddModalDialog(DialogControlBase control)
         control.AddHandler(DialogControlBase.LayerChangedEvent, OnDialogLayerChanged);
         if (!IsAnimationDisabled) MaskAppearAnimation.RunAsync(mask);
 
-        var element = control.GetVisualDescendants().OfType<InputElement>().FirstOrDefault(a => a.Focusable);
+        var element = control.GetVisualDescendants().OfType<InputElement>()
+                             .FirstOrDefault(FocusHelper.GetDialogFocusHint);
+        if (element is null)
+        {
+            element = control.GetVisualDescendants().OfType<InputElement>().FirstOrDefault(a => a.Focusable);
+        }
         element?.Focus();
         _modalCount++;
         IsInModalStatus = _modalCount > 0;
         control.IsClosed = false;
-        control.Focus();
+        // control.Focus();
     }
 
     // Handle dialog layer change event
diff --git a/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Drawer.cs b/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Drawer.cs
index 438d8a92..3c3a35b6 100644
--- a/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Drawer.cs
+++ b/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Drawer.cs
@@ -9,6 +9,7 @@
 using Ursa.Common;
 using Ursa.Controls.OverlayShared;
 using Ursa.EventArgs;
+using Ursa.Helpers;
 
 namespace Ursa.Controls;
 
@@ -69,7 +70,13 @@ internal async void AddModalDrawer(DrawerControlBase control)
         {
             await Task.WhenAll(animation.RunAsync(control), MaskAppearAnimation.RunAsync(mask));
         }
-        var element = control.GetVisualDescendants().OfType<InputElement>().FirstOrDefault(a => a.Focusable);
+
+        var element = control.GetVisualDescendants().OfType<InputElement>()
+                             .FirstOrDefault(FocusHelper.GetDialogFocusHint);
+        if (element is null)
+        {
+            element = control.GetVisualDescendants().OfType<InputElement>().FirstOrDefault(a => a.Focusable);
+        }
         element?.Focus();
     }
 
diff --git a/src/Ursa/Helpers/FocusHelper.cs b/src/Ursa/Helpers/FocusHelper.cs
new file mode 100644
index 00000000..2139c810
--- /dev/null
+++ b/src/Ursa/Helpers/FocusHelper.cs
@@ -0,0 +1,13 @@
+using Avalonia;
+using Avalonia.Input;
+
+namespace Ursa.Helpers;
+
+public class FocusHelper
+{
+    public static readonly AttachedProperty<bool> DialogFocusHintProperty =
+        AvaloniaProperty.RegisterAttached<FocusHelper, InputElement, bool>("DialogFocusHint");
+
+    public static void SetDialogFocusHint(InputElement obj, bool value) => obj.SetValue(DialogFocusHintProperty, value);
+    public static bool GetDialogFocusHint(InputElement obj) => obj.GetValue(DialogFocusHintProperty);
+}
\ No newline at end of file

From 27a3e5fb41fd772f9cacfb67d45864f5f254ddc6 Mon Sep 17 00:00:00 2001
From: rabbitism <popmessiah@hotmail.com>
Date: Wed, 13 Nov 2024 00:34:21 +0800
Subject: [PATCH 2/4] feat: revert mask animation for dialog.

---
 src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs b/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs
index 86d9c6ec..5a8a4f28 100644
--- a/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs
+++ b/src/Ursa/Controls/OverlayShared/OverlayDialogHost.Dialog.cs
@@ -112,6 +112,7 @@ internal void AddModalDialog(DialogControlBase control)
         SetToPosition(control);
         control.AddHandler(OverlayFeedbackElement.ClosedEvent, OnDialogControlClosing);
         control.AddHandler(DialogControlBase.LayerChangedEvent, OnDialogLayerChanged);
+        // Notice: mask animation here is not really awaited, because currently dialogs appears immediately.
         if (!IsAnimationDisabled) MaskAppearAnimation.RunAsync(mask);
 
         var element = control.GetVisualDescendants().OfType<InputElement>()

From 05cc27e3080a24812ae4f993eaed8eb441d1dc9b Mon Sep 17 00:00:00 2001
From: rabbitism <popmessiah@hotmail.com>
Date: Wed, 13 Nov 2024 00:58:18 +0800
Subject: [PATCH 3/4] feat: add headless tests.

---
 .../Dialog_Primary_Focus/FocusDialog.axaml    | 10 +++++
 .../Dialog_Primary_Focus/FocusDialog.axaml.cs | 13 ++++++
 .../Dialog_Primary_Focus/NormalDialog.axaml   |  9 +++++
 .../NormalDialog.axaml.cs                     | 13 ++++++
 .../Dialog_Primary_Focus/Test.cs              | 40 +++++++++++++++++++
 .../Dialog_Primary_Focus/TestWindow.axaml     | 11 +++++
 .../Dialog_Primary_Focus/TestWindow.axaml.cs  | 23 +++++++++++
 7 files changed, 119 insertions(+)
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml.cs
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml.cs
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml
 create mode 100644 tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs

diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml
new file mode 100644
index 00000000..e1903bdb
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml
@@ -0,0 +1,10 @@
+<UserControl xmlns="https://github.com/avaloniaui"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             xmlns:u="https://irihi.tech/ursa"
+             mc:Ignorable="d" d:DesignWidth="800"
+             d:DesignHeight="450"
+             x:Class="HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus.FocusDialog">
+    <TextBox Text="Hello" u:FocusHelper.DialogFocusHint="True"></TextBox>
+</UserControl>
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml.cs b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml.cs
new file mode 100644
index 00000000..fc70acdb
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/FocusDialog.axaml.cs
@@ -0,0 +1,13 @@
+using Avalonia;
+using Avalonia.Controls;
+using Avalonia.Markup.Xaml;
+
+namespace HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus;
+
+public partial class FocusDialog : UserControl
+{
+    public FocusDialog()
+    {
+        InitializeComponent();
+    }
+}
\ No newline at end of file
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml
new file mode 100644
index 00000000..3d4c3e67
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml
@@ -0,0 +1,9 @@
+<UserControl xmlns="https://github.com/avaloniaui"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             mc:Ignorable="d" d:DesignWidth="800"
+             d:DesignHeight="450"
+             x:Class="HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus.NormalDialog">
+    <TextBox Text="Hello"></TextBox>
+</UserControl>
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml.cs b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml.cs
new file mode 100644
index 00000000..0d0025d4
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/NormalDialog.axaml.cs
@@ -0,0 +1,13 @@
+using Avalonia;
+using Avalonia.Controls;
+using Avalonia.Markup.Xaml;
+
+namespace HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus;
+
+public partial class NormalDialog : UserControl
+{
+    public NormalDialog()
+    {
+        InitializeComponent();
+    }
+}
\ No newline at end of file
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs
new file mode 100644
index 00000000..a5edd8f1
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs
@@ -0,0 +1,40 @@
+using Avalonia.Controls;
+using Avalonia.Headless.XUnit;
+using Avalonia.Threading;
+using Avalonia.VisualTree;
+using Ursa.Controls;
+
+namespace HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus;
+
+public class Test
+{
+    [AvaloniaFact]
+    public void Normal_Dialog_Focus_On_Border()
+    {
+        var ursaWindow = new TestWindow();
+        ursaWindow.Show();
+        ursaWindow.InvokeNormalDialog();
+        Dispatcher.UIThread.RunJobs();
+        var dialog = ursaWindow.GetVisualDescendants().OfType<DefaultDialogControl>().FirstOrDefault();
+        Assert.NotNull(dialog);
+        var border = dialog.GetVisualDescendants().OfType<Border>().FirstOrDefault(a=>a.Name == "PART_Border");
+        var text = dialog.GetVisualDescendants().OfType<TextBox>().FirstOrDefault();
+        Assert.True(border?.IsFocused);
+        Assert.False(text?.IsFocused);
+    }
+    
+    [AvaloniaFact]
+    public void Focus_Dialog_Focus_On_Primary()
+    {
+        var ursaWindow = new TestWindow();
+        ursaWindow.Show();
+        ursaWindow.InvokeFocusDialog();
+        Dispatcher.UIThread.RunJobs();
+        var dialog = ursaWindow.GetVisualDescendants().OfType<DefaultDialogControl>().FirstOrDefault();
+        Assert.NotNull(dialog);
+        var border = dialog.GetVisualDescendants().OfType<Border>().FirstOrDefault(a=>a.Name == "PART_Border");
+        var text = dialog.GetVisualDescendants().OfType<TextBox>().FirstOrDefault();
+        Assert.False(border?.IsFocused);
+        Assert.True(text?.IsFocused);
+    }
+}
\ No newline at end of file
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml
new file mode 100644
index 00000000..855ec1dc
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml
@@ -0,0 +1,11 @@
+<u:UrsaWindow xmlns="https://github.com/avaloniaui"
+            xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+            xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+            xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+            xmlns:u="https://irihi.tech/ursa"
+            mc:Ignorable="d" d:DesignWidth="800"
+            d:DesignHeight="450"
+            x:Class="HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus.TestWindow"
+            Title="TestWindow">
+    Welcome to Avalonia!
+</u:UrsaWindow>
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs
new file mode 100644
index 00000000..90153ad6
--- /dev/null
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs
@@ -0,0 +1,23 @@
+using Ursa.Controls;
+
+namespace HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus;
+
+public partial class TestWindow : UrsaWindow
+{
+    public TestWindow()
+    {
+        InitializeComponent();
+    }
+
+    public void InvokeNormalDialog()
+    {
+        OverlayDialog.ShowModal<NormalDialog, object>("Hello World",
+            options: new OverlayDialogOptions { TopLevelHashCode = GetHashCode() });
+    }
+
+    public void InvokeFocusDialog()
+    {
+        OverlayDialog.ShowModal<FocusDialog, object>("Hello World",
+            options: new OverlayDialogOptions { TopLevelHashCode = GetHashCode() });
+    }
+}
\ No newline at end of file

From 9835fe0ba4e02cd0fd19cc5dae6fe58b13c1189d Mon Sep 17 00:00:00 2001
From: rabbitism <popmessiah@hotmail.com>
Date: Wed, 13 Nov 2024 01:11:59 +0800
Subject: [PATCH 4/4] feat: add drawer and dialog headless test.

---
 .../Dialog_Primary_Focus/Test.cs              | 38 ++++++++++++++++++-
 .../Dialog_Primary_Focus/TestWindow.axaml.cs  | 17 ++++++++-
 2 files changed, 51 insertions(+), 4 deletions(-)

diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs
index a5edd8f1..bc86188c 100644
--- a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/Test.cs
@@ -9,12 +9,45 @@ namespace HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus;
 public class Test
 {
     [AvaloniaFact]
-    public void Normal_Dialog_Focus_On_Border()
+    public async Task Normal_Drawer_Focus_On_Border()
+    {
+        var ursaWindow = new TestWindow();
+        ursaWindow.Show();
+        ursaWindow.InvokeNormalDrawer();
+        Dispatcher.UIThread.RunJobs();
+        await Task.Delay(500);
+        var dialog = ursaWindow.GetVisualDescendants().OfType<DefaultDrawerControl>().FirstOrDefault();
+        Assert.NotNull(dialog);
+        var border = dialog.GetVisualDescendants().OfType<Border>().FirstOrDefault(a=>a.Name == "PART_Root");
+        var text = dialog.GetVisualDescendants().OfType<TextBox>().FirstOrDefault();
+        Assert.True(border?.IsFocused);
+        Assert.False(text?.IsFocused);
+    }
+    
+    [AvaloniaFact]
+    public async Task Focus_Drawer_Focus_On_Primary()
+    {
+        var ursaWindow = new TestWindow();
+        ursaWindow.Show();
+        ursaWindow.InvokeFocusDrawer();
+        Dispatcher.UIThread.RunJobs();
+        await Task.Delay(500);
+        var dialog = ursaWindow.GetVisualDescendants().OfType<DefaultDrawerControl>().FirstOrDefault();
+        Assert.NotNull(dialog);
+        var border = dialog.GetVisualDescendants().OfType<Border>().FirstOrDefault(a=>a.Name == "PART_Root");
+        var text = dialog.GetVisualDescendants().OfType<TextBox>().FirstOrDefault();
+        Assert.False(border?.IsFocused);
+        Assert.True(text?.IsFocused);
+    }
+    
+    [AvaloniaFact]
+    public async Task Normal_Dialog_Focus_On_Border()
     {
         var ursaWindow = new TestWindow();
         ursaWindow.Show();
         ursaWindow.InvokeNormalDialog();
         Dispatcher.UIThread.RunJobs();
+        await Task.Delay(100);
         var dialog = ursaWindow.GetVisualDescendants().OfType<DefaultDialogControl>().FirstOrDefault();
         Assert.NotNull(dialog);
         var border = dialog.GetVisualDescendants().OfType<Border>().FirstOrDefault(a=>a.Name == "PART_Border");
@@ -24,12 +57,13 @@ public void Normal_Dialog_Focus_On_Border()
     }
     
     [AvaloniaFact]
-    public void Focus_Dialog_Focus_On_Primary()
+    public async Task Focus_Dialog_Focus_On_Primary() 
     {
         var ursaWindow = new TestWindow();
         ursaWindow.Show();
         ursaWindow.InvokeFocusDialog();
         Dispatcher.UIThread.RunJobs();
+        await Task.Delay(100);
         var dialog = ursaWindow.GetVisualDescendants().OfType<DefaultDialogControl>().FirstOrDefault();
         Assert.NotNull(dialog);
         var border = dialog.GetVisualDescendants().OfType<Border>().FirstOrDefault(a=>a.Name == "PART_Border");
diff --git a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs
index 90153ad6..a8352125 100644
--- a/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs
+++ b/tests/HeadlessTest.Ursa/Controls/OverlayShared/Dialog_Primary_Focus/TestWindow.axaml.cs
@@ -1,4 +1,5 @@
 using Ursa.Controls;
+using Ursa.Controls.Options;
 
 namespace HeadlessTest.Ursa.Controls.OverlayShared.Dialog_Primary_Focus;
 
@@ -9,15 +10,27 @@ public TestWindow()
         InitializeComponent();
     }
 
+    public void InvokeNormalDrawer()
+    {
+        Drawer.ShowModal<NormalDialog, object>("Hello World",
+            options: new DrawerOptions() { TopLevelHashCode = GetHashCode() });
+    }
+
+    public void InvokeFocusDrawer()
+    {
+        Drawer.ShowModal<FocusDialog, object>("Hello World",
+            options: new DrawerOptions() { TopLevelHashCode = GetHashCode() });
+    }
+    
     public void InvokeNormalDialog()
     {
         OverlayDialog.ShowModal<NormalDialog, object>("Hello World",
-            options: new OverlayDialogOptions { TopLevelHashCode = GetHashCode() });
+            options: new OverlayDialogOptions() { TopLevelHashCode = GetHashCode() });
     }
 
     public void InvokeFocusDialog()
     {
         OverlayDialog.ShowModal<FocusDialog, object>("Hello World",
-            options: new OverlayDialogOptions { TopLevelHashCode = GetHashCode() });
+            options: new OverlayDialogOptions() { TopLevelHashCode = GetHashCode() });
     }
 }
\ No newline at end of file