pixeldrain/gocqlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

gocqlxtest: add test keyspace name flag

Browse Source
This commit is contained in:
Serhii Polishchuk
2021-12-31 11:55:37 +02:00
committed by GitHub
parent 2b885ac61b
commit 6ca5e2ae91
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gocqlxtest/gocqlxtest.go
View File

@@ -18,6 +18,7 @@ import (
var ( var (
flagCluster = flag.String("cluster", "127.0.0.1", "a comma-separated list of host:port tuples") flagCluster = flag.String("cluster", "127.0.0.1", "a comma-separated list of host:port tuples")
flagKeyspace = flag.String("keyspace", "gocqlx_test", "keyspace name")
flagProto = flag.Int("proto", 0, "protcol version") flagProto = flag.Int("proto", 0, "protcol version")
flagCQL = flag.String("cql", "3.0.0", "CQL version") flagCQL = flag.String("cql", "3.0.0", "CQL version")
flagRF = flag.Int("rf", 1, "replication factor for test keyspace") flagRF = flag.Int("rf", 1, "replication factor for test keyspace")
@@ -92,15 +93,18 @@ func CreateKeyspace(cluster *gocql.ClusterConfig, keyspace string) error {
} }
func createSessionFromCluster(cluster *gocql.ClusterConfig, tb testing.TB) gocqlx.Session { func createSessionFromCluster(cluster *gocql.ClusterConfig, tb testing.TB) gocqlx.Session {
if !flag.Parsed() {
flag.Parse()
}
// Drop and re-create the keyspace once. Different tests should use their own // Drop and re-create the keyspace once. Different tests should use their own
// individual tables, but can assume that the table does not exist before. // individual tables, but can assume that the table does not exist before.
initOnce.Do(func() { initOnce.Do(func() {
if err := CreateKeyspace(cluster, "gocqlx_test"); err != nil { if err := CreateKeyspace(cluster, *flagKeyspace); err != nil {
tb.Fatal(err) tb.Fatal(err)
} }
}) })
cluster.Keyspace = "gocqlx_test" cluster.Keyspace = *flagKeyspace
session, err := gocqlx.WrapSession(cluster.CreateSession()) session, err := gocqlx.WrapSession(cluster.CreateSession())
if err != nil { if err != nil {
tb.Fatal("CreateSession:", err) tb.Fatal("CreateSession:", err)