From 828cf770d6b536e3c316564025d31a9a11cdb282 Mon Sep 17 00:00:00 2001
From: Dominik Przybysz <dominik.przybysz@snowflake.com>
Date: Wed, 5 Jun 2024 10:13:38 +0200
Subject: [PATCH] SNOW-1446174: Allow 513 in SFTrustManager tests

---
 src/test/java/net/snowflake/client/core/SFTrustManagerIT.java | 2 +-
 src/test/java/net/snowflake/client/jdbc/RestRequestTest.java  | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/test/java/net/snowflake/client/core/SFTrustManagerIT.java b/src/test/java/net/snowflake/client/core/SFTrustManagerIT.java
index 13f3f8f09..76c6d6531 100644
--- a/src/test/java/net/snowflake/client/core/SFTrustManagerIT.java
+++ b/src/test/java/net/snowflake/client/core/SFTrustManagerIT.java
@@ -182,7 +182,7 @@ private static void accessHost(String host, HttpClient client) throws IOExceptio
     assertThat(
         String.format("response code for %s", host),
         statusCode,
-        anyOf(equalTo(200), equalTo(403), equalTo(400)));
+        anyOf(equalTo(200), equalTo(403), equalTo(400), equalTo(513)));
   }
 
   /**
diff --git a/src/test/java/net/snowflake/client/jdbc/RestRequestTest.java b/src/test/java/net/snowflake/client/jdbc/RestRequestTest.java
index ed6e165d1..608f69d02 100644
--- a/src/test/java/net/snowflake/client/jdbc/RestRequestTest.java
+++ b/src/test/java/net/snowflake/client/jdbc/RestRequestTest.java
@@ -272,6 +272,7 @@ class TestCase {
     testCases.add(new TestCase(509, false, false));
     testCases.add(new TestCase(510, false, false));
     testCases.add(new TestCase(511, false, false));
+    testCases.add(new TestCase(513, false, false));
     // do retry on HTTP 403 option
     testCases.add(new TestCase(100, true, true));
     testCases.add(new TestCase(101, true, true));
@@ -325,6 +326,7 @@ class TestCase {
     testCases.add(new TestCase(509, true, false));
     testCases.add(new TestCase(510, true, false));
     testCases.add(new TestCase(511, true, false));
+    testCases.add(new TestCase(513, true, false));
 
     for (TestCase t : testCases) {
       if (t.result) {