diff --git a/src/test/java/com/alibaba/druid/bvt/pool/basic/TestDataSourceBasic.java b/src/test/java/com/alibaba/druid/bvt/pool/basic/TestDataSourceBasic.java new file mode 100644 index 0000000000..c9ffb4f172 --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/pool/basic/TestDataSourceBasic.java @@ -0,0 +1,56 @@ +package com.alibaba.druid.bvt.pool.basic; + +import java.sql.Connection; +import java.sql.PreparedStatement; + +import com.alibaba.druid.mock.MockDriver; +import com.alibaba.druid.pool.DruidDataSource; +import com.alibaba.druid.stat.DruidDataSourceStatManager; +import com.mysql.jdbc.ResultSet; + +import junit.framework.Assert; +import junit.framework.TestCase; + +public class TestDataSourceBasic extends TestCase { + + private MockDriver driver; + private DruidDataSource dataSource; + + protected void setUp() throws Exception { + Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size()); + + driver = new MockDriver(); + + dataSource = new DruidDataSource(); + dataSource.setUrl("jdbc:mock:xxx"); + dataSource.setDriver(driver); + dataSource.setInitialSize(1); + dataSource.setMaxActive(2); + dataSource.setMaxIdle(2); + dataSource.setMinIdle(1); + dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10 + dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10 + dataSource.setTestWhileIdle(true); + dataSource.setTestOnBorrow(false); + dataSource.setValidationQuery("SELECT 1"); + dataSource.setFilters("stat"); + dataSource.setRemoveAbandoned(true); + } + + protected void tearDown() throws Exception { + dataSource.close(); + Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size()); + } + + public void test_prepare() throws Exception { + Connection conn = dataSource.getConnection(); + + Assert.assertEquals(1, dataSource.getActiveConnectionStackTrace().size()); + Assert.assertEquals(1, dataSource.getActiveConnections().size()); + + conn.close(); + + Assert.assertEquals(0, dataSource.getActiveConnectionStackTrace().size()); + Assert.assertEquals(0, dataSource.getActiveConnections().size()); + } +}