package info.globalcode.sql.dk;

import info.globalcode.sql.dk.CLIParser.Tokens;
import static info.globalcode.sql.dk.CLIParser.TYPE_NAME_SEPARATOR;
import java.sql.Types;
import java.util.Collection;
import static org.testng.Assert.*;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

/**
 *
 * @author Ing. František Kučera (frantovo.cz)
 */
public class CLIParserTest {

	private static final String DATABASE_NAME_1 = "some database 1";
	private static final String SQL_1 = "SELECT * FROM table1";
	private static final String DATA_1 = "aaa";
	private static final String DATA_2 = "bbb";
	private static final String DATA_3 = "ccc";
	private static final String NAME_1 = "param1";
	private static final String NAME_2 = "param2";
	private static final String NAME_3 = "param3";
	private CLIParser parser;

	@BeforeMethod
	public void setUpMethod() throws Exception {
		parser = new CLIParser();
	}

	@Test
	public void testParseOptions_QueryNow_NoParams() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.DB, DATABASE_NAME_1,
			Tokens.SQL, SQL_1};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
		assertEquals(options.getSql(), SQL_1);
		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
		assertTrue(options.getNamedParameters().isEmpty(), "Named parameters should be empty.");
		assertTrue(options.getNumberedParameters().isEmpty(), "Numbered parameters should be empty.");
	}

	@Test
	public void testParseOptions_QueryNow_Numbered() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.DB, DATABASE_NAME_1,
			Tokens.SQL, SQL_1,
			Tokens.DATA, DATA_1, DATA_2, DATA_3};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
		assertEquals(options.getSql(), SQL_1);
		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
		assertEquals(options.getNumberedParameters().size(), 3);
		assertEquals(options.getNumberedParameters().get(0).getValue(), DATA_1);
		assertEquals(options.getNumberedParameters().get(1).getValue(), DATA_2);
		assertEquals(options.getNumberedParameters().get(2).getValue(), DATA_3);
		assertEquals(options.getNumberedParameters().get(0).getType(), Parameter.DEFAULT_TYPE);
		assertEquals(options.getNumberedParameters().get(1).getType(), Parameter.DEFAULT_TYPE);
		assertEquals(options.getNumberedParameters().get(2).getType(), Parameter.DEFAULT_TYPE);
	}

	@Test
	public void testParseOptions_QueryNow_Numbered_withTypes() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.DB, DATABASE_NAME_1,
			Tokens.SQL, SQL_1,
			Tokens.TYPES, " int,string, boolean",
			Tokens.DATA, DATA_1, DATA_2, DATA_3};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
		assertEquals(options.getSql(), SQL_1);
		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
		assertEquals(options.getNumberedParameters().size(), 3);
		assertEquals(options.getNumberedParameters().get(0).getValue(), DATA_1);
		assertEquals(options.getNumberedParameters().get(1).getValue(), DATA_2);
		assertEquals(options.getNumberedParameters().get(2).getValue(), DATA_3);
		assertEquals(options.getNumberedParameters().get(0).getType(), Types.INTEGER);
		assertEquals(options.getNumberedParameters().get(1).getType(), Types.VARCHAR);
		assertEquals(options.getNumberedParameters().get(2).getType(), Types.BOOLEAN);
	}

	@Test
	public void testParseOptions_QueryNow_Named() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.DB, DATABASE_NAME_1,
			Tokens.SQL, SQL_1,
			Tokens.DATA, TYPE_NAME_SEPARATOR + NAME_1, DATA_1, TYPE_NAME_SEPARATOR + NAME_2, DATA_2, TYPE_NAME_SEPARATOR + NAME_3, DATA_3};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
		assertEquals(options.getSql(), SQL_1);
		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
		assertEquals(options.getNamedParameters().size(), 3);
		assertNamedParameter(options.getNamedParameters(), NAME_1, DATA_1, Parameter.DEFAULT_TYPE);
		assertNamedParameter(options.getNamedParameters(), NAME_2, DATA_2, Parameter.DEFAULT_TYPE);
		assertNamedParameter(options.getNamedParameters(), NAME_3, DATA_3, Parameter.DEFAULT_TYPE);
	}

	@Test
	public void testParseOptions_QueryNow_Named_withTypes() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.DB, DATABASE_NAME_1,
			Tokens.SQL, SQL_1,
			Tokens.NAME_PREFIX, "$",
			Tokens.TYPES, " " + NAME_1 + TYPE_NAME_SEPARATOR + "int" + "," + NAME_2 + TYPE_NAME_SEPARATOR + "string" + "," + NAME_3 + TYPE_NAME_SEPARATOR + "boolean",
			Tokens.DATA, "$" + NAME_1, DATA_1, "$" + NAME_2, DATA_2, "$" + NAME_3, DATA_3};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
		assertEquals(options.getSql(), SQL_1);
		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
		assertEquals(options.getNamedParameters().size(), 3);
		assertNamedParameter(options.getNamedParameters(), NAME_1, DATA_1, Types.INTEGER);
		assertNamedParameter(options.getNamedParameters(), NAME_2, DATA_2, Types.VARCHAR);
		assertNamedParameter(options.getNamedParameters(), NAME_3, DATA_3, Types.BOOLEAN);
	}

	private void assertNamedParameter(Collection<NamedParameter> params, String name, Object value, int type) {
		for (NamedParameter p : params) {
			if (name.equals(p.getName())) {
				assertEquals(p.getValue(), value, "value does not match – name: " + name);
				assertEquals(p.getType(), type, "value does not match – name: " + name);
				return;
			}
		}
		fail("Named parameter not found: " + name);
	}

	@Test
	public void testParseOptions_PrepareBatch() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.BATCH,
			Tokens.SQL, SQL_1};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getSql(), SQL_1);
		assertEquals(options.getMode(), CLIOptions.MODE.PREPARE_BATCH);
	}

	@Test
	public void testParseOptions_ExecuteBatch() throws InvalidOptionsException, CLIParserException {
		String[] args = new String[]{
			Tokens.BATCH,
			Tokens.DB, DATABASE_NAME_1};
		CLIOptions options = parser.parseOptions(args);
		options.validate();

		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
		assertEquals(options.getMode(), CLIOptions.MODE.EXECUTE_BATCH);
	}
}