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

/*
 * Copyright 2011 Alibaba Group.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
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);
		}
	}
}