Add TransportConnection IConnectionEndPointFeature tests

Co-authored-by: BrennanConroy <[email protected]>

Author: Copilot

src/Servers/Kestrel/Core/test/TransportConnectionFeatureCollectionTests.cs

@@ -0,0 +1,103 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System;
+using System.Net;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Connections;
+using Microsoft.AspNetCore.Connections.Features;
+using Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Infrastructure;
+using Microsoft.AspNetCore.InternalTesting;
+using Moq;
+using Xunit;
+
+namespace Microsoft.AspNetCore.Server.Kestrel.Core.Tests;
+
+public class TransportConnectionFeatureCollectionTests
+{
+    [Fact]
+    public void IConnectionEndPointFeature_IsAvailableInFeatureCollection()
+    {
+        var serviceContext = new TestServiceContext();
+        var connection = new Mock<DefaultConnectionContext> { CallBase = true }.Object;
+        var transportConnectionManager = new TransportConnectionManager(serviceContext.ConnectionManager);
+        var kestrelConnection = CreateKestrelConnection(serviceContext, connection, transportConnectionManager);
+        
+        var endpointFeature = kestrelConnection.TransportConnection.Features.Get<IConnectionEndPointFeature>();
+        
+        Assert.NotNull(endpointFeature);
+    }
+
+    [Fact]
+    public void IConnectionEndPointFeature_ReturnsCorrectLocalEndPoint()
+    {
+        var serviceContext = new TestServiceContext();
+        var connection = new Mock<DefaultConnectionContext> { CallBase = true }.Object;
+        var expectedLocalEndPoint = new IPEndPoint(IPAddress.Parse("127.0.0.1"), 8080);
+        connection.LocalEndPoint = expectedLocalEndPoint;
+        var transportConnectionManager = new TransportConnectionManager(serviceContext.ConnectionManager);
+        var kestrelConnection = CreateKestrelConnection(serviceContext, connection, transportConnectionManager);
+        
+        var endpointFeature = kestrelConnection.TransportConnection.Features.Get<IConnectionEndPointFeature>();
+        
+        Assert.NotNull(endpointFeature);
+        Assert.Equal(expectedLocalEndPoint, endpointFeature.LocalEndPoint);
+    }
+
+    [Fact]
+    public void IConnectionEndPointFeature_ReturnsCorrectRemoteEndPoint()
+    {
+        var serviceContext = new TestServiceContext();
+        var connection = new Mock<DefaultConnectionContext> { CallBase = true }.Object;
+        var expectedRemoteEndPoint = new IPEndPoint(IPAddress.Parse("192.168.1.100"), 54321);
+        connection.RemoteEndPoint = expectedRemoteEndPoint;
+        var transportConnectionManager = new TransportConnectionManager(serviceContext.ConnectionManager);
+        var kestrelConnection = CreateKestrelConnection(serviceContext, connection, transportConnectionManager);
+        
+        var endpointFeature = kestrelConnection.TransportConnection.Features.Get<IConnectionEndPointFeature>();
+        
+        Assert.NotNull(endpointFeature);
+        Assert.Equal(expectedRemoteEndPoint, endpointFeature.RemoteEndPoint);
+    }
+
+    [Fact]
+    public void IConnectionEndPointFeature_AllowsSettingLocalEndPoint()
+    {
+        var serviceContext = new TestServiceContext();
+        var connection = new Mock<DefaultConnectionContext> { CallBase = true }.Object;
+        var transportConnectionManager = new TransportConnectionManager(serviceContext.ConnectionManager);
+        var kestrelConnection = CreateKestrelConnection(serviceContext, connection, transportConnectionManager);
+        var newLocalEndPoint = new IPEndPoint(IPAddress.Parse("127.0.0.1"), 9090);
+        
+        var endpointFeature = kestrelConnection.TransportConnection.Features.Get<IConnectionEndPointFeature>();
+        endpointFeature.LocalEndPoint = newLocalEndPoint;
+        
+        Assert.Equal(newLocalEndPoint, kestrelConnection.TransportConnection.LocalEndPoint);
+        Assert.Equal(newLocalEndPoint, endpointFeature.LocalEndPoint);
+    }
+
+    [Fact]
+    public void IConnectionEndPointFeature_AllowsSettingRemoteEndPoint()
+    {
+        var serviceContext = new TestServiceContext();
+        var connection = new Mock<DefaultConnectionContext> { CallBase = true }.Object;
+        var transportConnectionManager = new TransportConnectionManager(serviceContext.ConnectionManager);
+        var kestrelConnection = CreateKestrelConnection(serviceContext, connection, transportConnectionManager);
+        var newRemoteEndPoint = new IPEndPoint(IPAddress.Parse("10.0.0.1"), 12345);
+        
+        var endpointFeature = kestrelConnection.TransportConnection.Features.Get<IConnectionEndPointFeature>();
+        endpointFeature.RemoteEndPoint = newRemoteEndPoint;
+        
+        Assert.Equal(newRemoteEndPoint, kestrelConnection.TransportConnection.RemoteEndPoint);
+        Assert.Equal(newRemoteEndPoint, endpointFeature.RemoteEndPoint);
+    }
+
+    private static KestrelConnection<ConnectionContext> CreateKestrelConnection(TestServiceContext serviceContext, DefaultConnectionContext connection, TransportConnectionManager transportConnectionManager, Func<ConnectionContext, Task> connectionDelegate = null)
+    {
+        connectionDelegate ??= _ => Task.CompletedTask;
+
+        return new KestrelConnection<ConnectionContext>(
+            id: 0, serviceContext, transportConnectionManager, connectionDelegate, connection, serviceContext.Log, TestContextFactory.CreateMetricsContext(connection));
+    }
+}