Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

improve java.sql.Wrapper support

Browse Source
This commit is contained in:
wenshao
2012-12-19 14:58:23 +08:00
parent ce9c120dbe
commit 2a028bdc05
28 changed files with 1203 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/bvt/pool/TestOracleWall2.java
+33
View File
@@ -23,6 +23,9 @@ import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidPooledConnection;
import com.alibaba.druid.pool.DruidPooledPreparedStatement;
import com.alibaba.druid.pool.DruidPooledResultSet;
import com.alibaba.druid.test.util.OracleMockDriver;
import com.alibaba.druid.util.JdbcUtils;
import com.alibaba.druid.wall.WallConfig;
@@ -61,9 +64,39 @@ public class TestOracleWall2 extends TestCase {
{
Connection conn = dataSource.getConnection();
Assert.assertTrue(conn.isWrapperFor(DruidPooledConnection.class));
Assert.assertNotNull(conn.unwrap(DruidPooledConnection.class));
Assert.assertFalse(conn.isWrapperFor(oracle.jdbc.OracleConnection.class));
Assert.assertNull(conn.unwrap(oracle.jdbc.OracleConnection.class));
Assert.assertFalse(conn.isWrapperFor(java.sql.Connection.class));
Assert.assertNull(conn.unwrap(java.sql.Connection.class));
// /////////////
PreparedStatement stmt = conn.prepareStatement(sql);
Assert.assertNull(stmt.unwrap(oracle.jdbc.OraclePreparedStatement.class));
Assert.assertFalse(stmt.isWrapperFor(oracle.jdbc.OraclePreparedStatement.class));
Assert.assertTrue(stmt.isWrapperFor(DruidPooledPreparedStatement.class));
Assert.assertNotNull(stmt.unwrap(DruidPooledPreparedStatement.class));
Assert.assertFalse(stmt.isWrapperFor(java.sql.PreparedStatement.class));
Assert.assertNull(stmt.unwrap(java.sql.PreparedStatement.class));
ResultSet rs = stmt.executeQuery();
Assert.assertNull(rs.unwrap(oracle.jdbc.OracleResultSet.class));
Assert.assertFalse(rs.isWrapperFor(oracle.jdbc.OracleResultSet.class));
Assert.assertTrue(rs.isWrapperFor(DruidPooledResultSet.class));
Assert.assertNotNull(rs.unwrap(DruidPooledResultSet.class));
Assert.assertFalse(rs.isWrapperFor(java.sql.ResultSet.class));
Assert.assertNull(rs.unwrap(java.sql.ResultSet.class));
rs.next();
rs.close();