gopkg/pgx
2
0
Fork 0
Code Activity

Extract connection dead on server test

Browse Source
This commit is contained in:
Jack Christensen
2017-02-04 16:03:20 -06:00
parent 24193ee322
commit a9e7e3acbc
3 changed files with 19 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
query_test.go
+2 -16
View File
@@ -1513,14 +1513,7 @@ func TestQueryContextCancelationCancelsQuery(t *testing.T) {
t.Fatal("Expected context.Canceled error, got %v", rows.Err())
}
checkConn := mustConnect(t, *defaultConnConfig)
defer closeConn(t, checkConn)
var found bool
err = checkConn.QueryRow("select true from pg_stat_activity where pid=$1", conn.Pid).Scan(&found)
if err != pgx.ErrNoRows {
t.Fatal("Expected context canceled connection to be disconnected from server, but it wasn't")
}
ensureConnDeadOnServer(t, conn, *defaultConnConfig)
}
func TestQueryRowContextSuccess(t *testing.T) {
@@ -1580,12 +1573,5 @@ func TestQueryRowContextCancelationCancelsQuery(t *testing.T) {
t.Fatal("Expected context.Canceled error, got %v", err)
}
checkConn := mustConnect(t, *defaultConnConfig)
defer closeConn(t, checkConn)
var found bool
err = checkConn.QueryRow("select true from pg_stat_activity where pid=$1", conn.Pid).Scan(&found)
if err != pgx.ErrNoRows {
t.Fatal("Expected context canceled connection to be disconnected from server, but it wasn't")
}
ensureConnDeadOnServer(t, conn, *defaultConnConfig)
}