franta-hg@16: /**
franta-hg@16: * SQL-DK
franta-hg@16: * Copyright © 2013 František Kučera (frantovo.cz)
franta-hg@16: *
franta-hg@16: * This program is free software: you can redistribute it and/or modify
franta-hg@16: * it under the terms of the GNU General Public License as published by
franta-hg@16: * the Free Software Foundation, either version 3 of the License, or
franta-hg@16: * (at your option) any later version.
franta-hg@16: *
franta-hg@16: * This program is distributed in the hope that it will be useful,
franta-hg@16: * but WITHOUT ANY WARRANTY; without even the implied warranty of
franta-hg@16: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
franta-hg@16: * GNU General Public License for more details.
franta-hg@16: *
franta-hg@16: * You should have received a copy of the GNU General Public License
franta-hg@16: * along with this program. If not, see .
franta-hg@16: */
franta-hg@6: package info.globalcode.sql.dk;
franta-hg@6:
franta-hg@6: import info.globalcode.sql.dk.CLIParser.Tokens;
franta-hg@12: import static info.globalcode.sql.dk.CLIParser.TYPE_NAME_SEPARATOR;
franta-hg@72: import info.globalcode.sql.dk.InfoLister.InfoType;
franta-hg@168: import java.io.ByteArrayInputStream;
franta-hg@12: import java.util.Collection;
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@12: private static final String NAME_1 = "param1";
franta-hg@12: private static final String NAME_2 = "param2";
franta-hg@12: private static final String NAME_3 = "param3";
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@168: private CLIOptions parseOptions(String[] args) throws CLIParserException {
franta-hg@168: return parser.parseOptions(args, new ByteArrayInputStream("".getBytes()));
franta-hg@168: }
franta-hg@168:
franta-hg@6: @Test
franta-hg@12: public void testParseOptions_QueryNow_NoParams() throws InvalidOptionsException, CLIParserException {
franta-hg@12: String[] args = new String[]{
franta-hg@12: Tokens.DB, DATABASE_NAME_1,
franta-hg@12: Tokens.SQL, SQL_1};
franta-hg@168: CLIOptions options = parseOptions(args);
franta-hg@12: options.validate();
franta-hg@12:
franta-hg@12: assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
franta-hg@12: assertEquals(options.getSql(), SQL_1);
franta-hg@12: assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
franta-hg@12: assertTrue(options.getNamedParameters().isEmpty(), "Named parameters should be empty.");
franta-hg@12: assertTrue(options.getNumberedParameters().isEmpty(), "Numbered parameters should be empty.");
franta-hg@12: }
franta-hg@12:
franta-hg@12: @Test
franta-hg@9: public void testParseOptions_QueryNow_Numbered() throws InvalidOptionsException, CLIParserException {
franta-hg@11: String[] args = new String[]{
franta-hg@11: Tokens.DB, DATABASE_NAME_1,
franta-hg@11: Tokens.SQL, SQL_1,
franta-hg@11: Tokens.DATA, DATA_1, DATA_2, DATA_3};
franta-hg@168: CLIOptions options = 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@68: Tokens.TYPES, " INTEGER,VARCHAR, BOOLEAN",
franta-hg@10: Tokens.DATA, DATA_1, DATA_2, DATA_3};
franta-hg@168: CLIOptions options = 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@68: assertEquals(options.getNumberedParameters().get(0).getType(), SQLType.INTEGER);
franta-hg@68: assertEquals(options.getNumberedParameters().get(1).getType(), SQLType.VARCHAR);
franta-hg@68: assertEquals(options.getNumberedParameters().get(2).getType(), SQLType.BOOLEAN);
franta-hg@10: }
franta-hg@10:
franta-hg@10: @Test
franta-hg@9: public void testParseOptions_QueryNow_Named() throws InvalidOptionsException, CLIParserException {
franta-hg@11: String[] args = new String[]{
franta-hg@11: Tokens.DB, DATABASE_NAME_1,
franta-hg@12: Tokens.SQL, SQL_1,
franta-hg@52: Tokens.DATA_NAMED, NAME_1, DATA_1, NAME_2, DATA_2, NAME_3, DATA_3};
franta-hg@168: CLIOptions options = 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@12: assertEquals(options.getNamedParameters().size(), 3);
franta-hg@12: assertNamedParameter(options.getNamedParameters(), NAME_1, DATA_1, Parameter.DEFAULT_TYPE);
franta-hg@12: assertNamedParameter(options.getNamedParameters(), NAME_2, DATA_2, Parameter.DEFAULT_TYPE);
franta-hg@12: assertNamedParameter(options.getNamedParameters(), NAME_3, DATA_3, Parameter.DEFAULT_TYPE);
franta-hg@12: }
franta-hg@12:
franta-hg@12: @Test
franta-hg@12: public void testParseOptions_QueryNow_Named_withTypes() throws InvalidOptionsException, CLIParserException {
franta-hg@12: String[] args = new String[]{
franta-hg@12: Tokens.DB, DATABASE_NAME_1,
franta-hg@12: Tokens.SQL, SQL_1,
franta-hg@12: Tokens.NAME_PREFIX, "$",
franta-hg@68: Tokens.TYPES, " " + NAME_1 + TYPE_NAME_SEPARATOR + "INTEGER" + "," + NAME_3 + TYPE_NAME_SEPARATOR + "BOOLEAN",
franta-hg@52: Tokens.DATA_NAMED, NAME_1, DATA_1, NAME_2, DATA_2, NAME_3, DATA_3};
franta-hg@168: CLIOptions options = parseOptions(args);
franta-hg@12: options.validate();
franta-hg@12:
franta-hg@12: assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
franta-hg@12: assertEquals(options.getSql(), SQL_1);
franta-hg@12: assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
franta-hg@12: assertEquals(options.getNamedParameters().size(), 3);
franta-hg@68: assertNamedParameter(options.getNamedParameters(), NAME_1, DATA_1, SQLType.INTEGER);
franta-hg@13: assertNamedParameter(options.getNamedParameters(), NAME_2, DATA_2, Parameter.DEFAULT_TYPE);
franta-hg@68: assertNamedParameter(options.getNamedParameters(), NAME_3, DATA_3, SQLType.BOOLEAN);
franta-hg@12: }
franta-hg@12:
franta-hg@68: private void assertNamedParameter(Collection params, String name, Object value, SQLType type) {
franta-hg@12: for (NamedParameter p : params) {
franta-hg@12: if (name.equals(p.getName())) {
franta-hg@12: assertEquals(p.getValue(), value, "value does not match – name: " + name);
franta-hg@12: assertEquals(p.getType(), type, "value does not match – name: " + name);
franta-hg@12: return;
franta-hg@12: }
franta-hg@12: }
franta-hg@12: fail("Named parameter not found: " + name);
franta-hg@7: }
franta-hg@7:
franta-hg@7: @Test
franta-hg@9: public void testParseOptions_PrepareBatch() throws InvalidOptionsException, CLIParserException {
franta-hg@11: String[] args = new String[]{
franta-hg@11: Tokens.BATCH,
franta-hg@11: Tokens.SQL, SQL_1};
franta-hg@168: CLIOptions options = 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@11: String[] args = new String[]{
franta-hg@11: Tokens.BATCH,
franta-hg@11: Tokens.DB, DATABASE_NAME_1};
franta-hg@168: CLIOptions options = 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@14:
franta-hg@14: @Test
franta-hg@14: public void testParseOptions_ShowInfo_Help() throws InvalidOptionsException, CLIParserException {
franta-hg@14: String[] args = new String[]{Tokens.INFO_HELP};
franta-hg@168: CLIOptions options = parseOptions(args);
franta-hg@14: options.validate();
franta-hg@14:
franta-hg@14: assertEquals(options.getMode(), CLIOptions.MODE.JUST_SHOW_INFO);
franta-hg@14: assertEquals(options.getShowInfo().size(), 1);
franta-hg@72: assertTrue(options.getShowInfo().contains(InfoType.HELP));
franta-hg@14: }
franta-hg@6: }