DruidDataSource支持Filter

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@52 b9813039-fb51-4c41-a8b9-e21c2acb5095
2023-03-04 19:17:16 +00:00 · 2011-05-13 08:51:40 +00:00
parent 4af8848041
commit d387694ee0
3 changed files with 105 additions and 4 deletions
@@ -0,0 +1,96 @@
+package com.alibaba.druid.bvt.pool;
+
+import java.sql.Connection;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.pool.DruidDataSource;
+import com.alibaba.druid.stat.JdbcStatManager;
+
+public class ParamTest extends TestCase {
+
+    public void test_default() throws Exception {
+        DruidDataSource dataSource = new DruidDataSource();
+        dataSource.setUrl("jdbc:mock:");
+
+        Assert.assertEquals(0, dataSource.getFilters().size());
+
+        dataSource.setFilters("stat");
+
+        JdbcStatManager.getInstance().reset();
+
+        Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+        Assert.assertEquals(1, dataSource.getFilters().size());
+
+        for (int i = 0; i < 2; ++i) {
+            Connection conn = dataSource.getConnection();
+
+            Assert.assertEquals(1, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+            Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getCloseCount());
+
+            conn.close();
+
+            Assert.assertEquals(1, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+            Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getCloseCount()); // logic
+                                                                                                       // close不会导致计数器＋1
+        }
+
+        dataSource.close();
+
+        Assert.assertEquals(1, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+        Assert.assertEquals(1, JdbcStatManager.getInstance().getConnectionstat().getCloseCount());
+
+        JdbcStatManager.getInstance().reset();
+
+        Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+        Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getCloseCount());
+    }
+
+    public void test_zero() throws Exception {
+        DruidDataSource dataSource = new DruidDataSource();
+        dataSource.setUrl("jdbc:mock:");
+        dataSource.setInitialSize(0);
+        dataSource.setMaxActive(0);
+        dataSource.setMinIdle(0);
+        dataSource.setMaxIdle(0);
+
+        Assert.assertEquals(0, dataSource.getFilters().size());
+
+        dataSource.setFilters("stat");
+
+        JdbcStatManager.getInstance().reset();
+
+        Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+        Assert.assertEquals(1, dataSource.getFilters().size());
+
+        Exception error = null;
+        try {
+            dataSource.getConnection();
+        } catch (Exception ex) {
+            error = ex;
+        }
+        Assert.assertNotNull(error);
+    }
+
+    public void test_1() throws Exception {
+        DruidDataSource dataSource = new DruidDataSource();
+        dataSource.setUrl("jdbc:mock:");
+        dataSource.setInitialSize(10);
+        dataSource.setMaxActive(10);
+        dataSource.setMinIdle(0);
+        dataSource.setMaxIdle(10);
+
+        dataSource.setFilters("stat");
+        Assert.assertEquals(1, dataSource.getFilters().size());
+
+        JdbcStatManager.getInstance().reset();
+
+        Assert.assertEquals(0, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+        
+        Connection conn = dataSource.getConnection();
+        conn.close();        
+        
+        Assert.assertEquals(10, JdbcStatManager.getInstance().getConnectionstat().getConnectCount());
+    }
+}