mockit.internal.util.GeneratedClasses 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 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.

There is a newer version: 1.49

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.util;

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

public final class GeneratedClasses
{
   private static final String SUBCLASS_PREFIX = "$Subclass_";
   private static final String IMPLCLASS_PREFIX = "$Impl_";

   private GeneratedClasses() {}

   @Nonnull
   public static String getNameForGeneratedClass(@Nonnull Class aClass, @Nullable String suffix) {
      String prefix = aClass.isInterface() ? IMPLCLASS_PREFIX : SUBCLASS_PREFIX;
      StringBuilder name = new StringBuilder(60).append(prefix).append(aClass.getSimpleName());

      if (suffix != null) {
         name.append('_').append(suffix);
      }

      if (aClass.getClassLoader() != null) {
         Package targetPackage = aClass.getPackage();

         if (targetPackage != null && !targetPackage.isSealed()) {
            name.insert(0, '.').insert(0, targetPackage.getName());
         }
      }

      return name.toString();
   }

   public static boolean isGeneratedImplementationClass(@Nonnull Class mockedType) {
      return isGeneratedImplementationClass(mockedType.getName());
   }

   public static boolean isGeneratedSubclass(@Nonnull String className) { return className.contains(SUBCLASS_PREFIX); }

   public static boolean isExternallyGeneratedSubclass(@Nonnull String className) {
      int p = className.indexOf('$') + 1;

      //noinspection SimplifiableIfStatement
      if (p < 2 || p == className.length() || className.charAt(p) != '$') {
         return false;
      }

      return className.contains("_$$_javassist_") || className.contains("_$$_jvst") || className.contains("CGLIB$$");
   }

   public static boolean isGeneratedImplementationClass(@Nonnull String className) {
      return className.contains(IMPLCLASS_PREFIX);
   }

   public static boolean isGeneratedClass(@Nonnull String className) {
      return isGeneratedSubclass(className) || isGeneratedImplementationClass(className);
   }

   @Nonnull
   public static Class getMockedClassOrInterfaceType(@Nonnull Class aClass) {
      if (Proxy.isProxyClass(aClass) || isGeneratedImplementationClass(aClass)) {
         return getImplementedInterface(aClass);
      }

      if (isGeneratedSubclass(aClass.getName())) {
         return aClass.getSuperclass();
      }

      return aClass;
   }

   @Nonnull
   public static Class getImplementedInterface(@Nonnull Class implementationClass) {
      // Assumes that a proxy class implements a single interface.
      return implementationClass.getInterfaces()[0];
   }

   @Nonnull
   public static Class getMockedClass(@Nonnull Object mock) {
      return getMockedClassOrInterfaceType(mock.getClass());
   }
}