mirror of https://github.com/ppy/osu
71 lines
2.4 KiB
C#
71 lines
2.4 KiB
C#
// 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.
|
|
|
|
#nullable disable
|
|
|
|
using System;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using NUnit.Framework;
|
|
using osu.Framework.Allocation;
|
|
using osu.Framework.Extensions;
|
|
using osu.Framework.Platform;
|
|
using osu.Game.Collections;
|
|
using osu.Game.Tests.Resources;
|
|
|
|
namespace osu.Game.Tests
|
|
{
|
|
public abstract class ImportTest
|
|
{
|
|
protected virtual TestOsuGameBase LoadOsuIntoHost(GameHost host, bool withBeatmap = false)
|
|
{
|
|
var osu = new TestOsuGameBase(withBeatmap);
|
|
Task.Factory.StartNew(() => host.Run(osu), TaskCreationOptions.LongRunning)
|
|
.ContinueWith(t => Assert.Fail($"Host threw exception {t.Exception}"), TaskContinuationOptions.OnlyOnFaulted);
|
|
|
|
waitForOrAssert(() => osu.IsLoaded, @"osu! failed to start in a reasonable amount of time");
|
|
|
|
bool ready = false;
|
|
// wait for two update frames to be executed. this ensures that all components have had a change to run LoadComplete and hopefully avoid
|
|
// database access (GlobalActionContainer is one to do this).
|
|
host.UpdateThread.Scheduler.Add(() => host.UpdateThread.Scheduler.Add(() => ready = true));
|
|
|
|
waitForOrAssert(() => ready, @"osu! failed to start in a reasonable amount of time");
|
|
|
|
return osu;
|
|
}
|
|
|
|
private void waitForOrAssert(Func<bool> result, string failureMessage, int timeout = 60000)
|
|
{
|
|
Task task = Task.Run(() =>
|
|
{
|
|
while (!result()) Thread.Sleep(200);
|
|
});
|
|
|
|
Assert.IsTrue(task.Wait(timeout), failureMessage);
|
|
}
|
|
|
|
public class TestOsuGameBase : OsuGameBase
|
|
{
|
|
public CollectionManager CollectionManager { get; private set; }
|
|
|
|
private readonly bool withBeatmap;
|
|
|
|
public TestOsuGameBase(bool withBeatmap)
|
|
{
|
|
this.withBeatmap = withBeatmap;
|
|
}
|
|
|
|
[BackgroundDependencyLoader]
|
|
private void load()
|
|
{
|
|
// Beatmap must be imported before the collection manager is loaded.
|
|
if (withBeatmap)
|
|
BeatmapManager.Import(TestResources.GetTestBeatmapForImport()).WaitSafely();
|
|
|
|
AddInternal(CollectionManager = new CollectionManager(Storage));
|
|
}
|
|
}
|
|
}
|
|
}
|