franta-hg@6: package info.globalcode.sql.dk; franta-hg@6: franta-hg@6: import info.globalcode.sql.dk.CLIParser.Tokens; 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@7: public void testParseOptions_QueryNow_Numbered() throws InvalidOptionsException { 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@7: public void testParseOptions_QueryNow_Named() throws InvalidOptionsException { franta-hg@7: String[] args = new String[]{Tokens.DB, DATABASE_NAME_1, Tokens.SQL, SQL_1, Tokens.TYPES}; 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@7: public void testParseOptions_PrepareBatch() throws InvalidOptionsException { 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@7: public void testParseOptions_ExecuteBatch() throws InvalidOptionsException { 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: }