Ez2Lazer/osu.Game.Tests/Visual/Multiplayer/TestSceneMultiplayerSkipOverlay.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.Linq;
using NUnit.Framework;
using osu.Framework.Extensions;
using osu.Framework.Graphics;
using osu.Framework.Testing;
using osu.Game.Online.API.Requests.Responses;
using osu.Game.Online.Multiplayer;
using osu.Game.Rulesets.Osu;
using osu.Game.Screens.OnlinePlay.Multiplayer;
using osu.Game.Screens.Play;

namespace osu.Game.Tests.Visual.Multiplayer
{
    public partial class TestSceneMultiplayerSkipOverlay : MultiplayerTestScene
    {
        public override void SetUpSteps()
        {
            base.SetUpSteps();

            AddStep("join room", () => JoinRoom(CreateDefaultRoom()));
            WaitForJoined();

            AddStep("add skip overlay", () =>
            {
                GameplayClockContainer gameplayClockContainer;

                var working = CreateWorkingBeatmap(CreateBeatmap(new OsuRuleset().RulesetInfo));

                Child = gameplayClockContainer = new MasterGameplayClockContainer(working, 0)
                {
                    RelativeSizeAxes = Axes.Both,
                    Children = new Drawable[]
                    {
                        new MultiplayerSkipOverlay(120000)
                        {
                            RequestSkip = () => MultiplayerClient.VoteToSkipIntro().WaitSafely(),
                        }
                    },
                };

                gameplayClockContainer.Start();
            });

            AddStep("set playing state", () => MultiplayerClient.ChangeUserState(API.LocalUser.Value.OnlineID, MultiplayerUserState.Playing));
        }

        [Test]
        public void TestSkip()
        {
            for (int i = 0; i < 4; i++)
            {
                int userId = i;

                AddStep($"join user {userId}", () =>
                {
                    MultiplayerClient.AddUser(new APIUser
                    {
                        Id = userId,
                        Username = $"User {userId}"
                    });

                    MultiplayerClient.ChangeUserState(userId, MultiplayerUserState.Playing);
                });
            }

            AddStep("user 0 votes", () => MultiplayerClient.UserVoteToSkipIntro(0).WaitSafely());
            AddStep("local user votes", () => this.ChildrenOfType<MultiplayerSkipOverlay.Button>().Single().TriggerClick());
            AddStep("user 1 votes", () => MultiplayerClient.UserVoteToSkipIntro(1).WaitSafely());
        }

        [Test]
        public void TestLeavingBeforeLocalVote()
        {
            for (int i = 0; i < 4; i++)
            {
                int userId = i;

                AddStep($"join user {userId}", () =>
                {
                    MultiplayerClient.AddUser(new APIUser
                    {
                        Id = userId,
                        Username = $"User {userId}"
                    });

                    MultiplayerClient.ChangeUserState(userId, MultiplayerUserState.Playing);
                });
            }

            AddStep("user 0 votes", () => MultiplayerClient.UserVoteToSkipIntro(0).WaitSafely());
            AddStep("user 1 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 1 }));
            AddStep("user 2 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 2 }));
            AddStep("user 3 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 3 }));
            AddStep("user 0 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 0 }));
        }

        [Test]
        public void TestLeavingAfterLocalVote()
        {
            for (int i = 0; i < 4; i++)
            {
                int userId = i;

                AddStep($"join user {userId}", () =>
                {
                    MultiplayerClient.AddUser(new APIUser
                    {
                        Id = userId,
                        Username = $"User {userId}"
                    });

                    MultiplayerClient.ChangeUserState(userId, MultiplayerUserState.Playing);
                });
            }

            AddStep("local user votes", () => this.ChildrenOfType<MultiplayerSkipOverlay.Button>().Single().TriggerClick());
            AddStep("user 0 votes", () => MultiplayerClient.UserVoteToSkipIntro(0).WaitSafely());
            AddStep("user 1 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 1 }));
            AddStep("user 2 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 2 }));
            AddStep("user 3 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 3 }));
            AddStep("user 0 leaves", () => MultiplayerClient.RemoveUser(new APIUser { Id = 0 }));
        }

        public partial class TestMultiplayerSkipOverlay : MultiplayerSkipOverlay
        {
            public TestMultiplayerSkipOverlay()
                : base(120000)
            {
            }
        }
    }
}