WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -594,7 +594,43 @@ func TestSessionsProse(t *testing.T) {
 		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)
 	})
 }