FlexPMD/language-module/parser/parser-impl/src/test/java/org/apache/flex/pmd/parser/impl/TestForStatement.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.flex.pmd.parser.impl;

import org.apache.flex.pmd.parser.exceptions.TokenException;
import org.junit.Test;

public class TestForStatement extends AbstractStatementTestBase {

    @Test
    public void testSimpleFor() throws TokenException {
        assertStatement("1",
                "for( var i : int = 0; i < length; i++ ){ trace( i ); }",
                "<for line=\"1\"><init line=\"1\">"
                        + "<var-list line=\"1\"><name-type-init line=\"1\""
                        + "><name line=\"1\">i</name><type line=\"1\">int</type><init line=\"1\">"
                        + "<primary line=\"1\">0</primary></init>"
                        + "</name-type-init></var-list></init>"
                        + "<cond line=\"1\"><relation line=\"1\">"
                        + "<primary line=\"1\">i</primary><op line=\"1\""
                        + ">&lt;</op><primary line=\"1\">length"
                        + "</primary></relation></cond><iter line=\"1\">"
                        + "<post-inc line=\"1\"><primary line=\"1\">i"
                        + "</primary></post-inc></iter><block line=\"1\"><call line=\"1\""
                        + "><primary line=\"1\">trace"
                        + "</primary><arguments line=\"1\"><primary line=\"1\">i"
                        + "</primary></arguments></call></block></for>");

        assertStatement("",
                "        for (i = 0; i < n; i++)",
                "<for line=\"1\"><init line=\"1\">"
                        + "<assign line=\"1\"><primary line=\"1\">i</primary><op line=\"1\">=</op>"
                        + "<primary line=\"1\">0</primary></assign></init>"
                        + "<cond line=\"1\"><relation line=\"1\"><primary line=\"1\">i</primary>"
                        + "<op line=\"1\">&lt;</op><primary line=\"1\">n"
                        + "</primary></relation></cond><iter line=\"1\">"
                        + "<post-inc line=\"1\"><primary line=\"1\">i"
                        + "</primary></post-inc></iter><primary line=\"2\">__END__</primary></for>");
    }

    @Test
    public void testSimpleForEach() throws TokenException {
        assertStatement("1",
                "for each( var obj : Object in list ){ obj.print( i ); }",
                "<foreach line=\"1\"><var line=\"1\">"
                        + "<name-type-init line=\"1\"><name line=\"1\""
                        + ">obj</name><type line=\"1\">Object"
                        + "</type></name-type-init></var><in line=\"1\">"
                        + "<primary line=\"1\">list</primary></in>"
                        + "<block line=\"1\"><dot line=\"1\">"
                        + "<primary line=\"1\">obj</primary><call line=\"1\""
                        + "><primary line=\"1\">print</primary>"
                        + "<arguments line=\"1\"><primary line=\"1\">"
                        + "i</primary></arguments></call></dot></block></foreach>");

        assertStatement("1",
                "for each( obj in list ){}",
                "<foreach line=\"1\"><name line=\"1\">obj</name>"
                        + "<in line=\"1\"><primary line=\"1\">list</primary>"
                        + "</in><block line=\"1\"></block></foreach>");

        // assertStatement(
        // "", "for each (var a:XML in classInfo..accessor) {}", "" );
    }

    @Test
    public void testSimpleForIn() throws TokenException {
        assertStatement("1",
                "for( var s : String in obj ){ trace( s, obj[ s ]); }",
                "<forin line=\"1\"><init line=\"1\">"
                        + "<var-list line=\"1\"><name-type-init line=\"1\""
                        + "><name line=\"1\">s</name>"
                        + "<type line=\"1\">String</type></name-type-init>"
                        + "</var-list></init><in line=\"1\"><primary line=\"1\""
                        + ">obj</primary></in></forin>");

        assertStatement("for in",
                "            for (p in events);",
                "<forin line=\"1\"><init line=\"1\">"
                        + "<primary line=\"1\">p</primary></init><in line=\"1\""
                        + "><primary line=\"1\">events</primary></in></forin>");
    }
}