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

Streamline test

pull/483/head
Jack Christensen 2019-01-12 12:04:44 -06:00
parent 7f373ee92b
commit 08bd2b612f
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
conn_test.go
View File

@ -247,15 +247,11 @@ func TestExecContextCancelationCancelsQuery(t *testing.T) {
conn := mustConnectString(t, os.Getenv("PGX_TEST_DATABASE"))
defer closeConn(t, conn)
ctx, cancelFunc := context.WithCancel(context.Background())
go func() {
time.Sleep(500 * time.Millisecond)
cancelFunc()
}()
_, err := conn.Exec(ctx, "select pg_sleep(60)")
if err != context.Canceled {
t.Fatalf("Expected context.Canceled err, got %v", err)
ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
_, err := conn.Exec(ctx, "select pg_sleep(1)")
cancel()
if err != context.DeadlineExceeded {
t.Fatalf("Expected context.DeadlineExceeded err, got %v", err)
}
if !conn.LastStmtSent() {
t.Error("Expected LastStmtSent to return true")