iterator_test.go

package gockle

import (
	"context"
	"fmt"
	"reflect"
	"testing"
)

func TestIterator(t *testing.T) {
	var s = newSession(t)

	defer s.Close()

	var exec = func(q string) {
		if err := s.Exec(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)

	if i := s.ScanIterator("select * from gockle_test.test"); i == nil {
		t.Error("Actual iterator nil, expected not nil")
	} else {
		var id, n int

		if !i.Scan(&id, &n) {
			t.Errorf("Actual more false, expected true")
		}

		if i.WillSwitchPage() {
			t.Errorf("Actual WillSwitchPage true, expected false")
		}

		if state := i.PageState(); state != nil {
			t.Errorf("Actual PageState not nil, expected nil")
		}

		if id != 1 {
			t.Errorf("Actual id %v, expected 1", id)
		}

		if n != 2 {
			t.Errorf("Actual n %v, expected 2", n)
		}

		if err := i.Close(); err != nil {
			t.Errorf("Actual error %v, expected no error", err)
		}
	}

	if i := s.ScanIterator("select * from gockle_test.test"); i == nil {
		t.Error("Actual iterator nil, expected not nil")
	} else {
		var a = map[string]interface{}{}

		if !i.ScanMap(a) {
			t.Errorf("Actual more false, expected true")
		}

		if e := (map[string]interface{}{"id": 1, "n": 2}); !reflect.DeepEqual(a, e) {
			t.Errorf("Actual map %v, expected %v", a, e)
		}

		if err := i.Close(); err != nil {
			t.Errorf("Actual error %v, expected no error", err)
		}
	}
}

func TestIteratorPaging(t *testing.T) {
	var s = newSession(t)
	defer s.Close()
	var exec = func(q string) {
		if err := s.Exec(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)
	exec(rowInsert2)
	q := s.Query("select * from gockle_test.test").
		PageSize(1).
		WithContext(context.Background())
	var state []byte
	if i := q.Iter(); i == nil {
		t.Error("Actual iterator nil, expected not nil")
	} else {
		var id, n int
		// First row
		if !i.Scan(&id, &n) {
			t.Errorf("Actual more false, expected true")
		}
		if !i.WillSwitchPage() {
			t.Errorf("Actual WillSwitchPage false, expected true")
		}
		state = i.PageState()
		if state == nil {
			t.Errorf("Actual PageState nil, expected not nil")
		}
		if id != 1 {
			t.Errorf("Actual id %v, expected 1", id)
		}
		if n != 2 {
			t.Errorf("Actual n %v, expected 2", n)
		}
	}
	// second row
	q = q.PageState(state)
	if i := q.Iter(); i == nil {
		t.Error("Actual iterator nil, expected not nil")
	} else {
		var id, n int
		if !i.Scan(&id, &n) {
			t.Errorf("Actual more false, expected true")
		}
		if i.WillSwitchPage() {
			t.Errorf("Actual WillSwitchPage true, expected false")
		}
		state = i.PageState()
		if state == nil {
			t.Errorf("Actual PageState nil, expected not nil")
		}
		if id != 3 {
			t.Errorf("Actual id %v, expected 3", id)
		}
		if n != 4 {
			t.Errorf("Actual n %v, expected 4", n)
		}
		if err := i.Close(); err != nil {
			t.Errorf("Actual error %v, expected no error", err)
		}
	}
	// third row (not existing)
	q = q.PageState(state)
	if i := q.Iter(); i == nil {
		t.Error("Actual iterator nil, expected not nil")
	} else {
		var id, n int
		if i.Scan(&id, &n) {
			t.Errorf("Actual more true, expected false")
		}
		if i.WillSwitchPage() {
			t.Errorf("Actual WillSwitchPage true, expected false")
		}
		state = i.PageState()
		if state != nil {
			t.Errorf("Actual PageState not nil, expected nil")
		}
	}
}

func TestIteratorMock(t *testing.T) {
	var m, e = &IteratorMock{}, fmt.Errorf("e")

	testMock(t, m, &m.Mock, []struct {
		method    string
		arguments []interface{}
		results   []interface{}
	}{
		{"Close", nil, []interface{}{nil}},
		{"Close", nil, []interface{}{e}},
		{"Scan", []interface{}{[]interface{}(nil)}, []interface{}{false}},
		{"Scan", []interface{}{[]interface{}{1}}, []interface{}{true}},
		{"ScanMap", []interface{}{map[string]interface{}(nil)}, []interface{}{false}},
		{"ScanMap", []interface{}{map[string]interface{}{"a": 1}}, []interface{}{true}},
		{"WillSwitchPage", nil, []interface{}{false}},
		{"WillSwitchPage", nil, []interface{}{true}},
		{"PageState", nil, []interface{}{[]byte{}}},
		{"PageState", nil, []interface{}{[]byte{1}}},
	})
}