// Copyright (c) ppy Pty Ltd . Licensed under the MIT Licence. // See the LICENCE file in the repository root for full licence text. #nullable disable using System.Linq; using NUnit.Framework; using osu.Framework.Graphics; using osu.Game.Online; using osu.Game.Online.API.Requests.Responses; using osu.Framework.Allocation; using osu.Framework.Bindables; using osu.Framework.Extensions; using osu.Framework.Testing; using osu.Game.Beatmaps; using osu.Game.Database; using osu.Game.Graphics.UserInterface; using osu.Game.Rulesets; using osu.Game.Rulesets.Osu; using osu.Game.Scoring; using osu.Game.Screens.Ranking; using osu.Game.Tests.Resources; using osuTK.Input; using APIUser = osu.Game.Online.API.Requests.Responses.APIUser; namespace osu.Game.Tests.Visual.Gameplay { [TestFixture] public class TestSceneReplayDownloadButton : OsuManualInputManagerTestScene { private const long online_score_id = 2553163309; [Resolved] private RulesetStore rulesets { get; set; } private TestReplayDownloadButton downloadButton; [Resolved] private BeatmapManager beatmapManager { get; set; } [Resolved] private ScoreManager scoreManager { get; set; } [BackgroundDependencyLoader] private void load() { beatmapManager.Import(TestResources.GetQuickTestBeatmapForImport()).WaitSafely(); } [SetUpSteps] public void SetUpSteps() { AddStep("delete previous imports", () => { scoreManager.Delete(s => s.OnlineID == online_score_id); }); } [Test] public void TestDisplayStates() { AddStep(@"create button with replay", () => { Child = downloadButton = new TestReplayDownloadButton(getScoreInfo(true)) { Anchor = Anchor.Centre, Origin = Anchor.Centre, }; }); AddUntilStep("wait for load", () => downloadButton.IsLoaded); AddStep(@"downloading state", () => downloadButton.SetDownloadState(DownloadState.Downloading)); AddStep(@"locally available state", () => downloadButton.SetDownloadState(DownloadState.LocallyAvailable)); AddStep(@"not downloaded state", () => downloadButton.SetDownloadState(DownloadState.NotDownloaded)); } [Test] public void TestButtonWithReplayStartsDownload() { bool downloadStarted = false; bool downloadFinished = false; AddStep(@"create button with replay", () => { downloadStarted = false; downloadFinished = false; Child = downloadButton = new TestReplayDownloadButton(getScoreInfo(true)) { Anchor = Anchor.Centre, Origin = Anchor.Centre, }; downloadButton.State.BindValueChanged(state => { switch (state.NewValue) { case DownloadState.Downloading: downloadStarted = true; break; } switch (state.OldValue) { case DownloadState.Downloading: downloadFinished = true; break; } }); }); AddUntilStep("wait for load", () => downloadButton.IsLoaded); AddAssert("state is available", () => downloadButton.State.Value == DownloadState.NotDownloaded); AddStep("click button", () => { InputManager.MoveMouseTo(downloadButton); InputManager.Click(MouseButton.Left); }); AddAssert("state entered downloading", () => downloadStarted); AddUntilStep("state left downloading", () => downloadFinished); } [Test] public void TestButtonWithoutReplay() { AddStep("create button without replay", () => { Child = downloadButton = new TestReplayDownloadButton(getScoreInfo(false)) { Anchor = Anchor.Centre, Origin = Anchor.Centre, }; }); AddUntilStep("wait for load", () => downloadButton.IsLoaded); AddAssert("state is not downloaded", () => downloadButton.State.Value == DownloadState.NotDownloaded); AddAssert("button is not enabled", () => !downloadButton.ChildrenOfType().First().Enabled.Value); } [Test] public void TestScoreImportThenDelete() { Live imported = null; AddStep("create button without replay", () => { Child = downloadButton = new TestReplayDownloadButton(getScoreInfo(false)) { Anchor = Anchor.Centre, Origin = Anchor.Centre, }; }); AddUntilStep("wait for load", () => downloadButton.IsLoaded); AddUntilStep("state is not downloaded", () => downloadButton.State.Value == DownloadState.NotDownloaded); AddStep("import score", () => imported = scoreManager.Import(getScoreInfo(true))); AddUntilStep("state is available", () => downloadButton.State.Value == DownloadState.LocallyAvailable); AddAssert("button is enabled", () => downloadButton.ChildrenOfType().First().Enabled.Value); AddStep("delete score", () => scoreManager.Delete(imported.Value)); AddUntilStep("state is not downloaded", () => downloadButton.State.Value == DownloadState.NotDownloaded); AddAssert("button is not enabled", () => !downloadButton.ChildrenOfType().First().Enabled.Value); } [Test] public void CreateButtonWithNoScore() { AddStep("create button with null score", () => { Child = downloadButton = new TestReplayDownloadButton(null) { Anchor = Anchor.Centre, Origin = Anchor.Centre, }; }); AddUntilStep("wait for load", () => downloadButton.IsLoaded); AddAssert("state is not downloaded", () => downloadButton.State.Value == DownloadState.NotDownloaded); AddAssert("button is not enabled", () => !downloadButton.ChildrenOfType().First().Enabled.Value); } private ScoreInfo getScoreInfo(bool replayAvailable) { return new APIScore { OnlineID = online_score_id, RulesetID = 0, Beatmap = CreateAPIBeatmapSet(new OsuRuleset().RulesetInfo).Beatmaps.First(), HasReplay = replayAvailable, User = new APIUser { Id = 39828, Username = @"WubWoofWolf", } }.CreateScoreInfo(rulesets, beatmapManager.GetAllUsableBeatmapSets().First().Beatmaps.First()); } private class TestReplayDownloadButton : ReplayDownloadButton { public void SetDownloadState(DownloadState state) => State.Value = state; public new Bindable State => base.State; public TestReplayDownloadButton(ScoreInfo score) : base(score) { } } } }