add testcase & bug fixed

2023-03-04 19:17:16 +00:00 · 2012-08-26 09:18:16 +08:00
parent 994b773242
commit 21c1b1da5d
7 changed files with 235 additions and 2 deletions
@@ -0,0 +1,149 @@
+package com.alibaba.druid.bvt.pool;
+
+import java.sql.CallableStatement;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+import java.sql.SQLFeatureNotSupportedException;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.filter.FilterAdapter;
+import com.alibaba.druid.filter.FilterChain;
+import com.alibaba.druid.mock.MockCallableStatement;
+import com.alibaba.druid.pool.DruidDataSource;
+import com.alibaba.druid.pool.DruidPooledCallableStatement;
+import com.alibaba.druid.proxy.jdbc.CallableStatementProxy;
+
+public class DruidPooledCallableStatementTest 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 boolean callableStatement_wasNull(FilterChain chain, CallableStatementProxy statement)
+                                                                                                         throws SQLException {
+                throw new SQLException();
+            }
+        });
+    }
+
+    protected void tearDown() throws Exception {
+        dataSource.close();
+    }
+
+    public void test_wasNull() throws Exception {
+        Connection conn = dataSource.getConnection();
+        CallableStatement stmt = conn.prepareCall("select 1");
+        stmt.execute();
+
+        Assert.assertEquals(0, dataSource.getErrorCount());
+        
+        Exception error = null;
+        try {
+            stmt.wasNull();
+        } catch (Exception e) {
+            error = e;
+        }
+        Assert.assertNotNull(error);
+        
+        Assert.assertEquals(1, dataSource.getErrorCount());
+        
+        stmt.close();
+        conn.close();
+        
+        Assert.assertEquals(1, dataSource.getPoolingCount());
+    }
+    
+    public void test_getObject() throws Exception {
+        Connection conn = dataSource.getConnection();
+        DruidPooledCallableStatement stmt = (DruidPooledCallableStatement) conn.prepareCall("select 1");
+        stmt.execute();
+
+        Assert.assertEquals(0, dataSource.getErrorCount());
+        
+        Exception error = null;
+        try {
+            stmt.getObject(1, String.class);
+        } catch (SQLFeatureNotSupportedException e) {
+            error = e;
+        }
+        Assert.assertNotNull(error);
+        
+        Assert.assertEquals(0, dataSource.getErrorCount());
+        
+        stmt.close();
+        conn.close();
+        
+        Assert.assertEquals(1, dataSource.getPoolingCount());
+    }
+    
+    public void test_getObject_1() throws Exception {
+        Connection conn = dataSource.getConnection();
+        DruidPooledCallableStatement stmt = (DruidPooledCallableStatement) conn.prepareCall("select 1");
+        stmt.execute();
+        
+        Assert.assertEquals(0, dataSource.getErrorCount());
+        
+        Exception error = null;
+        try {
+            stmt.getObject("1", String.class);
+        } catch (SQLFeatureNotSupportedException e) {
+            error = e;
+        }
+        Assert.assertNotNull(error);
+        
+        Assert.assertEquals(0, dataSource.getErrorCount());
+        
+        stmt.close();
+        conn.close();
+        
+        Assert.assertEquals(1, dataSource.getPoolingCount());
+    }
+    
+    public void test_wrap() throws Exception {
+        
+        Connection conn = dataSource.getConnection();
+        CallableStatement stmt = conn.prepareCall("select 1");
+        
+        Assert.assertNotNull(stmt.unwrap(CallableStatement.class));
+        Assert.assertEquals(MockCallableStatement.class, stmt.unwrap(CallableStatement.class).getClass());
+        
+        stmt.close();
+        conn.close();
+    }
+    
+    public void test_wrap_1() throws Exception {
+        
+        Connection conn = dataSource.getConnection();
+        CallableStatement stmt = conn.prepareCall("select 1");
+        
+        Assert.assertNotNull(stmt.unwrap(PreparedStatement.class));
+        Assert.assertEquals(MockCallableStatement.class, stmt.unwrap(CallableStatement.class).getClass());
+        
+        stmt.close();
+        conn.close();
+    }
+    
+    public void test_wrap_2() throws Exception {
+        dataSource.getProxyFilters().clear();
+        
+        Connection conn = dataSource.getConnection();
+        CallableStatement stmt = conn.prepareCall("select 1");
+        
+        Assert.assertNotNull(stmt.unwrap(PreparedStatement.class));
+        Assert.assertEquals(MockCallableStatement.class, stmt.unwrap(CallableStatement.class).getClass());
+        
+        stmt.close();
+        conn.close();
+    }
+}