franta-hg@6: package info.globalcode.sql.dk;
franta-hg@6: 
franta-hg@6: import info.globalcode.sql.dk.CLIParser.Tokens;
franta-hg@10: import java.sql.Types;
franta-hg@6: import static org.testng.Assert.*;
franta-hg@6: import org.testng.annotations.BeforeMethod;
franta-hg@6: import org.testng.annotations.Test;
franta-hg@6: 
franta-hg@6: /**
franta-hg@6:  *
franta-hg@6:  * @author Ing. František Kučera (frantovo.cz)
franta-hg@6:  */
franta-hg@6: public class CLIParserTest {
franta-hg@6: 
franta-hg@7: 	private static final String DATABASE_NAME_1 = "some database 1";
franta-hg@7: 	private static final String SQL_1 = "SELECT * FROM table1";
franta-hg@7: 	private static final String DATA_1 = "aaa";
franta-hg@7: 	private static final String DATA_2 = "bbb";
franta-hg@7: 	private static final String DATA_3 = "ccc";
franta-hg@6: 	private CLIParser parser;
franta-hg@6: 
franta-hg@6: 	@BeforeMethod
franta-hg@6: 	public void setUpMethod() throws Exception {
franta-hg@6: 		parser = new CLIParser();
franta-hg@6: 	}
franta-hg@6: 
franta-hg@6: 	@Test
franta-hg@9: 	public void testParseOptions_QueryNow_Numbered() throws InvalidOptionsException, CLIParserException {
franta-hg@7: 		String[] args = new String[]{Tokens.DB, DATABASE_NAME_1, Tokens.SQL, SQL_1, Tokens.DATA, DATA_1, DATA_2, DATA_3};
franta-hg@6: 		CLIOptions options = parser.parseOptions(args);
franta-hg@6: 		options.validate();
franta-hg@6: 
franta-hg@6: 		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
franta-hg@6: 		assertEquals(options.getSql(), SQL_1);
franta-hg@7: 		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
franta-hg@7: 		assertEquals(options.getNumberedParameters().size(), 3);
franta-hg@7: 		assertEquals(options.getNumberedParameters().get(0).getValue(), DATA_1);
franta-hg@7: 		assertEquals(options.getNumberedParameters().get(1).getValue(), DATA_2);
franta-hg@7: 		assertEquals(options.getNumberedParameters().get(2).getValue(), DATA_3);
franta-hg@7: 		assertEquals(options.getNumberedParameters().get(0).getType(), Parameter.DEFAULT_TYPE);
franta-hg@7: 		assertEquals(options.getNumberedParameters().get(1).getType(), Parameter.DEFAULT_TYPE);
franta-hg@7: 		assertEquals(options.getNumberedParameters().get(2).getType(), Parameter.DEFAULT_TYPE);
franta-hg@7: 	}
franta-hg@7: 
franta-hg@7: 	@Test
franta-hg@10: 	public void testParseOptions_QueryNow_Numbered_withTypes() throws InvalidOptionsException, CLIParserException {
franta-hg@10: 		String[] args = new String[]{
franta-hg@10: 			Tokens.DB, DATABASE_NAME_1,
franta-hg@10: 			Tokens.SQL, SQL_1,
franta-hg@10: 			Tokens.TYPES, " int,string, boolean",
franta-hg@10: 			Tokens.DATA, DATA_1, DATA_2, DATA_3};
franta-hg@10: 		CLIOptions options = parser.parseOptions(args);
franta-hg@10: 		options.validate();
franta-hg@10: 
franta-hg@10: 		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
franta-hg@10: 		assertEquals(options.getSql(), SQL_1);
franta-hg@10: 		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
franta-hg@10: 		assertEquals(options.getNumberedParameters().size(), 3);
franta-hg@10: 		assertEquals(options.getNumberedParameters().get(0).getValue(), DATA_1);
franta-hg@10: 		assertEquals(options.getNumberedParameters().get(1).getValue(), DATA_2);
franta-hg@10: 		assertEquals(options.getNumberedParameters().get(2).getValue(), DATA_3);
franta-hg@10: 		assertEquals(options.getNumberedParameters().get(0).getType(), Types.INTEGER);
franta-hg@10: 		assertEquals(options.getNumberedParameters().get(1).getType(), Types.VARCHAR);
franta-hg@10: 		assertEquals(options.getNumberedParameters().get(2).getType(), Types.BOOLEAN);
franta-hg@10: 	}
franta-hg@10: 
franta-hg@10: 	@Test
franta-hg@9: 	public void testParseOptions_QueryNow_Named() throws InvalidOptionsException, CLIParserException {
franta-hg@9: 		String[] args = new String[]{Tokens.DB, DATABASE_NAME_1, Tokens.SQL, SQL_1};
franta-hg@7: 		CLIOptions options = parser.parseOptions(args);
franta-hg@7: 		options.validate();
franta-hg@7: 
franta-hg@7: 		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
franta-hg@7: 		assertEquals(options.getSql(), SQL_1);
franta-hg@7: 		assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
franta-hg@7: 	}
franta-hg@7: 
franta-hg@7: 	@Test
franta-hg@9: 	public void testParseOptions_PrepareBatch() throws InvalidOptionsException, CLIParserException {
franta-hg@7: 		String[] args = new String[]{Tokens.BATCH, Tokens.SQL, SQL_1};
franta-hg@7: 		CLIOptions options = parser.parseOptions(args);
franta-hg@7: 		options.validate();
franta-hg@7: 
franta-hg@7: 		assertEquals(options.getSql(), SQL_1);
franta-hg@7: 		assertEquals(options.getMode(), CLIOptions.MODE.PREPARE_BATCH);
franta-hg@7: 	}
franta-hg@7: 
franta-hg@7: 	@Test
franta-hg@9: 	public void testParseOptions_ExecuteBatch() throws InvalidOptionsException, CLIParserException {
franta-hg@7: 		String[] args = new String[]{Tokens.BATCH, Tokens.DB, DATABASE_NAME_1};
franta-hg@7: 		CLIOptions options = parser.parseOptions(args);
franta-hg@7: 		options.validate();
franta-hg@7: 
franta-hg@7: 		assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
franta-hg@7: 		assertEquals(options.getMode(), CLIOptions.MODE.EXECUTE_BATCH);
franta-hg@6: 	}
franta-hg@6: }