fix test

Author: sfc-gh-bli

src/main/java/net/snowflake/client/core/QueryExecDTO.java

@@ -30,6 +30,31 @@ public class QueryExecDTO {
   // Boolean value that, if true, indicates query should be asynchronous
   private boolean asyncExec;
 
+  public QueryExecDTO(
+      String sqlText,
+      boolean describeOnly,
+      Integer sequenceId,
+      Map<String, ParameterBindingDTO> bindings,
+      String bindStage,
+      Map<String, Object> parameters,
+      QueryContextDTO queryContext,
+      long querySubmissionTime,
+      boolean internal,
+      boolean asyncExec) {
+    this(
+        sqlText,
+        null,
+        describeOnly,
+        sequenceId,
+        bindings,
+        bindStage,
+        parameters,
+        queryContext,
+        querySubmissionTime,
+        internal,
+        asyncExec);
+  }
+
   public QueryExecDTO(
       String sqlText,
       String dataframeAst,

src/main/java/net/snowflake/client/core/SFBaseStatement.java

@@ -92,13 +92,16 @@ public abstract SFBaseResultSet execute(
    * @throws SFException exception raised from Snowflake components
    * @throws SQLException if SQL error occurs
    */
-  public abstract SFBaseResultSet execute(
+  public SFBaseResultSet execute(
       String sql,
       String dataframeAst,
       Map<String, ParameterBindingDTO> parametersBinding,
       CallingMethod caller,
       ExecTimeTelemetryData execTimeData)
-      throws SQLException, SFException;
+      throws SQLException, SFException {
+    // only used internally, not a public API
+    throw new UnsupportedOperationException();
+  }
 
   /**
    * Execute sql asynchronously. Note that at a minimum, this does not have to be supported; if

src/main/java/net/snowflake/client/core/SFStatement.java

@@ -315,6 +315,33 @@ public Void call() throws SQLException {
     executor.schedule(new TimeBombTask(this), this.queryTimeout, TimeUnit.SECONDS);
   }
 
+  /**
+   * A helper method to build URL and submit the SQL to snowflake for exec
+   *
+   * @param sql sql statement
+   * @param mediaType media type
+   * @param bindValues map of binding values
+   * @param describeOnly whether only show the result set metadata
+   * @param internal run internal query not showing up in history
+   * @param asyncExec is async execute
+   * @param execTimeData ExecTimeTelemetryData
+   * @return raw json response
+   * @throws SFException if query is canceled
+   * @throws SnowflakeSQLException if query is already running
+   */
+  public Object executeHelper(
+      String sql,
+      String mediaType,
+      Map<String, ParameterBindingDTO> bindValues,
+      boolean describeOnly,
+      boolean internal,
+      boolean asyncExec,
+      ExecTimeTelemetryData execTimeData)
+      throws SnowflakeSQLException, SFException {
+    return executeHelper(
+        sql, null, mediaType, bindValues, describeOnly, internal, asyncExec, execTimeData);
+  }
+
   /**
    * A helper method to build URL and submit the SQL to snowflake for exec
    *
@@ -706,7 +733,7 @@ public SFBaseResultSet execute(
       CallingMethod caller,
       ExecTimeTelemetryData execTimeData)
       throws SQLException, SFException {
-    return execute(sql, null, false, parametersBinding, caller, execTimeData);
+    return execute(sql, false, parametersBinding, caller, execTimeData);
   }
 
   @Override
@@ -762,6 +789,29 @@ private void cancelHelper(String sql, String mediaType, CancellationReason cance
     }
   }
 
+  /**
+   * Execute sql
+   *
+   * @param sql sql statement.
+   * @param asyncExec is async exec
+   * @param parametersBinding parameters to bind
+   * @param caller the JDBC interface method that called this method, if any
+   * @param execTimeData ExecTimeTelemetryData
+   * @return whether there is result set or not
+   * @throws SQLException if failed to execute sql
+   * @throws SFException exception raised from Snowflake components
+   * @throws SQLException if SQL error occurs
+   */
+  public SFBaseResultSet execute(
+      String sql,
+      boolean asyncExec,
+      Map<String, ParameterBindingDTO> parametersBinding,
+      CallingMethod caller,
+      ExecTimeTelemetryData execTimeData)
+      throws SQLException, SFException {
+    return execute(sql, null, asyncExec, parametersBinding, caller, execTimeData);
+  }
+
   /**
    * Execute sql
    *

src/test/java/net/snowflake/client/jdbc/DataframeAstIT.java

@@ -52,7 +52,7 @@ public void testSendAst() throws SQLException, IOException {
                       anyBoolean(),
                       any(),
                       any()))
-          .thenReturn("dummy");
+          .thenReturn("{\"result\":\"dummy\"}");
       SnowflakeStatementV1 stmt =
           new SnowflakeStatementV1(
               mockedConn,