add testcase & bug fixed

This commit is contained in:
wenshao
2012-08-25 14:22:56 +08:00
parent 66f7a4818f
commit 76eced3873
3 changed files with 154 additions and 0 deletions
@@ -0,0 +1,41 @@
package com.alibaba.druid.bvt.pool;
import java.sql.SQLException;
import junit.framework.TestCase;
import com.alibaba.druid.filter.FilterAdapter;
import com.alibaba.druid.filter.FilterChain;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.proxy.jdbc.ConnectionProxy;
/**
* 这个场景测试initialSize > maxActive
*
* @author wenshao<szujobs@hotmail.com>
*/
public class DruidDataSourceTest_closeError extends TestCase {
private DruidDataSource dataSource;
protected void setUp() throws Exception {
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:xxx");
dataSource.setTestOnBorrow(false);
dataSource.setInitialSize(1);
dataSource.getProxyFilters().add(new FilterAdapter() {
public void connection_close(FilterChain chain, ConnectionProxy connection) throws SQLException {
throw new SQLException();
}
});
dataSource.init();
}
protected void tearDown() throws Exception {
dataSource.close();
}
public void test_closeError() throws Exception {
}
}
@@ -0,0 +1,34 @@
package com.alibaba.druid.bvt.pool;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.pool.DruidDataSource;
/**
* 这个场景测试initialSize > maxActive
*
* @author wenshao<szujobs@hotmail.com>
*/
public class DruidDataSourceTest_getProperties extends TestCase {
private DruidDataSource dataSource;
protected void setUp() throws Exception {
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:xxx");
dataSource.setTestOnBorrow(false);
dataSource.getConnectProperties().put("user", "jobs");
dataSource.getConnectProperties().put("password", "xxx");
dataSource.init();
}
protected void tearDown() throws Exception {
dataSource.close();
}
public void test_error() throws Exception {
Assert.assertEquals(-1, dataSource.getProperties().indexOf("xxx"));
Assert.assertEquals(true, dataSource.getProperties().indexOf("******") != -1);
}
}
@@ -0,0 +1,79 @@
package com.alibaba.druid.bvt.pool;
import java.sql.Connection;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicInteger;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.pool.DruidDataSource;
/**
* 这个场景测试initialSize > maxActive
*
* @author wenshao<szujobs@hotmail.com>
*/
public class DruidDataSourceTest_maxWait extends TestCase {
private DruidDataSource dataSource;
protected void setUp() throws Exception {
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:xxx");
dataSource.setTestOnBorrow(false);
dataSource.setMaxActive(1);
dataSource.setMaxWait(30);
dataSource.setInitialSize(1);
dataSource.init();
}
protected void tearDown() throws Exception {
dataSource.close();
}
public void test_maxWait() throws Exception {
Connection conn = dataSource.getConnection();
final AtomicInteger errorCount = new AtomicInteger();
final CountDownLatch endLatch = new CountDownLatch(2);
Thread t1 = new Thread("t-1") {
public void run() {
try {
dataSource.getConnection();
} catch (Exception e) {
errorCount.incrementAndGet();
} finally {
endLatch.countDown();
}
}
};
Thread t2 = new Thread("t-2") {
public void run() {
try {
dataSource.getConnection();
} catch (Exception e) {
errorCount.incrementAndGet();
} finally {
endLatch.countDown();
}
}
};
t1.start();
t2.start();
Thread.sleep(10);
t1.interrupt();
endLatch.await();
Assert.assertEquals(2, errorCount.get());
conn.close();
}
}