Share project "druid" into "http://code.alibabatech.com/svn/druid"

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@2 b9813039-fb51-4c41-a8b9-e21c2acb5095

src/test/java/com/alibaba/druid/bvt/proxy/ConnectionTest.java

@@ -0,0 +1,166 @@
+package com.alibaba.druid.bvt.proxy;
+
+import java.sql.Blob;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLClientInfoException;
+import java.sql.SQLException;
+import java.sql.SQLFeatureNotSupportedException;
+import java.sql.Savepoint;
+import java.sql.Statement;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.util.JdbcUtils;
+
+public class ConnectionTest extends TestCase {
+	private static String create_url = "jdbc:wrap-jdbc:filters=default,commonLogging,log4j:name=demo:jdbc:derby:memory:connectionTestDB;create=true";
+
+	protected void setUp() throws Exception {
+		Class.forName("com.alibaba.druid.proxy.DruidDriver");
+
+		Connection conn = DriverManager.getConnection(create_url);
+
+		createTable();
+
+		conn.close();
+	}
+
+	private void createTable() throws SQLException {
+		Connection conn = DriverManager.getConnection(create_url);
+		Statement stmt = conn.createStatement();
+		stmt.execute("CREATE TABLE T_BLOB (ID INTEGER, DATA BLOB)");
+		stmt.close();
+		conn.close();
+	}
+
+	private void dropTable() throws SQLException {
+		Connection conn = DriverManager.getConnection(create_url);
+		Statement stmt = conn.createStatement();
+		stmt.execute("DROP TABLE T_BLOB");
+		stmt.close();
+		conn.close();
+	}
+
+	protected void tearDown() throws Exception {
+		dropTable();
+	}
+
+	@SuppressWarnings("deprecation")
+	public void test_connection() throws Exception {
+		Connection conn = null;
+		PreparedStatement pstmt = null;
+		Statement stmt = null;
+		ResultSet rs = null;
+
+		try {
+			conn = DriverManager.getConnection(create_url);
+
+			conn.setCatalog(conn.getCatalog());
+			conn.setClientInfo(conn.getClientInfo());
+			conn.setHoldability(conn.getHoldability());
+			conn.setReadOnly(conn.isReadOnly());
+			conn.setTransactionIsolation(conn.getTransactionIsolation());
+			conn.setTypeMap(conn.getTypeMap());
+			
+			try {
+			conn.setClientInfo("name", "value");
+			} catch (SQLClientInfoException ex) {
+				
+			}
+			
+			try {
+				conn.createArrayOf("VARCHAR", new String[] { "A", "B" });
+			} catch (SQLFeatureNotSupportedException ex) {
+			}
+
+			try {
+				conn.createNClob();
+			} catch (SQLFeatureNotSupportedException ex) {
+			}
+
+			try {
+				conn.createSQLXML();
+			} catch (SQLFeatureNotSupportedException ex) {
+			}
+
+			try {
+				conn.createStruct("VARCHAR", new String[] { "A", "B" });
+			} catch (SQLFeatureNotSupportedException ex) {
+			}
+
+			conn.setAutoCommit(false);
+			Savepoint savePoint = conn.setSavepoint("XX");
+			conn.releaseSavepoint(savePoint);
+
+			pstmt = conn.prepareStatement("INSERT INTO T_BLOB (ID, DATA) VALUES (?, ?)");
+
+			Blob blob = conn.createBlob();
+
+			blob.setBytes(1, new byte[100]);
+
+			pstmt.setInt(1, 1);
+			pstmt.setBlob(2, blob);
+
+			int updateCount = pstmt.executeUpdate();
+			Assert.assertEquals(1, updateCount);
+
+			stmt = conn.createStatement();
+
+			conn.nativeSQL("SELECT ID, DATA FROM T_BLOB");
+			// //////
+			rs = stmt.executeQuery("SELECT ID, DATA FROM T_BLOB");
+			rs.getStatement(); // just call
+			while (rs.next()) {
+				Blob readBlob = rs.getBlob(2);
+				readBlob.length();
+				readBlob.getBinaryStream(1, 100).close();
+				readBlob.getBinaryStream().close();
+				readBlob.free();
+				try {
+					rs.getUnicodeStream(1).close();
+				} catch (SQLFeatureNotSupportedException ex) {
+				}
+				try {
+					rs.getUnicodeStream("DATA").close();
+				} catch (SQLFeatureNotSupportedException ex) {
+				}
+			}
+			JdbcUtils.close(rs);
+
+			rs = stmt.executeQuery("SELECT ID, DATA FROM T_BLOB");
+			while (rs.next()) {
+				rs.getBinaryStream(2).close();
+			}
+			JdbcUtils.close(rs);
+
+			rs = stmt.executeQuery("SELECT ID, DATA FROM T_BLOB");
+			while (rs.next()) {
+				rs.getBinaryStream("DATA").close();
+			}
+			JdbcUtils.close(rs);
+
+			rs = stmt.executeQuery("SELECT ID, DATA FROM T_BLOB");
+			while (rs.next()) {
+				rs.getBytes(2);
+			}
+			JdbcUtils.close(rs);
+
+			rs = stmt.executeQuery("SELECT ID, DATA FROM T_BLOB");
+			while (rs.next()) {
+				rs.getBytes("DATA");
+			}
+			JdbcUtils.close(rs);
+
+			conn.setAutoCommit(true);
+		} finally {
+			JdbcUtils.close(rs);
+			JdbcUtils.close(stmt);
+			JdbcUtils.close(pstmt);
+			JdbcUtils.close(conn);
+		}
+	}
+}