Add overloads to DestinationState and ClusterState ctors (#2198)

src/ReverseProxy/Model/ClusterState.cs

@@ -3,6 +3,8 @@
 
 using System;
 using System.Collections.Concurrent;
+using Microsoft.AspNetCore.Http;
+using Yarp.ReverseProxy.Configuration;
 using Yarp.ReverseProxy.Utilities;
 
 namespace Yarp.ReverseProxy.Model;
@@ -24,6 +26,16 @@ public ClusterState(string clusterId)
         ClusterId = clusterId ?? throw new ArgumentNullException(nameof(clusterId));
     }
 
+    /// <summary>
+    /// Constructor overload to additionally initialize the <see cref="ClusterModel"/> for tests and infrastructure,
+    /// such as updating the <see cref="ReverseProxyFeature"/> via <see cref="HttpContextFeaturesExtensions"/>
+    /// </summary>
+    /// <exception cref="ArgumentNullException"><paramref name="model"/> is <see langword="null"/>.</exception>
+    public ClusterState(string clusterId, ClusterModel model) : this(clusterId)
+    {
+        Model = model ?? throw new ArgumentNullException(nameof(model));
+    }
+
     /// <summary>
     /// The cluster's unique id.
     /// </summary>

src/ReverseProxy/Model/DestinationState.cs

@@ -4,6 +4,7 @@
 using System;
 using System.Collections;
 using System.Collections.Generic;
+using Microsoft.AspNetCore.Http;
 using Yarp.ReverseProxy.Utilities;
 
 namespace Yarp.ReverseProxy.Model;
@@ -28,6 +29,16 @@ public DestinationState(string destinationId)
         DestinationId = destinationId;
     }
 
+    /// <summary>
+    /// Constructor overload to additionally initialize the <see cref="DestinationModel"/> for tests and infrastructure,
+    /// such as updating the <see cref="ReverseProxyFeature"/> via <see cref="HttpContextFeaturesExtensions"/>
+    /// </summary>
+    /// <exception cref="ArgumentNullException"><paramref name="model"/> is <see langword="null"/>.</exception>
+    public DestinationState(string destinationId, DestinationModel model) : this(destinationId)
+    {
+        Model = model ?? throw new ArgumentNullException(nameof(model));
+    }
+
     /// <summary>
     /// The destination's unique id.
     /// </summary>