Pārlūkot izejas kodu

close sessions after the tests

Chris Bannister 10 gadi atpakaļ
vecāks
revīzija
09a942bb64
2 mainītis faili ar 4 papildinājumiem un 0 dzēšanām
  1. 1 0
      cass1batch_test.go
  2. 3 0
      session_test.go

+ 1 - 0
cass1batch_test.go

@@ -12,6 +12,7 @@ func TestProto1BatchInsert(t *testing.T) {
 	if err := session.Query("CREATE TABLE large (id int primary key)").Exec(); err != nil {
 		t.Fatal("create table:", err)
 	}
+	defer session.Close()
 
 	begin := "BEGIN BATCH"
 	end := "APPLY BATCH"

+ 3 - 0
session_test.go

@@ -16,6 +16,7 @@ func TestSessionAPI(t *testing.T) {
 	}
 
 	s := NewSession(pool, cfg)
+	defer s.Close()
 
 	s.SetConsistency(All)
 	if s.cons != All {
@@ -161,6 +162,8 @@ func TestBatchBasicAPI(t *testing.T) {
 	}
 
 	s := NewSession(pool, cfg)
+	defer s.Close()
+
 	b := s.NewBatch(UnloggedBatch)
 	if b.Type != UnloggedBatch {
 		t.Fatalf("expceted batch.Type to be '%v', got '%v'", UnloggedBatch, b.Type)