From 7a2fcf097573d0f620a81eada75b32c329904369 Mon Sep 17 00:00:00 2001
From: Luke Bakken <luke@bakken.io>
Date: Wed, 15 Nov 2023 10:53:58 -0800
Subject: [PATCH] Add test

---
 projects/Unit/TestConnectionFactory.cs | 30 ++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/projects/Unit/TestConnectionFactory.cs b/projects/Unit/TestConnectionFactory.cs
index 225c94f99d..1344a46fd8 100644
--- 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()
         {