WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -542,20 +542,30 @@ func TestSessionsProse(t *testing.T) {
 			},
 		}
 
+		var pingStartedCommands []bson.Raw
+		var pingSucceededCommands []bson.Raw
+		commandMonitor := &event.CommandMonitor{
+			Started: func(_ context.Context, cse *event.CommandStartedEvent) {
+				fmt.Println("Command started:", cse.CommandName, cse.Command)
+				if cse.CommandName == "ping" {
+					pingStartedCommands = append(pingStartedCommands, cse.Command)
+				}
+			},
+			Succeeded: func(_ context.Context, cse *event.CommandSucceededEvent) {
+				fmt.Println("Command succeeded:", cse.CommandName, cse.Reply)
+				if cse.CommandName == "ping" {
+					pingSucceededCommands = append(pingSucceededCommands, cse.Reply)
+				}
+			},
+		}
+
 		opts := options.Client().
 			ApplyURI(mtest.ClusterURI()).
 			SetHosts([]string{mtest.ClusterConnString().Hosts[0]}).
 			SetDirect(true).
 			SetHeartbeatInterval(500 * time.Millisecond). // Minimum interval
-			SetMonitor(&event.CommandMonitor{
-				Started: func(_ context.Context, cse *event.CommandStartedEvent) {
-					fmt.Println("Command started:", cse.CommandName, cse.Command)
-				},
-				Succeeded: func(_ context.Context, cse *event.CommandSucceededEvent) {
-					fmt.Println("Command succeeded:", cse.CommandName, cse.Reply)
-				},
-			}).
-			SetServerMonitor(serverMonitor)
+			SetServerMonitor(serverMonitor).
+			SetMonitor(commandMonitor)
 
 		client, err := mongo.Connect(opts)
 		require.NoError(mt, err, "expected no error connecting to client, got: %v", err)
@@ -564,12 +574,8 @@ func TestSessionsProse(t *testing.T) {
 			require.NoError(mt, err, "expected no error disconnecting client, got: %v", err)
 		}()
 
-		res, err := client.Database("admin").RunCommand(context.Background(), bson.D{
-			{"ping", 1},
-		}).Raw()
+		err = client.Ping(context.Background(), readpref.Primary())
 		require.NoError(mt, err, "expected no error, got: %v", err)
-		replyClusterTime, err := res.LookupErr("$clusterTime")
-		require.NoError(mt, err, "$clusterTime not found in response")
 
 		_, 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)
@@ -578,13 +584,17 @@ func TestSessionsProse(t *testing.T) {
 		<-heartbeatStarted
 		<-heartbeatSucceeded
 
-		res, err = client.Database("admin").RunCommand(context.Background(), bson.D{
-			{"ping", 1},
-		}).Raw()
+		err = client.Ping(context.Background(), readpref.Primary())
 		require.NoError(mt, err, "expected no error, got: %v", err)
-		replyClusterTimeNew, err := res.LookupErr("$clusterTime")
+
+		require.Len(mt, pingStartedCommands, 2, "expected 2 pings started, got: %v", len(pingStartedCommands))
+		require.Len(mt, pingSucceededCommands, 2, "expected 2 pings succeeded, got: %v", len(pingSucceededCommands))
+		initialClusterTime, err := pingStartedCommands[0].LookupErr("$clusterTime")
 		require.NoError(mt, err, "$clusterTime not found in response")
-		mt.Fatalf("$clusterTime: %v %v", replyClusterTime, replyClusterTimeNew)
+		currentClusterTime, err := pingStartedCommands[1].LookupErr("$clusterTime")
+		require.NoError(mt, err, "$clusterTime not found in command")
+		assert.Equal(mt, initialClusterTime, currentClusterTime, "expected same cluster time, got %v and %v", initialClusterTime, currentClusterTime)
+		mt.Fatalf("$clusterTime: %v %v", initialClusterTime, currentClusterTime)
 	})
 }