2016-09-01 3 views
0

Я попытался создать TestSuite в JavaPoet api. У меня есть список с моими классами junit в testClasses, поэтому я хочу добавить аннотацию участника, но addmember позволяет нам добавлять только один объект вовремя, так как я могу добавить весь список?Добавить пользователей в JavaPoet

.addAnnotation(AnnotationSpec.builder(SuiteClasses.class) 
.addMember("value", "$T.class", testClasses.toArray()).build()) 

ответ

1

Попробуйте извлечь переменную для строителя.

AnnotationSpec.Builder ab = AnnotationSpec.builder(SuiteClasses.class); 
for (String s : strings) { 
    ab.addMember(s, "$S", lookup(s)); 
} 
AnnotationSpec annotation = ab.build(); 
+0

взгляд моего ответа –

0
import com.squareup.javapoet.AnnotationSpec; 
import com.squareup.javapoet.JavaFile; 
import com.squareup.javapoet.MethodSpec; 
import com.squareup.javapoet.TypeSpec; 
import org.junit.Test; 

import javax.lang.model.element.Modifier; 
import java.io.IOException; 
import java.nio.file.Path; 
import java.nio.file.Paths; 
import java.util.List; 

public class Javapoet { 

    public void generateTestSuite(List<Class> testClasses) throws IOException { 
     MethodSpec test = MethodSpec.methodBuilder("test").addModifiers(Modifier.PUBLIC).returns(void.class) 
       .addAnnotation(Test.class).build(); 

     TypeSpec testSuite = TypeSpec.classBuilder("TestSuite").addModifiers(Modifier.PUBLIC).addAnnotation(

       AnnotationSpec.Builder ab = AnnotationSpec.builder(SuiteClasses.class); 
     for (String s : strings) { 
      ab.addMember(s, "$S", lookup(s)); 
     } 
     AnnotationSpec annotation = ab.build(); 


     JavaFile javaFile = JavaFile.builder("", testSuite).build(); 

     Path path = Paths.get("src/test/java"); 
     javaFile.writeTo(path); 
    } 
}