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 ae9f3b716c..fe06a10661 100644 --- a/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java +++ b/src/test/java/com/alibaba/druid/bvt/stat/DruidStatServiceTest.java @@ -48,7 +48,7 @@ public class DruidStatServiceTest extends TestCase { } @SuppressWarnings("unchecked") - public void test_statService() throws Exception { + public void test_statService_getSqlList() throws Exception { String sql = "select 1"; Connection conn = dataSource.getConnection(); @@ -74,4 +74,29 @@ public class DruidStatServiceTest extends TestCase { 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")); + } }