目前，我正在创建规则，该规则应检查方法是否包含@Test和@TestInfo批注。如果是，则@TestInfo应该没有空的参数testCaseId。
有几种可能的方法来填充testCaseId。例子：
@Test
@TestInfo(testCaseId = { "ABC-123", "DEF-154" }, component = "Component")
public int foo() {
return 0;
}
@Test
@TestInfo(testCaseId = ("ABC-123"), component = "Component")
public int foo() {
return 0;
}
@Test
@TestInfo(testCaseId = "ABC-123", component = "Component")
public int foo() {
return 0;
}
我已经讨论过@TestInfo不存在，没有testCaseId参数或testCaseId等于“”的情况。
但是我发现很难找到如何获得testCaseId的全部价值。
这是我找到参数的方式：
Arguments arguments = annotationTree.arguments();
if (!arguments.isEmpty()) {
for (int i = 0; i < arguments.size(); i++) {
checkArgument(arguments.get(i));
}
}
if (!annotationsContainsTestCaseId) {
reportIssue(arguments, "Method annotated with @TestInfo should have not empty testCaseId parameter");
}
private void checkArgument(ExpressionTree argument) {
String parameter = argument.firstToken().text();
String parameterValue = argument.lastToken().text();
if (isParameterTestCaseId(parameter)) {
annotationsContainsTestCaseId = true;
if (isTestCaseIdEmpty(parameterValue)) {
reportIssue(argument, "Method annotated with @TestInfo should have not empty testCaseId parameter");
}
}
}
private boolean isParameterTestCaseId(String parameter) {
return parameter.matches("testCaseId");
}
private boolean isTestCaseIdEmpty(String parameterValue) {
// if "" is empty, length is 2
return parameterValue.length() == 2;
}
但是，当testCaseId位于{}的（）中时，lastToken.text（）仅返回）或}。
有没有一种方法可以在（）或{}中获得参数值？
最佳答案
我想到了。我改用FileScannerContext。
首先，我检查方法是否用@Test和@TestInfo注释，然后获取@TestInfo所在的行。后来，它只是在String行中找到我需要的东西。
这是我最终的工作解决方案：
private static final String TEST_ANNOTATION_PATH = "org.testng.annotations.Test";
private static final String TEST_INFO_ANNOTATION_PATH = "toolkit.utils.TestInfo";
private static final String REPORT_EMPTY_TEST_CASE_ID = "Method annotated with @TestInfo should have not empty testCaseId parameter";
private static final String REPORT_MISSING_TEST_INFO = "Method annotated with @Test should also be annotated with @TestInfo";
private static final String TEST_CASE_ID_SPLIT_PLACE = "testCaseId=";
private int methodStartLine = 0;
private int methodEndLine = 0;
private int annotationLine;
private String annotationLineText;
@Override
public List<Tree.Kind> nodesToVisit() {
return ImmutableList.of(Tree.Kind.METHOD, Tree.Kind.ANNOTATION);
}
@Override
public void visitNode(Tree tree) {
if (tree.is(Tree.Kind.METHOD)) {
MethodTree methodTree = (MethodTree) tree;
scanMethodTree(methodTree);
} else if (tree.is(Tree.Kind.ANNOTATION)) {
AnnotationTree annotationTree = (AnnotationTree) tree;
scanAnnotationTree(annotationTree);
}
}
private void scanMethodTree(MethodTree methodTree) {
if (methodTree.symbol().metadata().isAnnotatedWith(TEST_ANNOTATION_PATH)) {
if (methodTree.symbol().metadata().isAnnotatedWith(TEST_INFO_ANNOTATION_PATH)) {
methodStartLine = methodTree.firstToken().line();
methodEndLine = methodTree.lastToken().line();
} else {
reportIssue(methodTree.simpleName(), REPORT_MISSING_TEST_INFO);
}
}
}
private void scanAnnotationTree(AnnotationTree annotationTree) {
if (annotationTree.symbolType().fullyQualifiedName().equals(TEST_INFO_ANNOTATION_PATH)) {
annotationLine = annotationTree.firstToken().line();
if (annotationLine >= methodStartLine && annotationLine < methodEndLine) {
annotationLineText = context.getFileLines().get(annotationLine - 1).replaceAll("\\s", "");
if (annotationLineText.contains("testCaseId")) {
// {"ABC-123","DEF-456"} OR {("ABC-123"), ("DEF-456")}
if ((annotationLineText.contains("{(\"") || annotationLineText.contains("{\""))
&& (annotationLineText.contains("\")}") || annotationLineText.contains("\"}"))) {
reportCase1();
}
// ("ABC-123")
else if (annotationLineText.contains("(\"") && annotationLineText.contains("\")")) {
reportCase2();
}
// "ABC-123"
else if (annotationLineText.contains("\"") && annotationLineText.contains("\",")) {
reportCase3();
}
} else {
addIssue(annotationLine, REPORT_EMPTY_TEST_CASE_ID);
}
}
}
}
/**
* Method report an issue if testCaseId = {"ABC-123","DEF-456",...,"AEF-159"} or
* testCaseId = {("ABC-123"),("DEF-456"),...,("AEF-159")} has no values between
* quotes.
*/
private void reportCase1() {
String testCaseId = annotationLineText.split(TEST_CASE_ID_SPLIT_PLACE)[1].split("}")[0];
testCaseId = testCaseId.replaceAll("\"", "").replaceAll("\\{", "").replaceAll(",", "").replaceAll("\\(", "")
.replaceAll("\\)", "");
if (testCaseId.length() == 0) {
addIssue(annotationLine, REPORT_EMPTY_TEST_CASE_ID);
}
}
/**
* Method report an issue if testCaseId = ("ABC-123") has no value between
* quotes.
*/
private void reportCase2() {
String testCaseId = annotationLineText.split(TEST_CASE_ID_SPLIT_PLACE)[1].split("\"\\)")[0];
testCaseId = testCaseId.replaceAll("\"", "").replaceAll("\\(", "").replaceAll(",", "");
if (testCaseId.length() == 0) {
addIssue(annotationLine, REPORT_EMPTY_TEST_CASE_ID);
}
}
/**
* Method report an issue if testCaseId = "ABC-123" has no value between quotes.
*/
private void reportCase3() {
String testCaseId = annotationLineText.split(TEST_CASE_ID_SPLIT_PLACE)[1].split("\",")[0];
testCaseId = testCaseId.replaceAll("\"", "").replaceAll(",", "");
if (testCaseId.length() == 0) {
addIssue(annotationLine, REPORT_EMPTY_TEST_CASE_ID);
}
}