Examples with MockClassLoader org.powermock.core.classloader.MockClassLoader used on opensource projects

Example 31 with MockClassLoader

use of org.powermock.core.classloader.MockClassLoader in project powermock by powermock.

the class ObjenesisClassloaderExecutorTest method worksWithObjectHierarchy.

@Test
public void worksWithObjectHierarchy() throws Exception {
    final MockClassLoader classloader = createClassloader();
    final MyHierarchicalFieldHolder tested = new MyHierarchicalFieldHolder();
    assertSame(tested.getMyArgument1(), tested.getMyArgument2());
    assertEquals(tested.getMyArgument3(), tested.getMyArgument2());
    new SingleClassloaderExecutor(classloader).execute(new Runnable() {

        public void run() {
            assertEquals(MockClassLoader.class.getName(), this.getClass().getClassLoader().getClass().getName());
            assertSame(tested.getMyArgument1(), tested.getMyArgument2());
            assertEquals(tested.getMyArgument3(), tested.getMyArgument2());
        }
    });
}

Example 32 with MockClassLoader

use of org.powermock.core.classloader.MockClassLoader in project powermock by powermock.

the class ObjenesisClassloaderExecutorTest method usesReferenceCloningWhenTwoFieldsPointToSameInstance.

@Test
public void usesReferenceCloningWhenTwoFieldsPointToSameInstance() throws Exception {
    final MockClassLoader classloader = createClassloader();
    final MyReferenceFieldHolder tested = new MyReferenceFieldHolder();
    assertSame(tested.getMyArgument1(), tested.getMyArgument2());
    assertSame(tested.getMyArgument1(), MyReferenceFieldHolder.MY_ARGUMENT);
    new SingleClassloaderExecutor(classloader).execute(new Runnable() {

        public void run() {
            assertEquals(MockClassLoader.class.getName(), this.getClass().getClassLoader().getClass().getName());
            assertEquals(tested.getMyArgument1(), tested.getMyArgument2());
            assertEquals(tested.getMyArgument1(), MyReferenceFieldHolder.MY_ARGUMENT);
            assertSame(tested.getMyArgument1(), tested.getMyArgument2());
        // FIXME: This assertion should work:
        // assertSame(tested.getMyArgument1(), MyReferenceFieldHolder.MY_ARGUMENT);
        }
    });
}

Example 33 with MockClassLoader

use of org.powermock.core.classloader.MockClassLoader in project powermock by powermock.

the class PowerMock method doMock.

@SuppressWarnings("unchecked")
private static <T> T doMock(Class<T> type, boolean isStatic, MockStrategy mockStrategy, ConstructorArgs constructorArgs, Method... methods) {
    if (type == null) {
        throw new IllegalArgumentException("The class to mock cannot be null");
    }
    /*
         * Clear the EasyMock state after the test method is executed.
         */
    MockRepository.addAfterMethodRunner(new Runnable() {

        @Override
        public void run() {
            LastControl.reportLastControl(null);
        }
    });
    IMocksControl control = mockStrategy.createMockControl(type);
    MockRepository.addAfterMethodRunner(new EasyMockStateCleaner());
    T mock;
    if (type.isInterface()) {
        mock = control.createMock(type);
    } else if (type.getName().startsWith("java.") && Modifier.isFinal(type.getModifiers())) {
        Class<?> replicaType = createReplicaType(type, isStatic, constructorArgs);
        final Object replica = doCreateMock(replicaType, constructorArgs, control, methods);
        control = mockStrategy.createMockControl(replicaType);
        MockInvocationHandler h = new MockInvocationHandler((MocksControl) control);
        final Set<Method> methodsToMock = toSet(methods);
        if (isStatic) {
            MockRepository.putStaticMethodInvocationControl(type, new EasyMockMethodInvocationControl<Object>(h, methodsToMock, replica));
            MockRepository.addObjectsToAutomaticallyReplayAndVerify(type);
            return null;
        } else {
            final T newInstance;
            if (constructorArgs == null) {
                newInstance = Whitebox.newInstance(type);
                DefaultFieldValueGenerator.fillWithDefaultValues(newInstance);
            } else {
                try {
                    newInstance = (T) constructorArgs.getConstructor().newInstance(constructorArgs.getInitArgs());
                } catch (Exception e) {
                    throw new RuntimeException("Internal error", e);
                }
            }
            MockRepository.putInstanceMethodInvocationControl(newInstance, new EasyMockMethodInvocationControl<Object>(h, methodsToMock, replica));
            if (!(newInstance instanceof InvocationSubstitute<?>)) {
                MockRepository.addObjectsToAutomaticallyReplayAndVerify(newInstance);
            }
            return newInstance;
        }
    } else {
        mock = doCreateMock(type, constructorArgs, control, methods);
    }
    MockInvocationHandler h = new MockInvocationHandler((MocksControl) control);
    final Set<Method> methodsToMock = toSet(methods);
    if (isStatic) {
        MockRepository.putStaticMethodInvocationControl(type, new EasyMockMethodInvocationControl<T>(h, methodsToMock, mock));
        MockRepository.addObjectsToAutomaticallyReplayAndVerify(type);
    } else {
        MockRepository.putInstanceMethodInvocationControl(mock, new EasyMockMethodInvocationControl<T>(h, methodsToMock));
        if (!(mock instanceof InvocationSubstitute<?>)) {
            MockRepository.addObjectsToAutomaticallyReplayAndVerify(mock);
        }
    }
    ClassLoader classLoader = mock.getClass().getClassLoader();
    if (classLoader instanceof MockClassLoader) {
        ((MockClassLoader) classLoader).cache(mock.getClass());
    }
    return mock;
}

Example 34 with MockClassLoader

use of org.powermock.core.classloader.MockClassLoader in project powermock by powermock.

the class DefaultMockCreator method createMethodInvocationControl.

@SuppressWarnings("unchecked")
private static <T> MockData<T> createMethodInvocationControl(final String mockName, Class<T> type, Method[] methods, boolean isSpy, Object delegator, MockSettings mockSettings) {
    final MockSettingsImpl settings;
    if (mockSettings == null) {
        // We change the context classloader to the current CL in order for the Mockito
        // framework to load it's plugins (such as MockMaker) correctly.
        final ClassLoader originalCL = Thread.currentThread().getContextClassLoader();
        Thread.currentThread().setContextClassLoader(DefaultMockCreator.class.getClassLoader());
        try {
            settings = (MockSettingsImpl) Mockito.withSettings();
            Plugins.getMockMaker();
        } finally {
            Thread.currentThread().setContextClassLoader(originalCL);
        }
    } else {
        settings = (MockSettingsImpl) mockSettings;
    }
    if (isSpy) {
        settings.defaultAnswer(Mockito.CALLS_REAL_METHODS);
    }
    settings.setMockName(new MockNameImpl(mockName));
    settings.setTypeToMock(type);
    InternalMockHandler mockHandler = MockHandlerFactory.createMockHandler(settings);
    MethodInterceptorFilter filter = new PowerMockMethodInterceptorFilter(mockHandler, settings);
    final T mock = new ClassImposterizer(new DefaultInstantiatorProvider().getInstantiator(settings)).imposterise(filter, type);
    ClassLoader classLoader = mock.getClass().getClassLoader();
    if (classLoader instanceof MockClassLoader) {
        MockClassLoader mcl = (MockClassLoader) classLoader;
        mcl.cache(mock.getClass());
    }
    final MockitoMethodInvocationControl invocationControl = new MockitoMethodInvocationControl(filter, isSpy && delegator == null ? new Object() : delegator, mock, methods);
    return new MockData<T>(invocationControl, mock);
}

Example 35 with MockClassLoader

use of org.powermock.core.classloader.MockClassLoader in project powermock by powermock.

the class PowerMockMaker method createMock.

@Override
public <T> T createMock(MockCreationSettings<T> settings, MockHandler handler) {
    T mock = cglibMockMaker.createMock(settings, handler);
    ClassLoader classLoader = cglibMockMaker.getClass().getClassLoader();
    if (classLoader instanceof MockClassLoader) {
        MockClassLoader mcl = (MockClassLoader) classLoader;
        // The generated class is not picked up by PowerMock so we cache it here
        mcl.cache(mock.getClass());
    }
    return mock;
}