3 * Copyright © 2013 František Kučera (frantovo.cz)
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package info.globalcode.sql.dk;
20 import info.globalcode.sql.dk.CLIParser.Tokens;
21 import static info.globalcode.sql.dk.CLIParser.TYPE_NAME_SEPARATOR;
22 import static info.globalcode.sql.dk.CLIOptions.DEFAULT_NAME_PREFIX;
23 import java.sql.Types;
24 import java.util.Collection;
25 import static org.testng.Assert.*;
26 import org.testng.annotations.BeforeMethod;
27 import org.testng.annotations.Test;
31 * @author Ing. František Kučera (frantovo.cz)
33 public class CLIParserTest {
35 private static final String DATABASE_NAME_1 = "some database 1";
36 private static final String SQL_1 = "SELECT * FROM table1";
37 private static final String DATA_1 = "aaa";
38 private static final String DATA_2 = "bbb";
39 private static final String DATA_3 = "ccc";
40 private static final String NAME_1 = "param1";
41 private static final String NAME_2 = "param2";
42 private static final String NAME_3 = "param3";
43 private CLIParser parser;
46 public void setUpMethod() throws Exception {
47 parser = new CLIParser();
51 public void testParseOptions_QueryNow_NoParams() throws InvalidOptionsException, CLIParserException {
52 String[] args = new String[]{
53 Tokens.DB, DATABASE_NAME_1,
55 CLIOptions options = parser.parseOptions(args);
58 assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
59 assertEquals(options.getSql(), SQL_1);
60 assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
61 assertTrue(options.getNamedParameters().isEmpty(), "Named parameters should be empty.");
62 assertTrue(options.getNumberedParameters().isEmpty(), "Numbered parameters should be empty.");
66 public void testParseOptions_QueryNow_Numbered() throws InvalidOptionsException, CLIParserException {
67 String[] args = new String[]{
68 Tokens.DB, DATABASE_NAME_1,
70 Tokens.DATA, DATA_1, DATA_2, DATA_3};
71 CLIOptions options = parser.parseOptions(args);
74 assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
75 assertEquals(options.getSql(), SQL_1);
76 assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
77 assertEquals(options.getNumberedParameters().size(), 3);
78 assertEquals(options.getNumberedParameters().get(0).getValue(), DATA_1);
79 assertEquals(options.getNumberedParameters().get(1).getValue(), DATA_2);
80 assertEquals(options.getNumberedParameters().get(2).getValue(), DATA_3);
81 assertEquals(options.getNumberedParameters().get(0).getType(), Parameter.DEFAULT_TYPE);
82 assertEquals(options.getNumberedParameters().get(1).getType(), Parameter.DEFAULT_TYPE);
83 assertEquals(options.getNumberedParameters().get(2).getType(), Parameter.DEFAULT_TYPE);
87 public void testParseOptions_QueryNow_Numbered_withTypes() throws InvalidOptionsException, CLIParserException {
88 String[] args = new String[]{
89 Tokens.DB, DATABASE_NAME_1,
91 Tokens.TYPES, " int,string, boolean",
92 Tokens.DATA, DATA_1, DATA_2, DATA_3};
93 CLIOptions options = parser.parseOptions(args);
96 assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
97 assertEquals(options.getSql(), SQL_1);
98 assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
99 assertEquals(options.getNumberedParameters().size(), 3);
100 assertEquals(options.getNumberedParameters().get(0).getValue(), DATA_1);
101 assertEquals(options.getNumberedParameters().get(1).getValue(), DATA_2);
102 assertEquals(options.getNumberedParameters().get(2).getValue(), DATA_3);
103 assertEquals(options.getNumberedParameters().get(0).getType(), Types.INTEGER);
104 assertEquals(options.getNumberedParameters().get(1).getType(), Types.VARCHAR);
105 assertEquals(options.getNumberedParameters().get(2).getType(), Types.BOOLEAN);
109 public void testParseOptions_QueryNow_Named() throws InvalidOptionsException, CLIParserException {
110 String[] args = new String[]{
111 Tokens.DB, DATABASE_NAME_1,
113 Tokens.DATA, DEFAULT_NAME_PREFIX + NAME_1, DATA_1, DEFAULT_NAME_PREFIX + NAME_2, DATA_2, DEFAULT_NAME_PREFIX + NAME_3, DATA_3};
114 CLIOptions options = parser.parseOptions(args);
117 assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
118 assertEquals(options.getSql(), SQL_1);
119 assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
120 assertEquals(options.getNamedParameters().size(), 3);
121 assertNamedParameter(options.getNamedParameters(), NAME_1, DATA_1, Parameter.DEFAULT_TYPE);
122 assertNamedParameter(options.getNamedParameters(), NAME_2, DATA_2, Parameter.DEFAULT_TYPE);
123 assertNamedParameter(options.getNamedParameters(), NAME_3, DATA_3, Parameter.DEFAULT_TYPE);
127 public void testParseOptions_QueryNow_Named_withTypes() throws InvalidOptionsException, CLIParserException {
128 String[] args = new String[]{
129 Tokens.DB, DATABASE_NAME_1,
131 Tokens.NAME_PREFIX, "$",
132 Tokens.TYPES, " " + NAME_1 + TYPE_NAME_SEPARATOR + "int" + "," + NAME_3 + TYPE_NAME_SEPARATOR + "boolean",
133 Tokens.DATA, "$" + NAME_1, DATA_1, "$" + NAME_2, DATA_2, "$" + NAME_3, DATA_3};
134 CLIOptions options = parser.parseOptions(args);
137 assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
138 assertEquals(options.getSql(), SQL_1);
139 assertEquals(options.getMode(), CLIOptions.MODE.QUERY_NOW);
140 assertEquals(options.getNamedParameters().size(), 3);
141 assertNamedParameter(options.getNamedParameters(), NAME_1, DATA_1, Types.INTEGER);
142 assertNamedParameter(options.getNamedParameters(), NAME_2, DATA_2, Parameter.DEFAULT_TYPE);
143 assertNamedParameter(options.getNamedParameters(), NAME_3, DATA_3, Types.BOOLEAN);
146 private void assertNamedParameter(Collection<NamedParameter> params, String name, Object value, int type) {
147 for (NamedParameter p : params) {
148 if (name.equals(p.getName())) {
149 assertEquals(p.getValue(), value, "value does not match – name: " + name);
150 assertEquals(p.getType(), type, "value does not match – name: " + name);
154 fail("Named parameter not found: " + name);
158 public void testParseOptions_PrepareBatch() throws InvalidOptionsException, CLIParserException {
159 String[] args = new String[]{
162 CLIOptions options = parser.parseOptions(args);
165 assertEquals(options.getSql(), SQL_1);
166 assertEquals(options.getMode(), CLIOptions.MODE.PREPARE_BATCH);
170 public void testParseOptions_ExecuteBatch() throws InvalidOptionsException, CLIParserException {
171 String[] args = new String[]{
173 Tokens.DB, DATABASE_NAME_1};
174 CLIOptions options = parser.parseOptions(args);
177 assertEquals(options.getDatabaseName(), DATABASE_NAME_1);
178 assertEquals(options.getMode(), CLIOptions.MODE.EXECUTE_BATCH);
182 public void testParseOptions_ShowInfo_Help() throws InvalidOptionsException, CLIParserException {
183 String[] args = new String[]{Tokens.INFO_HELP};
184 CLIOptions options = parser.parseOptions(args);
187 assertEquals(options.getMode(), CLIOptions.MODE.JUST_SHOW_INFO);
188 assertEquals(options.getShowInfo().size(), 1);
189 assertTrue(options.getShowInfo().contains(CLIOptions.INFO_TYPE.HELP));