Selaa lähdekoodia

create an empty session

Chris Bannister 10 vuotta sitten
vanhempi
commit
c9232c7e16
1 muutettua tiedostoa jossa 8 lisäystä ja 4 poistoa
  1. 8 4
      session_test.go

+ 8 - 4
session_test.go

@@ -159,13 +159,17 @@ func TestQueryShouldPrepare(t *testing.T) {
 
 func TestBatchBasicAPI(t *testing.T) {
 
-	cfg := NewCluster("127.0.0.1")
-	cfg.RetryPolicy = &SimpleRetryPolicy{NumRetries: 2}
-
-	s, err := cfg.CreateSession()
+	cfg := &ClusterConfig{RetryPolicy: &SimpleRetryPolicy{NumRetries: 2}}
+	pool, err := NewSimplePool(cfg)
 	if err != nil {
 		t.Fatal(err)
 	}
+
+	s := &Session{
+		Pool: pool,
+		cfg:  *cfg,
+		cons: Quorum,
+	}
 	defer s.Close()
 
 	b := s.NewBatch(UnloggedBatch)