osu-lazer/osu.Game.Tests/Visual/Navigation/TestSceneBeatmapEditorNavigation.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 DeepEqual.Syntax;
using NUnit.Framework;
using osu.Framework.Screens;
using osu.Game.Beatmaps;
using osu.Game.Screens.Edit;
using osu.Game.Screens.Menu;

namespace osu.Game.Tests.Visual.Navigation
{
    public partial class TestSceneBeatmapEditorNavigation : OsuGameTestScene
    {
        [Test]
        public void TestCancelNavigationToEditor()
        {
            BeatmapSetInfo[]? beatmapSets = null;

            AddStep("Timestamp current beatmapsets", () =>
            {
                Game.Realm.Run(realm =>
                {
                    beatmapSets = realm.All<BeatmapSetInfo>().Where(x => !x.DeletePending).ToArray();
                });
            });

            AddStep("Set current beatmap to default", () =>
            {
                Game.Beatmap.SetDefault();
            });

            AddStep("Open editor loader", () =>
            {
                Game.ScreenStack.Push(new EditorLoader());
            });

            AddUntilStep("wait for editor", () => Game.ScreenStack.CurrentScreen is EditorLoader);

            AddStep("close editor while loading", () =>
            {
                Game.ScreenStack.CurrentScreen.Exit();
            });

            AddUntilStep("wait for editor", () => Game.ScreenStack.CurrentScreen is MainMenu);

            BeatmapSetInfo[]? currentSetInfos = null;

            AddStep("Get current beatmaps", () =>
            {
                Game.Realm.Run(realm =>
                {
                    currentSetInfos = realm.All<BeatmapSetInfo>().Where(x => !x.DeletePending).ToArray();
                });
            });

            AddAssert("dummy beatmap didn't appear", () => currentSetInfos.IsDeepEqual(beatmapSets) && currentSetInfos is not null);
        }
    }
}