test:connect all nodes

Author: ribaraka

go.mod

@@ -20,8 +20,6 @@ module github.com/gocql/gocql
 go 1.23
 
 require (
-	github.com/bitly/go-hostpool v0.1.0 // indirect
-	github.com/docker/docker v27.1.1+incompatible // indirect
 	github.com/golang/snappy v0.0.4
 	github.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed
 	github.com/testcontainers/testcontainers-go v0.33.0
@@ -32,13 +30,15 @@ require (
 	dario.cat/mergo v1.0.0 // indirect
 	github.com/Azure/go-ansiterm v0.0.0-20210617225240-d185dfc1b5a1 // indirect
 	github.com/Microsoft/go-winio v0.6.2 // indirect
+	github.com/bitly/go-hostpool v0.1.0 // indirect
 	github.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869 // indirect
 	github.com/cenkalti/backoff/v4 v4.2.1 // indirect
 	github.com/containerd/containerd v1.7.18 // indirect
 	github.com/containerd/log v0.1.0 // indirect
 	github.com/containerd/platforms v0.2.1 // indirect
 	github.com/cpuguy83/dockercfg v0.3.1 // indirect
 	github.com/distribution/reference v0.6.0 // indirect
+	github.com/docker/docker v27.1.1+incompatible // indirect
 	github.com/docker/go-connections v0.5.0 // indirect
 	github.com/docker/go-units v0.5.0 // indirect
 	github.com/felixge/httpsnoop v1.0.4 // indirect

integration_test.go

@@ -272,3 +272,28 @@ func TestUDF(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestAllNodesConnected(t *testing.T) {
+	cluster := createCluster()
+	cluster.PoolConfig.HostSelectionPolicy = RoundRobinHostPolicy()
+
+	session := createSessionFromCluster(cluster, t)
+	defer session.Close()
+
+	ids := make(map[string]bool)
+
+	// Loop to query system.local multiple times. If there is a cluster with more than 10 nodes add to the loop more iterations
+	for i := 0; i < 10; i++ {
+		var hostID string
+		err := session.Query("SELECT host_id FROM system.local").Scan(&hostID)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		ids[hostID] = true
+	}
+
+	if *clusterSize != len(ids) {
+		t.Fatalf("Expected to connect to %d unique nodes", *clusterSize)
+	}
+}