String (@NotNull PsiMethod method, @NotNull GrMethodCall methodCall) { PsiClass containingClass = method.getContainingClass(); if (containingClass == null) return null; String qualifiedName = containingClass.getQualifiedName(); if (!"junit.framework.Assert".equals(qualifiedName) && !"groovy.util.GroovyTestCase".equals(qualifiedName)) return null; String[] replacementStatements = Holder.ourStatementMap.get(method.getName()); if (replacementStatements == null) return null; GrArgumentList argumentList = methodCall.getArgumentList(); if (argumentList.getNamedArguments().length > 0) return null; GrExpression[] arguments = argumentList.getExpressionArguments(); if (arguments.length >= replacementStatements.length) return null; return replacementStatements[arguments.length]; }