Add test

rabbitmq · Nov 15, 2023 · 7a2fcf0 · 7a2fcf0
1 parent 55dfc54
commit 7a2fcf0
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/projects/Unit/TestConnectionFactory.cs b/projects/Unit/TestConnectionFactory.cs
@@ -30,7 +30,9 @@
 //---------------------------------------------------------------------------
 
 using System.Collections.Generic;
+using System.Net.Sockets;
 using RabbitMQ.Client.Exceptions;
+using RabbitMQ.Client.Impl;
 using Xunit;
 
 namespace RabbitMQ.Client.Unit
@@ -69,6 +71,34 @@ public void TestProperties()
             Assert.Equal(cf.Endpoint.MaxMessageSize, mms);
         }
 
+        [Fact]
+        public void TestConnectionFactoryWithCustomSocketFactory()
+        {
+            const int bufsz = 1024;
+
+            ConnectionFactory cf = new()
+            {
+                SocketFactory = (AddressFamily af) =>
+                {
+                    var socket = new Socket(af, SocketType.Stream, ProtocolType.Tcp)
+                    {
+                        SendBufferSize = bufsz,
+                        ReceiveBufferSize = bufsz,
+                        NoDelay = false
+                    };
+                    return new TcpClientAdapter(socket);
+                }
+            };
+
+            ITcpClient c = cf.SocketFactory(AddressFamily.InterNetwork);
+            Assert.IsType<TcpClientAdapter>(c);
+            TcpClientAdapter tcpClientAdapter = (TcpClientAdapter)c;
+            Socket s = tcpClientAdapter.Client;
+            Assert.Equal(bufsz, s.ReceiveBufferSize);
+            Assert.Equal(bufsz, s.SendBufferSize);
+            Assert.False(s.NoDelay);
+        }
+
         [Fact]
         public void TestCreateConnectionUsesSpecifiedPort()
         {