databend-go/rows_test.go at main · databendlabs/databend-go · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package godatabend

import (
	"context"
	"database/sql/driver"
	"reflect"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestTextRows(t *testing.T) {
	ptr1 := strPtr("1")
	ptr2 := strPtr("2")
	ptr3 := strPtr("2")
	dc := &DatabendConn{
		cfg: &Config{},
	}
	rows, err := dc.newNextRows(context.Background(), &QueryResponse{
		Data: [][]*string{{ptr1, ptr2, ptr3}, {ptr3, ptr2, ptr1}},
		Schema: &[]DataField{
			{Name: "age", Type: "Int32"},
			{Name: "height", Type: "Int64"},
			{Name: "score", Type: "String"},
		},
		State: "Succeeded",
	})
	if !assert.NoError(t, err) {
		return
	}
	assert.Equal(t, []string{"age", "height", "score"}, rows.Columns())
	assert.Equal(t, reflect.TypeOf(int32(0)), rows.ColumnTypeScanType(0))
	assert.Equal(t, reflect.TypeOf(""), rows.ColumnTypeScanType(2))
	assert.Equal(t, "Int32", rows.ColumnTypeDatabaseTypeName(0))
	assert.Equal(t, "String", rows.ColumnTypeDatabaseTypeName(2))
}

func TestNextRowsUsesTypedRowsWithoutTextData(t *testing.T) {
	dc := &DatabendConn{
		cfg: &Config{},
	}
	rows, err := dc.newNextRows(context.Background(), &QueryResponse{
		typedRows: [][]driver.Value{{"7", "alice"}},
		Schema: &[]DataField{
			{Name: "age", Type: "Int32"},
			{Name: "name", Type: "String"},
		},
		State: "Succeeded",
	})
	if !assert.NoError(t, err) {
		return
	}

	dest := make([]driver.Value, 2)
	if !assert.NoError(t, rows.Next(dest)) {
		return
	}
	assert.Equal(t, "7", dest[0])
	assert.Equal(t, "alice", dest[1])
}

func strPtr(s string) *string {
	return &s
}