From 5f58e0e4431311dc2a9aa376ef056a3553df9422 Mon Sep 17 00:00:00 2001 From: Jack Christensen Date: Sat, 13 Feb 2021 13:02:16 -0600 Subject: [PATCH] Cockroach DB uses different error code --- query_test.go | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/query_test.go b/query_test.go index 13472240..b9119cf8 100644 --- a/query_test.go +++ b/query_test.go @@ -627,8 +627,15 @@ func TestQueryEncodeError(t *testing.T) { if rows.Err() == nil { t.Error("Expected rows.Err() to return error, but it didn't") } - if !strings.Contains(rows.Err().Error(), "SQLSTATE 22P02") { - t.Error("Expected rows.Err() to return different error:", rows.Err()) + if conn.PgConn().ParameterStatus("crdb_version") != "" { + if !strings.Contains(rows.Err().Error(), "SQLSTATE 08P01") { + // CockroachDB returns protocol_violation instead of invalid_text_representation + t.Error("Expected rows.Err() to return different error:", rows.Err()) + } + } else { + if !strings.Contains(rows.Err().Error(), "SQLSTATE 22P02") { + t.Error("Expected rows.Err() to return different error:", rows.Err()) + } } }