wait more for schema when creating session and keyspace

common_test.go

@@ -96,25 +96,20 @@ func createKeyspace(tb testing.TB, cluster *ClusterConfig, keyspace string) {
 	defer session.Close()
 	defer log.Println("closing keyspace session")
 
-	err = session.control.query(`DROP KEYSPACE IF EXISTS ` + keyspace).Close()
+	err = createTable(session, `DROP KEYSPACE IF EXISTS `+keyspace)
 	if err != nil {
 		tb.Fatal(err)
 	}
 
-	err = session.control.query(fmt.Sprintf(`CREATE KEYSPACE %s
+	err = createTable(session, fmt.Sprintf(`CREATE KEYSPACE %s
 	WITH replication = {
 		'class' : 'SimpleStrategy',
 		'replication_factor' : %d
-	}`, keyspace, *flagRF)).Close()
+	}`, keyspace, *flagRF))
 
 	if err != nil {
 		tb.Fatal(err)
 	}
-
-	// lets just be sure
-	if err := session.control.awaitSchemaAgreement(); err != nil {
-		tb.Fatal(err)
-	}
 }
 
 func createSessionFromCluster(cluster *ClusterConfig, tb testing.TB) *Session {
@@ -130,6 +125,10 @@ func createSessionFromCluster(cluster *ClusterConfig, tb testing.TB) *Session {
 		tb.Fatal("createSession:", err)
 	}
 
+	if err := session.control.awaitSchemaAgreement(); err != nil {
+		tb.Fatal(err)
+	}
+
 	return session
 }