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

Test sending CopyData before CopyFrom responds with error

v5-dev
Jack Christensen 2022-09-03 09:28:42 -05:00
parent f8d088cfb6
commit 782133158f
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pgconn/pgconn_test.go
View File

@ -1779,7 +1779,18 @@ func TestConnCopyFromQuerySyntaxError(t *testing.T) {
srcBuf := &bytes.Buffer{}
res, err := pgConn.CopyFrom(context.Background(), srcBuf, "cropy foo to stdout")
// Send data even though the COPY FROM command will be rejected with a syntax error. This ensures that this does not
// break the connection. See https://github.com/jackc/pgconn/pull/127 for context.
inputRows := [][][]byte{}
for i := 0; i < 1000; i++ {
a := strconv.Itoa(i)
b := "foo " + a + " bar"
inputRows = append(inputRows, [][]byte{[]byte(a), []byte(b)})
_, err = srcBuf.Write([]byte(fmt.Sprintf("%s,\"%s\"\n", a, b)))
require.NoError(t, err)
}
res, err := pgConn.CopyFrom(context.Background(), srcBuf, "cropy foo FROM STDIN WITH (FORMAT csv)")
require.Error(t, err)
assert.IsType(t, &pgconn.PgError{}, err)
assert.Equal(t, int64(0), res.RowsAffected())