rows_test.go

package fireboltgosdk

import (
	"database/sql/driver"
	"encoding/json"
	"io"
	"reflect"
	"testing"
	"time"
)

func assert(val bool, t *testing.T, err string) {
	if !val {
		t.Errorf(err)
	}
}

func mockRows() driver.Rows {
	resultJson := "{\"query\":{\"query_id\":\"16FF2A0300ECA753\"},\"meta\":[{\"name\":\"int_col\",\"type\":\"Nullable(Int32)\"},{\"name\":\"bigint_col\",\"type\":\"Int64\"},{\"name\":\"float_col\",\"type\":\"Float32\"},{\"name\":\"double_col\",\"type\":\"Float64\"},{\"name\":\"text_col\",\"type\":\"String\"},{\"name\":\"date_col\",\"type\":\"Date\"},{\"name\":\"timestamp_col\",\"type\":\"DateTime\"},{\"name\":\"bool_col\",\"type\":\"UInt8\"},{\"name\":\"array_col\",\"type\":\"Array(Int32)\"},{\"name\":\"nested_array_col\",\"type\":\"Array(Array(String))\"}],\"data\":[[null,1,0.312321,123213.321321,\"text\",\"2080-12-31\",\"1989-04-15 01:02:03\",1,[1,2,3],[[]]],[2,1,0.312321,123213.321321,\"text\",\"1970-01-01\",\"1970-01-01 00:00:00\",1,[1,2,3],[[]]],[3,5,0.312321,123213.321321,\"text\",\"1970-01-01\",\"1970-01-01 00:00:00\",1,[5,2,3,2],[[\"TEST\",\"TEST1\"],[\"TEST3\"]]]],\"rows\":3,\"statistics\":{\"elapsed\":0.001797702,\"rows_read\":3,\"bytes_read\":293,\"time_before_execution\":0.001251613,\"time_to_execute\":0.000544098,\"scanned_bytes_cache\":2003,\"scanned_bytes_storage\":0}}"
	var response QueryResponse
	err := json.Unmarshal([]byte(resultJson), &response)
	if err != nil {
		panic("Error in test code")
	}
	return &fireboltRows{response, 0}
}

// TestRowsColumns checks, that correct column names are returned
func TestRowsColumns(t *testing.T) {
	rows := mockRows()

	columnNames := []string{"int_col", "bigint_col", "float_col", "double_col", "text_col", "date_col", "timestamp_col", "bool_col", "array_col", "nested_array_col"}
	if !reflect.DeepEqual(rows.Columns(), columnNames) {
		t.Errorf("column lists are not equal")
	}
}

// TestRowsClose checks Close method, and inability to use rows afterward
func TestRowsClose(t *testing.T) {
	rows := mockRows()
	if rows.Close() != nil {
		t.Errorf("Closing rows was not successful")
	}

	var dest []driver.Value
	if rows.Next(dest) != io.EOF {
		t.Errorf("Next of closed rows didn't return EOF")
	}
}

// TestRowsNext check Next method
func TestRowsNext(t *testing.T) {
	rows := mockRows()
	var dest = make([]driver.Value, 10)
	err := rows.Next(dest)
	loc, _ := time.LoadLocation("UTC")

	assert(err == nil, t, "Next shouldn't return an error")
	assert(dest[0] == nil, t, "results not equal for int32")
	assert(dest[1] == int64(1), t, "results not equal for int64")
	assert(dest[2] == float32(0.312321), t, "results not equal for float32")
	assert(dest[3] == float64(123213.321321), t, "results not equal for float64")
	assert(dest[4] == "text", t, "results not equal for string")
	assert(dest[5] == time.Date(2080, 12, 31, 0, 0, 0, 0, loc), t, "results not equal for date")
	assert(dest[6] == time.Date(1989, 04, 15, 1, 2, 3, 0, loc), t, "results not equal for datetime")

	err = rows.Next(dest)
	assert(err == nil, t, "Next shouldn't return an error")

	err = rows.Next(dest)
	assert(err == nil, t, "Next shouldn't return an error")

	assert(dest[0] == int32(3), t, "results not equal for int32")
	assert(dest[1] == int64(5), t, "results not equal for int64")
	assert(dest[2] == float32(0.312321), t, "results not equal for float32")
	assert(dest[3] == float64(123213.321321), t, "results not equal for float64")
	assert(dest[4] == "text", t, "results not equal for string")

	assert(io.EOF == rows.Next(dest), t, "Next should return io.EOF if no data available anymore")
}