ScopeAnalysisCommentsTest.java

/**
 * Copyright (c) 2005-2012 by Appcelerator, Inc. All Rights Reserved.
 * Licensed under the terms of the Eclipse Public License (EPL).
 * Please see the license.txt included with this distribution for details.
 * Any modifications to this file must keep this entire header intact.
 */
package com.python.pydev.analysis.scopeanalysis;

import java.util.ArrayList;
import java.util.List;

import org.python.pydev.parser.jython.ast.Name;
import org.python.pydev.parser.jython.ast.Str;
import org.python.pydev.parser.jython.ast.commentType;
import org.python.pydev.parser.visitors.scope.ASTEntry;

import junit.framework.TestCase;

public class ScopeAnalysisCommentsTest extends TestCase {

    public static void main(String[] args) {
        try {
            ScopeAnalysisCommentsTest test = new ScopeAnalysisCommentsTest();
            test.setUp();
            test.test5();
            test.tearDown();
            junit.textui.TestRunner.run(ScopeAnalysisCommentsTest.class);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void test1() throws Exception {
        ArrayList<Object> c = new ArrayList<Object>();
        commentType comment = new commentType("#this is foo");
        comment.beginLine = 10;
        comment.beginColumn = 20;
        c.add(comment);

        List<Name> names = ScopeAnalysis.checkComments(c, "this");
        assertEquals(1, names.size());

        assertEquals(10, names.get(0).beginLine);
        assertEquals(20 + 1, names.get(0).beginColumn);
    }

    public void test2() throws Exception {
        ArrayList<Object> c = new ArrayList<Object>();
        commentType comment = new commentType("\r\n#comment with RenFoo\r\n");
        comment.beginLine = 5;
        comment.beginColumn = 22;
        c.add(comment);

        List<Name> names = ScopeAnalysis.checkComments(c, "RenFoo");
        assertEquals(1, names.size());

        assertEquals(6, names.get(0).beginLine);
        assertEquals(15, names.get(0).beginColumn);
    }

    public void test3() throws Exception {
        ArrayList<Object> c = new ArrayList<Object>();
        commentType comment = new commentType("\r\n\r\n\n#comment with RenFoo\r\n");
        comment.beginLine = 5;
        comment.beginColumn = 22;
        c.add(comment);

        List<Name> names = ScopeAnalysis.checkComments(c, "RenFoo");
        assertEquals(1, names.size());

        assertEquals(8, names.get(0).beginLine);
        assertEquals(15, names.get(0).beginColumn);
    }

    public void test4() throws Exception {
        ArrayList<Object> c = new ArrayList<Object>();
        Str str = new Str("\nfoo\nfoo\n", Str.TripleSingle, false, false, false, false);
        str.beginLine = 5;
        str.beginColumn = 0;
        c.add(str);

        List<ASTEntry> names = ScopeAnalysis.getStringOccurrences("foo", str);
        assertEquals(2, names.size());

        assertEquals(6, names.get(0).node.beginLine);
        assertEquals(1, names.get(0).node.beginColumn);

        assertEquals(7, names.get(1).node.beginLine);
        assertEquals(1, names.get(1).node.beginColumn);
    }

    public void test5() throws Exception {
        ArrayList<Object> c = new ArrayList<Object>();
        Str str = new Str("\nfoo\n\n\nfoo\n", Str.TripleSingle, false, false, false, false);
        str.beginLine = 5;
        str.beginColumn = 0;
        c.add(str);

        List<ASTEntry> names = ScopeAnalysis.getStringOccurrences("foo", str);
        assertEquals(2, names.size());

        assertEquals(6, names.get(0).node.beginLine);
        assertEquals(1, names.get(0).node.beginColumn);

        assertEquals(9, names.get(1).node.beginLine);
        assertEquals(1, names.get(1).node.beginColumn);
    }
}