diff --git a/src/main/java/com/alibaba/testable/translator/TestableClassTestRoleTranslator.java b/src/main/java/com/alibaba/testable/translator/TestableClassTestRoleTranslator.java index 98a6ced..b7babb4 100644 --- a/src/main/java/com/alibaba/testable/translator/TestableClassTestRoleTranslator.java +++ b/src/main/java/com/alibaba/testable/translator/TestableClassTestRoleTranslator.java @@ -106,14 +106,19 @@ public class TestableClassTestRoleTranslator extends TreeTranslator { ListBuffer ndefs = new ListBuffer<>(); ndefs.addAll(jcClassDecl.defs); JCTree.JCModifiers mods = treeMaker.Modifiers(Modifier.PUBLIC, makeAnnotations()); - ndefs.add(treeMaker.MethodDef(mods, names.fromString("testableSetup"), null, + ndefs.add(treeMaker.MethodDef(mods, names.fromString("testableSetup"), treeMaker.Type(new Type.JCVoidType()), List.nil(), List.nil(), List.nil(), testableSetupBlock(), null)); jcClassDecl.defs = ndefs.toList(); } private List makeAnnotations() { - return List.nil(); + String[] elems = ANNOTATION_JUNIT5_TEST.split("\\."); + JCTree.JCExpression e = treeMaker.Ident(names.fromString(elems[0])); + for (int i = 1 ; i < elems.length ; i++) { + e = treeMaker.Select(e, names.fromString(elems[i])); + } + return List.of(treeMaker.Annotation(e, List.nil())); } private JCTree.JCBlock testableSetupBlock() {