mirror of
https://github.com/osukey/osukey.git
synced 2025-08-05 07:33:55 +09:00
Merge pull request #11551 from smoogipoo/fix-leave-room-race-2
Fix client potentially not leaving a room on exiting multiplayer
This commit is contained in:
111
osu.Game.Tests/NonVisual/TaskChainTest.cs
Normal file
111
osu.Game.Tests/NonVisual/TaskChainTest.cs
Normal file
@ -0,0 +1,111 @@
|
||||
// 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.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using NUnit.Framework;
|
||||
using osu.Game.Utils;
|
||||
|
||||
namespace osu.Game.Tests.NonVisual
|
||||
{
|
||||
[TestFixture]
|
||||
public class TaskChainTest
|
||||
{
|
||||
private TaskChain taskChain;
|
||||
private int currentTask;
|
||||
private CancellationTokenSource globalCancellationToken;
|
||||
|
||||
[SetUp]
|
||||
public void Setup()
|
||||
{
|
||||
globalCancellationToken = new CancellationTokenSource();
|
||||
taskChain = new TaskChain();
|
||||
currentTask = 0;
|
||||
}
|
||||
|
||||
[TearDown]
|
||||
public void TearDown()
|
||||
{
|
||||
globalCancellationToken?.Cancel();
|
||||
}
|
||||
|
||||
[Test]
|
||||
public async Task TestChainedTasksRunSequentially()
|
||||
{
|
||||
var task1 = addTask();
|
||||
var task2 = addTask();
|
||||
var task3 = addTask();
|
||||
|
||||
task3.mutex.Set();
|
||||
task2.mutex.Set();
|
||||
task1.mutex.Set();
|
||||
|
||||
await Task.WhenAll(task1.task, task2.task, task3.task);
|
||||
|
||||
Assert.That(task1.task.Result, Is.EqualTo(1));
|
||||
Assert.That(task2.task.Result, Is.EqualTo(2));
|
||||
Assert.That(task3.task.Result, Is.EqualTo(3));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public async Task TestChainedTaskWithIntermediateCancelRunsInSequence()
|
||||
{
|
||||
var task1 = addTask();
|
||||
var task2 = addTask();
|
||||
var task3 = addTask();
|
||||
|
||||
// Cancel task2, allow task3 to complete.
|
||||
task2.cancellation.Cancel();
|
||||
task2.mutex.Set();
|
||||
task3.mutex.Set();
|
||||
|
||||
// Allow task3 to potentially complete.
|
||||
Thread.Sleep(1000);
|
||||
|
||||
// Allow task1 to complete.
|
||||
task1.mutex.Set();
|
||||
|
||||
// Wait on both tasks.
|
||||
await Task.WhenAll(task1.task, task3.task);
|
||||
|
||||
Assert.That(task1.task.Result, Is.EqualTo(1));
|
||||
Assert.That(task2.task.IsCompleted, Is.False);
|
||||
Assert.That(task3.task.Result, Is.EqualTo(2));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public async Task TestChainedTaskDoesNotCompleteBeforeChildTasks()
|
||||
{
|
||||
var mutex = new ManualResetEventSlim(false);
|
||||
|
||||
var task = taskChain.Add(async () => await Task.Run(() => mutex.Wait(globalCancellationToken.Token)));
|
||||
|
||||
// Allow task to potentially complete
|
||||
Thread.Sleep(1000);
|
||||
|
||||
Assert.That(task.IsCompleted, Is.False);
|
||||
|
||||
// Allow the task to complete.
|
||||
mutex.Set();
|
||||
|
||||
await task;
|
||||
}
|
||||
|
||||
private (Task<int> task, ManualResetEventSlim mutex, CancellationTokenSource cancellation) addTask()
|
||||
{
|
||||
var mutex = new ManualResetEventSlim(false);
|
||||
var completionSource = new TaskCompletionSource<int>();
|
||||
|
||||
var cancellationSource = new CancellationTokenSource();
|
||||
var token = CancellationTokenSource.CreateLinkedTokenSource(cancellationSource.Token, globalCancellationToken.Token);
|
||||
|
||||
taskChain.Add(() =>
|
||||
{
|
||||
mutex.Wait(globalCancellationToken.Token);
|
||||
completionSource.SetResult(Interlocked.Increment(ref currentTask));
|
||||
}, token.Token);
|
||||
|
||||
return (completionSource.Task, mutex, cancellationSource);
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user