mockit.internal.state.ParameterNames Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation

JMockit is a Java toolkit for automated developer testing. It contains APIs for the creation of the objects to be tested, for mocking dependencies, and for faking external APIs; JUnit (4 & 5) and TestNG test runners are supported. It also contains an advanced code coverage tool.

The newest version!

/*
 * Copyright (c) 2006 JMockit developers
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.state;

import edu.umd.cs.findbugs.annotations.NonNull;
import edu.umd.cs.findbugs.annotations.Nullable;

import java.util.HashMap;
import java.util.Map;

import mockit.internal.util.TestMethod;

import org.checkerframework.checker.index.qual.NonNegative;

public final class ParameterNames {
    private static final Map> classesToMethodsToParameters = new HashMap<>();

    private ParameterNames() {
    }

    public static boolean hasNamesForClass(@NonNull String classDesc) {
        return classesToMethodsToParameters.containsKey(classDesc);
    }

    public static void register(@NonNull String classDesc, @NonNull String memberName, @NonNull String memberDesc,
            @NonNull String[] names) {
        Map methodsToParameters = classesToMethodsToParameters.get(classDesc);

        if (methodsToParameters == null) {
            methodsToParameters = new HashMap<>();
            classesToMethodsToParameters.put(classDesc, methodsToParameters);
        }

        String methodKey = memberName + memberDesc;
        methodsToParameters.put(methodKey, names);
    }

    @NonNull
    public static String getName(@NonNull TestMethod method, @NonNegative int index) {
        String name = getName(method.testClassDesc, method.testMethodDesc, index);
        return name == null ? "param" + index : name;
    }

    @Nullable
    public static String getName(@NonNull String classDesc, @NonNull String methodDesc, @NonNegative int index) {
        Map methodsToParameters = classesToMethodsToParameters.get(classDesc);

        if (methodsToParameters == null) {
            return null;
        }

        String[] parameterNames = methodsToParameters.get(methodDesc);
        return parameterNames[index];
    }
}