osu-lazer/osu.Game.Tests/Visual/Online/TestSceneBeatmapSetOverlaySuccessRate.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.Linq;
using NUnit.Framework;
using osu.Framework.Allocation;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Graphics.Shapes;
using osu.Framework.Testing;
using osu.Framework.Utils;
using osu.Game.Beatmaps;
using osu.Game.Graphics.UserInterface;
using osu.Game.Online.API.Requests.Responses;
using osu.Game.Overlays;
using osu.Game.Overlays.BeatmapSet;
using osu.Game.Screens.Select.Details;
using osuTK;
using osuTK.Graphics;

namespace osu.Game.Tests.Visual.Online
{
    public partial class TestSceneBeatmapSetOverlaySuccessRate : OsuTestScene
    {
        private GraphExposingSuccessRate successRate;

        [Cached]
        private OverlayColourProvider colourProvider = new OverlayColourProvider(OverlayColourScheme.Blue);

        [SetUp]
        public void Setup() => Schedule(() =>
        {
            Child = new Container
            {
                Anchor = Anchor.Centre,
                Origin = Anchor.Centre,
                Size = new Vector2(275, 220),
                Children = new Drawable[]
                {
                    new Box
                    {
                        RelativeSizeAxes = Axes.Both,
                        Colour = Color4.Gray,
                    },
                    successRate = new GraphExposingSuccessRate
                    {
                        Anchor = Anchor.Centre,
                        Origin = Anchor.Centre,
                        Size = new Vector2(275, 220),
                        Padding = new MarginPadding(20)
                    }
                }
            };
        });

        [Test]
        public void TestMetrics()
        {
            var firstBeatmap = createBeatmap();
            var secondBeatmap = createBeatmap();

            AddStep("set first set", () => successRate.Beatmap = firstBeatmap);
            AddAssert("ratings set", () => successRate.Graph.FailTimes == firstBeatmap.FailTimes);

            AddStep("set second set", () => successRate.Beatmap = secondBeatmap);
            AddAssert("ratings set", () => successRate.Graph.FailTimes == secondBeatmap.FailTimes);

            static APIBeatmap createBeatmap() => new APIBeatmap
            {
                FailTimes = new APIFailTimes
                {
                    Fails = Enumerable.Range(1, 100).Select(_ => RNG.Next(10)).ToArray(),
                    Retries = Enumerable.Range(-2, 100).Select(_ => RNG.Next(10)).ToArray(),
                },
                PassCount = RNG.Next(0, 999),
                PlayCount = RNG.Next(1000, 1999),
            };
        }

        [Test]
        public void TestOnlyFailMetrics()
        {
            AddStep("set beatmap", () => successRate.Beatmap = new APIBeatmap
            {
                FailTimes = new APIFailTimes
                {
                    Fails = Enumerable.Range(1, 100).ToArray(),
                }
            });

            AddAssert("graph max values correct", () => successRate.ChildrenOfType<BarGraph>().All(graph => graph.MaxValue == 100));
        }

        [Test]
        public void TestEmptyMetrics()
        {
            AddStep("set beatmap", () => successRate.Beatmap = new APIBeatmap
            {
                FailTimes = new APIFailTimes()
            });

            AddAssert("graph max values correct", () => successRate.ChildrenOfType<BarGraph>().All(graph => graph.MaxValue == 0));
        }

        private partial class GraphExposingSuccessRate : SuccessRate
        {
            public new FailRetryGraph Graph => base.Graph;
        }
    }
}