WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -589,12 +589,49 @@ func TestSessionsProse(t *testing.T) {
 
 		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")
+		initialClusterTime, err := pingSucceededCommands[0].LookupErr("$clusterTime")
 		require.NoError(mt, err, "$clusterTime not found in response")
 		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)
+	})
+
+	mt.Run("ping test", func(mt *mtest.T) {
+		serverMonitor := &event.ServerMonitor{
+			ServerHeartbeatStarted: func(e *event.ServerHeartbeatStartedEvent) {
+				fmt.Println("Server heartbeat started:", e.ConnectionID)
+			},
+			ServerHeartbeatSucceeded: func(e *event.ServerHeartbeatSucceededEvent) {
+				fmt.Println("Server heartbeat succeeded:", e.ConnectionID, e.Duration, e.Reply)
+			},
+		}
+
+		commandMonitor := &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)
+			},
+		}
+
+		opts := options.Client().
+			ApplyURI(mtest.ClusterURI()).
+			SetHosts([]string{mtest.ClusterConnString().Hosts[0]}).
+			SetDirect(true).
+			SetHeartbeatInterval(500 * time.Millisecond). // Minimum interval
+			SetServerMonitor(serverMonitor).
+			SetMonitor(commandMonitor)
+
+		client, err := mongo.Connect(opts)
+		require.NoError(mt, err, "expected no error connecting to client, got: %v", err)
+		defer func() {
+			err = client.Disconnect(context.Background())
+			require.NoError(mt, err, "expected no error disconnecting client, got: %v", err)
+		}()
+
+		err = client.Ping(context.Background(), readpref.Primary())
+		require.NoError(mt, err, "expected no error, got: %v", err)
 	})
 }