From f1df39a29d23ae4e5175b92c69697f2bf9b4e112 Mon Sep 17 00:00:00 2001
From: Jack Christensen <jack@jackchristensen.com>
Date: Sat, 4 Jun 2022 08:58:05 -0500
Subject: [PATCH] Initial passing tests

---
 pgconn/pgconn.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pgconn/pgconn.go b/pgconn/pgconn.go
index 01ad7eb5..9ecd8f11 100644
--- a/pgconn/pgconn.go
+++ b/pgconn/pgconn.go
@@ -1159,7 +1159,7 @@ func (pgConn *PgConn) CopyFrom(ctx context.Context, r io.Reader, sql string) (Co
 	buf[0] = 'd'
 
 	var readErr, pgErr error
-	for {
+	for pgErr == nil {
 		// Read chunk from r.
 		var n int
 		n, readErr = r.Read(buf[5:cap(buf)])
@@ -1182,7 +1182,7 @@ func (pgConn *PgConn) CopyFrom(ctx context.Context, r io.Reader, sql string) (Co
 		}
 
 		// Read messages until error or none available.
-		for {
+		for pgErr == nil {
 			msg, err := pgConn.receiveMessage()
 			if err != nil {
 				if errors.Is(err, nbbconn.ErrWouldBlock) {