Fixed test to check for retryCount set in connection string (#2308)

microsoft · Jan 30, 2024 · 183426a · 183426a
1 parent bb76a78
commit 183426a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/test/java/com/microsoft/sqlserver/jdbc/connection/TimeoutTest.java b/src/test/java/com/microsoft/sqlserver/jdbc/connection/TimeoutTest.java
@@ -275,7 +275,12 @@ public void testAzureEndpointRetry() {
                     } else if (TestUtils.isAzure(con)) {
                         assertTrue(retryCount == 2); // AZURE_SERVER_ENDPOINT_RETRY_COUNT_DEFAFULT
                     } else {
-                        assertTrue(retryCount == 1); // default connectRetryCount
+                        // default retryCount is 1 if not set in connection string
+                        String retryCountFromConnStr = TestUtils.getProperty(connectionString, "connectRetryCount");
+                        int expectedRetryCount = (retryCountFromConnStr != null) ? Integer
+                                .parseInt(retryCountFromConnStr) : 1;
+
+                        assertTrue(retryCount == expectedRetryCount); // default connectRetryCount
                     }
                 }
             }