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

Add test to validate CollectRows for empty Rows 

https://github.com/jackc/pgx/issues/1924
https://github.com/jackc/pgx/issues/1925
pull/1937/head
Felix 2024-03-05 04:07:44 +01:00 committed by Jack Christensen
parent da6f2c98f2
commit 0cc4c14e62
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
rows_test.go
View File

@ -175,6 +175,21 @@ func TestCollectRows(t *testing.T) {
}) })
} }
func TestCollectRowsEmpty(t *testing.T) {
defaultConnTestRunner.RunTest(context.Background(), t, func(ctx context.Context, t testing.TB, conn *pgx.Conn) {
rows, _ := conn.Query(ctx, `select n from generate_series(1, 0) n`)
numbers, err := pgx.CollectRows(rows, func(row pgx.CollectableRow) (int32, error) {
var n int32
err := row.Scan(&n)
return n, err
})
require.NoError(t, err)
require.NotNil(t, numbers)
assert.Empty(t, numbers)
})
}
// This example uses CollectRows with a manually written collector function. In most cases RowTo, RowToAddrOf, // This example uses CollectRows with a manually written collector function. In most cases RowTo, RowToAddrOf,
// RowToStructByPos, RowToAddrOfStructByPos, or another generic function would be used. // RowToStructByPos, RowToAddrOfStructByPos, or another generic function would be used.
func ExampleCollectRows() { func ExampleCollectRows() {