    1   /***
    2    * BSD-style license; for more info see http://pmd.sourceforge.net/license.html
    3    */
    4   package test.net.sourceforge.pmd.symboltable;
    5
    6   import junit.framework.TestCase;
    7   import net.sourceforge.pmd.ast.ASTVariableDeclaratorId;
    8   import net.sourceforge.pmd.symboltable.ImageFinderFunction;
    9   import net.sourceforge.pmd.symboltable.NameDeclaration;
    10  import net.sourceforge.pmd.symboltable.VariableNameDeclaration;
    11
    12  import java.util.ArrayList;
    13  import java.util.List;
    14
    15  public class ImageFinderFunctionTest extends TestCase {
    16
    17      public void testSingleImage() {
    18          ImageFinderFunction f = new ImageFinderFunction("foo");
    19          ASTVariableDeclaratorId node = new ASTVariableDeclaratorId(1);
    20          node.setImage("foo");
    21          NameDeclaration decl = new VariableNameDeclaration(node);
    22          f.applyTo(decl);
    23          assertEquals(decl, f.getDecl());
    24      }
    25
    26      public void testSeveralImages() {
    27          List imgs = new ArrayList();
    28          imgs.add("Foo.foo");
    29          imgs.add("foo");
    30          ImageFinderFunction f = new ImageFinderFunction(imgs);
    31          ASTVariableDeclaratorId node = new ASTVariableDeclaratorId(1);
    32          node.setImage("foo");
    33          NameDeclaration decl = new VariableNameDeclaration(node);
    34          f.applyTo(decl);
    35          assertEquals(decl, f.getDecl());
    36      }
    37  }