osu/osu.Game.Tests/Visual/Online/TestSceneVotePill.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.Collections.Generic;
using NUnit.Framework;
using osu.Framework.Graphics;
using osu.Game.Overlays.Comments;
using osu.Framework.Allocation;
using osu.Game.Online.API.Requests.Responses;
using osu.Game.Users;
using osu.Framework.MathUtils;

namespace osu.Game.Tests.Visual.Online
{
    [TestFixture]
    public class TestSceneVotePill : OsuTestScene
    {
        public override IReadOnlyList<Type> RequiredTypes => new[]
        {
            typeof(VotePill)
        };

        private VotePill votePill;

        [BackgroundDependencyLoader]
        private void load()
        {
            var randomComment = new Comment
            {
                IsVoted = false,
                UserId = 4444,
                VotesCount = 2,
            };

            AddStep("Log in", () => API.LocalUser.Value = new User
            {
                Id = RNG.Next(2, 100000)
            });
            AddStep("User comment", () => addVotePill(getUserComment()));
            AddStep("Click", () => votePill.Click());
            AddAssert("Not loading", () => !votePill.IsLoading);

            AddStep("Random comment", () => addVotePill(randomComment));
            AddStep("Click", () => votePill.Click());
            AddAssert("Loading", () => votePill.IsLoading);

            AddStep("Log out", API.Logout);
            AddStep("Random comment", () => addVotePill(randomComment));
            AddStep("Click", () => votePill.Click());
            AddAssert("Not loading", () => !votePill.IsLoading);
        }

        private Comment getUserComment() => new Comment
        {
            IsVoted = false,
            UserId = API.LocalUser.Value.Id,
            VotesCount = 10,
        };

        private void addVotePill(Comment comment)
        {
            Clear();
            Add(votePill = new VotePill(comment)
            {
                Anchor = Anchor.Centre,
                Origin = Anchor.Centre,
            });
        }
    }
}