summaryrefslogtreecommitdiffstats
path: root/vespajlib/src/test/java/com/yahoo/system/CommandLineParserTestCase.java
blob: a2a086e4c6595bf87beb447acbae2d54f6c6b084 (plain) (blame)
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.system;

import junit.framework.TestCase;

public class CommandLineParserTestCase extends TestCase {

    public void testParse1() {
        String[] args = new String[] {"-d", "-f", "hello.txt"};
        CommandLineParser parser = new CommandLineParser(args);
        parser.addLegalBinarySwitch("-f");
        parser.addLegalUnarySwitch("-d");
        parser.parse();
        assertNull(parser.getBinarySwitches().get("-g"));
        assertFalse(parser.getUnarySwitches().contains("-e"));
        assertEquals(parser.getBinarySwitches().get("-f"), "hello.txt");
        assertTrue(parser.getUnarySwitches().contains("-d"));
        assertFalse(parser.getArguments().contains("-d"));
        assertFalse(parser.getArguments().contains("-f"));
        assertFalse(parser.getArguments().contains("-hello.txt"));
        assertEquals(parser.getArguments().size(), 0);
    }

    public void testParse2() {
        String[] args = new String[] {"-d", "-f", "hello.txt", "-XX", "myName", "-o", "output file", "myLastField"};
        CommandLineParser parser = new CommandLineParser("progname", args);
        parser.setArgumentExplanation("Bla bla1");
        parser.setExtendedHelpText("Bla bla blaaaaaaa bla2");
        parser.addLegalBinarySwitch("-f");
        parser.addLegalBinarySwitch("-o");
        parser.addLegalUnarySwitch("-d");
        parser.addLegalUnarySwitch("-XX");
        parser.parse();
        assertNull(parser.getBinarySwitches().get("-g"));
        assertFalse(parser.getUnarySwitches().contains("-e"));
        assertEquals(parser.getBinarySwitches().get("-f"), "hello.txt");
        assertTrue(parser.getUnarySwitches().contains("-d"));
        assertTrue(parser.getUnarySwitches().contains("-XX"));
        assertEquals(parser.getBinarySwitches().get("-o"), "output file");
        assertTrue(parser.getArguments().contains("myName"));
        assertTrue(parser.getArguments().contains("myLastField"));
        assertEquals(parser.getUnarySwitches().size(), 2);
        assertEquals(parser.getBinarySwitches().size(), 2);
        assertEquals(parser.getArguments().size(), 2);
        assertEquals(parser.getArguments().get(0), "myName");
        assertEquals(parser.getArguments().get(1), "myLastField");
        assertEquals(parser.getUnarySwitches().get(0), "-d");
        assertEquals(parser.getUnarySwitches().get(1), "-XX");

        try {
            parser.usageAndThrow();
            fail("usageAndThrow didn't throw");
        } catch (Exception e) {
            assertTrue(e.getMessage().replaceAll("\n", "").matches(".*bla1.*"));
            assertTrue(e.getMessage().replaceAll("\n", "").matches(".*bla2.*"));
        }
    }

    public void testIllegal() {
        String[] args = new String[] {"-d", "-f", "hello.txt", "-XX", "myName", "-o", "output file", "myLastField"};
        CommandLineParser parser = new CommandLineParser(args);
        parser.addLegalBinarySwitch("-f");
        parser.addLegalBinarySwitch("-o");
        parser.addLegalUnarySwitch("-d");
        try {
            parser.parse();
            fail("Parse of cmd line with illegal arg worked");
        } catch (IllegalArgumentException e) {
            assertTrue(e.getMessage().startsWith("\nusage"));
        }

        args = new String[] {"-d", "-f", "hello.txt", "-XX", "myName", "-o", "output file", "myLastField"};
        parser = new CommandLineParser(args);
        parser.addLegalBinarySwitch("-f");
        parser.addLegalUnarySwitch("-d");
        parser.addLegalUnarySwitch("-XX");
        try {
            parser.parse();
            fail("Parse of cmd line with illegal arg worked");
        } catch (IllegalArgumentException e) {
            assertTrue(e.getMessage().startsWith("\nusage"));
        }
    }

    public void testRequired() {
        String[] args1 = new String[] {"-d", "-f", "hello.txt", "-XX", "myName", "-o", "output file", "myLastField"};
        String[] args2 = new String[] {"-XX", "myName", "-o", "output file", "myLastField"};
        CommandLineParser parser = new CommandLineParser(args1);
        parser.addLegalBinarySwitch("-f", "test1");
        parser.addRequiredBinarySwitch("-o", "test2");
        parser.addLegalUnarySwitch("-d", "test3");
        parser.addLegalUnarySwitch("-XX", "test4");
        parser.parse();

        parser = new CommandLineParser(args2);
        parser.addRequiredBinarySwitch("-o", "test2");
        parser.addLegalUnarySwitch("-XX", "test4");
        parser.parse();
        assertEquals(parser.getBinarySwitches().size(),1);
        assertEquals(parser.getUnarySwitches().size(),1);

        parser = new CommandLineParser(args2);
        parser.addLegalUnarySwitch("-XX", "test4");
        parser.addRequiredBinarySwitch("-f", "test5");
        parser.addRequiredBinarySwitch("-o", "test6");
        try {
            parser.parse();
            fail("Illegal cmd line parsed");
        } catch (Exception e) {
            assertTrue(e.getMessage().startsWith("\nusage"));
        }

        args1 = new String[] {"-d"};
        parser = new CommandLineParser(args1);
        parser.addRequiredUnarySwitch("-d", "(required, there are so many bugs)");
        try {
            parser.addLegalBinarySwitch("-d");
            fail("Switch clobber didn't throw");
        } catch (Exception e) {
            assertTrue(e.getMessage().matches(".*already.*"));
        }
        parser.parse();
        assertEquals(parser.getUnarySwitches().get(0), "-d");
    }
}