From 30d44c03699f810bf322faeccc3d5f5e4bee4656 Mon Sep 17 00:00:00 2001
From: Jack Christensen <jack@jackchristensen.com>
Date: Sat, 13 Feb 2021 14:18:03 -0600
Subject: [PATCH] Test code instead of error message for CockroachDB

---
 query_test.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/query_test.go b/query_test.go
index 76ea8079..e41eb31d 100644
--- a/query_test.go
+++ b/query_test.go
@@ -2006,7 +2006,13 @@ func TestQueryErrorWithNilStatementCacheMode(t *testing.T) {
 	require.NoError(t, err)
 	rows.Close()
 	err = rows.Err()
-	require.EqualError(t, err, `ERROR: duplicate key value violates unique constraint "t_unq_pkey" (SQLSTATE 23505)`)
+	require.Error(t, err)
+	var pgErr *pgconn.PgError
+	if errors.As(err, &pgErr) {
+		assert.Equal(t, "23505", pgErr.Code)
+	} else {
+		t.Errorf("err is not a *pgconn.PgError: %T", err)
+	}
 
 	ensureConnValid(t, conn)
 }