/*

 * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 * @test

 * @summary Test SignatureParser

 * @author Jean-Francois Denise

 * @modules java.base/jdk.internal.jimage.decompressor

 * @run main SignatureParserTest

 */

import java.util.Arrays;

import java.util.Objects;

import jdk.internal.jimage.decompressor.SignatureParser;

public class SignatureParserTest {

    private int passed = 0;

    private int failed = 0;

    public static void main(String[] args) {

        new SignatureParserTest().test();

    }

    private void test() {

        test("[Ljava/lang/String;", "[L;", "java/lang/String");

        test("[[[[[[[[[[Ljava/lang/String;", "[[[[[[[[[[L;", "java/lang/String");

        test("<T:Ljava/lang/Object;:Ljava/lang/Comparable<-TT;>;>" +

                        "(Ljava/lang/String;Ljava/lang/Class<TT;>;TT;Ljava/lang/Comparable<-TT;>;" +

                        "Ljava/lang/Comparable<-TT;>;ZZ)V",

                "<T:L;:L<-TT;>;>(L;L<TT;>;TT;L<-TT;>;L<-TT;>;ZZ)V",

                "java/lang/Object", "java/lang/Comparable", "java/lang/String",

                "java/lang/Class", "java/lang/Comparable", "java/lang/Comparable");

        test("(Ljava/lang/String;ZLjava/util/EventListener;TTK;)V",

                "(L;ZL;TTK;)V",

                "java/lang/String", "java/util/EventListener");

        test("<Y:Ljava/lang/String;>", "<Y:L;>", "java/lang/String");

        test("<Y:Ljava/lang/String;Z::Ljava/util/EventListener;>",

                "<Y:L;Z::L;>", "java/lang/String",

                "java/util/EventListener");

        test("<Y:Ljava/lang/String;Z::Ljava/util/EventListener;O::Ljava/lang/Comparable<Ljava/lang/String;>;>",

                "<Y:L;Z::L;O::L<L;>;>",

                "java/lang/String", "java/util/EventListener", "java/lang/Comparable", "java/lang/String");

        test("<Y:Ljava/lang/String;O::Ljava/lang/Comparable<Ljava/lang/String;Ljava/lang/Float;>;>",

                "<Y:L;O::L<L;L;>;>",

                "java/lang/String", "java/lang/Comparable", "java/lang/String", "java/lang/Float");

        test("<Y:Ljava/lang/String;O::Ljava/lang/Comparable<Ljava/lang/String;Ljava/lang/Float<Ljava/lang/Object;>;>;>",

                "<Y:L;O::L<L;L<L;>;>;>",

                "java/lang/String", "java/lang/Comparable", "java/lang/String", "java/lang/Float", "java/lang/Object");

        test("Ljava/util/Set;", "L;", "java/util/Set");

        test("Ljavaapplication20/Titi<[Ljava/lang/String;Ljava/lang/Integer;>;", "L<[L;L;>;",

                "javaapplication20/Titi",

                "java/lang/String", "java/lang/Integer");

        test("Ljava/lang/Comparable<TK;>;", "L<TK;>;", "java/lang/Comparable");

        test("Ljava/io/Serializable;Ljava/lang/Comparable<TK;>;", "L;L<TK;>;",

                "java/io/Serializable", "java/lang/Comparable");

        test("<Y:Ljava/lang/String;Z::Ljava/util/EventListener;K::Ljava/util/EventListener;O::"

                + "Ljava/lang/Comparable<Ljava/lang/String;>;>"

                + "Ljavaapplication20/Titi<[Ljava/lang/String;Ljava/lang/Integer;TZ;>;"

                + "Ljava/io/Serializable;Ljava/lang/Comparable<TK;>;",

                "<Y:L;Z::L;K::L;O::L<L;>;>L<[L;L;TZ;>;L;L<TK;>;",

                "java/lang/String", "java/util/EventListener", "java/util/EventListener", "java/lang/Comparable",

                "java/lang/String", "javaapplication20/Titi", "java/lang/String", "java/lang/Integer",

                "java/io/Serializable", "java/lang/Comparable");

        test("<PO:Ljava/lang/Object;>(Ljava/lang/Integer;TPO;)Ljava/lang/Integer;",

                "<PO:L;>(L;TPO;)L;",

                "java/lang/Object", "java/lang/Integer", "java/lang/Integer");

        test("<PO:Ljava/lang/Object;>(Ljava/lang/Integer;TPO;)TPO;", "<PO:L;>(L;TPO;)TPO;",

                "java/lang/Object", "java/lang/Integer");

        test("<T::Ljava/util/EventListener;>(Ljava/lang/Class<TT;>;)[TT;",

                "<T::L;>(L<TT;>;)[TT;",

                "java/util/EventListener", "java/lang/Class");

        test("<PO:LTiti;>(Ljava/lang/Integer;ITPO;)Z", "<PO:L;>(L;ITPO;)Z",

                "Titi", "java/lang/Integer");

        test("<K:Ljava/lang/Object;V:Ljava/lang/Object;>Ljava/lang/Object;",

                "<K:L;V:L;>L;",

                "java/lang/Object", "java/lang/Object", "java/lang/Object");

        test("Ljava/util/LinkedHashMap<TK;TV;>.LinkedHashIterator;Ljava/util/Iterator<TV;>;",

                "L<TK;TV;>.L;L<TV;>;",

                "java/util/LinkedHashMap",

                "inkedHashIterator",

                "java/util/Iterator");

        test("LToto<Ljava/lang/String;>;", "L<L;>;", "Toto",

                "java/lang/String");

        test("Ljavaapplication20/Titi<[Ljava/lang/String;Ljava/lang/Integer<LToto;>;TZ;>;",

                "L<[L;L<L;>;TZ;>;",

                "javaapplication20/Titi", "java/lang/String", "java/lang/Integer", "Toto");

        test("LX<[LQ;LW<LToto;>;TZ;>;", "L<[L;L<L;>;TZ;>;",

                "X", "Q", "W", "Toto");

        test("Ljava/lang/String<*>;", "L<*>;", "java/lang/String");

        test("Ljava/util/List<[B>;", "L<[B>;", "java/util/List");

        test("<T:Ljava/lang/Object;T_NODE::Ljava/util/stream/Node<TT;>;>Ljava/lang/Object;Ljava/util/stream/Node<TT;>;",

                "<T:L;T_NODE::L<TT;>;>L;L<TT;>;",

                "java/lang/Object", "java/util/stream/Node", "java/lang/Object", "java/util/stream/Node");

        test("Ljavaapplication20/Titi<[Ljava/lang/String;>;", "L<[L;>;",

                "javaapplication20/Titi", "java/lang/String");

        test("<A::Ljava/lang/annotation/Annotation;"

                        + "W::Lcom/sun/codemodel/internal/JAnnotationWriter<TA;>;>"

                        + "Ljava/lang/Object;Ljava/lang/reflect/InvocationHandler;"

                        + "Lcom/sun/codemodel/internal/JAnnotationWriter<TA;>;",

                "<A::L;W::L<TA;>;>L;L;L<TA;>;",

                "java/lang/annotation/Annotation", "com/sun/codemodel/internal/JAnnotationWriter",

                "java/lang/Object", "java/lang/reflect/InvocationHandler", "com/sun/codemodel/internal/JAnnotationWriter");

        test("<W::Lcom/sun/codemodel/internal/JAnnotationWriter<*>;>(Ljava/lang/Class<TW;>;" +

                "Lcom/sun/codemodel/internal/JAnnotatable;)TW;",

                "<W::L<*>;>(L<TW;>;L;)TW;",

                "com/sun/codemodel/internal/JAnnotationWriter", "java/lang/Class", "com/sun/codemodel/internal/JAnnotatable");

        test("Ljava/util/Set<Lcom/sun/tools/jdeps/JdepsTask$DotGraph<TT;>.Edge;>;",

                "L<L<TT;>.Edge;>;",

                "java/util/Set",

                "com/sun/tools/jdeps/JdepsTask$DotGraph");

        test("<E::Lcom/sun/xml/internal/rngom/ast/om/ParsedElementAnnotation;" +

                "L::Lcom/sun/xml/internal/rngom/ast/om/Location;" +

                "CL::Lcom/sun/xml/internal/rngom/ast/builder/CommentList<TL;>;>Ljava/lang/Object;",

                "<E::L;L::L;CL::L<TL;>;>L;",

                "com/sun/xml/internal/rngom/ast/om/ParsedElementAnnotation",

                "",

                "com/sun/xml/internal/rngom/ast/om/Location",

                "",

                "com/sun/xml/internal/rngom/ast/builder/CommentList",

                "",

                "java/lang/Object");

        test("(Ljava/util/List<Lcom/sun/xml/internal/rngom/nc/NameClass;>;TL;TA;)" +

                "Lcom/sun/xml/internal/rngom/nc/NameClass;",

                "(L<L;>;TL;TA;)L;",

                "java/util/List",

                "com/sun/xml/internal/rngom/nc/NameClass",

                "",

                "com/sun/xml/internal/rngom/nc/NameClass");

        test("[Ljava/util/List;", "[L;", "java/util/List");

        test("[Ljava/util/List<+Lcom/sun/jdi/request/EventRequest;>;",

                "[L<+L;>;",

                "java/util/List", "com/sun/jdi/request/EventRequest");

        test("Lcom/sun/xml/internal/bind/v2/util/QNameMap<TV;>.HashIterator" +

                "<Lcom/sun/xml/internal/bind/v2/util/QNameMap$Entry<TV;>;>;",

                "L<TV;>.HashIterator<L<TV;>;>;",

                "com/sun/xml/internal/bind/v2/util/QNameMap", "com/sun/xml/internal/bind/v2/util/QNameMap$Entry");

        test("[Ljava/lang/String;", "[L;", "java/lang/String");

        test("[Ljava/lang/String<Ljava/lang/Toto<Ljava/lang/Titi;>;>;",

                "[L<L<L;>;>;",

                "java/lang/String", "java/lang/Toto", "java/lang/Titi");

        test("<T::Ljava/util/EventListener;K:Ljava/util/BOO;>(ZCLjava/lang/Class<TT;>;IJS)[TT;",

                "<T::L;K:L;>(ZCL<TT;>;IJS)[TT;",

                "java/util/EventListener", "java/util/BOO", "java/lang/Class");

        test("<T:Ljava/lang/Object;>(TT;ILjava/lang/Long;)TT;",

                "<T:L;>(TT;IL;)TT;", "java/lang/Object", "java/lang/Long");

        test("<T:Ljava/lang/Object;>(TT;ILjava/lang/Long;)TT;^TT;",

                "<T:L;>(TT;IL;)TT;^TT;", "java/lang/Object", "java/lang/Long");

        test("<T:Ljava/lang/Object;>(TT;ILjava/lang/Long;)TT;^TT;^Ljava/lang/Exception;",

                "<T:L;>(TT;IL;)TT;^TT;^L;",

                "java/lang/Object", "java/lang/Long", "java/lang/Exception");

        if (passed + failed == 0) {

            throw new AssertionError("No tests were run");

        }

        String message = String.format("Passed: %d, failed: %d, total: %d", passed, failed, passed + failed);

        if (failed > 0) {

            throw new AssertionError("Test failed: " + message);

        } else {

            System.err.println(message);

        }

    }

    private void test(String type, String formatted, String...classNames) {

        try {

            SignatureParser.ParseResult result = SignatureParser.parseSignatureDescriptor(type);

            String[] parsedNames = parse(classNames);

            assertEquals(result.formatted, formatted, "Input: '" + type + "', checking 'formatted'");

            assertEquals(result.types.size(), 2 * classNames.length,

                    "Input: '" + type + "', checking the length of 'types':" +

                            "\nexpected: " + Arrays.toString(parsedNames) +

                            "\n     got: " + result.types);

            for (int i = 0; i < result.types.size(); ++i) {

                assertEquals(result.types.get(i), parsedNames[i],

                        "Input: '" + type + "', checking 'packageName' at index " + i / 2);

                ++i;

                assertEquals(result.types.get(i), parsedNames[i],

                        "Input: '" + type + "', checking 'simpleName' at index " + i / 2);

            }

            String reconstructed = SignatureParser.reconstruct(result.formatted, result.types);

            assertEquals(reconstructed, type, "Input: '" + type + "', checking reconstruction from: "

                    + result.formatted + " " + result.types);

            ++passed;

        } catch (Exception | AssertionError e) {

            e.printStackTrace();

            ++failed;

        }

    }

    private void assertEquals(Object actual, Object expected, String message) {

        if (!Objects.equals(actual, expected)) {

            throw new AssertionError(message + ": expected: " + expected + ", actual: " + actual);

        }

    }

    private String[] parse(String[] classNames) {

        String[] result = new String[2 * classNames.length];

        for (int i = 0; i < classNames.length; ++i) {

            int index = classNames[i].lastIndexOf("/");

            result[2 * i] = index == -1 ? "" : classNames[i].substring(0, index);

            result[2 *i + 1] = classNames[i].substring(index + 1);

        }

        return result;

    }

}