ydeng/gameservicewarden
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added automatic cancel to cancel tokens in tests.

Browse Source
This commit is contained in:
Harrison Deng 2021-04-09 01:30:10 -05:00
parent a44ad71f59
commit 1ae4de4f73
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/GameServiceWarden.Core.Tests/Modules/Games/ServiceManagerTest.cs
View File

@ -226,10 +226,11 @@ namespace GameServiceWarden.Core.Tests.Modules.Games
Thread.Sleep(1000); Thread.Sleep(1000);
//Then //Then
byte[] buffer = new byte[1024 * 8]; byte[] buffer = new byte[1024 * 8];
CancellationTokenSource cancelToken = new CancellationTokenSource(); CancellationTokenSource cancelToken = new CancellationTokenSource(2000);
ValueTask<int> task = clientStream.ReadAsync(buffer, cancelToken.Token); ValueTask<int> task = clientStream.ReadAsync(buffer, cancelToken.Token);
Assert.False(task.AsTask().Wait(1000)); Assert.False(task.AsTask().Wait(1000));
serviceManager.StopService(FAKE_SERVICE_NAME); serviceManager.StopService(FAKE_SERVICE_NAME);
cancelToken.Dispose();
} }
[Fact] [Fact]
@ -258,11 +259,12 @@ namespace GameServiceWarden.Core.Tests.Modules.Games
//Then //Then
using (StreamReader reader = new StreamReader(clientStream)) using (StreamReader reader = new StreamReader(clientStream))
{ {
CancellationTokenSource cancelToken = new CancellationTokenSource(); CancellationTokenSource cancelToken = new CancellationTokenSource(2000);
string message = null; string message = null;
Task task = Task.Run(() => message = reader.ReadLine(), cancelToken.Token); Task task = Task.Run(() => message = reader.ReadLine(), cancelToken.Token);
Assert.True(task.Wait(1000)); Assert.True(task.Wait(1000));
Assert.True(COMMAND.Equals(message), $"Received message \"{message}\" when expecting \"{COMMAND}\""); Assert.True(COMMAND.Equals(message), $"Received message \"{message}\" when expecting \"{COMMAND}\"");
cancelToken.Dispose();
} }
serviceManager.StopService(FAKE_SERVICE_NAME); serviceManager.StopService(FAKE_SERVICE_NAME);
} }
@ -299,11 +301,12 @@ namespace GameServiceWarden.Core.Tests.Modules.Games
{ {
using (StreamReader reader = new StreamReader(clientStreams[i])) using (StreamReader reader = new StreamReader(clientStreams[i]))
{ {
CancellationTokenSource cancelToken = new CancellationTokenSource(); CancellationTokenSource cancelToken = new CancellationTokenSource(15000);
string message = null; string message = null;
Task task = Task.Run(() => message = reader.ReadLine(), cancelToken.Token); Task task = Task.Run(() => message = reader.ReadLine(), cancelToken.Token);
Assert.True(task.Wait(10000)); Assert.True(task.Wait(10000));
Assert.True(COMMAND.Equals(message), $"Received message \"{message}\" when expecting \"{COMMAND}\""); Assert.True(COMMAND.Equals(message), $"Received message \"{message}\" when expecting \"{COMMAND}\"");
cancelToken.Dispose();
} }
} }
serviceManager.StopService(FAKE_SERVICE_NAME); serviceManager.StopService(FAKE_SERVICE_NAME);