WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -513,29 +513,31 @@ 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{})
-		// clusterTimeAdvanced := make(chan struct{})
+		clusterTimeAdvanced := make(chan struct{})
 		serverMonitor := &event.ServerMonitor{
 			ServerHeartbeatStarted: func(e *event.ServerHeartbeatStartedEvent) {
 				fmt.Println("Server heartbeat started:", e.ConnectionID)
-				// select {
-				// case _, ok := <-clusterTimeAdvanced:
-				// 	if !ok {
-				// 		heartbeatStarted <- struct{}{}
-				// 	}
-				// default:
-				// 	// NOOP
-				// }
+				select {
+				case _, ok := <-clusterTimeAdvanced:
+					fmt.Println("ServerHeartbeatStartedEvent: cluster time advanced", ok)
+					// if !ok {
+					// 	heartbeatStarted <- struct{}{}
+					// }
+				default:
+					// NOOP
+				}
 			},
 			ServerHeartbeatSucceeded: func(e *event.ServerHeartbeatSucceededEvent) {
 				fmt.Println("Server heartbeat succeeded:", e.ConnectionID, e.Duration, e.Reply)
-				// select {
-				// case _, ok := <-clusterTimeAdvanced:
-				// 	if !ok {
-				// 		heartbeatSucceeded <- struct{}{}
-				// 	}
-				// default:
-				// 	// NOOP
-				// }
+				select {
+				case _, ok := <-clusterTimeAdvanced:
+					fmt.Println("ServerHeartbeatSucceededEvent: cluster time advanced", ok)
+					// if !ok {
+					// 	heartbeatSucceeded <- struct{}{}
+					// }
+				default:
+					// NOOP
+				}
 			},
 		}
 
@@ -571,7 +573,7 @@ func TestSessionsProse(t *testing.T) {
 		_, err = mt.Client.Database("test").Collection("test").InsertOne(context.Background(), bson.D{{"advance", "$clusterTime"}})
 		require.NoError(mt, err, "expected no error inserting document, got: %v", err)
 
-		// close(clusterTimeAdvanced)
+		close(clusterTimeAdvanced)
 		// <-heartbeatStarted
 		// <-heartbeatSucceeded
 		time.Sleep(3 * time.Second)