/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.commons.dbcp2.cpdsadapter;
import java.util.Properties;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import javax.sql.DataSource;
import org.apache.commons.dbcp2.datasources.SharedPoolDataSource;
import org.junit.Assert;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
/**
* Tests for DriverAdapterCPDS
*
* @version $Revision:$ $Date:$
*/
public class TestDriverAdapterCPDS extends TestCase {
public TestDriverAdapterCPDS(String testName) {
super(testName);
}
public static Test suite() {
return new TestSuite(TestDriverAdapterCPDS.class);
}
private DriverAdapterCPDS pcds;
@Override
public void setUp() throws Exception {
pcds = new DriverAdapterCPDS();
pcds.setDriver("org.apache.commons.dbcp2.TesterDriver");
pcds.setUrl("jdbc:apache:commons:testdriver");
pcds.setUser("foo");
pcds.setPassword("bar");
pcds.setPoolPreparedStatements(false);
}
/**
* JIRA: DBCP-245
*/
public void testIncorrectPassword() throws Exception
{
pcds.getPooledConnection("u2", "p2").close();
try {
// Use bad password
pcds.getPooledConnection("u1", "zlsafjk");
fail("Able to retrieve connection with incorrect password");
} catch (SQLException e1) {
// should fail
}
// Use good password
pcds.getPooledConnection("u1", "p1").close();
try {
pcds.getPooledConnection("u1", "x");
fail("Able to retrieve connection with incorrect password");
}
catch (SQLException e) {
if (!e.getMessage().startsWith("x is not the correct password")) {
throw e;
}
// else the exception was expected
}
// Make sure we can still use our good password.
pcds.getPooledConnection("u1", "p1").close();
}
public void testSimple() throws Exception {
Connection conn = pcds.getPooledConnection().getConnection();
assertNotNull(conn);
PreparedStatement stmt = conn.prepareStatement("select * from dual");
assertNotNull(stmt);
ResultSet rset = stmt.executeQuery();
assertNotNull(rset);
assertTrue(rset.next());
rset.close();
stmt.close();
conn.close();
}
public void testSimpleWithUsername() throws Exception {
Connection conn = pcds.getPooledConnection("u1", "p1").getConnection();
assertNotNull(conn);
PreparedStatement stmt = conn.prepareStatement("select * from dual");
assertNotNull(stmt);
ResultSet rset = stmt.executeQuery();
assertNotNull(rset);
assertTrue(rset.next());
rset.close();
stmt.close();
conn.close();
}
public void testClosingWithUserName()
throws Exception {
Connection[] c = new Connection[10];
for (int i=0; i<c.length; i++) {
c[i] = pcds.getPooledConnection("u1", "p1").getConnection();
}
// close one of the connections
c[0].close();
assertTrue(c[0].isClosed());
// get a new connection
c[0] = pcds.getPooledConnection("u1", "p1").getConnection();
for (Connection element : c) {
element.close();
}
// open all the connections
for (int i=0; i<c.length; i++) {
c[i] = pcds.getPooledConnection("u1", "p1").getConnection();
}
for (Connection element : c) {
element.close();
}
}
public void testSetProperties() throws Exception {
// Set user property to bad value
pcds.setUser("bad");
// Supply correct value in connection properties
// This will overwrite field value
Properties properties = new Properties();
properties.put("user", "foo");
pcds.setConnectionProperties(properties);
pcds.getPooledConnection().close();
assertEquals("foo", pcds.getUser());
// Put bad password into properties
properties.put("password", "bad");
// This does not change local field
assertEquals("bar", pcds.getPassword());
// Supply correct password in getPooledConnection
// Call will succeed and overwrite property
pcds.getPooledConnection("foo", "bar").close();
assertEquals("bar", pcds.getConnectionProperties().getProperty("password"));
}
// https://issues.apache.org/jira/browse/DBCP-376
public void testDbcp367() throws Exception {
ThreadDbcp367[] threads = new ThreadDbcp367[200];
pcds.setPoolPreparedStatements(true);
pcds.setMaxPreparedStatements(-1);
pcds.setAccessToUnderlyingConnectionAllowed(true);
SharedPoolDataSource spds = new SharedPoolDataSource();
spds.setConnectionPoolDataSource(pcds);
spds.setMaxTotal(threads.length + 10);
spds.setDefaultMaxWaitMillis(-1);
spds.setDefaultMaxIdle(10);
spds.setDefaultAutoCommit(Boolean.FALSE);
spds.setValidationQuery("SELECT 1");
spds.setDefaultTimeBetweenEvictionRunsMillis(10000);
spds.setDefaultNumTestsPerEvictionRun(-1);
spds.setDefaultTestWhileIdle(true);
spds.setDefaultTestOnBorrow(true);
spds.setDefaultTestOnReturn(false);
for (int i = 0; i < threads.length; i++) {
threads[i] = new ThreadDbcp367(spds);
threads[i].start();
}
for (int i = 0; i < threads.length; i++) {
threads[i].join();
Assert.assertFalse("Thread " + i + " has failed",threads[i].isFailed());
}
}
private static class ThreadDbcp367 extends Thread {
private final DataSource ds;
private volatile boolean failed = false;
public ThreadDbcp367(DataSource ds) {
this.ds = ds;
}
@Override
public void run() {
Connection c = null;
try {
for (int j=0; j < 5000; j++) {
c = ds.getConnection();
c.close();
}
} catch (SQLException sqle) {
failed = true;
sqle.printStackTrace();
}
}
public boolean isFailed() {
return failed;
}
}
}