mockit.internal.util.MethodFormatter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation
Show all versions of jmockit Show documentation
JMockit is a Java toolkit for automated developer testing.
It contains mocking/faking APIs and a code coverage tool, supporting both JUnit and TestNG.
The mocking APIs allow all kinds of Java code, without testability restrictions, to be tested
in isolation from selected dependencies.
/*
* Copyright (c) 2006-2015 Rogério Liesenfeld
* This file is subject to the terms of the MIT license (see LICENSE.txt).
*/
package mockit.internal.util;
import java.util.*;
import javax.annotation.*;
import mockit.internal.state.*;
public final class MethodFormatter
{
@Nonnull private final StringBuilder out;
@Nonnull private final List parameterTypes;
@Nullable private final String classDesc;
@Nonnull private String methodDesc;
// Auxiliary fields for handling method parameters:
private int parameterIndex;
private int typeDescPos;
private char typeCode;
private int arrayDimensions;
public MethodFormatter(@Nullable String classDesc)
{
out = new StringBuilder();
parameterTypes = new ArrayList(5);
this.classDesc = classDesc;
methodDesc = "";
}
public MethodFormatter(@Nonnull String classDesc, @Nonnull String methodNameAndDesc)
{
this(classDesc);
methodDesc = methodNameAndDesc;
appendFriendlyMethodSignature();
}
@Override
public String toString() { return out.toString(); }
@Nonnull public List getParameterTypes() { return parameterTypes; }
@Nonnull
public String friendlyMethodSignatures(@Nonnull Collection methodNamesAndDescs)
{
String sep = "";
for (String methodNameAndDesc : methodNamesAndDescs) {
out.append(sep);
methodDesc = methodNameAndDesc;
appendFriendlyMethodSignature();
sep = ",\n";
}
return out.toString();
}
private void appendFriendlyMethodSignature()
{
String friendlyDesc = methodDesc;
if (classDesc != null) {
String className = classDesc.replace('/', '.');
out.append(className).append('#');
String constructorName = getConstructorName(className);
friendlyDesc = friendlyDesc.replace("", constructorName);
}
int leftParenNextPos = friendlyDesc.indexOf('(') + 1;
int rightParenPos = friendlyDesc.indexOf(')');
if (leftParenNextPos < rightParenPos) {
out.append(friendlyDesc.substring(0, leftParenNextPos));
String concatenatedParameterTypes = friendlyDesc.substring(leftParenNextPos, rightParenPos);
parameterIndex = 0;
appendFriendlyTypes(concatenatedParameterTypes);
out.append(')');
}
else {
out.append(friendlyDesc.substring(0, rightParenPos + 1));
}
}
@Nonnull
private static String getConstructorName(@Nonnull String className)
{
int p = className.lastIndexOf('.');
String constructorName = p < 0 ? className : className.substring(p + 1);
//noinspection ReuseOfLocalVariable
p = constructorName.lastIndexOf('$');
if (p > 0) {
constructorName = constructorName.substring(p + 1);
}
return constructorName;
}
private void appendFriendlyTypes(@Nonnull String typeDescs)
{
String sep = "";
for (String typeDesc : typeDescs.split(";")) {
out.append(sep);
if (typeDesc.charAt(0) == 'L') {
appendParameterType(friendlyReferenceType(typeDesc));
appendParameterName();
}
else {
appendFriendlyPrimitiveTypes(typeDesc);
}
sep = ", ";
}
}
@Nonnull
private static String friendlyReferenceType(@Nonnull String typeDesc)
{
return typeDesc.substring(1).replace("java/lang/", "").replace('/', '.');
}
private void appendParameterType(@Nonnull String friendlyTypeDesc)
{
out.append(friendlyTypeDesc);
parameterTypes.add(friendlyTypeDesc);
}
private void appendParameterName()
{
if (classDesc != null) {
String name = ParameterNames.getName(classDesc, methodDesc, parameterIndex);
if (name != null) {
out.append(' ').append(name);
}
}
parameterIndex++;
}
private void appendFriendlyPrimitiveTypes(String typeDesc)
{
String sep = "";
for (typeDescPos = 0; typeDescPos < typeDesc.length(); typeDescPos++) {
typeCode = typeDesc.charAt(typeDescPos);
advancePastArrayDimensionsIfAny(typeDesc);
out.append(sep);
String paramType = getTypeNameForTypeDesc(typeDesc) + getArrayBrackets();
appendParameterType(paramType);
appendParameterName();
sep = ", ";
}
}
@SuppressWarnings("OverlyComplexMethod")
@Nonnull
private String getTypeNameForTypeDesc(@Nonnull String typeDesc)
{
String paramType;
switch (typeCode) {
case 'B': return "byte";
case 'C': return "char";
case 'D': return "double";
case 'F': return "float";
case 'I': return "int";
case 'J': return "long";
case 'S': return "short";
case 'V': return "void";
case 'Z': return "boolean";
case 'L':
paramType = friendlyReferenceType(typeDesc.substring(typeDescPos));
typeDescPos = typeDesc.length();
break;
default:
paramType = typeDesc.substring(typeDescPos);
typeDescPos = typeDesc.length();
}
return paramType;
}
private void advancePastArrayDimensionsIfAny(@Nonnull String param)
{
arrayDimensions = 0;
while (typeCode == '[') {
typeDescPos++;
typeCode = param.charAt(typeDescPos);
arrayDimensions++;
}
}
@Nonnull
private String getArrayBrackets()
{
@SuppressWarnings("NonConstantStringShouldBeStringBuffer")
String result = "";
for (int i = 0; i < arrayDimensions; i++) {
//noinspection StringContatenationInLoop
result += "[]";
}
return result;
}
}