WIP

Author: qingyang-hu

internal/integration/sessions_test.go

@@ -597,7 +597,13 @@ func TestSessionsProse(t *testing.T) {
 		})
 	*/
 
+	firstServerAddr := mtest.GlobalTopology().Description().Servers[0].Addr
+	directConnectionOpts := options.Client().
+		ApplyURI(fmt.Sprintf("mongodb://%s", firstServerAddr)).
+		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) {
@@ -622,7 +628,6 @@ 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()).
 			SetDirect(true).
 			SetHeartbeatInterval(500 * time.Millisecond). // Minimum interval