From d77a15f29c1909d1a430eb97c90e0d326ca91575 Mon Sep 17 00:00:00 2001 From: wenshao Date: Fri, 24 Aug 2012 14:08:59 +0800 Subject: [PATCH] add testcase --- .../druid/bvt/stat/DruidStatServiceTest.java | 54 +++++++++++++++++++ 1 file changed, 54 insertions(+) create mode 100644 src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java diff --git a/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java b/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java new file mode 100644 index 0000000000..9b0783b98b --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java @@ -0,0 +1,54 @@ +package com.alibaba.druid.bvt.stat; + +import java.sql.Connection; +import java.sql.PreparedStatement; +import java.sql.ResultSet; +import java.util.List; +import java.util.Map; + +import junit.framework.Assert; +import junit.framework.TestCase; + +import com.alibaba.druid.pool.DruidDataSource; +import com.alibaba.druid.stat.DruidStatService; +import com.alibaba.druid.support.json.JSONUtils; +import com.alibaba.druid.util.JdbcUtils; + +public class DruidStatServiceTest extends TestCase { + + private DruidDataSource dataSource; + + protected void setUp() throws Exception { + dataSource = new DruidDataSource(); + + dataSource.setUrl("jdbc:mock:xxx"); + dataSource.setFilters("stat"); + dataSource.setTestOnBorrow(false); + + dataSource.init(); + } + + protected void tearDown() throws Exception { + JdbcUtils.close(dataSource); + } + + public void test_statService() throws Exception { + String sql = "select 1"; + Connection conn = dataSource.getConnection(); + + PreparedStatement stmt = conn.prepareStatement(sql); + ResultSet rs = stmt.executeQuery(); + rs.next(); + rs.close(); + stmt.close(); + + conn.close(); + + String result = DruidStatService.getInstance().service("/sql.json"); + Map resultMap = (Map) JSONUtils.parse(result); + + List> sqlList = (List>) resultMap.get("Content"); + + Assert.assertEquals(1, sqlList.size()); + } +}