/*
 * 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 java.text.ParseException;

import junit.framework.AssertionFailedError;
import junit.framework.TestCase;

import org.junit.Test;

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

public class TestCmdLineParser extends TestCase
{

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

@Test
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");
    }
}

@Test
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);
        fail("Should have thrown an AssertionError");
    } catch (AssertionError e) {
        assertEquals(e.getMessage(),
            "CmdLineParser:  You have already registered option alpha");
    }
}

@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(p.getNextOpt(), CmdLineParser.EndOfOpts);
}

@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(remainders[0], "a");
}

@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(p.getNextOpt(), 'a');
    assertEquals(p.getNextOpt(), 'b');
    assertEquals(p.getValStr(), "beth");
    assertEquals(p.getNextOpt(), 'c');
    assertNull(p.getValStr());
    assertEquals(p.getNextOpt(), CmdLineParser.EndOfOpts);
    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(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);
}

// 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(p.getNextOpt(), 'a');
    assertEquals(p.getNextOpt(), 'b');
    Integer ii = p.getValInt();
    assertEquals(ii.intValue(), 5);
    assertEquals(p.getNextOpt(), 'c');
    assertNull(p.getValInt());
    assertEquals(p.getNextOpt(), CmdLineParser.EndOfOpts);
    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(p.getNextOpt(), 'a');
    String[] r = p.getRemainingArgs();
    assertEquals(r.length, 1);
    assertEquals(r[0], "aleph");
}

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

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

@Test
public void testParseValueRequiredNotProvided2()
{
    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();
        fail("Should have thrown a ParseException");
    } catch (ParseException e) {
        assertEquals(e.getMessage(),
            "Option --alpha requires a value but you did not provide one.");
    }
}

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

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

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

}