private String getDisplayNameBottomUp(MethodInvocation invocation) {
if (invocation.parent() == null || !(invocation.parent() instanceof MethodInvocation))
return invocation.target().toString();
else
return getTestNameBottomUp((MethodInvocation) invocation.parent())
+ " "
+ invocation.target().toString();
}
private String getTestNameBottomUp(MethodInvocation invocation) {
String result = "";
while (invocation != null) {
MethodInvocation nameInvocation = invocation;
if (invocation.target().name().equals("taggedAs")
&& invocation.target() instanceof MethodInvocation) {
nameInvocation = (MethodInvocation) invocation.target();
}
if (!nameInvocation.target().canBePartOfTestName()
|| (!invocation.name().equals("in") && invocation.canBePartOfTestName())) return null;
String targetText =
(nameInvocation.name().equals("in") || nameInvocation.name().equals("taggedAs"))
? nameInvocation.target().toString()
: (nameInvocation.target().toString() + " " + nameInvocation.name());
result = targetText + " " + result;
if (invocation.parent() != null && invocation.parent() instanceof MethodInvocation)
invocation = (MethodInvocation) invocation.parent();
else invocation = null;
}
return result.trim();
}
public Selection find(AstNode node) {
Selection result = null;
while (result == null) {
if (node instanceof MethodInvocation) {
MethodInvocation invocation = (MethodInvocation) node;
String name = invocation.name();
AstNode parent = invocation.parent();
if (name.equals("in") && scopeSet.contains(parent.name())) {
String testName = getTestNameBottomUp(invocation);
result =
testName != null
? new Selection(invocation.className(), testName, new String[] {testName})
: null;
if (testName == null) {
if (node.parent() != null) {
node = node.parent();
} else break;
}
} else if (scopeSet.contains(name)) {
String displayName = getDisplayNameBottomUp(invocation);
List<String> testNames = getTestNamesTopDown(invocation);
result =
new Selection(
invocation.className(),
displayName,
testNames.toArray(new String[testNames.size()]));
}
}
if (result == null) {
if (node.parent() != null) node = node.parent();
else break;
}
}
return result;
}