From e8d20bcea15c1b56d816024eec19b74be4434192 Mon Sep 17 00:00:00 2001
From: Bing Li <bing.li@snowflake.com>
Date: Mon, 9 Oct 2023 13:50:34 -0700
Subject: [PATCH] fix testProdConnectivity

---
 src/test/java/net/snowflake/client/jdbc/ConnectionIT.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test/java/net/snowflake/client/jdbc/ConnectionIT.java b/src/test/java/net/snowflake/client/jdbc/ConnectionIT.java
index 9a89560d2..509b12ade 100644
--- a/src/test/java/net/snowflake/client/jdbc/ConnectionIT.java
+++ b/src/test/java/net/snowflake/client/jdbc/ConnectionIT.java
@@ -135,7 +135,7 @@ public void testProdConnectivity() throws SQLException {
         DriverManager.getConnection(url, properties);
         fail();
       } catch (SQLException e) {
-        assertThat(e.getErrorCode(), is(INVALID_CONNECTION_INFO_CODE));
+        assertThat(e.getErrorCode(), anyOf(is(INVALID_CONNECTION_INFO_CODE), is(BAD_REQUEST_GS_CODE)));
       }
     }
   }