improved the unit tests

kerryjiang · Jan 20, 2025 · f5f775f · f5f775f
1 parent 47dd09a
commit f5f775f
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/test/SuperSocket.Tests/ProtocolTestBase.cs b/test/SuperSocket.Tests/ProtocolTestBase.cs
@@ -44,7 +44,9 @@ public virtual async Task TestNormalRequest(Type hostConfiguratorType)
 
             using (var server = CreateServer(hostConfigurator))
             {
-                await server.StartAsync();
+                Assert.True(await server.StartAsync());
+
+                OutputHelper.WriteLine("The server has been started.");
 
                 using (var socket = CreateClient(hostConfigurator))
                 {
@@ -56,7 +58,7 @@ public virtual async Task TestNormalRequest(Type hostConfiguratorType)
                         writer.Write(CreateRequest(line));
                         writer.Flush();
 
-                        var receivedLine = await reader.ReadLineAsync();
+                        var receivedLine = await reader.ReadLineAsync(TestContext.Current.CancellationToken);
                         Assert.Equal(line, receivedLine);
                     }
                 }
@@ -78,7 +80,9 @@ public virtual async Task TestMiddleBreak(Type hostConfiguratorType)
 
             using (var server = CreateServer(hostConfigurator))
             {
-                await server.StartAsync();
+                Assert.True(await server.StartAsync());
+
+                OutputHelper.WriteLine("The server has been started.");
 
                 for (var i = 0; i < 100; i++)
                 {
@@ -114,7 +118,9 @@ public virtual async Task TestFragmentRequest(Type hostConfiguratorType)
 
             using (var server = CreateServer(hostConfigurator))
             {
-                await server.StartAsync();
+                Assert.True(await server.StartAsync());
+
+                OutputHelper.WriteLine("The server has been started.");
 
                 using (var socket = CreateClient(hostConfigurator))
                 {
@@ -133,7 +139,7 @@ public virtual async Task TestFragmentRequest(Type hostConfiguratorType)
                             await hostConfigurator.KeepSequence();
                         }
 
-                        var receivedLine = await reader.ReadLineAsync();
+                        var receivedLine = await reader.ReadLineAsync(TestContext.Current.CancellationToken);
                         Assert.Equal(line, receivedLine);
                     }
                 }
@@ -155,7 +161,9 @@ public virtual async Task TestBatchRequest(Type hostConfiguratorType)
 
             using (var server = CreateServer(hostConfigurator))
             {
-                await server.StartAsync();
+                Assert.True(await server.StartAsync());
+
+                OutputHelper.WriteLine("The server has been started.");
 
                 using (var socket = CreateClient(hostConfigurator))
                 {
@@ -181,7 +189,7 @@ public virtual async Task TestBatchRequest(Type hostConfiguratorType)
 
                         for (var i = 0; i < size; i++)
                         {
-                            var receivedLine = await reader.ReadLineAsync();
+                            var receivedLine = await reader.ReadLineAsync(TestContext.Current.CancellationToken);
                             Assert.Equal(lines[i], receivedLine);
                         }
                     }
@@ -204,7 +212,9 @@ public virtual async Task TestBreakRequest(Type hostConfiguratorType)
 
             using (var server = CreateServer(hostConfigurator))
             {
-                await server.StartAsync();
+                Assert.True(await server.StartAsync());
+
+                OutputHelper.WriteLine("The server has been started.");
 
                 using (var socket = CreateClient(hostConfigurator))
                 {
@@ -257,7 +267,7 @@ public virtual async Task TestBreakRequest(Type hostConfiguratorType)
 
                         for (var i = 0; i < size; i++)
                         {
-                            var receivedLine = await reader.ReadLineAsync();
+                            var receivedLine = await reader.ReadLineAsync(TestContext.Current.CancellationToken);
                             Assert.Equal(lines[i], receivedLine);
                         }
                     }