mockit.internal.reflection.RealMethodOrConstructor Maven / Gradle / Ivy

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.

There is a newer version: 1.53.0

Show newest version

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

import java.lang.reflect.*;
import javax.annotation.*;

import mockit.internal.util.*;

public final class RealMethodOrConstructor
{
   @Nonnull public final Member member;

   public RealMethodOrConstructor(@Nonnull String className, @Nonnull String memberNameAndDesc) throws NoSuchMethodException {
      this(ClassLoad.loadFromLoader(RealMethodOrConstructor.class.getClassLoader(), className), memberNameAndDesc);
   }

   public RealMethodOrConstructor(@Nonnull Class realClass, @Nonnull String memberNameAndDesc) throws NoSuchMethodException {
      int p = memberNameAndDesc.indexOf('(');
      String memberDesc = memberNameAndDesc.substring(p);

      if (memberNameAndDesc.charAt(0) == '<') {
         member = findConstructor(realClass, memberDesc);
      }
      else {
         String methodName = memberNameAndDesc.substring(0, p);
         member = findMethod(realClass, methodName, memberDesc);
      }
   }

   public RealMethodOrConstructor(
      @Nonnull Class realClass, @Nonnull String memberName, @Nonnull String memberDesc
   ) throws NoSuchMethodException {
      if (memberName.charAt(0) == '<') {
         member = findConstructor(realClass, memberDesc);
      }
      else {
         member = findMethod(realClass, memberName, memberDesc);
      }
   }

   @Nonnull
   private static Constructor findConstructor(@Nonnull Class realClass, @Nonnull String constructorDesc) {
      Class[] parameterTypes = TypeDescriptor.getParameterTypes(constructorDesc);
      return ConstructorReflection.findSpecifiedConstructor(realClass, parameterTypes);
   }

   @Nonnull
   private static Method findMethod(
      @Nonnull Class realClass, @Nonnull String methodName, @Nonnull String methodDesc
   ) throws NoSuchMethodException {
      Class[] parameterTypes = TypeDescriptor.getParameterTypes(methodDesc);
      Class ownerClass = realClass;

      while (true) {
         try {
            Method method = ownerClass.getDeclaredMethod(methodName, parameterTypes);

            if (method.isBridge()) {
               ownerClass = ownerClass.getSuperclass();
               continue;
            }

            return method;
         }
         catch (NoSuchMethodException e) {
            Method interfaceMethod = findInterfaceMethod(ownerClass, methodName, parameterTypes);

            if (interfaceMethod != null) {
               return interfaceMethod;
            }

            ownerClass = ownerClass.getSuperclass();

            if (ownerClass == null || ownerClass == Object.class) {
               throw e;
            }
         }
      }
   }

   @Nullable
   private static Method findInterfaceMethod(@Nonnull Class aType, @Nonnull String methodName, @Nonnull Class[] parameterTypes) {
      for (Class anInterface : aType.getInterfaces()) {
         try { return anInterface.getMethod(methodName, parameterTypes); } catch (NoSuchMethodException ignore) {}
      }

      return null;
   }

   @Nonnull
   public  M getMember() {
      //noinspection unchecked
      return (M) member;
   }
}