SNOW-1935764: Explicitly pick account for heartbeat tests on jenkins (#…

…2085)
snowflakedb · Feb 20, 2025 · 7f998d9 · 7f998d9
1 parent 808407a
commit 7f998d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/test/java/net/snowflake/client/jdbc/HeartbeatAsyncLatestIT.java b/src/test/java/net/snowflake/client/jdbc/HeartbeatAsyncLatestIT.java
@@ -43,7 +43,7 @@ protected void submitQuery(boolean useKeepAliveSession, int queryIdx)
         "CLIENT_SESSION_KEEP_ALIVE",
         useKeepAliveSession ? Boolean.TRUE.toString() : Boolean.FALSE.toString());
 
-    try (Connection connection = getConnection(sessionParams);
+    try (Connection connection = getConnection("s3testaccount", sessionParams);
         Statement stmt = connection.createStatement();
         // Query will take 5 seconds to run, but ResultSet will be returned immediately
         ResultSet resultSet =
@@ -83,7 +83,7 @@ public void testAsynchronousQueryFailure() throws Exception {
   @Test
   @DontRunOnGithubActions
   public void testIsValidWithInvalidSession() throws Exception {
-    try (Connection connection = getConnection()) {
+    try (Connection connection = getConnection("s3testaccount")) {
       // assert that connection starts out valid
       assertTrue(connection.isValid(5));
       Thread.sleep(61000); // sleep 61 seconds to await session expiration time

diff --git a/src/test/java/net/snowflake/client/jdbc/HeartbeatIT.java b/src/test/java/net/snowflake/client/jdbc/HeartbeatIT.java
@@ -92,7 +92,7 @@ protected void submitQuery(boolean useKeepAliveSession, int queryIdx)
         "CLIENT_SESSION_KEEP_ALIVE",
         useKeepAliveSession ? Boolean.TRUE.toString() : Boolean.FALSE.toString());
 
-    try (Connection connection = getConnection(sessionParams);
+    try (Connection connection = getConnection("s3testaccount", sessionParams);
         Statement statement = connection.createStatement()) {
 
       Thread.sleep(61000); // sleep 61 seconds