CSHARP-5608: CSOT: Command Execution (#1709)

Author: sanych-sun

src/MongoDB.Driver/Core/Clusters/Cluster.cs

@@ -22,9 +22,9 @@ namespace MongoDB.Driver.Core.Clusters.ServerSelectors
 {
     internal sealed class OperationsCountServerSelector : IServerSelector
     {
-        private readonly IEnumerable<IClusterableServer> _clusterableServers;
+        private IReadOnlyList<IClusterableServer> _clusterableServers;
 
-        public OperationsCountServerSelector(IEnumerable<IClusterableServer> clusterableServers)
+        public OperationsCountServerSelector(IReadOnlyList<IClusterableServer> clusterableServers)
         {
             _clusterableServers = clusterableServers;
         }
@@ -58,6 +58,11 @@ public IEnumerable<ServerDescription> SelectServers(ClusterDescription cluster,
             }
         }
 
+        public void PopulateServers(IReadOnlyList<IClusterableServer> clusterableServers)
+        {
+            _clusterableServers = clusterableServers;
+        }
+
         /// <inheritdoc/>
         public override string ToString() =>
             nameof(OperationsCountServerSelector);

src/MongoDB.Driver/Core/Clusters/ServerSelectors/OperationsCountServerSelector.cs

@@ -289,7 +289,7 @@ private IServerSelector CreateWritableServerAndEndPointSelector(EndPoint endPoin
         private void ForceClusterId(MultiServerCluster cluster, ClusterId clusterId)
         {
             Reflector.SetFieldValue(cluster, "_clusterId", clusterId);
-            Reflector.SetFieldValue(cluster, "_descriptionWithChangedTaskCompletionSource", new Cluster.ClusterDescriptionChangeSource(ClusterDescription.CreateInitial(clusterId, __directConnection)));
+            Reflector.SetFieldValue(cluster, "_expirableClusterDescription", new Cluster.ExpirableClusterDescription(cluster, ClusterDescription.CreateInitial(clusterId, __directConnection)));
         }
 
         private void SetupServerMonitorConnection(

tests/MongoDB.Driver.Tests/Core/Jira/CSharp3173Tests.cs

@@ -270,7 +270,7 @@ private IServerSelector CreateWritableServerAndEndPointSelector(EndPoint endPoin
         private void ForceClusterId(MultiServerCluster cluster, ClusterId clusterId)
         {
             Reflector.SetFieldValue(cluster, "_clusterId", clusterId);
-            Reflector.SetFieldValue(cluster, "_descriptionWithChangedTaskCompletionSource", new Cluster.ClusterDescriptionChangeSource(ClusterDescription.CreateInitial(clusterId, __directConnection)));
+            Reflector.SetFieldValue(cluster, "_expirableClusterDescription", new Cluster.ExpirableClusterDescription(cluster, ClusterDescription.CreateInitial(clusterId, __directConnection)));
         }
 
         private void SetupServerMonitorConnection(

tests/MongoDB.Driver.Tests/Core/Jira/CSharp3302Tests.cs

@@ -94,7 +94,7 @@ private static IEnumerable<object[]> EventsData()
                 (new LogCategories.Command(), new CommandStartedEvent("test", new Bson.BsonDocument(), new DatabaseNamespace("test"), 1, 1, connectionId)),
                 (new LogCategories.Connection(), new ConnectionCreatedEvent(connectionId, null, 1)),
                 (new LogCategories.SDAM(), new ServerHeartbeatStartedEvent(connectionId, true)),
-                (new LogCategories.ServerSelection(), new ClusterSelectingServerEvent(clusterDescription, new RandomServerSelector(), default, default))
+                (new LogCategories.ServerSelection(), new ClusterSelectingServerEvent(clusterDescription, Mock.Of<IServerSelector>(), default, default))
             };
 
             var booleanValues = new[] { true, false };

tests/MongoDB.Driver.Tests/Core/Logging/EventLoggerTests.cs

@@ -447,7 +447,7 @@ public void BypassSpawningMongocryptdTest(
                 var exception = Record.Exception(() => adminDatabase.RunCommand<BsonDocument>(legacyHelloCommand));
 
                 exception.Should().BeOfType<TimeoutException>();
-                exception.Message.Should().Contain("A timeout occurred after 1000ms selecting a server").And.Contain("localhost:27021");
+                exception.Message.Should().MatchRegex(@".*A timeout occurred after \d+ms selecting a server.*").And.Contain("localhost:27021");
             }
 
             IMongoClient EnsureEnvironmentAndConfigureTestClientEncrypted()