From ab57cbf6f5a1d438963469f4dea27974755559e7 Mon Sep 17 00:00:00 2001
From: Pasi4K5 <pascal.pomper03@gmail.com>
Date: Mon, 25 Sep 2023 12:53:17 +0200
Subject: [PATCH] Add `TestSceneDifficultySearch`

---
 .../SongSelect/TestSceneDifficultySearch.cs   | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 osu.Game.Tests/SongSelect/TestSceneDifficultySearch.cs

diff --git a/osu.Game.Tests/SongSelect/TestSceneDifficultySearch.cs b/osu.Game.Tests/SongSelect/TestSceneDifficultySearch.cs
new file mode 100644
index 0000000000..63963d1101
--- /dev/null
+++ b/osu.Game.Tests/SongSelect/TestSceneDifficultySearch.cs
@@ -0,0 +1,71 @@
+// 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.Collections.Generic;
+using System.Linq;
+using NUnit.Framework;
+using osu.Game.Beatmaps;
+using osu.Game.Screens.Select;
+using osu.Game.Screens.Select.Carousel;
+using osu.Game.Tests.Visual;
+
+namespace osu.Game.Tests.SongSelect
+{
+    public partial class TestSceneDifficultySearch : OsuTestScene
+    {
+        private static readonly (string title, string difficultyName)[] beatmaps =
+        {
+            ("Title1", "Diff1"),
+            ("Title1", "Diff2"),
+            ("My[Favourite]Song", "Expert"),
+            ("Title", "My Favourite Diff"),
+            ("Another One", "diff ]with [[] brackets]]]"),
+        };
+
+        [TestCase("[1]", new[] { 0 })]
+        [TestCase("[1", new[] { 0 })]
+        [TestCase("My[Favourite", new[] { 2 })]
+        [TestCase("My[Favourite]", new[] { 2 })]
+        [TestCase("My[Favourite]Song", new[] { 2 })]
+        [TestCase("Favourite]", new[] { 2 })]
+        [TestCase("[Diff", new[] { 0, 1, 3, 4 })]
+        [TestCase("[Diff]", new[] { 0, 1, 3, 4 })]
+        [TestCase("[Favourite]", new[] { 3 })]
+        [TestCase("Title1 [Diff]", new[] { 0, 1 })]
+        [TestCase("Title1[Diff]", new int[] { })]
+        [TestCase("[diff ]with]", new[] { 4 })]
+        [TestCase("[diff ]with [[] brackets]]]]", new[] { 4 })]
+        public void TestDifficultySearch(string query, int[] expectedBeatmapIndexes)
+        {
+            var carouselBeatmaps = createCarouselBeatmaps().ToList();
+
+            AddStep("filter beatmaps", () =>
+            {
+                var criteria = new FilterCriteria();
+                FilterQueryParser.ApplyQueries(criteria, query);
+                carouselBeatmaps.ForEach(b => b.Filter(criteria));
+            });
+
+            AddAssert("filtered correctly", () => carouselBeatmaps.All(b =>
+            {
+                int index = carouselBeatmaps.IndexOf(b);
+
+                bool filtered = b.Filtered.Value;
+
+                return filtered != expectedBeatmapIndexes.Contains(index);
+            }));
+        }
+
+        private static IEnumerable<CarouselBeatmap> createCarouselBeatmaps()
+        {
+            return beatmaps.Select(info => new CarouselBeatmap(new BeatmapInfo
+            {
+                Metadata = new BeatmapMetadata
+                {
+                    Title = info.title
+                },
+                DifficultyName = info.difficultyName
+            }));
+        }
+    }
+}