From 9e495df1d52a4dfa2e6a3281ede2fad24f675242 Mon Sep 17 00:00:00 2001
From: Jack Christensen <jack@jackchristensen.com>
Date: Sat, 29 Feb 2020 09:48:04 -0600
Subject: [PATCH] Add test for rollback failure

---
 tx_test.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/tx_test.go b/tx_test.go
index 47880bb0..7a640995 100644
--- a/tx_test.go
+++ b/tx_test.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/jackc/pgconn"
 	"github.com/jackc/pgx/v4"
+	"github.com/stretchr/testify/require"
 )
 
 func TestTransactionSuccessfulCommit(t *testing.T) {
@@ -188,6 +189,25 @@ func TestTransactionSuccessfulRollback(t *testing.T) {
 	}
 }
 
+func TestTransactionRollbackFailsClosesConnection(t *testing.T) {
+	t.Parallel()
+
+	conn := mustConnectString(t, os.Getenv("PGX_TEST_DATABASE"))
+	defer closeConn(t, conn)
+
+	ctx, cancel := context.WithCancel(context.Background())
+
+	tx, err := conn.Begin(ctx)
+	require.NoError(t, err)
+
+	cancel()
+
+	err = tx.Rollback(ctx)
+	require.Error(t, err)
+
+	require.True(t, conn.IsClosed())
+}
+
 func TestBeginIsoLevels(t *testing.T) {
 	t.Parallel()