WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -599,7 +599,7 @@ func TestSessionsProse(t *testing.T) {
 
 	pingOpts := mtest.NewOptions().
 		CreateCollection(false).
-		Topologies(mtest.ReplicaSet) // Read preference isn't sent to standalones so we can test on replica sets.
+		Topologies(mtest.ReplicaSet, mtest.LoadBalanced)
 	mt.RunOpts("ping test", pingOpts, func(mt *mtest.T) {
 		serverMonitor := &event.ServerMonitor{
 			ServerHeartbeatStarted: func(e *event.ServerHeartbeatStartedEvent) {
@@ -619,11 +619,9 @@ func TestSessionsProse(t *testing.T) {
 			},
 		}
 
-		firstServerAddr := mtest.GlobalTopology().Description().Servers[0].Addr
 		opts := options.Client().
-			ApplyURI(fmt.Sprintf("mongodb://%s", firstServerAddr)).
-			SetHeartbeatInterval(500 * time.Millisecond). // Minimum interval
-			SetReadPreference(readpref.Primary()).
+			ApplyURI(mtest.ClusterURI()).
+			SetHosts([]string{mtest.ClusterConnString().Hosts[0]}).
 			SetDirect(true).
 			SetHeartbeatInterval(500 * time.Millisecond). // Minimum interval
 			SetServerMonitor(serverMonitor).