Add TestIterator

iterator_test.go

@@ -5,6 +5,53 @@ import (
  "testing"
 )
 
+func TestIterator(t *testing.T) {
+ var s = newSession(t)
+
+ defer s.Close()
+
+ var exec = func(q string) {
+ if err := s.QueryExec(q); err != nil {
+ t.Fatalf("Actual error %v, expected no error", err)
+ }
+ }
+
+ exec(ksDropIf)
+ exec(ksCreate)
+
+ defer exec(ksDrop)
+
+ exec(tabCreate)
+
+ defer exec(tabDrop)
+
+ exec(rowInsert)
+
+ var i = s.QueryIterator("select * from gockle_test.test")
+
+ if i == nil {
+ t.Fatal("Actual iterator nil, expected not nil")
+ }
+
+ var id, n int
+
+ if !i.Scan(&id, &n) {
+ t.Fatal("Actual more false, expected true")
+ }
+
+ if id != 1 {
+ t.Fatalf("Actual id %v, expected 1", id)
+ }
+
+ if n != 2 {
+ t.Fatalf("Actual n %v, expected 2", n)
+ }
+
+ if err := i.Close(); err != nil {
+ t.Fatalf("Actual error %v, expected no error", err)
+ }
+}
+
 func TestIteratorMock(t *testing.T) {
  var m, e = &IteratorMock{}, fmt.Errorf("e")
 

session_test.go

@@ -55,10 +55,17 @@ func TestSessionMetadata(t *testing.T) {
 
  var s = newSession(t)
 
+ defer s.Close()
+
  exec(s, ksDropIf)
  exec(s, ksCreate)
+
+ defer exec(s, ksDrop)
+
  exec(s, tabCreate)
 
+ defer exec(s, tabDrop)
+
  s = newSession(t)
 
  if a, err := s.Tables("gockle_test"); err == nil {
@@ -88,11 +95,6 @@ func TestSessionMetadata(t *testing.T) {
  } else {
  t.Errorf("Actual error %v, expected no error", err)
  }
-
- exec(s, tabDrop)
- exec(s, ksDrop)
-
- s.Close()
 }
 
 func TestSessionMock(t *testing.T) {