/*
    *    Copyright 2009-2023 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
    *
    *       https://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.jupiter.api.Assertions.assertEquals;
  
  import java.net.URL;
  import java.net.URLClassLoader;
  import java.sql.Driver;
  import java.sql.DriverManager;
  import java.util.Enumeration;
  
  import org.junit.jupiter.api.Disabled;
  import org.junit.jupiter.api.Test;
  
  class UnpooledDataSourceTest {
  
    @Test
    void shouldNotRegisterTheSameDriverMultipleTimes() throws Exception {
      // https://github.com/mybatis/old-google-code-issues/issues/430
      UnpooledDataSource dataSource = new UnpooledDataSource("org.hsqldb.jdbcDriver", "jdbc:hsqldb:mem:multipledrivers",
          "sa", "");
      dataSource.getConnection().close();
      int before = countRegisteredDrivers();
      dataSource = new UnpooledDataSource("org.hsqldb.jdbcDriver", "jdbc:hsqldb:mem:multipledrivers", "sa", "");
      dataSource.getConnection().close();
      assertEquals(before, countRegisteredDrivers());
    }
  
    @Disabled("Requires MySQL server and a driver.")
    @Test
    void shouldRegisterDynamicallyLoadedDriver() throws Exception {
      int before = countRegisteredDrivers();
      ClassLoader driverClassLoader = new URLClassLoader(
          new URL[] { new URL("jar:file:/PATH_TO/mysql-connector-java-5.1.25.jar!/") });
      UnpooledDataSource dataSource = new UnpooledDataSource(driverClassLoader, "com.mysql.jdbc.Driver",
          "jdbc:mysql://127.0.0.1/test", "root", "");
      dataSource.getConnection().close();
      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().close();
      assertEquals(before + 1, countRegisteredDrivers());
    }
  
    int countRegisteredDrivers() {
      Enumeration<Driver> drivers = DriverManager.getDrivers();
      int count = 0;
      while (drivers.hasMoreElements()) {
        drivers.nextElement();
        count++;
      }
      return count;
    }
  
  }