osu-lazer/osu.Game.Tests/Visual/Playlists/TestSceneAddPlaylistToCollectionButton.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.Diagnostics;
using System.Linq;
using osu.Framework.Allocation;
using osu.Framework.Audio;
using osu.Framework.Graphics;
using osu.Framework.Platform;
using osu.Framework.Testing;
using osu.Game.Beatmaps;
using osu.Game.Database;
using osu.Game.Online.Rooms;
using osu.Game.Overlays;
using osu.Game.Rulesets;
using osu.Game.Rulesets.Osu;
using osu.Game.Screens.OnlinePlay.Playlists;
using osuTK;

namespace osu.Game.Tests.Visual.Playlists
{
    public partial class TestSceneAddPlaylistToCollectionButton : OsuTestScene
    {
        private BeatmapManager manager = null!;
        private BeatmapSetInfo importedBeatmap = null!;
        private Room room = null!;

        [BackgroundDependencyLoader]
        private void load(GameHost host, AudioManager audio)
        {
            Dependencies.Cache(new RealmRulesetStore(Realm));
            Dependencies.Cache(manager = new BeatmapManager(LocalStorage, Realm, API, audio, Resources, host, Beatmap.Default));
            Dependencies.Cache(Realm);
        }

        [Cached(typeof(INotificationOverlay))]
        private NotificationOverlay notificationOverlay = new NotificationOverlay
        {
            Anchor = Anchor.TopRight,
            Origin = Anchor.TopRight,
        };

        [SetUpSteps]
        public void SetUpSteps()
        {
            importBeatmap();

            setupRoom();

            AddStep("create button", () =>
            {
                AddRange(new Drawable[]
                {
                    notificationOverlay,
                    new AddPlaylistToCollectionButton(room)
                    {
                        Anchor = Anchor.Centre,
                        Origin = Anchor.Centre,
                        Size = new Vector2(300, 40),
                    }
                });
            });
        }

        private void importBeatmap() => AddStep("import beatmap", () =>
        {
            var beatmap = CreateBeatmap(new OsuRuleset().RulesetInfo);

            Debug.Assert(beatmap.BeatmapInfo.BeatmapSet != null);

            importedBeatmap = manager.Import(beatmap.BeatmapInfo.BeatmapSet)!.Value.Detach();
        });

        private void setupRoom() => AddStep("setup room", () =>
        {
            room = new Room
            {
                Name = "my awesome room",
                MaxAttempts = 5,
                Host = API.LocalUser.Value
            };
            room.RecentParticipants = [room.Host];
            room.EndDate = DateTimeOffset.Now.AddMinutes(5);
            room.Playlist =
            [
                new PlaylistItem(importedBeatmap.Beatmaps.First())
                {
                    RulesetID = new OsuRuleset().RulesetInfo.OnlineID
                }
            ];
        });
    }
}