WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -512,20 +512,32 @@ func TestSessionsProse(t *testing.T) {
 	})
 
 	mt.Run("20 Drivers do not gossip $clusterTime on SDAM commands", func(mt *mtest.T) {
-		// heartbeatStarted := make(chan struct{})
-		// heartbeatSucceeded := make(chan struct{})
+		heartbeatStarted := make(chan struct{})
+		heartbeatSucceeded := make(chan struct{})
 		var clusterTimeAdvanced uint32
 		serverMonitor := &event.ServerMonitor{
 			ServerHeartbeatStarted: func(e *event.ServerHeartbeatStartedEvent) {
 				fmt.Println("Server heartbeat started:", e.ConnectionID)
 				if atomic.LoadUint32(&clusterTimeAdvanced) == 1 {
 					fmt.Println("ServerHeartbeatStartedEvent: cluster time advanced")
+					select {
+					case heartbeatStarted <- struct{}{}:
+						// NOOP
+					default:
+						// NOOP
+					}
 				}
 			},
 			ServerHeartbeatSucceeded: func(e *event.ServerHeartbeatSucceededEvent) {
 				fmt.Println("Server heartbeat succeeded:", e.ConnectionID, e.Duration, e.Reply)
 				if atomic.LoadUint32(&clusterTimeAdvanced) == 1 {
 					fmt.Println("ServerHeartbeatSucceededEvent: cluster time advanced")
+					select {
+					case heartbeatSucceeded <- struct{}{}:
+						// NOOP
+					default:
+						// NOOP
+					}
 				}
 			},
 		}
@@ -563,9 +575,8 @@ func TestSessionsProse(t *testing.T) {
 		require.NoError(mt, err, "expected no error inserting document, got: %v", err)
 
 		atomic.StoreUint32(&clusterTimeAdvanced, 1)
-		// <-heartbeatStarted
-		// <-heartbeatSucceeded
-		time.Sleep(3 * time.Second)
+		<-heartbeatStarted
+		<-heartbeatSucceeded
 
 		res, err = client.Database("admin").RunCommand(context.Background(), bson.D{
 			{"ping", 1},