2
0

Add batch test for QueryRow without any rows

refs #1150
This commit is contained in:
Jack Christensen
2022-02-05 20:11:58 -06:00
parent 9eccdd6a81
commit d02b2ed013
+7
View File
@@ -2,6 +2,7 @@ package pgx_test
import ( import (
"context" "context"
"errors"
"os" "os"
"testing" "testing"
@@ -33,6 +34,7 @@ func TestConnSendBatch(t *testing.T) {
batch.Queue("insert into ledger(description, amount) values($1, $2)", "q3", 3) batch.Queue("insert into ledger(description, amount) values($1, $2)", "q3", 3)
batch.Queue("select id, description, amount from ledger order by id") batch.Queue("select id, description, amount from ledger order by id")
batch.Queue("select id, description, amount from ledger order by id") batch.Queue("select id, description, amount from ledger order by id")
batch.Queue("select * from ledger where false")
batch.Queue("select sum(amount) from ledger") batch.Queue("select sum(amount) from ledger")
br := conn.SendBatch(context.Background(), batch) br := conn.SendBatch(context.Background(), batch)
@@ -127,6 +129,11 @@ func TestConnSendBatch(t *testing.T) {
t.Error(err) t.Error(err)
} }
err = br.QueryRow().Scan(&id, &description, &amount)
if !errors.Is(err, pgx.ErrNoRows) {
t.Errorf("expected pgx.ErrNoRows but got: %v", err)
}
err = br.QueryRow().Scan(&amount) err = br.QueryRow().Scan(&amount)
if err != nil { if err != nil {
t.Error(err) t.Error(err)