ShutdownOnExitTest.java

/*
 * Copyright 2006-2012 The Scriptella Project Team.
 *
 * 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 scriptella.driver.hsqldb;

import scriptella.AbstractTestCase;
import scriptella.configuration.MockConnectionEl;
import scriptella.jdbc.JdbcConnection;
import scriptella.jdbc.JdbcUtils;
import scriptella.spi.ConnectionParameters;
import scriptella.spi.MockDriverContext;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

/**
 * Tests shutdown on exit functionality.
 *
 * @author Fyodor Kupolov
 * @version 1.0
 */
public class ShutdownOnExitTest extends AbstractTestCase {
    public void test() throws SQLException {
        Driver drv = new Driver();
        Map<String, String> props = new HashMap<String, String>();

        final String url = "jdbc:hsqldb:mem:shutdowntest";
        MockConnectionEl conf = new MockConnectionEl(props, url);
        conf.setUser("sa");
        ConnectionParameters params = new ConnectionParameters(conf, MockDriverContext.INSTANCE);
        JdbcConnection con = drv.connect(params);
        Connection nc = con.getNativeConnection();
        Statement st = nc.createStatement();

        st.execute("        CREATE TABLE Test (ID INT);");
        props.put("ifexists", "true"); //do not create new database if not exists
        MockConnectionEl conf2 = new MockConnectionEl(props, url);
        conf.setUser("sa");
        ConnectionParameters params2 = new ConnectionParameters(conf2, MockDriverContext.INSTANCE);

        JdbcConnection con2 = drv.connect(params2);
        con2.close();
        con.close();
        Driver.HOOK.run(); //emulates shutdown hook
        assertDatabaseShutdown(url);
    }

    private void assertDatabaseShutdown(String url) {
        Properties props = new Properties();
        props.put("ifexists", "true"); //do not create new database if not exists
        props.put("user", "sa");
        try {
            final Connection con = DriverManager.getConnection(url, props);
            JdbcUtils.closeSilent(con);
        } catch (SQLException e) {
            assertTrue("Driver.HOOK must shutdown the database " + url, e.getErrorCode() == -94);
            return; //OK
        } catch (Exception e) {
            e.printStackTrace();
        }
        fail("Driver.HOOK must shutdown the database");
    }


    /**
     * Tests if no failures occurs on attempt to close connection for shutdown database
     *
     * @throws SQLException
     */
    public void testAlreadyClosed() throws SQLException {
        Driver drv = new Driver();
        Map<String, String> props = new HashMap<String, String>();

        final String url = "jdbc:hsqldb:mem:alreadyClosed";
        MockConnectionEl conf = new MockConnectionEl(props, url);
        conf.setUser("sa");
        ConnectionParameters params = new ConnectionParameters(conf, MockDriverContext.INSTANCE);
        JdbcConnection con = drv.connect(params);
        Connection nc = con.getNativeConnection();
        nc.createStatement().execute("SHUTDOWN;");
        con.close();
        Driver.HOOK.run(); //emulates shutdown hook
    }

    /**
     * Tests if different databases are independently shutdown, i.e. closing one db
     * does not affect other.
     *
     * @throws SQLException
     */
    public void testDifferentDbs() throws SQLException {
        Driver drv = new Driver();
        Map<String, String> props = new HashMap<String, String>();

        //Create first db and obtain 2 connections
        String url1 = "jdbc:hsqldb:mem:DifferentDbs1";
        MockConnectionEl conf = new MockConnectionEl(props, url1);
        conf.setUser("sa");
        ConnectionParameters params = new ConnectionParameters(conf, MockDriverContext.INSTANCE);
        JdbcConnection con1 = drv.connect(params);
        JdbcConnection con11 = drv.connect(params);
        //Create second db and obtain 2 connections
        String url2 = "jdbc:hsqldb:mem:DifferentDbs2";
        conf = new MockConnectionEl(props, url2);
        conf.setUser("sa");
        params = new ConnectionParameters(conf, MockDriverContext.INSTANCE);
        JdbcConnection con2 = drv.connect(params);
        JdbcConnection con22 = drv.connect(params);
        //close everything
        con1.close();
        con2.close();
        con11.close();
        con22.close();
        Driver.HOOK.run(); //emulates shutdown hook
        //all databases should be correctly shutdown
        assertDatabaseShutdown(url1);
        assertDatabaseShutdown(url2);
    }


}