osu-lazer/osu.Game.Tests/Visual/UserInterface/TestSceneButtonSystem.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

#nullable disable

using System;
using System.Linq;
using NUnit.Framework;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Colour;
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 partial class TestSceneButtonSystem : OsuManualInputManagerTestScene
    {
        private OsuLogo logo;
        private ButtonSystem buttons;

        [SetUp]
        public void SetUp() => Schedule(() =>
        {
            Children = new Drawable[]
            {
                new Box
                {
                    Colour = ColourInfo.GradientVertical(Color4.Gray, Color4.WhiteSmoke),
                    RelativeSizeAxes = Axes.Both,
                },
                buttons = new ButtonSystem(),
                logo = new OsuLogo
                {
                    RelativePositionAxes = Axes.Both,
                    Position = new Vector2(0.5f)
                }
            };

            buttons.SetOsuLogo(logo);
        });

        [Test]
        public void TestAllStates()
        {
            foreach (var s in Enum.GetValues(typeof(ButtonSystemState)).OfType<ButtonSystemState>().Skip(1))
                AddStep($"State to {s}", () => buttons.State = s);

            AddStep("Enter mode", performEnterMode);

            AddStep("Return to menu", () =>
            {
                buttons.State = ButtonSystemState.Play;
                buttons.FadeIn(MainMenu.FADE_IN_DURATION, Easing.OutQuint);
                buttons.MoveTo(new Vector2(0), MainMenu.FADE_IN_DURATION, Easing.OutQuint);
                logo.FadeColour(Color4.White, 100, Easing.OutQuint);
                logo.FadeIn(100, Easing.OutQuint);
            });
        }

        [Test]
        public void TestSmoothExit()
        {
            AddStep("Enter mode", performEnterMode);
        }

        [TestCase(Key.P, Key.P)]
        [TestCase(Key.M, Key.M, Key.L)]
        [TestCase(Key.M, Key.M, Key.M)]
        [TestCase(Key.L, Key.L)]
        [TestCase(Key.B, Key.E, Key.B)]
        [TestCase(Key.S, Key.E, Key.S)]
        [TestCase(Key.D)]
        [TestCase(Key.Q)]
        [TestCase(Key.O)]
        public void TestShortcutKeys(params Key[] keys)
        {
            int activationCount = -1;
            AddStep("set up action", () =>
            {
                activationCount = 0;
                void action() => activationCount++;

                switch (keys.First())
                {
                    case Key.P:
                        buttons.OnSolo = action;
                        break;

                    case Key.M:
                        buttons.OnMultiplayer = action;
                        break;

                    case Key.L:
                        buttons.OnPlaylists = action;
                        break;

                    case Key.B:
                        buttons.OnEditBeatmap = action;
                        break;

                    case Key.S:
                        buttons.OnEditSkin = action;
                        break;

                    case Key.D:
                        buttons.OnBeatmapListing = action;
                        break;

                    case Key.Q:
                        buttons.OnExit = _ => action();
                        break;

                    case Key.O:
                        buttons.OnSettings = action;
                        break;
                }
            });

            // trigger out of idle state
            AddStep($"press {keys.First()}", () => InputManager.Key(keys.First()));
            AddAssert("state is top level", () => buttons.State == ButtonSystemState.TopLevel);

            for (int i = 0; i < keys.Length; i++)
            {
                var key = keys[i];
                AddStep($"press {key}", () => InputManager.Key(key));

                if (i > 0)
                    AddAssert("state is not top menu", () => buttons.State != ButtonSystemState.TopLevel);
            }

            AddAssert("action triggered", () => activationCount == 1);
        }

        private void performEnterMode()
        {
            buttons.State = ButtonSystemState.EnteringMode;
            buttons.FadeOut(MainMenu.FADE_OUT_DURATION, Easing.InSine);
            buttons.MoveTo(new Vector2(-800, 0), MainMenu.FADE_OUT_DURATION, Easing.InSine);
            logo.FadeOut(300, Easing.InSine)
                .ScaleTo(0.2f, 300, Easing.InSine);
        }
    }
}