osu-lazer/osu.Game.Tests/Visual/DailyChallenge/TestSceneDailyChallenge.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;
using System.Linq;
using NUnit.Framework;
using osu.Framework.Allocation;
using osu.Game.Online.API;
using osu.Game.Online.Metadata;
using osu.Game.Online.Rooms;
using osu.Game.Overlays;
using osu.Game.Rulesets.Osu.Mods;
using osu.Game.Tests.Resources;
using osu.Game.Tests.Visual.Metadata;
using osu.Game.Tests.Visual.OnlinePlay;

namespace osu.Game.Tests.Visual.DailyChallenge
{
    public partial class TestSceneDailyChallenge : OnlinePlayTestScene
    {
        [Cached(typeof(MetadataClient))]
        private TestMetadataClient metadataClient = new TestMetadataClient();

        [Cached(typeof(INotificationOverlay))]
        private NotificationOverlay notificationOverlay = new NotificationOverlay();

        [BackgroundDependencyLoader]
        private void load()
        {
            base.Content.Add(notificationOverlay);
        }

        [Test]
        public void TestDailyChallenge()
        {
            var room = new Room
            {
                RoomID = { Value = 1234 },
                Name = { Value = "Daily Challenge: June 4, 2024" },
                Playlist =
                {
                    new PlaylistItem(TestResources.CreateTestBeatmapSetInfo().Beatmaps.First())
                    {
                        RequiredMods = [new APIMod(new OsuModTraceable())],
                        AllowedMods = [new APIMod(new OsuModDoubleTime())]
                    }
                },
                EndDate = { Value = DateTimeOffset.Now.AddHours(12) },
                Category = { Value = RoomCategory.DailyChallenge }
            };

            AddStep("add room", () => API.Perform(new CreateRoomRequest(room)));
            AddStep("push screen", () => LoadScreen(new Screens.OnlinePlay.DailyChallenge.DailyChallenge(room)));
        }

        [Test]
        public void TestNotifications()
        {
            var room = new Room
            {
                RoomID = { Value = 1234 },
                Name = { Value = "Daily Challenge: June 4, 2024" },
                Playlist =
                {
                    new PlaylistItem(CreateAPIBeatmapSet().Beatmaps.First())
                    {
                        RequiredMods = [new APIMod(new OsuModTraceable())],
                        AllowedMods = [new APIMod(new OsuModDoubleTime())]
                    }
                },
                EndDate = { Value = DateTimeOffset.Now.AddHours(12) },
                Category = { Value = RoomCategory.DailyChallenge }
            };

            AddStep("add room", () => API.Perform(new CreateRoomRequest(room)));
            AddStep("set daily challenge info", () => metadataClient.DailyChallengeInfo.Value = new DailyChallengeInfo { RoomID = 1234 });
            AddStep("push screen", () => LoadScreen(new Screens.OnlinePlay.DailyChallenge.DailyChallenge(room)));
            AddStep("daily challenge ended", () => metadataClient.DailyChallengeInfo.Value = null);

            Func<APIRequest, bool>? previousHandler = null;

            AddStep("install custom handler", () =>
            {
                previousHandler = ((DummyAPIAccess)API).HandleRequest;
                ((DummyAPIAccess)API).HandleRequest = req =>
                {
                    switch (req)
                    {
                        case GetRoomRequest r:
                        {
                            r.TriggerSuccess(new Room
                            {
                                RoomID = { Value = 1235, },
                                Name = { Value = "Daily Challenge: June 5, 2024" },
                                Playlist =
                                {
                                    new PlaylistItem(CreateAPIBeatmapSet().Beatmaps.First())
                                    {
                                        RequiredMods = [new APIMod(new OsuModTraceable())],
                                        AllowedMods = [new APIMod(new OsuModDoubleTime())]
                                    }
                                },
                                EndDate = { Value = DateTimeOffset.Now.AddHours(12) },
                                Category = { Value = RoomCategory.DailyChallenge }
                            });
                            return true;
                        }

                        default:
                            return false;
                    }
                };
            });
            AddStep("next daily challenge started", () => metadataClient.DailyChallengeInfo.Value = new DailyChallengeInfo { RoomID = 1235 });

            AddStep("restore previous handler", () => ((DummyAPIAccess)API).HandleRequest = previousHandler);
        }
    }
}