org.geoserver.data.jdbc.GenericUnWrapperTest.java Source code

Introduction

Here is the source code for org.geoserver.data.jdbc.GenericUnWrapperTest.java

Source

/* (c) 2014 Open Source Geospatial Foundation - all rights reserved
 * (c) 2001 - 2013 OpenPlans
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.data.jdbc;

import static org.junit.Assert.*;

import java.io.IOException;
import java.sql.Connection;
import java.sql.SQLException;

import org.geotools.data.jdbc.datasource.DataSourceFinder;
import org.geotools.data.jdbc.datasource.UnWrapper;
import org.junit.Before;
import org.junit.Test;

/**
 * Uses the known target org.apache.commons.dbcp.DelegatingStatement to test out GenericUnWrapper in
 * isolation.
 * 
 * @author Jody Garnett (Boundless)
 */
public class GenericUnWrapperTest {
    GenericUnWrapper unwrapper = new GenericUnWrapper();

    @Before
    public void reset() {
        // reset generic wrapper state
        GenericUnWrapper.CONNECTION_METHODS.remove(WrapperConnection.class);
    }

    @Test
    public void testUnwrapConnection() throws SQLException, NoSuchMethodException, SecurityException {
        Connection connection = new TestConnection();
        Connection wrapper = new WrapperConnection(connection);
        assertTrue(wrapper.isWrapperFor(Connection.class));
        Connection unwrap = wrapper.unwrap(Connection.class);

        assertSame(connection, unwrap);

        UnWrapper unwrapper = new GenericUnWrapper();

        assertFalse(unwrapper.canUnwrap(wrapper));
        try {
            assertNull(unwrapper.unwrap(wrapper));
            fail("Cannot unwrap yet");
        } catch (Exception expected) {
        }
        GenericUnWrapper.CONNECTION_METHODS.put(WrapperConnection.class,
                WrapperConnection.class.getMethod("getUnderlyingConnection", null));

        assertTrue(unwrapper.canUnwrap(wrapper));
        assertSame(connection, unwrapper.unwrap(wrapper));
    }

    @Test
    public void testSPIRegistration() throws Exception {
        Connection connection = new TestConnection();
        Connection wrapper = new WrapperConnection(connection);

        UnWrapper uw = DataSourceFinder.getUnWrapper(wrapper);
        if (uw != null) {
            assertTrue("picked up by jdbc4 extractor", uw instanceof SpringUnWrapper);
        }

        GenericUnWrapper.CONNECTION_METHODS.put(WrapperConnection.class,
                WrapperConnection.class.getMethod("getUnderlyingConnection", null));

        uw = DataSourceFinder.getUnWrapper(wrapper);
        assertNotNull("registed and canUnwrap", uw);
        if (uw instanceof GenericUnWrapper) {
            assertSame("Generic unwrapper is working", connection, uw.unwrap(wrapper));
        }
    }

}