spork-master/test/org/apache/pig/test/TestCmdLineParser.java
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.pig.test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;

import java.text.ParseException;

import org.apache.pig.tools.cmdline.CmdLineParser;
import org.junit.Test;

public class TestCmdLineParser {

    @Test(expected = AssertionError.class)
    public void testRegisterDash() {
        String[] args = { "a", "b", "c" };
        CmdLineParser p = new CmdLineParser(args);
        try {
            p.registerOpt('-', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        } catch (AssertionError e) {
            assertEquals(e.getMessage(),
                    "CmdLineParser:  '-' is not a legal single character designator.");
            throw e;
        }
    }

    @Test(expected = AssertionError.class)
    public void testDoubleRegisterShort() {
        String[] args = { "a", "b", "c" };
        CmdLineParser p = new CmdLineParser(args);
        try {
            p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
            p.registerOpt('a', "beta", CmdLineParser.ValueExpected.NOT_ACCEPTED);
            fail("Should have thrown an AssertionError");
        } catch (AssertionError e) {
            assertEquals(e.getMessage(),
                    "CmdLineParser:  You have already registered option a");
            throw e;
        }
    }

    @Test(expected = AssertionError.class)
    public void testDoubleRegisterLong() {
        String[] args = { "a", "b", "c" };
        CmdLineParser p = new CmdLineParser(args);
        try {
            p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
            p.registerOpt('b', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        } catch (AssertionError e) {
            assertEquals(e.getMessage(),
                    "CmdLineParser:  You have already registered option alpha");
            throw e;
        }
    }

    @Test
    public void testRegister() {
        String[] args = { "a", "b", "c" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        p.registerOpt('b', "beta", CmdLineParser.ValueExpected.REQUIRED);
        p.registerOpt('c', null, CmdLineParser.ValueExpected.OPTIONAL);
    }

    @Test
    public void testParseNoArgs() throws ParseException {
        String[] args = {};
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        assertEquals(CmdLineParser.EndOfOpts, p.getNextOpt());
    }

    @Test
    public void testParseNoDash() throws ParseException {
        String[] args = { "a" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        assertEquals(p.getNextOpt(), CmdLineParser.EndOfOpts);
        String[] remainders = p.getRemainingArgs();
        assertEquals("a", remainders[0]);
    }

    @Test
    public void testParseLongShortNoLeftover() throws ParseException {
        String[] args = { "-a", "--beta", "beth", "--c" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        p.registerOpt('b', "beta", CmdLineParser.ValueExpected.REQUIRED);
        p.registerOpt('c', null, CmdLineParser.ValueExpected.OPTIONAL);
        assertEquals('a', p.getNextOpt());
        assertEquals('b', p.getNextOpt());
        assertEquals("beth", p.getValStr());
        assertEquals('c', p.getNextOpt());
        assertNull(p.getValStr());
        assertEquals(CmdLineParser.EndOfOpts, p.getNextOpt());
        assertNull(p.getRemainingArgs());
    }

    @Test
    public void testParseLongShortLeftover1() throws ParseException {
        String[] args = { "-a", "--beta", "beth", "--c", "gimel", "-", "hi", "i'm", "left",
                        "over" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        p.registerOpt('b', "beta", CmdLineParser.ValueExpected.REQUIRED);
        p.registerOpt('c', null, CmdLineParser.ValueExpected.OPTIONAL);
        assertEquals('a', p.getNextOpt());
        assertEquals('b', p.getNextOpt());
        assertEquals("beth", p.getValStr());
        assertEquals('c', p.getNextOpt());
        assertEquals("gimel", p.getValStr());
        assertEquals(CmdLineParser.EndOfOpts, p.getNextOpt());
        String[] r = p.getRemainingArgs();
        assertEquals(4, r.length);
    }

    // has two dashes instead of one for end of args
    @Test
    public void testParseLongShortLeftover2() throws ParseException {
        String[] args = { "-a", "-beta", "beth", "--c", "gimel", "--", "hi", "i'm", "left",
                        "over" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        p.registerOpt('b', "beta", CmdLineParser.ValueExpected.REQUIRED);
        p.registerOpt('c', null, CmdLineParser.ValueExpected.OPTIONAL);
        assertEquals(p.getNextOpt(), 'a');
        assertEquals(p.getNextOpt(), 'b');
        assertEquals(p.getValStr(), "beth");
        assertEquals(p.getNextOpt(), 'c');
        assertEquals(p.getValStr(), "gimel");
        assertEquals(p.getNextOpt(), CmdLineParser.EndOfOpts);
        String[] r = p.getRemainingArgs();
        assertEquals(r.length, 4);
    }

    @Test
    public void testParseLongShortLeftover3() throws ParseException {
        String[] args = { "-a", "--beta", "5", "--c", "--" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        p.registerOpt('b', "beta", CmdLineParser.ValueExpected.REQUIRED);
        p.registerOpt('c', null, CmdLineParser.ValueExpected.OPTIONAL);
        assertEquals('a', p.getNextOpt());
        assertEquals('b', p.getNextOpt());
        Integer ii = p.getValInt();
        assertEquals(5, ii.intValue());
        assertEquals('c', p.getNextOpt());
        assertNull(p.getValInt());
        assertEquals(CmdLineParser.EndOfOpts, p.getNextOpt());
        String[] r = p.getRemainingArgs();
        assertNull(p.getRemainingArgs());
    }

    @Test
    public void testParseValueNotAcceptedProvided1() throws ParseException {
        String[] args = { "-a", "aleph" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        assertEquals('a', p.getNextOpt());
        String[] r = p.getRemainingArgs();
        assertEquals(1, r.length);
        assertEquals("aleph", r[0]);
    }

    @Test
    public void testParseValueNotAcceptedProvided2() throws ParseException {
        String[] args = { "-alpha", "aleph" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        assertEquals('a', p.getNextOpt());
        String[] r = p.getRemainingArgs();
        assertEquals(1, r.length);
        assertEquals("aleph", r[0]);
    }

    @Test(expected = ParseException.class)
    public void testParseValueRequiredNotProvided1() throws Exception {
        String[] args = { "-a" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.REQUIRED);
        try {
            p.getNextOpt();
        } catch (ParseException e) {
            assertEquals(e.getMessage(),
                    "Option -a requires a value but you did not provide one.");
            throw e;
        }
    }

    @Test(expected = ParseException.class)
    public void testParseValueRequiredNotProvided2() throws Exception {
        String[] args = { "--alpha", "-b" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.REQUIRED);
        p.registerOpt('b', "beta", CmdLineParser.ValueExpected.NOT_ACCEPTED);
        try {
            p.getNextOpt();
        } catch (ParseException e) {
            assertEquals(e.getMessage(),
                    "Option --alpha requires a value but you did not provide one.");
            throw e;
        }
    }

    @Test(expected = NumberFormatException.class)
    public void testParseValueStrForInt() throws ParseException {
        String[] args = { "-alpha", "b" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.REQUIRED);
        p.getNextOpt();
        Integer ii = p.getValInt();
    }

    @Test(expected = ParseException.class)
    public void testParseUnknownShort() throws Exception {
        String[] args = { "-alpha", "b", "-z" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.REQUIRED);
        try {
            p.getNextOpt();
            assertEquals("b", p.getValStr());
            p.getNextOpt();
        } catch (ParseException e) {
            assertEquals(e.getMessage(),
                    "Found unknown option (-z) at position 3");
            throw e;
        }
    }

    @Test(expected = ParseException.class)
    public void testParseUnknownLong() throws Exception {
        String[] args = { "--zeta" };
        CmdLineParser p = new CmdLineParser(args);
        p.registerOpt('a', "alpha", CmdLineParser.ValueExpected.REQUIRED);
        try {
            p.getNextOpt();
        } catch (ParseException e) {
            assertEquals(e.getMessage(),
                    "Found unknown option (--zeta) at position 1");
            throw e;
        }
    }
}
