01: package jargs.test.gnu;
02:
03: import jargs.gnu.CmdLineParser;
04:
05: import java.text.DateFormat;
06: import java.text.ParseException;
07: import java.util.Calendar;
08: import java.util.Date;
09: import java.util.Locale;
10:
11: import junit.framework.TestCase;
12:
13: public class CustomOptionTestCase extends TestCase {
14:
15: public CustomOptionTestCase(String name) {
16: super (name);
17: }
18:
19: public void testCustomOption() throws Exception {
20: Calendar calendar = Calendar.getInstance();
21: CmdLineParser parser = new CmdLineParser();
22: CmdLineParser.Option date = parser
23: .addOption(new ShortDateOption('d', "date"));
24:
25: parser.parse(new String[] { "-d", "11/03/2003" }, Locale.UK);
26: Date d = (Date) parser.getOptionValue(date);
27: calendar.setTime(d);
28: assertEquals(11, calendar.get(Calendar.DAY_OF_MONTH));
29: assertEquals(Calendar.MARCH, calendar.get(Calendar.MONTH));
30: assertEquals(2003, calendar.get(Calendar.YEAR));
31:
32: parser.parse(new String[] { "-d", "11/03/2003" }, Locale.US);
33: d = (Date) parser.getOptionValue(date);
34: calendar.setTime(d);
35: assertEquals(3, calendar.get(Calendar.DAY_OF_MONTH));
36: assertEquals(Calendar.NOVEMBER, calendar.get(Calendar.MONTH));
37: assertEquals(2003, calendar.get(Calendar.YEAR));
38: }
39:
40: public void testIllegalCustomOption() throws Exception {
41: CmdLineParser parser = new CmdLineParser();
42: CmdLineParser.Option date = parser
43: .addOption(new ShortDateOption('d', "date"));
44: try {
45: parser.parse(new String[] { "-d", "foobar" }, Locale.US);
46: fail("Expected IllegalOptionValueException");
47: } catch (CmdLineParser.IllegalOptionValueException e) {
48: //pass
49: }
50: }
51:
52: public static class ShortDateOption extends CmdLineParser.Option {
53: public ShortDateOption(char shortForm, String longForm) {
54: super (shortForm, longForm, true);
55: }
56:
57: protected Object parseValue(String arg, Locale locale)
58: throws CmdLineParser.IllegalOptionValueException {
59: try {
60: DateFormat dateFormat = DateFormat.getDateInstance(
61: DateFormat.SHORT, locale);
62: return dateFormat.parse(arg);
63: } catch (ParseException e) {
64: throw new CmdLineParser.IllegalOptionValueException(
65: this, arg);
66: }
67: }
68: }
69:
70: }
|