Tx.Status handles in transaction error

refs #421

tx_test.go

@@ -354,6 +354,39 @@ func TestTxStatus(t *testing.T) {
 	}
 }
 
+func TestTxStatusErrorInTransactions(t *testing.T) {
+	t.Parallel()
+
+	conn := mustConnect(t, *defaultConnConfig)
+	defer closeConn(t, conn)
+
+	tx, err := conn.Begin()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if status := tx.Status(); status != pgx.TxStatusInProgress {
+		t.Fatalf("Expected status to be %v, but it was %v", pgx.TxStatusInProgress, status)
+	}
+
+	_, err = tx.Exec("syntax error")
+	if err == nil {
+		t.Fatal("expected an error but did not get one")
+	}
+
+	if status := tx.Status(); status != pgx.TxStatusInFailure {
+		t.Fatalf("Expected status to be %v, but it was %v", pgx.TxStatusInFailure, status)
+	}
+
+	if err := tx.Rollback(); err != nil {
+		t.Fatal(err)
+	}
+
+	if status := tx.Status(); status != pgx.TxStatusRollbackSuccess {
+		t.Fatalf("Expected status to be %v, but it was %v", pgx.TxStatusRollbackSuccess, status)
+	}
+}
+
 func TestTxErr(t *testing.T) {
 	t.Parallel()