| 220 | } |
| 221 | |
| 222 | func TestCollectOneRow(t *testing.T) { |
| 223 | defaultConnTestRunner.RunTest(context.Background(), t, func(ctx context.Context, t testing.TB, conn *pgx.Conn) { |
| 224 | rows, _ := conn.Query(ctx, `select 42`) |
| 225 | n, err := pgx.CollectOneRow(rows, func(row pgx.CollectableRow) (int32, error) { |
| 226 | var n int32 |
| 227 | err := row.Scan(&n) |
| 228 | return n, err |
| 229 | }) |
| 230 | assert.NoError(t, err) |
| 231 | assert.Equal(t, int32(42), n) |
| 232 | }) |
| 233 | } |
| 234 | |
| 235 | func TestCollectOneRowNotFound(t *testing.T) { |
| 236 | defaultConnTestRunner.RunTest(context.Background(), t, func(ctx context.Context, t testing.TB, conn *pgx.Conn) { |