org.etlunit.feature.sql_server_database.SqlCmd Maven / Gradle / Ivy

Go to download
package org.etlunit.feature.sql_server_database;

import org.etlunit.ProcessDescription;
import org.etlunit.ProcessFacade;
import org.etlunit.RuntimeSupport;
import org.etlunit.TestExecutionError;

import java.io.File;

public class SqlCmd {
	private final File sqlCmd;
	private final RuntimeSupport runtimeSupport;

	public SqlCmd(File sqlBinDir, RuntimeSupport runtimeSupport) {
		this.sqlCmd = new File(sqlBinDir, "sqlcmd.exe");

		if (!sqlCmd.exists())
		{
			throw new IllegalArgumentException("sqlcmd.exe not found");
		}

		this.runtimeSupport = runtimeSupport;
	}

	public void executeScript(
		String serverName,
		String databaseName,
		File scriptFile
	) throws Exception
	{
		ProcessDescription pb = new ProcessDescription(
			sqlCmd.getAbsolutePath())
			.argument("-b")
			.argument("-E")
			.argument("-S")
			.argument(serverName)
			.argument("-d")
			.argument(databaseName)
			.argument("-i")
			.argument(scriptFile.getAbsolutePath());

		ProcessFacade res = runtimeSupport.execute(pb);
		res.waitForCompletion();

		if (res.getCompletionCode() != 0)
		{
			throw new TestExecutionError("Could not execute script: " + scriptFile.getAbsolutePath() + " : " + res.getInputBuffered());
		}
	}
}