fixe

Author: lilgreenbird

src/test/java/com/microsoft/sqlserver/jdbc/connection/TimeoutTest.java

@@ -26,6 +26,7 @@
 
 import com.microsoft.sqlserver.jdbc.RandomUtil;
 import com.microsoft.sqlserver.jdbc.SQLServerDataSource;
+import com.microsoft.sqlserver.jdbc.SQLServerDriver;
 import com.microsoft.sqlserver.jdbc.SQLServerStatement;
 import com.microsoft.sqlserver.jdbc.TestResource;
 import com.microsoft.sqlserver.jdbc.TestUtils;
@@ -275,7 +276,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
                     }
                 }
             }