All Downloads are FREE. Search and download functionalities are using the official Maven repository.

mockit.internal.expectations.injection.TestedObjectCreation Maven / Gradle / Ivy

Go to download

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 Rogério Liesenfeld
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.expectations.injection;

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

import mockit.external.asm.*;
import mockit.internal.classGeneration.*;
import mockit.internal.expectations.mocking.*;
import mockit.internal.state.*;

final class TestedObjectCreation
{
   @Nonnull private final InjectionState injectionState;
   @Nullable private final FullInjection fullInjection;
   @Nonnull private final Class declaredTestedClass;
   @Nonnull private final Class actualTestedClass;
   @Nonnull private final TestedClass testedClass;

   TestedObjectCreation(
      @Nonnull InjectionState injectionState, @Nullable FullInjection fullInjection, @Nonnull Field testedField)
   {
      this.injectionState = injectionState;
      this.fullInjection = fullInjection;
      declaredTestedClass = testedField.getType();
      actualTestedClass =
         isAbstract(declaredTestedClass.getModifiers()) ?
            generateSubclass(testedField.getGenericType()) : declaredTestedClass;
      testedClass = new TestedClass(declaredTestedClass);
   }

   @Nonnull
   private Class generateSubclass(@Nonnull final Type testedType)
   {
      Class generatedSubclass = new ImplementationClass(declaredTestedClass) {
         @Nonnull @Override
         protected ClassVisitor createMethodBodyGenerator(@Nonnull ClassReader typeReader)
         {
            return
               new SubclassGenerationModifier(declaredTestedClass, testedType, typeReader, generatedClassName, true);
         }
      }.generateClass();

      TestRun.mockFixture().registerMockedClass(generatedSubclass);
      return generatedSubclass;
   }

   TestedObjectCreation(
      @Nonnull InjectionState injectionState, @Nullable FullInjection fullInjection,
      @Nonnull Class implementationClass)
   {
      this.injectionState = injectionState;
      this.fullInjection = fullInjection;
      declaredTestedClass = implementationClass;
      actualTestedClass = implementationClass;
      testedClass = new TestedClass(implementationClass);
   }

   @Nonnull
   Object create()
   {
      ConstructorSearch constructorSearch =
         new ConstructorSearch(injectionState, actualTestedClass, fullInjection != null);
      Constructor constructor = constructorSearch.findConstructorAccordingToAccessibilityAndAvailableInjectables();

      if (constructor == null) {
         throw new IllegalArgumentException(
            "No constructor in tested class that can be satisfied by available injectables" + constructorSearch);
      }

      ConstructorInjection constructorInjection =
         new ConstructorInjection(testedClass, injectionState, fullInjection, constructor);

      return constructorInjection.instantiate(constructorSearch.parameterProviders);
   }
}