From cdfef088464046ee468ca016579d8362441a8c3e Mon Sep 17 00:00:00 2001
From: Susko3 <Susko3@protonmail.com>
Date: Sat, 23 Apr 2022 18:44:50 +0200
Subject: [PATCH] Add tests for shortcut keys

---
 .../UserInterface/TestSceneButtonSystem.cs    | 63 ++++++++++++++++++-
 1 file changed, 62 insertions(+), 1 deletion(-)

diff --git a/osu.Game.Tests/Visual/UserInterface/TestSceneButtonSystem.cs b/osu.Game.Tests/Visual/UserInterface/TestSceneButtonSystem.cs
index 1bb5cadc6a..1a879e2e70 100644
--- a/osu.Game.Tests/Visual/UserInterface/TestSceneButtonSystem.cs
+++ b/osu.Game.Tests/Visual/UserInterface/TestSceneButtonSystem.cs
@@ -10,11 +10,12 @@ using osu.Framework.Graphics.Shapes;
 using osu.Game.Screens.Menu;
 using osuTK;
 using osuTK.Graphics;
+using osuTK.Input;
 
 namespace osu.Game.Tests.Visual.UserInterface
 {
     [TestFixture]
-    public class TestSceneButtonSystem : OsuTestScene
+    public class TestSceneButtonSystem : OsuManualInputManagerTestScene
     {
         private OsuLogo logo;
         private ButtonSystem buttons;
@@ -64,6 +65,66 @@ namespace osu.Game.Tests.Visual.UserInterface
             AddStep("Enter mode", performEnterMode);
         }
 
+        [TestCase(Key.P, true)]
+        [TestCase(Key.M, true)]
+        [TestCase(Key.L, true)]
+        [TestCase(Key.E, false)]
+        [TestCase(Key.D, false)]
+        [TestCase(Key.Q, false)]
+        [TestCase(Key.O, false)]
+        public void TestShortcutKeys(Key key, bool entersPlay)
+        {
+            int activationCount = -1;
+            AddStep("set up action", () =>
+            {
+                activationCount = 0;
+                void action() => activationCount++;
+
+                switch (key)
+                {
+                    case Key.P:
+                        buttons.OnSolo = action;
+                        break;
+
+                    case Key.M:
+                        buttons.OnMultiplayer = action;
+                        break;
+
+                    case Key.L:
+                        buttons.OnPlaylists = action;
+                        break;
+
+                    case Key.E:
+                        buttons.OnEdit = action;
+                        break;
+
+                    case Key.D:
+                        buttons.OnBeatmapListing = action;
+                        break;
+
+                    case Key.Q:
+                        buttons.OnExit = action;
+                        break;
+
+                    case Key.O:
+                        buttons.OnSettings = action;
+                        break;
+                }
+            });
+
+            AddStep($"press {key}", () => InputManager.Key(key));
+            AddAssert("state is top level", () => buttons.State == ButtonSystemState.TopLevel);
+
+            if (entersPlay)
+            {
+                AddStep("press P", () => InputManager.Key(Key.P));
+                AddAssert("state is play", () => buttons.State == ButtonSystemState.Play);
+            }
+
+            AddStep($"press {key}", () => InputManager.Key(key));
+            AddAssert("action triggered", () => activationCount == 1);
+        }
+
         private void performEnterMode()
         {
             buttons.State = ButtonSystemState.EnteringMode;