From a73597d5c99f2cbffa5c591f8efb86e556f61b0f Mon Sep 17 00:00:00 2001 From: ascrutae Date: Fri, 28 Jul 2017 11:49:31 +0800 Subject: [PATCH] add prepared statement test case --- .../plugin/jdbc/SwPreparedStatementTest.java | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) 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 fb75d2029..4a5658128 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"); -- GitLab