ConfigurationTest.java

package com.vladmihalcea.flexypool.metric.codahale;

import com.codahale.metrics.ExponentiallyDecayingReservoir;
import com.codahale.metrics.Metric;
import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.Reservoir;
import com.vladmihalcea.flexypool.adaptor.PoolAdapter;
import com.vladmihalcea.flexypool.adaptor.PoolAdapterFactory;
import com.vladmihalcea.flexypool.common.ConfigurationProperties;
import com.vladmihalcea.flexypool.config.Configuration;
import com.vladmihalcea.flexypool.connection.ConnectionProxyFactory;
import com.vladmihalcea.flexypool.metric.Metrics;
import com.vladmihalcea.flexypool.metric.MetricsFactory;
import org.junit.Test;
import org.mockito.Mockito;

import javax.sql.DataSource;

import static org.junit.Assert.*;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.when;

/**
 * ConfigurationTest - Configuration Test
 *
 * @author Vlad Mihalcea
 */
public class ConfigurationTest {

    @Test
    public void testBuilder() {
        final MetricRegistry metricRegistry = Mockito.mock(MetricRegistry.class);

        DataSource dataSource = Mockito.mock(DataSource.class);
        PoolAdapterFactory<DataSource> poolAdapterFactory = Mockito.mock(PoolAdapterFactory.class);
        ConnectionProxyFactory connectionProxyFactory = Mockito.mock(ConnectionProxyFactory.class);
        Metrics metrics = Mockito.mock(Metrics.class);
        PoolAdapter poolAdapter = Mockito.mock(PoolAdapter.class);
        when(poolAdapterFactory.newInstance(any(ConfigurationProperties.class))).thenReturn(poolAdapter);
        Configuration<DataSource> configuration = new Configuration.Builder<DataSource>(
                "unique", dataSource, poolAdapterFactory)
                .setConnectionProxyFactory(connectionProxyFactory)
                .setJmxAutoStart(true)
                .setJmxEnabled(true)
                .setMetricLogReporterMillis(120)
                .setMetricsFactory(new MetricsFactory() {
                    @Override
                    public Metrics newInstance(ConfigurationProperties configurationProperties) {
                        return new CodahaleMetrics(configurationProperties, metricRegistry, new ReservoirFactory() {
                            @Override
                            public Reservoir newInstance(Class<? extends Metric> metricClass, String metricName) {
                                return new ExponentiallyDecayingReservoir();
                            }
                        });
                    }
                })
                .build();
        assertSame("unique", configuration.getUniqueName());
        assertSame(connectionProxyFactory, configuration.getConnectionProxyFactory());
        assertTrue(configuration.isJmxAutoStart());
        assertTrue(configuration.isJmxEnabled());
        assertEquals(120, configuration.getMetricLogReporterMillis());
        assertSame(poolAdapter, configuration.getPoolAdapter());
        assertSame(dataSource, configuration.getTargetDataSource());
    }
}