WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -13,6 +13,7 @@ import (
 	"fmt"
 	"reflect"
 	"sync"
+	"sync/atomic"
 	"testing"
 	"time"
 
@@ -513,29 +514,19 @@ 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{})
+		var clusterTimeAdvanced uint32
 		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
-				// }
+				if atomic.LoadUint32(&clusterTimeAdvanced) == 1 {
+					fmt.Println("ServerHeartbeatStartedEvent: cluster time advanced")
+				}
 			},
 			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
-				// }
+				if atomic.LoadUint32(&clusterTimeAdvanced) == 1 {
+					fmt.Println("ServerHeartbeatSucceededEvent: cluster time advanced")
+				}
 			},
 		}
 
@@ -571,7 +562,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)
+		atomic.StoreUint32(&clusterTimeAdvanced, 1)
 		// <-heartbeatStarted
 		// <-heartbeatSucceeded
 		time.Sleep(3 * time.Second)