Update test for Synapse OnDemand servers (#2302)

microsoft · Jan 15, 2024 · 066aeeb · 066aeeb
1 parent 0f04ce6
commit 066aeeb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/test/java/com/microsoft/sqlserver/jdbc/TestUtils.java b/src/test/java/com/microsoft/sqlserver/jdbc/TestUtils.java
@@ -237,6 +237,15 @@ public static boolean isAzureMI(Connection con) {
         return ((SQLServerConnection) con).isAzureMI();
     }
 
+    /**
+     * Checks if connection is established to Azure Synapse OnDemand server
+     * 
+     */
+    public static boolean isAzureSynapseOnDemand(Connection con) {
+        isAzure(con);
+        return ((SQLServerConnection) con).isAzureSynapseOnDemandEndpoint();
+    }
+
     /**
      * Checks if connection is established to server that supports AEv2.
      * 

diff --git a/src/test/java/com/microsoft/sqlserver/jdbc/connection/TimeoutTest.java b/src/test/java/com/microsoft/sqlserver/jdbc/connection/TimeoutTest.java
@@ -266,12 +266,12 @@ public void testAzureEndpointRetry() {
         try (Connection con = PrepUtil.getConnection(connectionString)) {
             Field fields[] = con.getClass().getSuperclass().getDeclaredFields();
             for (Field f : fields) {
-                if (f.getName() == "connectRetryCount") {
+                if (f.getName().equals("connectRetryCount")) {
                     f.setAccessible(true);
                     int retryCount = f.getInt(con);
 
-                    if (TestUtils.isAzureDW(con)) {
-                        assertTrue(retryCount == 5); // AZURE_SYNAPSE_ONDEMAND_ENDPOINT_RETRY_COUNT_DEFAFULT
+                    if (TestUtils.isAzureSynapseOnDemand(con)) {
+                        assertTrue(retryCount == 5); // AZURE_SYNAPSE_ONDEMAND_ENDPOINT_RETRY_COUNT_DEFAULT
                     } else if (TestUtils.isAzure(con)) {
                         assertTrue(retryCount == 2); // AZURE_SERVER_ENDPOINT_RETRY_COUNT_DEFAFULT
                     } else {