WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -17,7 +17,6 @@ import (
 	"time"
 
 	"go.mongodb.org/mongo-driver/v2/bson"
-	"go.mongodb.org/mongo-driver/v2/event"
 	"go.mongodb.org/mongo-driver/v2/internal/assert"
 	"go.mongodb.org/mongo-driver/v2/internal/integration/mtest"
 	"go.mongodb.org/mongo-driver/v2/internal/mongoutil"
@@ -597,42 +596,20 @@ func TestSessionsProse(t *testing.T) {
 		})
 	*/
 
-	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)
+	firstServerAddr := mtest.GlobalTopology().Description().Servers[0].Addr
+	directConnectionOpts := options.Client().
+		ApplyURI(fmt.Sprintf("mongodb://%s", firstServerAddr)).
+		SetHeartbeatInterval(500 * time.Millisecond). // Minimum interval
+		SetReadPreference(readpref.Primary()).
+		SetDirect(true)
+	pingOpts := mtest.NewOptions().
+		ClientOptions(directConnectionOpts).
+		CreateCollection(false).
+		Topologies(mtest.ReplicaSet) // Read preference isn't sent to standalones so we can test on replica sets.
+	mt.RunOpts("ping test", pingOpts, func(mt *mtest.T) {
+		err := mt.Client.Ping(context.Background(), readpref.Primary())
+		assert.NoError(mt, err, "expected no error, got: %v", err)
+		mt.Fatal(mtest.ClusterURI(), mtest.ClusterConnString().Hosts, mtest.GlobalTopology().Description().Servers)
 	})
 }