From 32d1d5a34a874e9c868bcf414ae1478c9e6b58bc Mon Sep 17 00:00:00 2001
From: OpenSauce <opensauce04@gmail.com>
Date: Wed, 11 Jan 2023 12:04:52 +0000
Subject: [PATCH] Added tests for new Taiko touch control schemes

---
 .../TestSceneDrumTouchInputArea.cs            | 44 ++++++++++---------
 .../UI/DrumTouchInputArea.cs                  |  6 ++-
 2 files changed, 28 insertions(+), 22 deletions(-)

diff --git a/osu.Game.Rulesets.Taiko.Tests/TestSceneDrumTouchInputArea.cs b/osu.Game.Rulesets.Taiko.Tests/TestSceneDrumTouchInputArea.cs
index ee780204a2..6b02b0078a 100644
--- a/osu.Game.Rulesets.Taiko.Tests/TestSceneDrumTouchInputArea.cs
+++ b/osu.Game.Rulesets.Taiko.Tests/TestSceneDrumTouchInputArea.cs
@@ -3,7 +3,7 @@
 
 using NUnit.Framework;
 using osu.Framework.Graphics;
-using osu.Framework.Testing;
+using osu.Game.Rulesets.Taiko.Configuration;
 using osu.Game.Rulesets.Taiko.UI;
 using osu.Game.Tests.Visual;
 
@@ -14,35 +14,37 @@ namespace osu.Game.Rulesets.Taiko.Tests
     {
         private DrumTouchInputArea drumTouchInputArea = null!;
 
-        [SetUpSteps]
-        public void SetUpSteps()
+        private void createDrum(TaikoTouchControlScheme _forcedControlScheme)
         {
-            AddStep("create drum", () =>
+            Child = new TaikoInputManager(new TaikoRuleset().RulesetInfo)
             {
-                Child = new TaikoInputManager(new TaikoRuleset().RulesetInfo)
-                {
-                    RelativeSizeAxes = Axes.Both,
-                    Children = new Drawable[]
+                RelativeSizeAxes = Axes.Both,
+                Children = new Drawable[]
+    {
+                    new InputDrum
                     {
-                        new InputDrum
-                        {
-                            Anchor = Anchor.TopCentre,
-                            Origin = Anchor.TopCentre,
-                            Height = 0.2f,
-                        },
-                        drumTouchInputArea = new DrumTouchInputArea
-                        {
-                            Anchor = Anchor.BottomCentre,
-                            Origin = Anchor.BottomCentre,
-                        },
+                        Anchor = Anchor.TopCentre,
+                        Origin = Anchor.TopCentre,
+                        Height = 0.2f,
                     },
-                };
-            });
+                    drumTouchInputArea = new DrumTouchInputArea
+                    {
+                        Anchor = Anchor.BottomCentre,
+                        Origin = Anchor.BottomCentre,
+                        ForceControlScheme = _forcedControlScheme
+                    }
+                }
+            };
         }
 
         [Test]
         public void TestDrum()
         {
+            AddStep("create drum (kddk)", () => createDrum(TaikoTouchControlScheme.KDDK));
+            AddStep("show drum", () => drumTouchInputArea.Show());
+            AddStep("create drum (ddkk)", () => createDrum(TaikoTouchControlScheme.DDKK));
+            AddStep("show drum", () => drumTouchInputArea.Show());
+            AddStep("create drum (kkdd)", () => createDrum(TaikoTouchControlScheme.KKDD));
             AddStep("show drum", () => drumTouchInputArea.Show());
         }
 
diff --git a/osu.Game.Rulesets.Taiko/UI/DrumTouchInputArea.cs b/osu.Game.Rulesets.Taiko/UI/DrumTouchInputArea.cs
index b7a0d46d51..8208aaba39 100644
--- a/osu.Game.Rulesets.Taiko/UI/DrumTouchInputArea.cs
+++ b/osu.Game.Rulesets.Taiko/UI/DrumTouchInputArea.cs
@@ -24,6 +24,7 @@ namespace osu.Game.Rulesets.Taiko.UI
     /// </summary>
     public partial class DrumTouchInputArea : VisibilityContainer
     {
+        public TaikoTouchControlScheme? ForceControlScheme { get; set; }
         // visibility state affects our child. we always want to handle input.
         public override bool PropagatePositionalInputSubTree => true;
         public override bool PropagateNonPositionalInputSubTree => true;
@@ -55,7 +56,10 @@ namespace osu.Game.Rulesets.Taiko.UI
 
             const float centre_region = 0.80f;
 
-            config.BindWith(TaikoRulesetSetting.TouchControlScheme, configTouchControlScheme);
+            if (ForceControlScheme == null)
+                config.BindWith(TaikoRulesetSetting.TouchControlScheme, configTouchControlScheme);
+            else
+                configTouchControlScheme.Value = (TaikoTouchControlScheme)ForceControlScheme;
 
             Children = new Drawable[]
             {