osukey/osu.Game.Tests/Visual/UserInterface/TestSceneModCustomizationSettings.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.

using System.Linq;
using osu.Framework.Allocation;
using osu.Framework.Bindables;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Game.Graphics.UserInterface;
using osu.Game.Overlays.Mods;
using osu.Game.Overlays.Settings;
using osu.Game.Rulesets.Mods;

namespace osu.Game.Tests.Visual.UserInterface
{
    public class TestSceneModCustomizationSettings : OsuTestScene
    {
        private TestModSelectOverlay modSelect;

        [BackgroundDependencyLoader]
        private void load()
        {
            Add(modSelect = new TestModSelectOverlay
            {
                RelativeSizeAxes = Axes.X,
                Origin = Anchor.BottomCentre,
                Anchor = Anchor.BottomCentre,
            });

            var testMod = new TestModCustomizable();

            AddStep("open", modSelect.Show);
            AddAssert("button disabled", () => !modSelect.CustomizeButton.Enabled.Value);
            AddStep("select mod", () => modSelect.SelectMod(testMod));
            AddAssert("button enabled", () => modSelect.CustomizeButton.Enabled.Value);
            AddStep("open customization", () => modSelect.CustomizeButton.Click());
            AddAssert("controls exist", () => modSelect.GetControlSection(testMod) != null);
            AddStep("deselect mod", () => modSelect.SelectMod(testMod));
            AddAssert("controls hidden", () => modSelect.ModSettingsContainer.Alpha == 0);
        }

        private class TestModSelectOverlay : ModSelectOverlay
        {
            public new Container ModSettingsContainer => base.ModSettingsContainer;
            public new TriangleButton CustomizeButton => base.CustomizeButton;

            public void SelectMod(Mod mod) =>
                ModSectionsContainer.Children.Single((s) => s.ModType == mod.Type)
                    .ButtonsContainer.OfType<ModButton>().Single(b => b.Mods.Any(m => m.GetType() == mod.GetType())).SelectNext(1);

            public ModControlSection GetControlSection(Mod mod) =>
                ModSettingsContent.Children.FirstOrDefault((s) => s.Mod == mod);

            protected override void LoadComplete()
            {
                base.LoadComplete();

                foreach (var section in ModSectionsContainer)
                    if (section.ModType == ModType.Conversion)
                        section.Mods = new Mod[] { new TestModCustomizable() };
                    else
                        section.Mods = new Mod[] { };
            }
        }

        private class TestModCustomizable : Mod, IModHasSettings
        {
            public override string Name => "Customizable Mod";

            public override string Acronym => "CM";

            public override double ScoreMultiplier => 1.0;

            public override ModType Type => ModType.Conversion;

            public readonly BindableFloat sliderBindable = new BindableFloat
            {
                MinValue = 0,
                MaxValue = 10,
            };

            public readonly BindableBool tickBindable = new BindableBool();

            public Drawable[] CreateControls()
            {
                BindableFloat sliderControl = new BindableFloat();
                BindableBool tickControl = new BindableBool();

                sliderControl.BindTo(sliderBindable);
                tickControl.BindTo(tickBindable);

                return new Drawable[]
                {
                    new SettingsSlider<float>
                    {
                        LabelText = "Slider",
                        Bindable = sliderControl
                    },
                    new SettingsCheckbox
                    {
                        LabelText = "Checkbox",
                        Bindable = tickControl
                    }
                };
            }
        }
    }
}