session: Increase timeout to 5 seconds

scylladb · Jul 19, 2018 · 57e7951 · 57e7951
1 parent 52505d1
commit 57e7951
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/session.go b/session.go
@@ -1,6 +1,8 @@
 package gemini
 
 import (
+	"time"
+
 	"github.com/gocql/gocql"
 	"github.com/google/go-cmp/cmp"
 
@@ -14,12 +16,14 @@ type Session struct {
 
 func NewSession(testClusterHost string, oracleClusterHost string) *Session {
 	testCluster := gocql.NewCluster(testClusterHost)
+	testCluster.Timeout = 5 * time.Second
 	testSession, err := testCluster.CreateSession()
 	if err != nil {
 		panic(err)
 	}
 
 	oracleCluster := gocql.NewCluster(oracleClusterHost)
+	oracleCluster.Timeout = 5 * time.Second
 	oracleSession, err := oracleCluster.CreateSession()
 	if err != nil {
 		panic(err)