mockit.internal.faking.CaptureOfFakedImplementations 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.

There is a newer version: 1.49

Show newest version

/*
 * Copyright (c) 2006 Rogério Liesenfeld
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.faking;

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

import mockit.*;
import mockit.external.asm.*;
import mockit.internal.*;
import mockit.internal.capturing.*;
import static mockit.internal.util.Utilities.getClassType;

public final class CaptureOfFakedImplementations extends CaptureOfImplementations
{
   private final FakeClassSetup fakeClassSetup;

   public CaptureOfFakedImplementations(@Nonnull MockUp fake, @Nonnull Type baseType)
   {
      Class baseClassType = getClassType(baseType);
      fakeClassSetup = new FakeClassSetup(baseClassType, baseType, fake, null);
   }

   @Nonnull @Override
   protected BaseClassModifier createModifier(
      @Nullable ClassLoader cl, @Nonnull ClassReader cr, @Nonnull Class baseType, Void typeMetadata)
   {
      return fakeClassSetup.createClassModifier(cr);
   }

   @Override
   protected void redefineClass(@Nonnull Class realClass, @Nonnull byte[] modifiedClass)
   {
      fakeClassSetup.applyClassModifications(realClass, modifiedClass);
   }

   @Nullable
   public  Class apply()
   {
      @SuppressWarnings("unchecked") Class baseType = (Class) fakeClassSetup.realClass;
      Class baseClassType = baseType;
      Class fakedClass = null;

      if (baseType.isInterface()) {
         fakedClass = new FakedImplementationClass(fakeClassSetup.fake).createImplementation(baseType);
         baseClassType = fakedClass;
      }

      if (baseClassType != Object.class) {
         redefineClass(baseClassType, baseType, null);
      }

      makeSureAllSubtypesAreModified(baseType, false, null);
      return fakedClass;
   }
}