/** * This method will add methods, fields and import statement to existing java file * * @throws IOException * @throws ParseException */ public void insertCode() throws IOException, ParseException { CompilationUnit cuResult = JavaParser.parse(baseFile); if (cuResult.getImports() != null) { List<ImportDeclaration> importsFromBaseFile = cuResult.getImports(); for (ImportDeclaration eachImport : importsFromExtendedFile) { if (!importAlreadyPresent(importsFromBaseFile, eachImport)) { importsFromBaseFile.add(eachImport); } } cuResult.setImports(importsFromBaseFile); } String code = cuResult.toString(); BufferedWriter b = new BufferedWriter(new FileWriter(baseFile)); b.write(code); b.close(); }
public void build(DumpVisitorFactory factory, List<String> excludedMethods) { UnitTestVisitor unitTestVisitor = new UnitTestVisitor(name); // Iterator<ImportDeclaration> itImports = compilationUnit.getImports().iterator(); while (itImports.hasNext()) { if (GroovyTestGeneration.class.getName().equals(itImports.next().getName().toString())) { itImports.remove(); } } // for (String dep : deps) { int i = dep.lastIndexOf("."); String depPackage = dep.substring(0, i) + ".groovy"; String depImport = depPackage + dep.substring(i); compilationUnit.getImports().add(new ImportDeclaration(new NameExpr(depImport), false, false)); } // unitTestVisitor.visit(compilationUnit, excludedMethods); sb.append(compilationUnit.toString(factory)); }
public void assertSave() { assertSave(cu.toString()); }
public void assertSave() { assertSave(cu.toString()); }
public void build() { UnitChromatticVisitor unitChromatticVisitor = new UnitChromatticVisitor(); compilationUnit.getPackage().getName().setName(compilationUnit.getPackage().getName().getName() + ".groovy"); unitChromatticVisitor.visit(compilationUnit, null); sb.append(compilationUnit.toString(new GroovyCompatibilityFactory())); }