Add prepared statement query for xml assertions (#34111)

apache · Dec 20, 2024 · 3a9a490 · 3a9a490
1 parent c2e05d9
commit 3a9a490
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/...sql/src/test/java/org/apache/shardingsphere/test/e2e/engine/type/dql/GeneralDQLE2EIT.java b/...sql/src/test/java/org/apache/shardingsphere/test/e2e/engine/type/dql/GeneralDQLE2EIT.java
@@ -69,6 +69,14 @@ private void assertExecuteQueryWithXmlExpected(final AssertionTestParameter test
                 || "proxy".equals(testParam.getAdapter()) && "empty_storage_units".equalsIgnoreCase(testParam.getScenario())) {
             return;
         }
+        if (SQLExecuteType.LITERAL == context.getSqlExecuteType()) {
+            assertQueryForStatementWithXmlExpected(context);
+        } else {
+            assertQueryForPreparedStatementWithXmlExpected(context);
+        }
+    }
+
+    private void assertQueryForStatementWithXmlExpected(final E2ETestContext context) throws SQLException {
         try (
                 Connection connection = getEnvironmentEngine().getTargetDataSource().getConnection();
                 Statement statement = connection.createStatement();
@@ -77,6 +85,19 @@ private void assertExecuteQueryWithXmlExpected(final AssertionTestParameter test
         }
     }
 
+    private void assertQueryForPreparedStatementWithXmlExpected(final E2ETestContext context) throws SQLException {
+        try (
+                Connection connection = getEnvironmentEngine().getTargetDataSource().getConnection();
+                PreparedStatement preparedStatement = connection.prepareStatement(context.getSQL())) {
+            for (SQLValue each : context.getAssertion().getSQLValues()) {
+                preparedStatement.setObject(each.getIndex(), each.getValue());
+            }
+            try (ResultSet resultSet = preparedStatement.executeQuery()) {
+                assertResultSet(context, resultSet);
+            }
+        }
+    }
+
     private void assertExecuteQueryWithExpectedDataSource(final AssertionTestParameter testParam, final E2ETestContext context) throws SQLException {
         try (
                 Connection actualConnection = getEnvironmentEngine().getTargetDataSource().getConnection();
@@ -143,6 +164,14 @@ private void assertExecuteWithXmlExpected(final AssertionTestParameter testParam
         if ("jdbc".equals(testParam.getAdapter())) {
             return;
         }
+        if (SQLExecuteType.LITERAL == context.getSqlExecuteType()) {
+            assertExecuteForStatementWithXmlExpected(context);
+        } else {
+            assertExecuteForPreparedStatementWithXmlExpected(context);
+        }
+    }
+
+    private void assertExecuteForStatementWithXmlExpected(final E2ETestContext context) throws SQLException {
         try (
                 Connection connection = getEnvironmentEngine().getTargetDataSource().getConnection();
                 Statement statement = connection.createStatement()) {
@@ -152,6 +181,20 @@ private void assertExecuteWithXmlExpected(final AssertionTestParameter testParam
         }
     }
 
+    private void assertExecuteForPreparedStatementWithXmlExpected(final E2ETestContext context) throws SQLException {
+        try (
+                Connection connection = getEnvironmentEngine().getTargetDataSource().getConnection();
+                PreparedStatement preparedStatement = connection.prepareStatement(context.getSQL())) {
+            for (SQLValue each : context.getAssertion().getSQLValues()) {
+                preparedStatement.setObject(each.getIndex(), each.getValue());
+            }
+            assertTrue(preparedStatement.execute(), "Not a query preparedStatement.");
+            try (ResultSet resultSet = preparedStatement.getResultSet()) {
+                assertResultSet(context, resultSet);
+            }
+        }
+    }
+
     private void assertExecuteWithExpectedDataSource(final AssertionTestParameter testParam, final E2ETestContext context) throws SQLException {
         try (
                 Connection actualConnection = getEnvironmentEngine().getTargetDataSource().getConnection();