SNOW-1957486 fix test (#2119)

Author: sfc-gh-pmotacki

src/test/java/net/snowflake/client/jdbc/ConnectionIT.java

@@ -58,6 +58,7 @@ public class ConnectionIT extends BaseJDBCWithSharedConnectionIT {
   private static final int SESSION_CREATION_OBJECT_DOES_NOT_EXIST_NOT_AUTHORIZED = 390201;
   private static final int ROLE_IN_CONNECT_STRING_DOES_NOT_EXIST = 390189;
   public static final int BAD_REQUEST_GS_CODE = 390400;
+  public static final int NETWORK_ERROR_CODE = 200015;
 
   public static final int WAIT_FOR_TELEMETRY_REPORT_IN_MILLISECS = 5000;
 

src/test/java/net/snowflake/client/jdbc/ConnectionLatestIT.java

@@ -3,6 +3,7 @@
 import static net.snowflake.client.core.SessionUtil.CLIENT_SESSION_KEEP_ALIVE_HEARTBEAT_FREQUENCY;
 import static net.snowflake.client.jdbc.ConnectionIT.BAD_REQUEST_GS_CODE;
 import static net.snowflake.client.jdbc.ConnectionIT.INVALID_CONNECTION_INFO_CODE;
+import static net.snowflake.client.jdbc.ConnectionIT.NETWORK_ERROR_CODE;
 import static net.snowflake.client.jdbc.ConnectionIT.WAIT_FOR_TELEMETRY_REPORT_IN_MILLISECS;
 import static org.awaitility.Awaitility.await;
 import static org.hamcrest.CoreMatchers.equalTo;
@@ -592,7 +593,10 @@ public void testHttpsLoginTimeoutWithOutSSL() throws InterruptedException {
         assertThat(
             "Communication error",
             e.getErrorCode(),
-            anyOf(equalTo(INVALID_CONNECTION_INFO_CODE), equalTo(BAD_REQUEST_GS_CODE)));
+            anyOf(
+                equalTo(INVALID_CONNECTION_INFO_CODE),
+                equalTo(BAD_REQUEST_GS_CODE),
+                equalTo(NETWORK_ERROR_CODE)));
 
         // since it returns normal response,
         // the telemetry does not create new event