osu-lazer/osu.Game.Tests/Visual/UserInterface/TestScenePaginatedContainerHeader.cs

using NUnit.Framework;
using osu.Game.Overlays.Profile.Sections;
using osu.Framework.Testing;
using System.Linq;
using osu.Framework.Graphics;
using osu.Game.Overlays;
using osu.Framework.Allocation;

namespace osu.Game.Tests.Visual.UserInterface
{
    public class TestScenePaginatedContainerHeader : OsuTestScene
    {
        [Cached]
        private readonly OverlayColourProvider colourProvider = new OverlayColourProvider(OverlayColourScheme.Pink);

        private PaginatedContainerHeader header;

        [Test]
        public void TestHiddenCounter()
        {
            AddStep("Create header", () => createHeader("Header with hidden counter", CounterVisibilityState.AlwaysHidden));
            AddAssert("Value is 0", () => header.Current.Value == 0);
            AddAssert("Counter is hidden", () => header.ChildrenOfType<CounterPill>().First().Alpha == 0);
            AddStep("Set count 10", () => header.Current.Value = 10);
            AddAssert("Value is 10", () => header.Current.Value == 10);
            AddAssert("Counter is hidden", () => header.ChildrenOfType<CounterPill>().First().Alpha == 0);
        }

        [Test]
        public void TestVisibleCounter()
        {
            AddStep("Create header", () => createHeader("Header with visible counter", CounterVisibilityState.AlwaysVisible));
            AddAssert("Value is 0", () => header.Current.Value == 0);
            AddAssert("Counter is visible", () => header.ChildrenOfType<CounterPill>().First().Alpha == 1);
            AddStep("Set count 10", () => header.Current.Value = 10);
            AddAssert("Value is 10", () => header.Current.Value == 10);
            AddAssert("Counter is visible", () => header.ChildrenOfType<CounterPill>().First().Alpha == 1);
        }

        [Test]
        public void TestVisibleWhenZeroCounter()
        {
            AddStep("Create header", () => createHeader("Header with visible when zero counter", CounterVisibilityState.VisibleWhenNonZero));
            AddAssert("Value is 0", () => header.Current.Value == 0);
            AddAssert("Counter is visible", () => header.ChildrenOfType<CounterPill>().First().Alpha == 1);
            AddStep("Set count 10", () => header.Current.Value = 10);
            AddAssert("Value is 10", () => header.Current.Value == 10);
            AddAssert("Counter is hidden", () => header.ChildrenOfType<CounterPill>().First().Alpha == 0);
            AddStep("Set count 0", () => header.Current.Value = 0);
            AddAssert("Value is 0", () => header.Current.Value == 0);
            AddAssert("Counter is visible", () => header.ChildrenOfType<CounterPill>().First().Alpha == 1);
        }

        [Test]
        public void TestInitialVisibility()
        {
            AddStep("Create header with 0 value", () => createHeader("Header with visible when zero counter", CounterVisibilityState.VisibleWhenNonZero, 0));
            AddAssert("Value is 0", () => header.Current.Value == 0);
            AddAssert("Counter is visible", () => header.ChildrenOfType<CounterPill>().First().Alpha == 1);

            AddStep("Create header with 1 value", () => createHeader("Header with visible when zero counter", CounterVisibilityState.VisibleWhenNonZero, 1));
            AddAssert("Value is 1", () => header.Current.Value == 1);
            AddAssert("Counter is hidden", () => header.ChildrenOfType<CounterPill>().First().Alpha == 0);
        }

        private void createHeader(string text, CounterVisibilityState state, int initialValue = 0)
        {
            Clear();
            Add(header = new PaginatedContainerHeader(text, state)
            {
                Anchor = Anchor.Centre,
                Origin = Anchor.Centre,
                Current = { Value = initialValue }
            });
        }
    }
}