OutputToFileIntegrationTest.java

package com.dbdeploy.integration;

import com.dbdeploy.DbDeploy;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.*;
import org.junit.Test;
import static org.junit.Assert.fail;

import java.io.File;
import java.sql.SQLException;
import java.util.List;

public class OutputToFileIntegrationTest {
	@Test
	public void shouldSuccessfullyApplyAValidSetOfDeltas() throws Exception {
		Database db = new Database("success_test");
		db.createSchemaVersionTable();

		File outputFile = File.createTempFile("success",".sql");

		DbDeploy dbDeploy = new DbDeploy();
		db.applyDatabaseSettingsTo(dbDeploy);
		dbDeploy.setScriptdirectory(findScriptDirectory("src/it/db/deltas"));
		dbDeploy.setOutputfile(outputFile);
		dbDeploy.go();

		db.applyScript(outputFile);

		assertThat(db.getChangelogEntries(), hasItems(1L, 2L));

		List<Object[]> results = db.executeQuery("select id from Test");
		assertThat(results.size(), is(1));
		assertThat((Integer) results.get(0)[0], is(6));
	}

	@Test
	public void shouldBeAbleToRecoverFromBadScriptsJustByRunningCorrectedScriptsAgain() throws Exception {
		File outputFile = File.createTempFile("recovery",".sql");

		Database db = new Database("failure_recovery_test");
		db.createSchemaVersionTable();

		DbDeploy dbDeploy = new DbDeploy();
		db.applyDatabaseSettingsTo(dbDeploy);
		dbDeploy.setScriptdirectory(findScriptDirectory("src/it/db/invalid_deltas"));
		dbDeploy.setOutputfile(outputFile);
		dbDeploy.go();

		try {
			db.applyScript(outputFile);
		} catch (SQLException ex) {
			//expected
			assertThat(ex.getMessage(), startsWith("Column count does not match in statement"));
		}

		// script 2 failed, so it should not be considered applied to the database
		assertThat(db.getChangelogEntries(), hasItems(1L));
		assertThat(db.getChangelogEntries(), not(hasItems(2L)));

		List<Object[]> results = db.executeQuery("select id from Test");
		assertThat(results.size(), is(0));

		// now run dbdeploy again with valid scripts, should recover
		dbDeploy.setScriptdirectory(findScriptDirectory("src/it/db/deltas"));
		dbDeploy.setOutputfile(outputFile);
		dbDeploy.go();

		db.applyScript(outputFile);

		assertThat(db.getChangelogEntries(), hasItems(1L, 2L));

		results = db.executeQuery("select id from Test");
		assertThat(results.size(), is(1));
	}

   @Test
    public void shouldUseSpecifiedChangeLogTable() throws Exception {
        Database db = new Database("user_defined_changelog_test", "user_defined_changelog_table");
        db.createSchemaVersionTable();

        File outputFile = File.createTempFile("changelog_success", ".sql");

        DbDeploy dbDeploy = new DbDeploy();
        db.applyDatabaseSettingsTo(dbDeploy);
        dbDeploy.setScriptdirectory(findScriptDirectory("src/it/db/deltas"));
        dbDeploy.setOutputfile(outputFile);
        dbDeploy.setChangeLogTableName("user_defined_changelog_table");
        dbDeploy.go();

        db.applyScript(outputFile);

        assertThat(db.getChangelogEntries(), hasItems(1L, 2L));
    }

    @Test
	public void shouldNotAddCommaSeparatorsToHighNumberedScripts() throws Exception {
		Database db = new Database("high_number_test");
		db.createSchemaVersionTable();

		File outputFile = File.createTempFile("high_number_test",".sql");

		DbDeploy dbDeploy = new DbDeploy();
		db.applyDatabaseSettingsTo(dbDeploy);
		dbDeploy.setScriptdirectory(findScriptDirectory("src/it/db/high_numbers"));
		dbDeploy.setOutputfile(outputFile);
		dbDeploy.go();

		db.applyScript(outputFile);
	}

	private File findScriptDirectory(String directoryName) {
		File directoryWhenRunningUnderMaven = new File(directoryName);
		if (directoryWhenRunningUnderMaven.isDirectory()) {
			return directoryWhenRunningUnderMaven;
		}

		File directoryWhenRunningUnderIde = new File("dbdeploy-core", directoryName);
		if (directoryWhenRunningUnderIde.isDirectory()) {
			return directoryWhenRunningUnderIde;
		}

		fail("Could not find script directory: " + directoryName);

		return null;
	}

}