Bläddra i källkod

Make autodiscovery frequency configurable

Ben Hood 11 år sedan
förälder
incheckning
6c4ef1c964
1 ändrade filer med 2 tillägg och 1 borttagningar
  1. 2 1
      cassandra_test.go

+ 2 - 1
cassandra_test.go

@@ -30,6 +30,7 @@ var (
 	clusterSize  = flag.Int("clusterSize", 1, "the expected size of the cluster")
 	flagRetry    = flag.Int("retries", 5, "number of times to retry queries")
 	flagAutoWait = flag.Duration("autowait", 1000*time.Millisecond, "time to wait for autodiscovery to fill the hosts poll")
+	flagAutoFreq = flag.Duration("autofreq", 1000*time.Millisecond, "frequency for autodiscovery to discover new hosts")
 	clusterHosts []string
 )
 
@@ -106,7 +107,7 @@ func TestRingDiscovery(t *testing.T) {
 
 	cluster := NewCluster(clusterHosts[0])
 	cluster.DiscoverHosts = true
-	cluster.Discovery = DiscoveryConfig{Sleep: 1 * time.Second}
+	cluster.Discovery = DiscoveryConfig{Sleep: *flagAutoFreq}
 	session, err := cluster.CreateSession()
 	if err != nil {
 		t.Errorf("got error connecting to the cluster %v", err)