/***
 * ASM tests
 * Copyright (c) 2000-2011 INRIA, France Telecom
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the copyright holders nor the names of its
 *    contributors may be used to endorse or promote products derived from
 *    this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGE.
 */
package org.objectweb.asm.util;

import junit.framework.TestCase;
import junit.framework.TestSuite;

import org.objectweb.asm.signature.SignatureReader;
import org.objectweb.asm.signature.SignatureVisitor;
import org.objectweb.asm.signature.SignatureWriter;
import org.objectweb.asm.util.TraceSignatureVisitorUnitTest.TestData;

/**
 * CheckSignatureAdapter tests.
 * 
 * @author Eric Bruneton
 */
public class CheckSignatureAdapterUnitTest extends TestCase {

    public static TestSuite suite() {
        TestSuite suite = new TestSuite(
                CheckSignatureAdapterUnitTest.class.getName());
        for (int i = 0; i < TraceSignatureVisitorUnitTest.DATA.length; i++) {
            suite.addTest(new CheckSignatureAdapterUnitTest(new TestData(
                    TraceSignatureVisitorUnitTest.DATA[i])));
        }
        suite.addTestSuite(CheckSignatureAdapterUnitTest.class);
        return suite;
    }

    private TestData data;

    private SignatureVisitor sv;

    public CheckSignatureAdapterUnitTest() {
    }

    private CheckSignatureAdapterUnitTest(final TestData data) {
        super("checkSignature");
        this.data = data;
    }

    public void checkSignature() {
        SignatureWriter wrt = new SignatureWriter();
        SignatureReader rdr = new SignatureReader(data.signature);
        switch (data.type) {
        case 'C':
            rdr.accept(new CheckSignatureAdapter(
                    CheckSignatureAdapter.CLASS_SIGNATURE, wrt));
            break;
        case 'M':
            rdr.accept(new CheckSignatureAdapter(
                    CheckSignatureAdapter.METHOD_SIGNATURE, wrt));
            break;
        case 'F':
            rdr.acceptType(new CheckSignatureAdapter(
                    CheckSignatureAdapter.TYPE_SIGNATURE, wrt));
            break;
        default:
            return;
        }
        assertEquals(data.signature, wrt.toString());
    }

    public void testIllegalFormalTypeParam() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitFormalTypeParameter("T");
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalClassBound() {
        setup(CheckSignatureAdapter.CLASS_SIGNATURE);
        try {
            sv.visitClassBound();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalInterfaceBound() {
        setup(CheckSignatureAdapter.CLASS_SIGNATURE);
        try {
            sv.visitInterfaceBound();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalSuperclass() {
        setup(CheckSignatureAdapter.METHOD_SIGNATURE);
        try {
            sv.visitSuperclass();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalInterface() {
        setup(CheckSignatureAdapter.CLASS_SIGNATURE);
        try {
            sv.visitInterface();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalParameterType() {
        setup(CheckSignatureAdapter.CLASS_SIGNATURE);
        try {
            sv.visitParameterType();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalReturnType() {
        setup(CheckSignatureAdapter.METHOD_SIGNATURE);
        try {
            sv.visitReturnType();
            sv.visitReturnType();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalExceptionType() {
        setup(CheckSignatureAdapter.METHOD_SIGNATURE);
        try {
            sv.visitExceptionType();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalBaseType() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitBaseType('I');
            sv.visitBaseType('I');
            fail();
        } catch (Exception e) {
        }
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitBaseType('V');
            fail();
        } catch (Exception e) {
        }
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitBaseType('A');
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalTypeVariable() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitTypeVariable("T");
            sv.visitTypeVariable("T");
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalArrayType() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitArrayType();
            sv.visitArrayType();
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalClassType() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitClassType("A");
            sv.visitClassType("A");
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalInnerClassType() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitInnerClassType("A");
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalTypeArgument() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitTypeArgument();
            fail();
        } catch (Exception e) {
        }
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitTypeArgument('+');
            fail();
        } catch (Exception e) {
        }
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitClassType("A");
            sv.visitTypeArgument('*');
            fail();
        } catch (Exception e) {
        }
    }

    public void testIllegalEnd() {
        setup(CheckSignatureAdapter.TYPE_SIGNATURE);
        try {
            sv.visitEnd();
            fail();
        } catch (Exception e) {
        }
    }

    @Override
    public String getName() {
        if (data == null) {
            return super.getName();
        } else {
            return super.getName() + " " + data.signature;
        }
    }

    private void setup(int type) {
        sv = new CheckSignatureAdapter(type, null);
    }
}
