Home | History | Annotate | Download | only in ast 
      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.printer.lexicalpreservation.transformations.ast;
     23 
     24 import com.github.javaparser.ast.PackageDeclaration;
     25 import com.github.javaparser.ast.expr.Name;
     26 import com.github.javaparser.printer.lexicalpreservation.AbstractLexicalPreservingTest;
     27 import org.junit.Test;
     28 
     29 import java.io.IOException;
     30 
     31 import static com.github.javaparser.utils.Utils.EOL;
     32 
     33 /**
     34  * Transforming CompilationUnit and verifying the LexicalPreservation works as expected.
     35  */
     36 public class CompilationUnitTransformationsTest extends AbstractLexicalPreservingTest {
     37 
     38     // packageDeclaration
     39 
     40     @Test
     41     public void addingPackageDeclaration() throws IOException {
     42         considerCode("class A {}");
     43         cu.setPackageDeclaration(new PackageDeclaration(new Name(new Name("foo"), "bar")));
     44         assertTransformedToString("package foo.bar;"+ EOL + EOL + "class A {}", cu);
     45     }
     46 
     47     @Test
     48     public void removingPackageDeclaration() throws IOException {
     49         considerCode("package foo.bar; class A {}");
     50         cu.removePackageDeclaration();
     51         assertTransformedToString("class A {}", cu);
     52     }
     53 
     54     @Test
     55     public void replacingPackageDeclaration() throws IOException {
     56         considerCode("package foo.bar; class A {}");
     57         cu.setPackageDeclaration(new PackageDeclaration(new Name(new Name("foo2"), "baz")));
     58         assertTransformedToString("package foo2.baz;" +
     59                 EOL + EOL +
     60                 " class A {}", cu);
     61     }
     62 
     63     // imports
     64 
     65     // types
     66 }
     67