From 7eb9d95a7871d77c876a243052da95db1d0abb67 Mon Sep 17 00:00:00 2001
From: wenshao <szujobs@hotmail.com>
Date: Mon, 5 Sep 2011 02:40:05 +0000
Subject: [PATCH] git-svn-id: http://code.alibabatech.com/svn/druid/trunk@633
 b9813039-fb51-4c41-a8b9-e21c2acb5095

---
 .../druid/bvt/pool/TestGraceShutdown.java     | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 src/test/java/com/alibaba/druid/bvt/pool/TestGraceShutdown.java

diff --git a/src/test/java/com/alibaba/druid/bvt/pool/TestGraceShutdown.java b/src/test/java/com/alibaba/druid/bvt/pool/TestGraceShutdown.java
new file mode 100644
index 0000000000..ef6247f822
--- /dev/null
+++ b/src/test/java/com/alibaba/druid/bvt/pool/TestGraceShutdown.java
@@ -0,0 +1,71 @@
+package com.alibaba.druid.bvt.pool;
+
+import java.sql.Connection;
+import java.util.Random;
+import java.util.concurrent.CountDownLatch;
+
+import com.alibaba.druid.mock.MockDriver;
+import com.alibaba.druid.pool.DruidDataSource;
+
+import junit.framework.TestCase;
+
+
+public class TestGraceShutdown extends TestCase {
+    private MockDriver      driver;
+    private DruidDataSource dataSource;
+
+    protected void setUp() throws Exception {
+        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");
+    }
+    
+    public void test_close() throws Exception {
+        final int threadCount = 100;
+        Thread[] threads = new Thread[threadCount];
+        final CountDownLatch startLatch = new CountDownLatch(1);
+        final CountDownLatch endLatch = new CountDownLatch(threadCount);
+
+        for (int i = 0; i < threadCount; ++i) {
+            threads[i] = new Thread("thread-" + i) {
+
+                public void run() {
+                    try {
+                        startLatch.await();
+                        Connection conn = dataSource.getConnection();
+                    } catch (Exception e) {
+                        e.printStackTrace();
+                    } finally {
+                        endLatch.countDown();
+                    }
+                }
+            };
+        }
+
+        startLatch.countDown();
+        for (int i = 0; i < threadCount; ++i) {
+            threads[i].start();
+        }
+        Thread.sleep(1000);
+        
+        new Thread("close thread") {
+            public void run() {
+                dataSource.close();
+            }
+        }.start();
+
+        endLatch.await();
+    }
+}