source: java2python/core/src/test/java/javaparser/AnnotatedForEachVar.java

package javaparser;

import static org.junit.Assert.assertTrue;

import java.util.Arrays;

import org.eclipse.jdt.annotation.NonNull;
import org.junit.Test;

import com.github.javaparser.JavaParser;
import com.github.javaparser.ParserConfiguration;
import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.Node;
import com.github.javaparser.ast.nodeTypes.NodeWithAnnotations;
import com.github.javaparser.symbolsolver.JavaSymbolSolver;
import com.github.javaparser.symbolsolver.resolution.typesolvers.CombinedTypeSolver;
import com.github.javaparser.symbolsolver.resolution.typesolvers.ReflectionTypeSolver;

/**
 * #285 the annotation disappears from foreach variable...
 *
 */
public class AnnotatedForEachVar {

        public static void main(String[] args) {
                for (final @NonNull String s : Arrays.asList("a", "b")) {
                        System.out.println(s);
                }
        }

        String CODE = "import org.eclipse.jdt.annotation.NonNull;\n"
                        + "public class AnnotatedForEachVar {\n" + "\n"
                        + "     public static void main(String[] args) {\n"
                        + "             for (final @NonNull String s : Arrays.asList(\"a\", \"b\")) {\n"
                        + "                     System.out.println(s);\n" + "           }\n" + "        }\n"
                        + "}\n";

        @Test
        public void testResolve() {
                ParserConfiguration conf = new ParserConfiguration();
                CombinedTypeSolver typeSolver = new CombinedTypeSolver();
                typeSolver.add(new ReflectionTypeSolver(false));
                JavaSymbolSolver symbolSolver = new JavaSymbolSolver(typeSolver);
                conf.setSymbolResolver(symbolSolver);
                JavaParser parser = new JavaParser(conf);

                CompilationUnit res = parser.parse(CODE).getResult().get();
                System.out.println(res);

                assertTrue(containsAnnotation(res));
        }

        /**
         * 
         * @param node a node in the parse tree
         * @return true if node or any of its children contains an annotation
         */
        private boolean containsAnnotation(Node node) {
                if (node instanceof NodeWithAnnotations
                                && ((NodeWithAnnotations) node).getAnnotations().isNonEmpty())
                        return true;

                for (Node child : node.getChildNodes())
                        if (containsAnnotation(child))
                                return true;
                return false;
        }

}