GOMODULES
/
pgx
mirror of https://github.com/jackc/pgx.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update TestRowToStructByName with snake case support

pull/1856/head
Tikhon Fedulov 2023-12-24 17:01:18 +03:00 committed by Jack Christensen
parent d2c9ebc2ef
commit 4367ee0598
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
rows_test.go
View File

@ -8,11 +8,12 @@ import (
"testing" "testing"
"time" "time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/jackc/pgx/v5" "github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgconn" "github.com/jackc/pgx/v5/pgconn"
"github.com/jackc/pgx/v5/pgxtest" "github.com/jackc/pgx/v5/pgxtest"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
) )
type testRowScanner struct { type testRowScanner struct {
@ -620,13 +621,14 @@ func TestRowToAddrOfStructPos(t *testing.T) {
func TestRowToStructByName(t *testing.T) { func TestRowToStructByName(t *testing.T) {
type person struct { type person struct {
Last string Last string
First string First string
Age int32 Age int32
AccountID string
} }
defaultConnTestRunner.RunTest(context.Background(), t, func(ctx context.Context, t testing.TB, conn *pgx.Conn) { defaultConnTestRunner.RunTest(context.Background(), t, func(ctx context.Context, t testing.TB, conn *pgx.Conn) {
rows, _ := conn.Query(ctx, `select 'John' as first, 'Smith' as last, n as age from generate_series(0, 9) n`) rows, _ := conn.Query(ctx, `select 'John' as first, 'Smith' as last, n as age, 'd5e49d3f' as account_id from generate_series(0, 9) n`)
slice, err := pgx.CollectRows(rows, pgx.RowToStructByName[person]) slice, err := pgx.CollectRows(rows, pgx.RowToStructByName[person])
assert.NoError(t, err) assert.NoError(t, err)
@ -635,6 +637,7 @@ func TestRowToStructByName(t *testing.T) {
assert.Equal(t, "Smith", slice[i].Last) assert.Equal(t, "Smith", slice[i].Last)
assert.Equal(t, "John", slice[i].First) assert.Equal(t, "John", slice[i].First)
assert.EqualValues(t, i, slice[i].Age) assert.EqualValues(t, i, slice[i].Age)
assert.Equal(t, "d5e49d3f", slice[i].AccountID)
} }
// check missing fields in a returned row // check missing fields in a returned row
@ -643,7 +646,7 @@ func TestRowToStructByName(t *testing.T) {
assert.ErrorContains(t, err, "cannot find field First in returned row") assert.ErrorContains(t, err, "cannot find field First in returned row")
// check missing field in a destination struct // check missing field in a destination struct
rows, _ = conn.Query(ctx, `select 'John' as first, 'Smith' as last, n as age, null as ignore from generate_series(0, 9) n`) rows, _ = conn.Query(ctx, `select 'John' as first, 'Smith' as last, n as age, 'd5e49d3f' as account_id, null as ignore from generate_series(0, 9) n`)
_, err = pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[person]) _, err = pgx.CollectRows(rows, pgx.RowToAddrOfStructByName[person])
assert.ErrorContains(t, err, "struct doesn't have corresponding row field ignore") assert.ErrorContains(t, err, "struct doesn't have corresponding row field ignore")
}) })