TestPropertySetter.java

/*
 * Copyright (C) 2013, 2014 Brett Wooldridge
 *
 * 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.zaxxer.hikari.pool;

import static com.zaxxer.hikari.pool.TestElf.newHikariConfig;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertThrows;
import static org.junit.Assert.assertTrue;

import java.io.ByteArrayOutputStream;
import java.io.PrintWriter;
import java.time.Duration;
import java.util.Properties;
import java.util.Set;

import javax.sql.DataSource;

import org.junit.Test;

import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.util.PropertyElf;

public class TestPropertySetter
{
   @Test
   public void testProperty1() throws Exception
   {
      Properties propfile1 = new Properties();
      propfile1.load(TestPropertySetter.class.getResourceAsStream("/propfile1.properties"));
      HikariConfig config = new HikariConfig(propfile1);
      config.validate();

      assertEquals(5, config.getMinimumIdle());
      assertEquals("SELECT 1", config.getConnectionTestQuery());
   }

   @Test
   public void testProperty2() throws Exception
   {
      Properties propfile2 = new Properties();
      propfile2.load(TestPropertySetter.class.getResourceAsStream("/propfile2.properties"));
      HikariConfig config = new HikariConfig(propfile2);
      config.validate();

      Class<?> clazz = this.getClass().getClassLoader().loadClass(config.getDataSourceClassName());
      DataSource dataSource = (DataSource) clazz.getDeclaredConstructor().newInstance();
      PropertyElf.setTargetFromProperties(dataSource, config.getDataSourceProperties());
   }

   @Test
   public void testObjectProperty() throws Exception
   {
      HikariConfig config = newHikariConfig();
      config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
      PrintWriter writer = new PrintWriter(new ByteArrayOutputStream());
      config.addDataSourceProperty("logWriter", writer);

      Class<?> clazz = this.getClass().getClassLoader().loadClass(config.getDataSourceClassName());
      DataSource dataSource = (DataSource) clazz.getDeclaredConstructor().newInstance();
      PropertyElf.setTargetFromProperties(dataSource, config.getDataSourceProperties());

      assertSame(PrintWriter.class, dataSource.getLogWriter().getClass());
   }

   @Test
   public void testPropertyUpperCase() throws Exception
   {
      Properties propfile3 = new Properties();
      propfile3.load(TestPropertySetter.class.getResourceAsStream("/propfile3.properties"));
      HikariConfig config = new HikariConfig(propfile3);
      config.validate();

      Class<?> clazz = this.getClass().getClassLoader().loadClass(config.getDataSourceClassName());
      DataSource dataSource = (DataSource) clazz.getDeclaredConstructor().newInstance();
      PropertyElf.setTargetFromProperties(dataSource, config.getDataSourceProperties());
   }

   @Test
   public void testDurationPropertiesSet() throws Exception
   {
      Properties durationProperties = new Properties();
      durationProperties.load(TestPropertySetter.class.getResourceAsStream("/duration-config.properties"));
      HikariConfig config = new HikariConfig(durationProperties);
      config.validate();

      assertEquals(Duration.ofMillis(2000), Duration.ofMillis(config.getConnectionTimeout()));
      assertEquals(Duration.ofSeconds(22), Duration.ofMillis(config.getValidationTimeout()));
      assertEquals(Duration.ofMinutes(33), Duration.ofMillis(config.getIdleTimeout()));
      assertEquals(Duration.ofHours(44), Duration.ofMillis(config.getLeakDetectionThreshold()));
      assertEquals(Duration.ofDays(55), Duration.ofMillis(config.getMaxLifetime()));

      Class<?> clazz = this.getClass().getClassLoader().loadClass(config.getDataSourceClassName());
      DataSource dataSource = (DataSource) clazz.getDeclaredConstructor().newInstance();
      PropertyElf.setTargetFromProperties(dataSource, config.getDataSourceProperties());

      assertEquals(Duration.ofMinutes(1), Duration.ofMillis(dataSource.getLoginTimeout()));
   }

   @Test
   public void testGetPropertyNames() throws Exception
   {
      Set<String> propertyNames = PropertyElf.getPropertyNames(HikariConfig.class);
      assertTrue(propertyNames.contains("dataSourceClassName"));
   }

   @Test
   public void testSetNonExistantPropertyName() throws Exception
   {
      RuntimeException e = assertThrows(RuntimeException.class, () -> {
         Properties props = new Properties();
         props.put("what", "happened");
         PropertyElf.setTargetFromProperties(new HikariConfig(), props);
      });
      assertEquals("Property what does not exist on target class com.zaxxer.hikari.HikariConfig", e.getMessage());
   }
}