/** * Copyright 2009-2015 the original author or authors. * * 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 org.apache.ibatis.datasource.unpooled; import static org.junit.Assert.*; import java.net.URL; import java.net.URLClassLoader; import java.sql.Driver; import java.sql.DriverManager; import java.util.Enumeration; import org.junit.Ignore; import org.junit.Test; public class UnpooledDataSourceTest { @Test public void shouldNotRegisterTheSameDriverMultipleTimes() throws Exception { // https://code.google.com/p/mybatis/issues/detail?id=430 UnpooledDataSource dataSource = null; dataSource = new UnpooledDataSource("org.hsqldb.jdbcDriver", "jdbc:hsqldb:mem:multipledrivers", "sa", ""); dataSource.getConnection(); int before = countRegisteredDrivers(); dataSource = new UnpooledDataSource("org.hsqldb.jdbcDriver", "jdbc:hsqldb:mem:multipledrivers", "sa", ""); dataSource.getConnection(); assertEquals(before, countRegisteredDrivers()); } @Ignore("Requires MySQL server and a driver.") @Test public void shouldRegisterDynamicallyLoadedDriver() throws Exception { int before = countRegisteredDrivers(); ClassLoader driverClassLoader = null; UnpooledDataSource dataSource = null; driverClassLoader = new URLClassLoader(new URL[] { new URL("jar:file:/PATH_TO/mysql-connector-java-5.1.25.jar!/") }); dataSource = new UnpooledDataSource(driverClassLoader, "com.mysql.jdbc.Driver", "jdbc:mysql://127.0.0.1/test", "root", ""); dataSource.getConnection(); assertEquals(before + 1, countRegisteredDrivers()); driverClassLoader = new URLClassLoader(new URL[] { new URL("jar:file:/PATH_TO/mysql-connector-java-5.1.25.jar!/") }); dataSource = new UnpooledDataSource(driverClassLoader, "com.mysql.jdbc.Driver", "jdbc:mysql://127.0.0.1/test", "root", ""); dataSource.getConnection(); assertEquals(before + 1, countRegisteredDrivers()); } protected int countRegisteredDrivers() { Enumeration<Driver> drivers = DriverManager.getDrivers(); int count = 0; while (drivers.hasMoreElements()) { drivers.nextElement(); count++; } return count; } }