1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
package com.beust.jcommander;
import org.testng.Assert;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
public class CmdTest {
@Parameters(commandNames = "--cmd-one")
public static class CmdOne {
}
@Parameters(commandNames = "--cmd-two")
class CmdTwo {
@Parameter
List<String> params = new java.util.LinkedList<String>();
}
public String parseArgs(boolean withDefault, String[] args) {
JCommander jc = new JCommander();
jc.addCommand(new CmdOne());
jc.addCommand(new CmdTwo());
if (withDefault) {
// First check if a command was given, when not prepend default
// command (--cmd-two")
// In version up to 1.23 JCommander throws an Exception in this
// line,
// which might be incorrect, at least its not reasonable if the
// method
// is named "WithoutValidation".
jc.parseWithoutValidation(args);
if (jc.getParsedCommand() == null) {
LinkedList<String> newArgs = new LinkedList<String>();
newArgs.add("--cmd-two");
newArgs.addAll(Arrays.asList(args));
jc.parse(newArgs.toArray(new String[0]));
}
} else {
jc.parse(args);
}
return jc.getParsedCommand();
}
@DataProvider
public Object[][] testData() {
return new Object[][] {
new Object[] { "--cmd-one", false, new String[] { "--cmd-one" } },
new Object[] { "--cmd-two", false, new String[] { "--cmd-two" } },
new Object[] { "--cmd-two", false,
new String[] { "--cmd-two", "param1", "param2" } },
// This is the relevant test case to test default commands
new Object[] { "--cmd-two", true,
new String[] { "param1", "param2" } } };
}
@Test(dataProvider = "testData")
public void testArgsWithoutDefaultCmd(String expected,
boolean requireDefault, String[] args) {
if (!requireDefault) {
Assert.assertEquals(parseArgs(false, args), expected);
}
}
@Test(dataProvider = "testData", expectedExceptions = MissingCommandException.class)
public void testArgsWithoutDefaultCmdFail(String expected,
boolean requireDefault, String[] args) {
if (requireDefault) {
parseArgs(false, args);
} else {
throw new MissingCommandException("irrelevant test case");
}
}
// We do not expect a MissingCommandException!
@Test(dataProvider = "testData")
public void testArgsWithDefaultCmd(String expected, boolean requireDefault,
String[] args) {
Assert.assertEquals(parseArgs(true, args), expected);
}
}
|
