@@ -58,6 +58,25 @@ public CompileResult(List<Diagnostic<? extends JavaFileObject>> diagnostics) {
5858 protected static final String SOURCE_FILE_SUFFIX = ".java" ;
5959 protected static final String OUTPUT_DIRECTORY = "target/generated-test" ;
6060
61+ public static void assertGeneratedClassContains (File output , String value ) {
62+ String [] outputContent = getContents (output );
63+ for (String line : outputContent ) {
64+ if (line .contains (value )) {
65+ return ;
66+ }
67+ }
68+ fail ("Expected value \" " + value + "\" couldn't be found in file " + output .getAbsolutePath ());
69+ }
70+
71+ public static void assertGeneratedClassDoesntContains (File output , String value ) {
72+ String [] outputContent = getContents (output );
73+ for (String line : outputContent ) {
74+ if (line .contains (value )) {
75+ fail ("Value \" " + value + "\" shouldn't be in file " + output .getAbsolutePath ());
76+ }
77+ }
78+ }
79+
6180 public static void assertOutput (File expectedResult , File output ) {
6281 String [] expectedContent = getContents (expectedResult );
6382 String [] outputContent = getContents (output );
@@ -245,6 +264,20 @@ public CompileResult compileFiles(Type... compilationUnits) {
245264 return compileFiles (files );
246265 }
247266
267+ public CompileResult compileFiles (Object ... elements ) {
268+ assert elements != null ;
269+
270+ List <File > files = new ArrayList <File >();
271+ for (Object element : elements ) {
272+ if (element instanceof Type ) {
273+ addCollection (files , (Type ) element );
274+ } else if (element instanceof String ) {
275+ files .add (new File ((String ) element ));
276+ }
277+ }
278+ return compileFiles (files );
279+ }
280+
248281 public CompileResult compileFiles (File ... compilationUnits ) {
249282 return compileFiles (Arrays .asList (compilationUnits ));
250283 }
@@ -334,20 +367,24 @@ private <T extends Type> void addCollection(List<File> files, T... compilationUn
334367 return ;
335368 }
336369 for (T element : compilationUnits ) {
337- assert element != null ;
370+ addCollection (files , element );
371+ }
372+ }
338373
339- if (element instanceof Class <?>) {
340- File file = toFile ((Class <?>) element );
341- if (file != null ) {
342- files .add (file );
343- } else {
344- // These are innerclasses, etc ... that should not be
345- // defined in this way
346- }
347- } else if (element instanceof Package ) {
348- ClassFinder classFinder = new ClassFinder ();
349- addCollection (files , classFinder .findClassesInPackage (((Package ) element ).getName ()));
374+ private <T extends Type > void addCollection (List <File > files , T element ) {
375+ assert element != null ;
376+
377+ if (element instanceof Class <?>) {
378+ File file = toFile ((Class <?>) element );
379+ if (file != null ) {
380+ files .add (file );
381+ } else {
382+ // These are innerclasses, etc ... that should not be
383+ // defined in this way
350384 }
385+ } else if (element instanceof Package ) {
386+ ClassFinder classFinder = new ClassFinder ();
387+ addCollection (files , classFinder .findClassesInPackage (((Package ) element ).getName ()));
351388 }
352389 }
353390
0 commit comments