Test all of session.Columns

willfaught · Mar 29, 2016 · 0e8f353 · 0e8f353
1 parent e867c25
commit 0e8f353
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/session_test.go b/session_test.go
@@ -55,8 +55,6 @@ func TestSessionMetadata(t *testing.T) {
 
  var s = newSession(t)
 
- defer s.Close()
-
  exec(s, ksDropIf)
  exec(s, ksCreate)
 
@@ -103,6 +101,16 @@ func TestSessionMetadata(t *testing.T) {
  } else {
  t.Fatalf("Actual error %v, expected no error", err)
  }
+
+ if _, err := s.Columns("gockle_test", "invalid"); err == nil {
+ t.Error("Actual no error, expected error")
+ }
+
+ s.Close()
+
+ if _, err := s.Columns("gockle_test", "test"); err == nil {
+ t.Error("Actual no error, expected error")
+ }
 }
 
 func TestSessionMock(t *testing.T) {