From a8d32a20611650c6bf3af729e0cb4c9bcfb79f19 Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Mon, 18 Apr 2022 23:37:08 +0900
Subject: [PATCH] Add test coverage of notification resume flow

---
 .../TestSceneFirstRunSetupOverlay.cs          | 45 ++++++++++++++++---
 1 file changed, 40 insertions(+), 5 deletions(-)

diff --git a/osu.Game.Tests/Visual/UserInterface/TestSceneFirstRunSetupOverlay.cs b/osu.Game.Tests/Visual/UserInterface/TestSceneFirstRunSetupOverlay.cs
index dfe7ba0252..77a37e7420 100644
--- a/osu.Game.Tests/Visual/UserInterface/TestSceneFirstRunSetupOverlay.cs
+++ b/osu.Game.Tests/Visual/UserInterface/TestSceneFirstRunSetupOverlay.cs
@@ -12,6 +12,7 @@ using osu.Framework.Screens;
 using osu.Framework.Testing;
 using osu.Game.Overlays;
 using osu.Game.Overlays.FirstRunSetup;
+using osu.Game.Overlays.Notifications;
 using osu.Game.Screens;
 using osuTK;
 using osuTK.Input;
@@ -22,20 +23,34 @@ namespace osu.Game.Tests.Visual.UserInterface
     {
         private FirstRunSetupOverlay overlay;
 
-        private readonly Mock<IPerformFromScreenRunner> perfomer = new Mock<IPerformFromScreenRunner>();
+        private readonly Mock<IPerformFromScreenRunner> performer = new Mock<IPerformFromScreenRunner>();
+
+        private readonly Mock<INotificationOverlay> notificationOverlay = new Mock<INotificationOverlay>();
+
+        private Notification lastNotification;
 
         [BackgroundDependencyLoader]
         private void load()
         {
-            Dependencies.CacheAs(perfomer.Object);
-
-            perfomer.Setup(g => g.PerformFromScreen(It.IsAny<Action<IScreen>>(), It.IsAny<IEnumerable<Type>>()))
-                    .Callback((Action<IScreen> action, IEnumerable<Type> types) => action(null));
+            Dependencies.CacheAs(performer.Object);
+            Dependencies.CacheAs(notificationOverlay.Object);
         }
 
         [SetUpSteps]
         public void SetUpSteps()
         {
+            AddStep("setup dependencies", () =>
+            {
+                performer.Reset();
+                notificationOverlay.Reset();
+
+                performer.Setup(g => g.PerformFromScreen(It.IsAny<Action<IScreen>>(), It.IsAny<IEnumerable<Type>>()))
+                         .Callback((Action<IScreen> action, IEnumerable<Type> types) => action(null));
+
+                notificationOverlay.Setup(n => n.Post(It.IsAny<Notification>()))
+                                   .Callback((Notification n) => lastNotification = n);
+            });
+
             AddStep("add overlay", () =>
             {
                 Child = overlay = new FirstRunSetupOverlay
@@ -58,6 +73,8 @@ namespace osu.Game.Tests.Visual.UserInterface
 
             AddUntilStep("wait for screens removed", () => !overlay.ChildrenOfType<Screen>().Any());
 
+            AddStep("no notifications", () => notificationOverlay.VerifyNoOtherCalls());
+
             AddStep("display again on demand", () => overlay.Show());
 
             AddUntilStep("back at start", () => overlay.CurrentScreen is ScreenWelcome);
@@ -108,5 +125,23 @@ namespace osu.Game.Tests.Visual.UserInterface
 
             AddAssert("overlay dismissed", () => overlay.State.Value == Visibility.Hidden);
         }
+
+        [Test]
+        public void TestResumeViaNotification()
+        {
+            AddStep("step to next", () => overlay.NextButton.TriggerClick());
+
+            AddAssert("is at known screen", () => overlay.CurrentScreen is ScreenUIScale);
+
+            AddStep("hide", () => overlay.Hide());
+            AddAssert("overlay hidden", () => overlay.State.Value == Visibility.Hidden);
+
+            AddStep("notification arrived", () => notificationOverlay.Verify(n => n.Post(It.IsAny<Notification>()), Times.Once));
+
+            AddStep("run notification action", () => lastNotification.Activated());
+
+            AddAssert("overlay shown", () => overlay.State.Value == Visibility.Visible);
+            AddAssert("is resumed", () => overlay.CurrentScreen is ScreenUIScale);
+        }
     }
 }