diff --git a/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java b/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java index d8dafc0fa6..fe06a10661 100644 --- a/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java +++ b/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java @@ -47,23 +47,56 @@ public class DruidStatServiceTest extends TestCase { JdbcUtils.close(dataSource); } - public void test_statService() throws Exception { + @SuppressWarnings("unchecked") + public void test_statService_getSqlList() 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()); + + Map sqlStat = sqlList.get(0); + + Assert.assertEquals(0, sqlStat.get("RunningCount")); + Assert.assertEquals(1, sqlStat.get("ExecuteCount")); + Assert.assertEquals(1, sqlStat.get("FetchRowCount")); + Assert.assertEquals(0, sqlStat.get("EffectedRowCount")); + } + + public void test_statService_getDataSourceList() 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("/datasource.json"); + Map resultMap = (Map) JSONUtils.parse(result); + + List> dataSourceList = (List>) resultMap.get("Content"); + + Assert.assertEquals(1, dataSourceList.size()); + + Map dataSourceStat = dataSourceList.get(0); + + Assert.assertEquals(1, dataSourceStat.get("PoolingCount")); + Assert.assertEquals(0, dataSourceStat.get("ActiveCount")); } }