123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972 |
- package sqlx
- import (
- "database/sql"
- "testing"
- "github.com/DATA-DOG/go-sqlmock"
- "github.com/stretchr/testify/assert"
- "github.com/tal-tech/go-zero/core/logx"
- )
- func TestUnmarshalRowBool(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("1")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value bool
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.True(t, value)
- })
- }
- func TestUnmarshalRowInt(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value int
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, 2, value)
- })
- }
- func TestUnmarshalRowInt8(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value int8
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, int8(3), value)
- })
- }
- func TestUnmarshalRowInt16(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("4")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value int16
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.Equal(t, int16(4), value)
- })
- }
- func TestUnmarshalRowInt32(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("5")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value int32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.Equal(t, int32(5), value)
- })
- }
- func TestUnmarshalRowInt64(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("6")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value int64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, int64(6), value)
- })
- }
- func TestUnmarshalRowUint(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value uint
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, uint(2), value)
- })
- }
- func TestUnmarshalRowUint8(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value uint8
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, uint8(3), value)
- })
- }
- func TestUnmarshalRowUint16(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("4")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value uint16
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, uint16(4), value)
- })
- }
- func TestUnmarshalRowUint32(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("5")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value uint32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, uint32(5), value)
- })
- }
- func TestUnmarshalRowUint64(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("6")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value uint64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, uint16(6), value)
- })
- }
- func TestUnmarshalRowFloat32(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("7")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value float32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, float32(7), value)
- })
- }
- func TestUnmarshalRowFloat64(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("8")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value float64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, float64(8), value)
- })
- }
- func TestUnmarshalRowString(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- const expect = "hello"
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString(expect)
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value string
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowStruct(t *testing.T) {
- var value = new(struct {
- Name string
- Age int
- })
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("liao,5")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- assert.Equal(t, "liao", value.Name)
- assert.Equal(t, 5, value.Age)
- })
- }
- func TestUnmarshalRowStructWithTags(t *testing.T) {
- var value = new(struct {
- Age int `db:"age"`
- Name string `db:"name"`
- })
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("liao,5")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- assert.Equal(t, "liao", value.Name)
- assert.Equal(t, 5, value.Age)
- })
- }
- func TestUnmarshalRowsBool(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []bool{true, false}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("1\n0")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []bool
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []int{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []int
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt8(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []int8{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []int8
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt16(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []int16{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []int16
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt32(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []int32{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []int32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt64(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []int64{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []int64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []uint{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []uint
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint8(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []uint8{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []uint8
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint16(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []uint16{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []uint16
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint32(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []uint32{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []uint32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint64(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []uint64{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []uint64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsFloat32(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []float32{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []float32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsFloat64(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []float64{2, 3}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []float64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsString(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []string{"hello", "world"}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("hello\nworld")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []string
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsBoolPtr(t *testing.T) {
- yes := true
- no := false
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*bool{&yes, &no}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("1\n0")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*bool
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsIntPtr(t *testing.T) {
- two := 2
- three := 3
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*int{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*int
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt8Ptr(t *testing.T) {
- two := int8(2)
- three := int8(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*int8{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*int8
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt16Ptr(t *testing.T) {
- two := int16(2)
- three := int16(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*int16{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*int16
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt32Ptr(t *testing.T) {
- two := int32(2)
- three := int32(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*int32{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*int32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsInt64Ptr(t *testing.T) {
- two := int64(2)
- three := int64(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*int64{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*int64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUintPtr(t *testing.T) {
- two := uint(2)
- three := uint(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*uint{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*uint
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint8Ptr(t *testing.T) {
- two := uint8(2)
- three := uint8(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*uint8{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*uint8
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint16Ptr(t *testing.T) {
- two := uint16(2)
- three := uint16(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*uint16{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*uint16
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint32Ptr(t *testing.T) {
- two := uint32(2)
- three := uint32(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*uint32{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*uint32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsUint64Ptr(t *testing.T) {
- two := uint64(2)
- three := uint64(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*uint64{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*uint64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsFloat32Ptr(t *testing.T) {
- two := float32(2)
- three := float32(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*float32{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*float32
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsFloat64Ptr(t *testing.T) {
- two := float64(2)
- three := float64(3)
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*float64{&two, &three}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("2\n3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*float64
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsStringPtr(t *testing.T) {
- hello := "hello"
- world := "world"
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- var expect = []*string{&hello, &world}
- rs := sqlmock.NewRows([]string{"value"}).FromCSVString("hello\nworld")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var value []*string
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select value from users where user=?", "anyone"))
- assert.EqualValues(t, expect, value)
- })
- }
- func TestUnmarshalRowsStruct(t *testing.T) {
- var expect = []struct {
- Name string
- Age int64
- }{
- {
- Name: "first",
- Age: 2,
- },
- {
- Name: "second",
- Age: 3,
- },
- }
- var value []struct {
- Name string
- Age int64
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("first,2\nsecond,3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, value[i].Age)
- }
- })
- }
- func TestUnmarshalRowsStructWithNullStringType(t *testing.T) {
- var expect = []struct {
- Name string
- NullString sql.NullString
- }{
- {
- Name: "first",
- NullString: sql.NullString{
- String: "firstnullstring",
- Valid: true,
- },
- },
- {
- Name: "second",
- NullString: sql.NullString{
- String: "",
- Valid: false,
- },
- },
- }
- var value []struct {
- Name string `db:"name"`
- NullString sql.NullString `db:"value"`
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "value"}).AddRow(
- "first", "firstnullstring").AddRow("second", nil)
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.NullString.String, value[i].NullString.String)
- assert.Equal(t, each.NullString.Valid, value[i].NullString.Valid)
- }
- })
- }
- func TestUnmarshalRowsStructWithTags(t *testing.T) {
- var expect = []struct {
- Name string
- Age int64
- }{
- {
- Name: "first",
- Age: 2,
- },
- {
- Name: "second",
- Age: 3,
- },
- }
- var value []struct {
- Age int64 `db:"age"`
- Name string `db:"name"`
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("first,2\nsecond,3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, value[i].Age)
- }
- })
- }
- func TestUnmarshalRowsStructAndEmbeddedAnonymousStructWithTags(t *testing.T) {
- type Embed struct {
- Value int64 `db:"value"`
- }
- var expect = []struct {
- Name string
- Age int64
- Value int64
- }{
- {
- Name: "first",
- Age: 2,
- Value: 3,
- },
- {
- Name: "second",
- Age: 3,
- Value: 4,
- },
- }
- var value []struct {
- Name string `db:"name"`
- Age int64 `db:"age"`
- Embed
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age", "value"}).FromCSVString("first,2,3\nsecond,3,4")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age, value from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, value[i].Age)
- assert.Equal(t, each.Value, value[i].Value)
- }
- })
- }
- func TestUnmarshalRowsStructAndEmbeddedStructPtrAnonymousWithTags(t *testing.T) {
- type Embed struct {
- Value int64 `db:"value"`
- }
- var expect = []struct {
- Name string
- Age int64
- Value int64
- }{
- {
- Name: "first",
- Age: 2,
- Value: 3,
- },
- {
- Name: "second",
- Age: 3,
- Value: 4,
- },
- }
- var value []struct {
- Name string `db:"name"`
- Age int64 `db:"age"`
- *Embed
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age", "value"}).FromCSVString("first,2,3\nsecond,3,4")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age, value from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, value[i].Age)
- assert.Equal(t, each.Value, value[i].Value)
- }
- })
- }
- func TestUnmarshalRowsStructPtr(t *testing.T) {
- var expect = []*struct {
- Name string
- Age int64
- }{
- {
- Name: "first",
- Age: 2,
- },
- {
- Name: "second",
- Age: 3,
- },
- }
- var value []*struct {
- Name string
- Age int64
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("first,2\nsecond,3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, value[i].Age)
- }
- })
- }
- func TestUnmarshalRowsStructWithTagsPtr(t *testing.T) {
- var expect = []*struct {
- Name string
- Age int64
- }{
- {
- Name: "first",
- Age: 2,
- },
- {
- Name: "second",
- Age: 3,
- },
- }
- var value []*struct {
- Age int64 `db:"age"`
- Name string `db:"name"`
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("first,2\nsecond,3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, value[i].Age)
- }
- })
- }
- func TestUnmarshalRowsStructWithTagsPtrWithInnerPtr(t *testing.T) {
- var expect = []*struct {
- Name string
- Age int64
- }{
- {
- Name: "first",
- Age: 2,
- },
- {
- Name: "second",
- Age: 3,
- },
- }
- var value []*struct {
- Age *int64 `db:"age"`
- Name string `db:"name"`
- }
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"name", "age"}).FromCSVString("first,2\nsecond,3")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRows(&value, rows, true)
- }, "select name, age from users where user=?", "anyone"))
- for i, each := range expect {
- assert.Equal(t, each.Name, value[i].Name)
- assert.Equal(t, each.Age, *value[i].Age)
- }
- })
- }
- func TestCommonSqlConn_QueryRowOptional(t *testing.T) {
- runOrmTest(t, func(db *sql.DB, mock sqlmock.Sqlmock) {
- rs := sqlmock.NewRows([]string{"age"}).FromCSVString("5")
- mock.ExpectQuery("select (.+) from users where user=?").WithArgs("anyone").WillReturnRows(rs)
- var r struct {
- User string `db:"user"`
- Age int `db:"age"`
- }
- assert.Nil(t, query(db, func(rows *sql.Rows) error {
- return unmarshalRow(&r, rows, false)
- }, "select age from users where user=?", "anyone"))
- assert.Empty(t, r.User)
- assert.Equal(t, 5, r.Age)
- })
- }
- func runOrmTest(t *testing.T, fn func(db *sql.DB, mock sqlmock.Sqlmock)) {
- logx.Disable()
- db, mock, err := sqlmock.New()
- if err != nil {
- t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
- }
- defer db.Close()
- fn(db, mock)
- if err := mock.ExpectationsWereMet(); err != nil {
- t.Errorf("there were unfulfilled expectations: %s", err)
- }
- }
|