From 129ff96567730c92ea6b94a846d1cf0d8c7541c5 Mon Sep 17 00:00:00 2001 From: Jack Christensen Date: Thu, 24 Mar 2016 16:22:27 -0500 Subject: [PATCH] TestFatalRxError expects multiple error types --- conn_test.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/conn_test.go b/conn_test.go index bcab2b7d..99f222b4 100644 --- a/conn_test.go +++ b/conn_test.go @@ -1214,8 +1214,10 @@ func TestFatalRxError(t *testing.T) { var n int32 var s string err := conn.QueryRow("select 1::int4, pg_sleep(10)::varchar").Scan(&n, &s) - if pgErr, ok := err.(pgx.PgError); !ok || pgErr.Severity != "FATAL" { - t.Fatalf("Expected QueryRow Scan to return fatal PgError, but instead received %v", err) + if err == pgx.ErrDeadConn { + } else if pgErr, ok := err.(pgx.PgError); ok && pgErr.Severity == "FATAL" { + } else { + t.Fatalf("Expected QueryRow Scan to return fatal PgError or ErrDeadConn, but instead received %v", err) } }()