CSHARP-5204: Test Heartbeats don't gossip cluster time (#1665)

Author: adelinowona

tests/MongoDB.Driver.Tests/Specifications/sessions/SessionsProseTests.cs

@@ -21,6 +21,7 @@
 using MongoDB.Bson;
 using MongoDB.Bson.TestHelpers;
 using MongoDB.Driver.Core;
+using MongoDB.Driver.Core.Clusters;
 using MongoDB.Driver.Core.Events;
 using MongoDB.Driver.Core.Misc;
 using MongoDB.Driver.Core.TestHelpers.Logging;
@@ -302,6 +303,55 @@ public async Task Ensure_server_session_are_allocated_only_on_connection_checkou
             await eventsTask.WithTimeout(1000);
         }
 
+        // https://specifications.readthedocs.io/en/latest/sessions/tests/#20-drivers-do-not-gossip-clustertime-on-sdam-commands
+        [Fact]
+        public void Ensure_cluster_times_are_not_gossiped_on_SDAM_commands()
+        {
+            RequireServer.Check().ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded);
+
+            var eventCapturer = new EventCapturer()
+                .Capture<ServerHeartbeatStartedEvent>()
+                .Capture<ServerHeartbeatSucceededEvent>()
+                .Capture<CommandStartedEvent>();
+
+            using var c1 = DriverTestConfiguration.CreateMongoClient(
+                settings =>
+                {
+                    settings.ClusterConfigurator = c => c.Subscribe(eventCapturer);
+                    settings.DirectConnection = true;
+                    settings.HeartbeatInterval = TimeSpan.FromMilliseconds(10);
+                    if (settings.Servers.Count() > 1)
+                    {
+                        settings.Servers = settings.Servers.Take(1);
+                    }
+                });
+
+            var pingCommand = new BsonDocument("ping", 1);
+            var pingResult = c1.GetDatabase("admin").RunCommand<BsonDocument>(pingCommand);
+
+            var clusterTime = pingResult["$clusterTime"];
+
+            var c2 = DriverTestConfiguration.Client;
+            c2.GetDatabase("test").GetCollection<BsonDocument>("test").InsertOne(new BsonDocument("advance", "$clusterTime"));
+
+            eventCapturer.Clear();
+
+            eventCapturer.WaitForOrThrowIfTimeout(
+                capturedEvents =>
+                {
+                    return capturedEvents
+                        .SkipWhile(e => e is not ServerHeartbeatStartedEvent)
+                        .Any(e => e is ServerHeartbeatSucceededEvent);
+                }, TimeSpan.FromSeconds(1), "Didn't get any server heartbeat pairs");
+
+            c1.GetDatabase("admin").RunCommand<BsonDocument>(pingCommand);
+
+            var commandStartedEvents = eventCapturer.Events.OfType<CommandStartedEvent>().ToArray();
+            commandStartedEvents.Length.Should().Be(1);
+            commandStartedEvents[0].CommandName.Should().Be("ping");
+            commandStartedEvents[0].Command["$clusterTime"].Should().Be(clusterTime);
+        }
+
         private sealed class MongocryptdContext : IDisposable
         {
             public IMongoClient MongoClient { get; }