Home | History | Annotate | Download | only in body
      1 /*
      2  * Copyright (C) 2007-2010 Jlio Vilmar Gesser.
      3  * Copyright (C) 2011, 2013-2016 The JavaParser Team.
      4  *
      5  * This file is part of JavaParser.
      6  *
      7  * JavaParser can be used either under the terms of
      8  * a) the GNU Lesser General Public License as published by
      9  *     the Free Software Foundation, either version 3 of the License, or
     10  *     (at your option) any later version.
     11  * b) the terms of the Apache License
     12  *
     13  * You should have received a copy of both licenses in LICENCE.LGPL and
     14  * LICENCE.APACHE. Please refer to those files for details.
     15  *
     16  * JavaParser is distributed in the hope that it will be useful,
     17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
     18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     19  * GNU Lesser General Public License for more details.
     20  */
     21 
     22 package com.github.javaparser.ast.body;
     23 
     24 import com.github.javaparser.ast.expr.Expression;
     25 import com.github.javaparser.ast.expr.IntegerLiteralExpr;
     26 import com.github.javaparser.ast.expr.SimpleName;
     27 import org.junit.Test;
     28 
     29 import static org.junit.Assert.assertEquals;
     30 import static org.junit.Assert.assertFalse;
     31 import static org.junit.Assert.assertTrue;
     32 
     33 public class AnnotationMemberDeclarationTest {
     34 
     35     @Test
     36     public void whenSettingNameTheParentOfNameIsAssigned() {
     37         AnnotationMemberDeclaration decl = new AnnotationMemberDeclaration();
     38         SimpleName name = new SimpleName("foo");
     39         decl.setName(name);
     40         assertTrue(name.getParentNode().isPresent());
     41         assertTrue(decl == name.getParentNode().get());
     42     }
     43 
     44     @Test
     45     public void removeDefaultValueWhenNoDefaultValueIsPresent() {
     46         AnnotationMemberDeclaration decl = new AnnotationMemberDeclaration();
     47         SimpleName name = new SimpleName("foo");
     48         decl.setName(name);
     49 
     50         decl.removeDefaultValue();
     51 
     52         assertFalse(decl.getDefaultValue().isPresent());
     53     }
     54 
     55     @Test
     56     public void removeDefaultValueWhenDefaultValueIsPresent() {
     57         AnnotationMemberDeclaration decl = new AnnotationMemberDeclaration();
     58         SimpleName name = new SimpleName("foo");
     59         decl.setName(name);
     60         Expression defaultValue = new IntegerLiteralExpr("2");
     61         decl.setDefaultValue(defaultValue);
     62 
     63         decl.removeDefaultValue();
     64 
     65         assertFalse(defaultValue.getParentNode().isPresent());
     66     }
     67 }
     68