JDBCPluginTest.java

/*
 * Copyright (c) 2014 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 io.werval.modules.jdbc;

import com.codahale.metrics.MetricRegistry;
import io.werval.test.WervalRule;
import java.sql.Connection;
import java.sql.SQLException;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.sql.DataSource;
import org.junit.ClassRule;
import org.junit.Test;
import io.werval.modules.metrics.Metrics;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.core.IsEqual.equalTo;
import static org.hamcrest.core.IsInstanceOf.instanceOf;
import static org.hamcrest.core.IsNull.notNullValue;
import static org.junit.Assert.assertThat;

/**
 * JDBC Plugin Test.
 */
public class JDBCPluginTest
{
    @ClassRule
    public static final WervalRule WERVAL = new WervalRule();

    @Test
    public void dataSourcesSetup()
    {
        JDBC jdbc = WERVAL.application().plugin( JDBC.class );
        assertThat( jdbc.dataSource(), equalTo( jdbc.dataSource( "default" ) ) );
        assertThat( jdbc.dataSource( "another" ), notNullValue() );
    }

    @Test
    public void dataSourceUsage()
        throws SQLException
    {
        DataSource dataSource = WERVAL.application().plugin( JDBC.class ).dataSource();
        try( Connection connection = dataSource.getConnection() )
        {
            connection.getMetaData().getTypeInfo();
        }
    }

    @Test
    public void connectionUsage()
        throws SQLException
    {
        try( Connection connection = WERVAL.application().plugin( JDBC.class ).connection() )
        {
            connection.getMetaData().getTypeInfo();
        }
    }

    @Test
    public void jndiLookup()
        throws NamingException
    {
        Object ds = new InitialContext().lookup( "defaultDS" );
        assertThat( ds, notNullValue() );
        assertThat( ds, instanceOf( DataSource.class ) );
    }

    @Test
    public void metrics()
    {
        MetricRegistry metrics = WERVAL.application().plugin( Metrics.class ).metrics();
        assertThat( metrics.getGauges().get( "default.pool.TotalConnections" ).getValue(), is( 10 ) );
    }
}