From f12be60d8d33471fc8acb4ee54c369f4f4e928e8 Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Mon, 22 Jan 2024 17:18:22 +0900
Subject: [PATCH] Make test actually test multiple icons

---
 .../Beatmaps/TestSceneDifficultyIcon.cs       | 39 +++++++++++++++++--
 1 file changed, 35 insertions(+), 4 deletions(-)

diff --git a/osu.Game.Tests/Visual/Beatmaps/TestSceneDifficultyIcon.cs b/osu.Game.Tests/Visual/Beatmaps/TestSceneDifficultyIcon.cs
index 79f9aec2e3..80320c138b 100644
--- a/osu.Game.Tests/Visual/Beatmaps/TestSceneDifficultyIcon.cs
+++ b/osu.Game.Tests/Visual/Beatmaps/TestSceneDifficultyIcon.cs
@@ -4,27 +4,58 @@
 #nullable disable
 
 using NUnit.Framework;
+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 fill;
+
+        protected override void LoadComplete()
+        {
+            base.LoadComplete();
+
+            Child = fill = new FillFlowContainer
+            {
+                AutoSizeAxes = Axes.Y,
+                Width = 300,
+                Direction = FillDirection.Full,
+                Anchor = Anchor.Centre,
+                Origin = Anchor.Centre,
+            };
+        }
+
         [Test]
         public void CreateDifficultyIcon()
         {
             DifficultyIcon difficultyIcon = null;
 
-            AddStep("create difficulty icon", () =>
+            AddRepeatStep("create difficulty icon", () =>
             {
-                Child = difficultyIcon = new DifficultyIcon(new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo, new OsuRuleset().RulesetInfo)
+                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(difficultyIcon = new DifficultyIcon(beatmapInfo, rulesetInfo)
                 {
+                    Scale = new Vector2(2),
                     ShowTooltip = true,
                     ShowExtendedTooltip = true
-                };
-            });
+                });
+            }, 10);
 
             AddStep("hide extended tooltip", () => difficultyIcon.ShowExtendedTooltip = false);