osu-lazer/osu.Game.Tests/Visual/Multiplayer/TestSceneDrawableRoomPlaylist.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;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using NUnit.Framework;
using osu.Framework.Allocation;
using osu.Framework.Audio;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Graphics.UserInterface;
using osu.Framework.Platform;
using osu.Framework.Testing;
using osu.Game.Beatmaps;
using osu.Game.Beatmaps.Drawables;
using osu.Game.Collections;
using osu.Game.Database;
using osu.Game.Graphics.Containers;
using osu.Game.Graphics.Cursor;
using osu.Game.Graphics.UserInterface;
using osu.Game.Models;
using osu.Game.Online.API;
using osu.Game.Online.Rooms;
using osu.Game.Overlays;
using osu.Game.Rulesets;
using osu.Game.Rulesets.Osu;
using osu.Game.Rulesets.Osu.Mods;
using osu.Game.Screens.OnlinePlay;
using osu.Game.Tests.Beatmaps;
using osu.Game.Users.Drawables;
using osuTK;
using osuTK.Input;

namespace osu.Game.Tests.Visual.Multiplayer
{
    public class TestSceneDrawableRoomPlaylist : MultiplayerTestScene
    {
        protected override Container<Drawable> Content { get; } = new Container { RelativeSizeAxes = Axes.Both };

        private TestPlaylist playlist;

        private BeatmapManager manager;
        private RulesetStore rulesets;
        private CollectionManager collections;

        private BeatmapSetOverlay beatmapOverlay;
        private ManageCollectionsDialog manageCollectionsDialog;

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

            base.Content.AddRange(new Drawable[]
            {
                collections = new CollectionManager(LocalStorage),
                Content
            });

            Dependencies.Cache(collections);
        }

        [Test]
        public void TestNonEditableNonSelectable()
        {
            createPlaylist();

            moveToItem(0);
            assertHandleVisibility(0, false);
            assertDeleteButtonVisibility(0, false);

            AddStep("click", () => InputManager.Click(MouseButton.Left));
            AddAssert("no item selected", () => playlist.SelectedItem.Value == null);

            AddStep("press down", () => InputManager.Key(Key.Down));
            AddAssert("no item selected", () => playlist.SelectedItem.Value == null);
        }

        [Test]
        public void TestEditable()
        {
            createPlaylist(p =>
            {
                p.AllowReordering = true;
                p.AllowDeletion = true;
            });

            moveToItem(0);
            assertHandleVisibility(0, true);
            assertDeleteButtonVisibility(0, true);

            AddStep("click", () => InputManager.Click(MouseButton.Left));
            AddAssert("no item selected", () => playlist.SelectedItem.Value == null);

            AddStep("press down", () => InputManager.Key(Key.Down));
            AddAssert("no item selected", () => playlist.SelectedItem.Value == null);
        }

        [Test]
        public void TestMarkInvalid()
        {
            createPlaylist(p =>
            {
                p.AllowReordering = true;
                p.AllowDeletion = true;
                p.AllowSelection = true;
            });

            AddStep("mark item 0 as invalid", () => playlist.Items[0].MarkInvalid());

            moveToItem(0);

            AddStep("click", () => InputManager.Click(MouseButton.Left));
            AddAssert("no item selected", () => playlist.SelectedItem.Value == null);

            AddStep("press down", () => InputManager.Key(Key.Down));
            AddAssert("no item selected", () => playlist.SelectedItem.Value == null);
        }

        [Test]
        public void TestSelectable()
        {
            createPlaylist(p => p.AllowSelection = true);

            moveToItem(0);
            assertHandleVisibility(0, false);
            assertDeleteButtonVisibility(0, false);

            AddStep("click", () => InputManager.Click(MouseButton.Left));

            AddAssert("item 0 is selected", () => playlist.SelectedItem.Value == playlist.Items[0]);
        }

        [Test]
        public void TestEditableSelectable()
        {
            createPlaylist(p =>
            {
                p.AllowReordering = true;
                p.AllowDeletion = true;
                p.AllowSelection = true;
            });

            moveToItem(0);
            assertHandleVisibility(0, true);
            assertDeleteButtonVisibility(0, true);

            AddStep("click", () => InputManager.Click(MouseButton.Left));

            AddAssert("item 0 is selected", () => playlist.SelectedItem.Value == playlist.Items[0]);
        }

        [Test]
        public void TestSelectionNotLostAfterRearrangement()
        {
            createPlaylist(p =>
            {
                p.AllowReordering = true;
                p.AllowDeletion = true;
                p.AllowSelection = true;
            });

            moveToItem(0);
            AddStep("click", () => InputManager.Click(MouseButton.Left));

            moveToDragger(0);
            AddStep("begin drag", () => InputManager.PressButton(MouseButton.Left));
            moveToDragger(1, new Vector2(0, 5));
            AddStep("end drag", () => InputManager.ReleaseButton(MouseButton.Left));

            AddAssert("item 1 is selected", () => playlist.SelectedItem.Value == playlist.Items[1]);
        }

        [Test]
        public void TestKeyboardSelection()
        {
            createPlaylist(p => p.AllowSelection = true);

            AddStep("press down", () => InputManager.Key(Key.Down));
            AddAssert("item 0 is selected", () => playlist.SelectedItem.Value == playlist.Items[0]);

            AddStep("press down", () => InputManager.Key(Key.Down));
            AddAssert("item 1 is selected", () => playlist.SelectedItem.Value == playlist.Items[1]);

            AddStep("press up", () => InputManager.Key(Key.Up));
            AddAssert("item 0 is selected", () => playlist.SelectedItem.Value == playlist.Items[0]);

            AddUntilStep("navigate to last item via keyboard", () =>
            {
                InputManager.Key(Key.Down);
                return playlist.SelectedItem.Value == playlist.Items.Last();
            });
            AddAssert("last item is selected", () => playlist.SelectedItem.Value == playlist.Items.Last());
            AddUntilStep("last item is scrolled into view", () =>
            {
                var drawableItem = playlist.ItemMap[playlist.Items.Last()];
                return playlist.ScreenSpaceDrawQuad.Contains(drawableItem.ScreenSpaceDrawQuad.TopLeft)
                       && playlist.ScreenSpaceDrawQuad.Contains(drawableItem.ScreenSpaceDrawQuad.BottomRight);
            });

            AddStep("press down", () => InputManager.Key(Key.Down));
            AddAssert("last item is selected", () => playlist.SelectedItem.Value == playlist.Items.Last());

            AddStep("press up", () => InputManager.Key(Key.Up));
            AddAssert("second last item is selected", () => playlist.SelectedItem.Value == playlist.Items.Reverse().ElementAt(1));
        }

        [Test]
        public void TestDownloadButtonHiddenWhenBeatmapExists()
        {
            Live<BeatmapSetInfo> imported = null;

            AddStep("import beatmap", () =>
            {
                var beatmap = new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo;

                Debug.Assert(beatmap.BeatmapSet != null);
                imported = manager.Import(beatmap.BeatmapSet);
            });

            createPlaylistWithBeatmaps(() => imported.PerformRead(s => s.Beatmaps.Detach()));

            assertDownloadButtonVisible(false);

            AddStep("delete beatmap set", () => imported.PerformWrite(s => s.DeletePending = true));
            assertDownloadButtonVisible(true);

            AddStep("undelete beatmap set", () => imported.PerformWrite(s => s.DeletePending = false));
            assertDownloadButtonVisible(false);

            void assertDownloadButtonVisible(bool visible) => AddUntilStep($"download button {(visible ? "shown" : "hidden")}",
                () => playlist.ChildrenOfType<BeatmapDownloadButton>().SingleOrDefault()?.Alpha == (visible ? 1 : 0));
        }

        [Test]
        public void TestDownloadButtonVisibleInitiallyWhenBeatmapDoesNotExist()
        {
            var byOnlineId = CreateAPIBeatmap();
            byOnlineId.OnlineID = 1337; // Some random ID that does not exist locally.

            var byChecksum = CreateAPIBeatmap();
            byChecksum.Checksum = "1337"; // Some random checksum that does not exist locally.

            createPlaylistWithBeatmaps(() => new[] { byOnlineId, byChecksum });

            AddAssert("download buttons shown", () => playlist.ChildrenOfType<BeatmapDownloadButton>().All(d => d.IsPresent));
        }

        [Test]
        public void TestExplicitBeatmapItem()
        {
            var beatmap = CreateAPIBeatmap();

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

            beatmap.BeatmapSet.HasExplicitContent = true;

            createPlaylistWithBeatmaps(() => new[] { beatmap });
        }

        [Test]
        public void TestExpiredItems()
        {
            createPlaylist(p =>
            {
                p.Items.Clear();
                p.Items.AddRange(new[]
                {
                    new PlaylistItem(new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo)
                    {
                        ID = 0,
                        RulesetID = new OsuRuleset().RulesetInfo.OnlineID,
                        Expired = true,
                        RequiredMods = new[]
                        {
                            new APIMod(new OsuModHardRock()),
                            new APIMod(new OsuModDoubleTime()),
                            new APIMod(new OsuModAutoplay())
                        }
                    },
                    new PlaylistItem(new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo)
                    {
                        ID = 1,
                        RulesetID = new OsuRuleset().RulesetInfo.OnlineID,
                        RequiredMods = new[]
                        {
                            new APIMod(new OsuModHardRock()),
                            new APIMod(new OsuModDoubleTime()),
                            new APIMod(new OsuModAutoplay())
                        }
                    }
                });
            });

            AddUntilStep("wait for items to load", () => playlist.ItemMap.Values.All(i => i.IsLoaded));
        }

        [TestCase(false)]
        [TestCase(true)]
        public void TestWithOwner(bool withOwner)
        {
            createPlaylist(p => p.ShowItemOwners = withOwner);

            AddAssert("owner visible", () => playlist.ChildrenOfType<UpdateableAvatar>().All(a => a.IsPresent == withOwner));
        }

        [Test]
        public void TestWithAllButtonsEnabled()
        {
            createPlaylist(p =>
            {
                p.AllowDeletion = true;
                p.AllowShowingResults = true;
                p.AllowEditing = true;
            });
        }

        [Test]
        public void TestContextMenuDetails()
        {
            OsuContextMenu contextMenu = null;

            createPlaylist();

            AddAssert("beatmap overlay hidden", () => beatmapOverlay.State.Value == Visibility.Hidden);

            moveToItem(0);
            AddStep("right click", () => InputManager.Click(MouseButton.Right));
            AddAssert("context menu open", () => (contextMenu = this.ChildrenOfType<OsuContextMenu>().SingleOrDefault())?.State == MenuState.Open);

            AddStep("click details", () =>
            {
                InputManager.MoveMouseTo(contextMenu.ChildrenOfType<DrawableOsuMenuItem>().First());
                InputManager.Click(MouseButton.Left);
            });
            AddAssert("beatmap overlay visible", () => beatmapOverlay.State.Value == Visibility.Visible);
        }

        [Test]
        public void TestContextMenuCollection()
        {
            OsuContextMenu contextMenu = null;
            BeatmapInfo beatmap = null;
            Live<BeatmapSetInfo> imported = null;

            AddStep("import beatmap", () =>
            {
                beatmap = new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo;

                Debug.Assert(beatmap.BeatmapSet != null);
                imported = manager.Import(beatmap.BeatmapSet);
            });

            createPlaylistWithBeatmaps(() => imported.PerformRead(s => s.Beatmaps.Detach()));

            AddStep("add two collections", () =>
            {
                collections.Collections.Clear();
                collections.Collections.AddRange(new[]
                {
                    new BeatmapCollection { Name = { Value = "Collection #1" }, BeatmapHashes = { beatmap.MD5Hash } },
                    new BeatmapCollection { Name = { Value = "Collection #2" } },
                });
            });

            moveToItem(0);
            AddStep("right click", () => InputManager.Click(MouseButton.Right));
            AddAssert("context menu open", () => (contextMenu = this.ChildrenOfType<OsuContextMenu>().SingleOrDefault())?.State == MenuState.Open);

            AddStep("select collections", () => InputManager.MoveMouseTo(contextMenu.ChildrenOfType<DrawableOsuMenuItem>().ElementAt(1)));
            AddAssert("collection 1 present and beatmap added", () =>
            {
                var item = (ToggleMenuItem)contextMenu.Items[1].Items[0];
                return item.Text.Value == "Collection #1" && item.State.Value;
            });
            AddAssert("collection 2 present", () =>
            {
                var item = (ToggleMenuItem)contextMenu.Items[1].Items[1];
                return item.Text.Value == "Collection #2" && !item.State.Value;
            });

            AddStep("select second collection", () =>
            {
                InputManager.MoveMouseTo(contextMenu.ChildrenOfType<DrawableStatefulMenuItem>().ElementAt(1));
                InputManager.Click(MouseButton.Left);
            });
            AddAssert("beatmap added to second collection", () => collections.Collections[1].BeatmapHashes.Contains(beatmap.MD5Hash));
            AddAssert("item state updated", () => ((ToggleMenuItem)contextMenu.Items[1].Items[1]).State.Value);
        }

        [Test]
        public void TestContextMenuManageCollections()
        {
            OsuContextMenu contextMenu = null;
            Live<BeatmapSetInfo> imported = null;

            AddStep("import beatmap", () =>
            {
                var beatmap = new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo;

                Debug.Assert(beatmap.BeatmapSet != null);
                imported = manager.Import(beatmap.BeatmapSet);
            });

            createPlaylistWithBeatmaps(() => imported.PerformRead(s => s.Beatmaps.Detach()));

            AddStep("clear collections", () => collections.Collections.Clear());
            AddAssert("manage collections dialog hidden", () => manageCollectionsDialog.State.Value == Visibility.Hidden);

            moveToItem(0);
            AddStep("right click", () => InputManager.Click(MouseButton.Right));
            AddAssert("context menu open", () => (contextMenu = this.ChildrenOfType<OsuContextMenu>().SingleOrDefault())?.State == MenuState.Open);

            AddStep("select collections", () => InputManager.MoveMouseTo(contextMenu.ChildrenOfType<DrawableOsuMenuItem>().ElementAt(1)));
            AddStep("click manage", () =>
            {
                InputManager.MoveMouseTo(contextMenu.ChildrenOfType<OsuContextMenu>().ElementAt(1).ChildrenOfType<DrawableOsuMenuItem>().Single());
                InputManager.Click(MouseButton.Left);
            });
            AddAssert("manage collections dialog open", () => manageCollectionsDialog.State.Value == Visibility.Visible);
        }

        private void moveToItem(int index, Vector2? offset = null)
            => AddStep($"move mouse to item {index}", () => InputManager.MoveMouseTo(playlist.ChildrenOfType<DrawableRoomPlaylistItem>().ElementAt(index), offset));

        private void moveToDragger(int index, Vector2? offset = null) => AddStep($"move mouse to dragger {index}", () =>
        {
            var item = playlist.ChildrenOfType<OsuRearrangeableListItem<PlaylistItem>>().ElementAt(index);
            InputManager.MoveMouseTo(item.ChildrenOfType<OsuRearrangeableListItem<PlaylistItem>.PlaylistItemHandle>().Single(), offset);
        });

        private void assertHandleVisibility(int index, bool visible)
            => AddAssert($"handle {index} {(visible ? "is" : "is not")} visible",
                () => (playlist.ChildrenOfType<OsuRearrangeableListItem<PlaylistItem>.PlaylistItemHandle>().ElementAt(index).Alpha > 0) == visible);

        private void assertDeleteButtonVisibility(int index, bool visible)
            => AddAssert($"delete button {index} {(visible ? "is" : "is not")} visible",
                () => (playlist.ChildrenOfType<DrawableRoomPlaylistItem.PlaylistRemoveButton>().ElementAt(2 + index * 2).Alpha > 0) == visible);

        private void createPlaylistWithBeatmaps(Func<IEnumerable<IBeatmapInfo>> beatmaps) => createPlaylist(p =>
        {
            int index = 0;

            p.Items.Clear();

            foreach (var b in beatmaps())
            {
                p.Items.Add(new PlaylistItem(b)
                {
                    ID = index++,
                    OwnerID = 2,
                    RulesetID = new OsuRuleset().RulesetInfo.OnlineID,
                    RequiredMods = new[]
                    {
                        new APIMod(new OsuModHardRock()),
                        new APIMod(new OsuModDoubleTime()),
                        new APIMod(new OsuModAutoplay())
                    }
                });
            }
        });

        private void createPlaylist(Action<TestPlaylist> setupPlaylist = null)
        {
            AddStep("create playlist", () =>
            {
                Child = new DependencyProvidingContainer
                {
                    RelativeSizeAxes = Axes.Both,
                    CachedDependencies = new (Type, object)[]
                    {
                        (typeof(BeatmapSetOverlay), beatmapOverlay = new BeatmapSetOverlay()),
                        (typeof(ManageCollectionsDialog), manageCollectionsDialog = new ManageCollectionsDialog()),
                    },
                    Children = new Drawable[]
                    {
                        new OsuContextMenuContainer
                        {
                            RelativeSizeAxes = Axes.Both,
                            Child = playlist = new TestPlaylist
                            {
                                Anchor = Anchor.Centre,
                                Origin = Anchor.Centre,
                                Size = new Vector2(500, 300),
                            },
                        },
                        beatmapOverlay,
                        manageCollectionsDialog,
                    },
                };

                for (int i = 0; i < 20; i++)
                {
                    playlist.Items.Add(new PlaylistItem(i % 2 == 1
                        ? new TestBeatmap(new OsuRuleset().RulesetInfo).BeatmapInfo
                        : new BeatmapInfo
                        {
                            Metadata = new BeatmapMetadata
                            {
                                Artist = "Artist",
                                Author = new RealmUser { Username = "Creator name here" },
                                Title = "Long title used to check background colour",
                            },
                            BeatmapSet = new BeatmapSetInfo()
                        })
                    {
                        ID = i,
                        OwnerID = 2,
                        RulesetID = new OsuRuleset().RulesetInfo.OnlineID,
                        RequiredMods = new[]
                        {
                            new APIMod(new OsuModHardRock()),
                            new APIMod(new OsuModDoubleTime()),
                            new APIMod(new OsuModAutoplay())
                        }
                    });
                }

                setupPlaylist?.Invoke(playlist);
            });

            AddUntilStep("wait for items to load", () => playlist.ItemMap.Values.All(i => i.IsLoaded));
        }

        private class TestPlaylist : DrawableRoomPlaylist
        {
            public new IReadOnlyDictionary<PlaylistItem, RearrangeableListItem<PlaylistItem>> ItemMap => base.ItemMap;
        }
    }
}