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.remove; 23 24 import com.github.javaparser.ast.CompilationUnit; 25 import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration; 26 import com.github.javaparser.ast.body.FieldDeclaration; 27 import com.github.javaparser.ast.body.MethodDeclaration; 28 import com.github.javaparser.ast.stmt.BlockStmt; 29 import com.github.javaparser.ast.stmt.Statement; 30 import org.junit.After; 31 import org.junit.Before; 32 import org.junit.Test; 33 34 import static org.junit.Assert.assertEquals; 35 36 public class NodeRemovalTest { 37 private final CompilationUnit cu = new CompilationUnit(); 38 39 @Test 40 public void testRemoveClassFromCompilationUnit() { 41 ClassOrInterfaceDeclaration testClass = cu.addClass("test"); 42 assertEquals(1, cu.getTypes().size()); 43 boolean remove = testClass.remove(); 44 assertEquals(true, remove); 45 assertEquals(0, cu.getTypes().size()); 46 } 47 48 @Test 49 public void testRemoveFieldFromClass() { 50 ClassOrInterfaceDeclaration testClass = cu.addClass("test"); 51 52 FieldDeclaration addField = testClass.addField(String.class, "test"); 53 assertEquals(1, testClass.getMembers().size()); 54 boolean remove = addField.remove(); 55 assertEquals(true, remove); 56 assertEquals(0, testClass.getMembers().size()); 57 } 58 59 @Test 60 public void testRemoveStatementFromMethodBody() { 61 ClassOrInterfaceDeclaration testClass = cu.addClass("testC"); 62 63 MethodDeclaration addMethod = testClass.addMethod("testM"); 64 BlockStmt methodBody = addMethod.createBody(); 65 Statement addStatement = methodBody.addAndGetStatement("test"); 66 assertEquals(1, methodBody.getStatements().size()); 67 boolean remove = addStatement.remove(); 68 assertEquals(true, remove); 69 assertEquals(0, methodBody.getStatements().size()); 70 } 71 } 72