AbstractDbTestTemplate.java

package org.jsoftware.fods;

import org.hsqldb.jdbc.JDBCDriver;
import org.jsoftware.fods.client.ext.Configuration;
import org.jsoftware.fods.impl.AbstractFoDataSourceFactory;
import org.jsoftware.fods.impl.PropertiesBasedConfigurationFactory;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;

import javax.sql.DataSource;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;

public abstract class AbstractDbTestTemplate {
	private static final String TEST_CONFIGURATION_PROPERTIES = "testConfiguration.properties";
	private DBHolder[] dbs = new DBHolder[] { new DBHolder(0), new DBHolder(1) };
	protected Configuration configuration;



	@BeforeClass
	public static void loadDriver() throws InstantiationException, IllegalAccessException, SQLException, IOException {
		JDBCDriver.class.newInstance();
		DriverManager.setLoginTimeout(10);
	}



	@Before
	public void startDbs() throws SQLException, IOException {
		for (DBHolder holder : dbs) {
			holder.start();
		}
	}



	@After
	public void stop() throws SQLException {
		for (DBHolder holder : dbs) {
			holder.stop();
		}
	}



	@Before
	public void prepareFODS() throws SQLException, IOException {
		PropertiesBasedConfigurationFactory propertiesConfigurationFactory = new PropertiesBasedConfigurationFactory();
		Properties properties = new Properties();
		int i = 4;
		String name = TEST_CONFIGURATION_PROPERTIES;
		InputStream ins = null;
		try {
			do {
				ins = AbstractDbTestTemplate.class.getResourceAsStream(name);
				i--;
				name = "../" + name;
			} while (i > 0 && ins == null);
			if (ins == null) {
				throw new IOException("Can not load " + TEST_CONFIGURATION_PROPERTIES);
			}
			properties.load(ins);
		} finally {
			if (ins != null) {
				ins.close();
			}
		}
		propertiesConfigurationFactory.setProperties(properties);
		configuration = propertiesConfigurationFactory.getConfiguration();
	}



	protected DataSource getFoDS() throws IOException {
		AbstractFoDataSourceFactory dsFactory = new AbstractFoDataSourceFactory() {
			@Override
			protected Configuration getConfiguration() throws IOException {
				return configuration;
			}
		};
		return dsFactory.getObjectInstance();
	}



	protected final void start(int i) throws SQLException, IOException {
		dbs[i].start();
	}



	protected final void stop(int i) throws SQLException {
		dbs[i].stop();
	}



	protected String getDbnameForConnection(Connection connection) throws SQLException {
		ResultSet rs = connection.createStatement().executeQuery("SELECT str_col FROM stable WHERE id=1");
		try {
			if (rs.next()) {
				String str = rs.getString(1);
				return str;
			} else {
				throw new IllegalStateException("Cannot fetch row from table `stable`");
			}
		} finally {
			rs.close();
		}

	}

}

class DBHolder {
	int index;
	private boolean working;



	public DBHolder(int index) {
		this.index = index;
	}



	public void start() throws SQLException, IOException {
		if (!working) {
			Connection connection = DriverManager.getConnection("jdbc:hsqldb:mem:db" + index, "sa", "");
			BufferedReader br = new BufferedReader(new InputStreamReader(getClass().getResourceAsStream("/db" + index + ".sql")));
			String s;
			Statement stm = connection.createStatement();
			try {
				while ((s = br.readLine()) != null) {
					stm.execute(s);
					working = true;
				}
			} finally {
				br.close();
				stm.close();
				connection.close();
			}
		}
	}



	public void stop() throws SQLException {
		if (working) {
			Connection connection = DriverManager.getConnection("jdbc:hsqldb:mem:db" + index, "sa", "");
			Statement stm = null;
			try {
				stm = connection.createStatement();
				stm.execute("shutdown");
			} finally {
				stm.close();
				connection.close();
			}
			working = false;
		}
	}
}