feat: add max connections testing

Author: insthync

SimpleNetworkManager.NET.Tests/Network/TcpTransport/TcpTransportTests.cs

@@ -42,30 +42,6 @@ public async Task TestSimpleConnection()
             Assert.False(client.IsConnected);
         }
 
-        [Fact]
-        public async Task TestReuseClientConnection()
-        {
-            var server = new TcpNetworkServer(_loggerFactoryMock.Object);
-            var client = new TcpNetworkClient(_loggerFactoryMock.Object);
-
-            var serverCancelSrc = new CancellationTokenSource();
-            await server.StartAsync(7890, serverCancelSrc.Token);
-            Assert.True(server.IsRunning);
-
-            for (int i = 0; i < 10; ++i)
-            {
-                var clientCancelSrc = new CancellationTokenSource();
-                await client.ConnectAsync("127.0.0.1", 7890, clientCancelSrc.Token);
-                Assert.True(client.IsConnected);
-                await client.DisconnectAsync();
-                Assert.False(client.IsConnected);
-            }
-
-            await server.StopAsync();
-
-            Assert.False(server.IsRunning);
-        }
-
         [Fact]
         public async Task TestClientDisconnectionFromServer()
         {
@@ -200,5 +176,82 @@ public async Task TestMessageFromServerHandling()
             Assert.False(server.IsRunning);
             Assert.False(client.IsConnected);
         }
+
+        [Fact]
+        public async Task TestReuseClientConnection()
+        {
+            var server = new TcpNetworkServer(_loggerFactoryMock.Object);
+            var client = new TcpNetworkClient(_loggerFactoryMock.Object);
+
+            var serverCancelSrc = new CancellationTokenSource();
+            await server.StartAsync(7895, serverCancelSrc.Token);
+            Assert.True(server.IsRunning);
+
+            for (int i = 0; i < 10; ++i)
+            {
+                var clientCancelSrc = new CancellationTokenSource();
+                await client.ConnectAsync("127.0.0.1", 7895, clientCancelSrc.Token);
+                // Wait a bit for connection acceptance
+                await Task.Delay(100);
+                Assert.True(client.IsConnected);
+                await client.DisconnectAsync();
+                // Wait a bit for disconnection
+                await Task.Delay(100);
+                Assert.False(client.IsConnected);
+            }
+
+            await server.StopAsync();
+
+            Assert.False(server.IsRunning);
+        }
+
+        [Fact]
+        public async Task TestClientMaxConnections()
+        {
+            var server = new TcpNetworkServer(_loggerFactoryMock.Object);
+            server.MaxConnections = 2;
+
+            var serverCancelSrc = new CancellationTokenSource();
+            await server.StartAsync(7896, serverCancelSrc.Token);
+            Assert.True(server.IsRunning);
+
+            // Client 1 - must be able to connection
+            var client1 = new TcpNetworkClient(_loggerFactoryMock.Object);
+            var clientCancelSrc = new CancellationTokenSource();
+            await client1.ConnectAsync("127.0.0.1", 7896, clientCancelSrc.Token);
+            // Wait a second for connection acceptance
+            await Task.Delay(1000);
+            Assert.True(client1.IsConnected);
+
+            // Client 2 - must be able to connection
+            var client2 = new TcpNetworkClient(_loggerFactoryMock.Object);
+            clientCancelSrc = new CancellationTokenSource();
+            await client2.ConnectAsync("127.0.0.1", 7896, clientCancelSrc.Token);
+            // Wait a second for connection acceptance
+            await Task.Delay(1000);
+            Assert.True(client2.IsConnected);
+
+            // Client 3 - must not be able to connection
+            var client3 = new TcpNetworkClient(_loggerFactoryMock.Object);
+            clientCancelSrc = new CancellationTokenSource();
+            await client3.ConnectAsync("127.0.0.1", 7896, clientCancelSrc.Token);
+            // Wait a second for connection acceptance
+            await Task.Delay(1000);
+            Assert.False(client3.IsConnected);
+
+            await client1.DisconnectAsync();
+            await client2.DisconnectAsync();
+            try
+            {
+                await client3.DisconnectAsync();
+            } catch (Exception ex)
+            {
+                Assert.IsType<InvalidOperationException>(ex);
+            }
+
+            await server.StopAsync();
+
+            Assert.False(server.IsRunning);
+        }
     }
 }