diff --git a/apm-sniffer/apm-sdk-plugin/jdbc-plugin/src/test/java/org/skywalking/apm/plugin/jdbc/SwPreparedStatementTest.java b/apm-sniffer/apm-sdk-plugin/jdbc-plugin/src/test/java/org/skywalking/apm/plugin/jdbc/SwPreparedStatementTest.java index fb75d202903d49b127b727a4a8887a9350d081c9..4a5658128e8ceea3afa75da567525764f7e18b40 100644 --- a/apm-sniffer/apm-sdk-plugin/jdbc-plugin/src/test/java/org/skywalking/apm/plugin/jdbc/SwPreparedStatementTest.java +++ b/apm-sniffer/apm-sdk-plugin/jdbc-plugin/src/test/java/org/skywalking/apm/plugin/jdbc/SwPreparedStatementTest.java @@ -298,6 +298,21 @@ public class SwPreparedStatementTest extends AbstractStatementTest { assertDBSpan(spans.get(0), "Mysql/JDBI/PreparedStatement/executeQuery", "SELECT * FROM test"); } + + @Test + public void testExecute() throws SQLException { + PreparedStatement preparedStatement = swConnection.prepareStatement("SELECT * FROM test", 1, 1, 1); + preparedStatement.execute(); + + preparedStatement.close(); + + verify(mysqlPreparedStatement, times(1)).execute(); + verify(mysqlPreparedStatement, times(1)).close(); + TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0); + List spans = SegmentHelper.getSpans(traceSegment); + assertThat(spans.size(), is(1)); + assertDBSpan(spans.get(0), "Mysql/JDBI/PreparedStatement/execute", "SELECT * FROM test"); + } @Test public void testQuerySqlWithSql() throws SQLException { PreparedStatement preparedStatement = swConnection.prepareStatement("SELECT * FROM test", 1); @@ -359,7 +374,7 @@ public class SwPreparedStatementTest extends AbstractStatementTest { } @Test - public void testExecute() throws SQLException { + public void testExecuteWithSQL() throws SQLException { PreparedStatement preparedStatement = swConnection.prepareStatement("UPDATE test SET a = ?"); preparedStatement.setString(1, "a"); boolean updateCount = preparedStatement.execute("UPDATE test SET a = 1");