osu-lazer/osu.Game.Tests/Visual/Beatmaps/TestSceneDifficultyIcon.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 NUnit.Framework;
using osu.Framework.Extensions.IEnumerableExtensions;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Utils;
using osu.Game.Beatmaps.Drawables;
using osu.Game.Rulesets.Osu;
using osu.Game.Tests.Beatmaps;
using osuTK;

namespace osu.Game.Tests.Visual.Beatmaps
{
    public partial class TestSceneDifficultyIcon : OsuTestScene
    {
        private FillFlowContainer<DifficultyIcon> fill = null!;

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

            Child = fill = new FillFlowContainer<DifficultyIcon>
            {
                AutoSizeAxes = Axes.Y,
                Width = 300,
                Direction = FillDirection.Full,
                Anchor = Anchor.Centre,
                Origin = Anchor.Centre,
            };
        }

        [Test]
        public void CreateDifficultyIcon()
        {
            AddRepeatStep("create difficulty icon", () =>
            {
                var rulesetInfo = new OsuRuleset().RulesetInfo;
                var beatmapInfo = new TestBeatmap(rulesetInfo).BeatmapInfo;

                beatmapInfo.Difficulty.ApproachRate = RNG.Next(0, 10);
                beatmapInfo.Difficulty.CircleSize = RNG.Next(0, 10);
                beatmapInfo.Difficulty.OverallDifficulty = RNG.Next(0, 10);
                beatmapInfo.Difficulty.DrainRate = RNG.Next(0, 10);
                beatmapInfo.StarRating = RNG.NextSingle(0, 10);
                beatmapInfo.BPM = RNG.Next(60, 300);

                fill.Add(new DifficultyIcon(beatmapInfo, rulesetInfo)
                {
                    Scale = new Vector2(2),
                });
            }, 10);

            AddStep("no tooltip", () => fill.ForEach(icon => icon.TooltipType = DifficultyIconTooltipType.None));
            AddStep("basic tooltip", () => fill.ForEach(icon => icon.TooltipType = DifficultyIconTooltipType.StarRating));
            AddStep("extended tooltip", () => fill.ForEach(icon => icon.TooltipType = DifficultyIconTooltipType.Extended));
        }
    }
}