From c693ec4373d1c49cd1e6ead04fab140284e74453 Mon Sep 17 00:00:00 2001
From: wenshao <szujobs@qq.com>
Date: Thu, 17 Nov 2011 15:47:07 +0800
Subject: [PATCH] improve testcase

---
 .../java/com/alibaba/druid/filter/stat/StatFilter.java    | 8 ++++++++
 .../alibaba/druid/bvt/pool/TestPoolPreparedStatement.java | 2 ++
 .../druid/bvt/pool/TestPoolPreparedStatement2.java        | 5 ++++-
 3 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/alibaba/druid/filter/stat/StatFilter.java b/src/main/java/com/alibaba/druid/filter/stat/StatFilter.java
index 5d7f58c092..f163333eb1 100644
--- a/src/main/java/com/alibaba/druid/filter/stat/StatFilter.java
+++ b/src/main/java/com/alibaba/druid/filter/stat/StatFilter.java
@@ -85,6 +85,14 @@ public class StatFilter extends FilterEventAdapter implements StatFilterMBean {
         return this.dataSourceStat;
     }
 
+    public int getMaxSqlStatCount() {
+        return maxSqlStatCount;
+    }
+
+    public void setMaxSqlStatCount(int maxSqlStatCount) {
+        this.maxSqlStatCount = maxSqlStatCount;
+    }
+
     public void reset() {
         dataSourceStat.reset();
 
diff --git a/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement.java b/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement.java
index c2e208a498..d9a5019ed8 100644
--- a/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement.java
+++ b/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement.java
@@ -6,6 +6,7 @@ import java.sql.PreparedStatement;
 import junit.framework.Assert;
 import junit.framework.TestCase;
 
+import com.alibaba.druid.filter.stat.StatFilter;
 import com.alibaba.druid.mock.MockDriver;
 import com.alibaba.druid.mock.MockPreparedStatement;
 import com.alibaba.druid.pool.DruidDataSource;
@@ -35,6 +36,7 @@ public class TestPoolPreparedStatement extends TestCase {
         dataSource.setValidationQuery("SELECT 1");
         dataSource.setFilters("stat");
         dataSource.setPoolPreparedStatements(true);
+        ((StatFilter) dataSource.getProxyFilters().get(0)).setMaxSqlStatCount(100);
     }
 
     protected void tearDown() throws Exception {
diff --git a/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement2.java b/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement2.java
index b57a6e6d9a..7edad1f731 100644
--- a/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement2.java
+++ b/src/test/java/com/alibaba/druid/bvt/pool/TestPoolPreparedStatement2.java
@@ -6,6 +6,7 @@ import java.sql.PreparedStatement;
 import junit.framework.Assert;
 import junit.framework.TestCase;
 
+import com.alibaba.druid.filter.stat.StatFilter;
 import com.alibaba.druid.mock.MockDriver;
 import com.alibaba.druid.pool.DruidDataSource;
 import com.alibaba.druid.pool.PoolableConnection;
@@ -36,6 +37,8 @@ public class TestPoolPreparedStatement2 extends TestCase {
         dataSource.setFilters("stat");
         dataSource.setPoolPreparedStatements(false);
         dataSource.setMaxPoolPreparedStatementPerConnectionSize(20);
+        
+        ((StatFilter) dataSource.getProxyFilters().get(0)).setMaxSqlStatCount(100);
     }
 
     protected void tearDown() throws Exception {
@@ -68,7 +71,7 @@ public class TestPoolPreparedStatement2 extends TestCase {
             }
         }
 
-        for (int i = 0; i < 1000 * 100 * 1; ++i) {
+        for (int i = 0; i < 1000 * 1; ++i) {
             Connection conn = dataSource.getConnection();
             PreparedStatement stmt = conn.prepareStatement("SELECT " + i);
             stmt.execute();