mockit.internal.faking.CaptureOfFakedImplementations Maven / Gradle / Ivy

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

import static mockit.internal.util.Utilities.getClassType;

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

import java.lang.reflect.Type;

import mockit.MockUp;
import mockit.asm.classes.ClassReader;
import mockit.internal.BaseClassModifier;
import mockit.internal.capturing.CaptureOfImplementations;

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;
    }
}