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

Move and rename test

pull/1372/head
Jack Christensen 2022-10-01 10:09:57 -05:00
parent fb83fb0cc3
commit 305c4ddbc7
1 changed files with 45 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
pgconn/pgconn_test.go
View File

@ -1858,6 +1858,51 @@ func TestConnCopyFromNoticeResponseReceivedMidStream(t *testing.T) {
require.NoError(t, err)
}
type delayedReader struct {
r io.Reader
}
func (d delayedReader) Read(p []byte) (int, error) {
// W/o sleep test passes, with sleep it fails.
time.Sleep(time.Millisecond)
return d.r.Read(p)
}
// https://github.com/jackc/pgconn/issues/128
func TestConnCopyFromDataWriteAfterErrorAndReturn(t *testing.T) {
connString := os.Getenv("PGX_TEST_CONN_STRING")
if connString == "" {
t.Skipf("Skipping due to missing environment variable %v", "PGX_TEST_CONN_STRING")
}
config, err := pgconn.ParseConfig(connString)
require.NoError(t, err)
pgConn, err := pgconn.ConnectConfig(context.Background(), config)
require.NoError(t, err)
if pgConn.ParameterStatus("crdb_version") != "" {
t.Skip("Server does support COPY FROM")
}
setupSQL := `create temporary table t (
id text primary key,
n int not null
);`
_, err = pgConn.Exec(context.Background(), setupSQL).ReadAll()
assert.NoError(t, err)
r1 := delayedReader{r: strings.NewReader(`id 0\n`)}
// Generate an error with a bogus COPY command
_, err = pgConn.CopyFrom(context.Background(), r1, "COPY nosuchtable FROM STDIN ")
assert.Error(t, err)
r2 := delayedReader{r: strings.NewReader(`id 0\n`)}
_, err = pgConn.CopyFrom(context.Background(), r2, "COPY t FROM STDIN")
assert.NoError(t, err)
}
func TestConnEscapeString(t *testing.T) {
t.Parallel()
@ -2814,47 +2859,3 @@ func TestSNISupport(t *testing.T) {
})
}
}
type delayedReader struct {
r io.Reader
}
func (d delayedReader) Read(p []byte) (int, error) {
// W/o sleep test passes, with sleep it fails.
time.Sleep(time.Millisecond)
return d.r.Read(p)
}
func TestCopyFrom(t *testing.T) {
connString := os.Getenv("PGX_TEST_CONN_STRING")
if connString == "" {
t.Skipf("Skipping due to missing environment variable %v", "PGX_TEST_CONN_STRING")
}
config, err := pgconn.ParseConfig(connString)
require.NoError(t, err)
pgConn, err := pgconn.ConnectConfig(context.Background(), config)
require.NoError(t, err)
if pgConn.ParameterStatus("crdb_version") != "" {
t.Skip("Server does support COPY FROM")
}
setupSQL := `create temporary table t (
id text primary key,
n int not null
);`
_, err = pgConn.Exec(context.Background(), setupSQL).ReadAll()
assert.NoError(t, err)
r1 := delayedReader{r: strings.NewReader(`id 0\n`)}
// Generate an error with a bogus COPY command
_, err = pgConn.CopyFrom(context.Background(), r1, "COPY nosuchtable FROM STDIN ")
assert.Error(t, err)
r2 := delayedReader{r: strings.NewReader(`id 0\n`)}
_, err = pgConn.CopyFrom(context.Background(), r2, "COPY t FROM STDIN")
assert.NoError(t, err)
}