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

org.mockito.internal.configuration.injection.SpyOnInjectedFieldsHandler Maven / Gradle / Ivy

/*
 * Copyright (c) 2007 Mockito contributors
 * This program is made available under the terms of the MIT License.
 */
package org.mockito.internal.configuration.injection;

import static org.mockito.Mockito.withSettings;

import java.lang.reflect.Field;
import java.util.Set;

import org.mockito.Mockito;
import org.mockito.Spy;
import org.mockito.exceptions.base.MockitoException;
import org.mockito.internal.configuration.plugins.Plugins;
import org.mockito.internal.util.MockUtil;
import org.mockito.internal.util.reflection.FieldReader;
import org.mockito.plugins.MemberAccessor;

/**
 * Handler for field annotated with @InjectMocks and @Spy.
 *
 * 

* The handler assumes that field initialization AND injection already happened. * So if the field is still null, then nothing will happen there. *

*/ public class SpyOnInjectedFieldsHandler extends MockInjectionStrategy { private final MemberAccessor accessor = Plugins.getMemberAccessor(); @Override protected boolean processInjection(Field field, Object fieldOwner, Set mockCandidates) { FieldReader fieldReader = new FieldReader(fieldOwner, field); // TODO refactor : code duplicated in SpyAnnotationEngine if (!fieldReader.isNull() && field.isAnnotationPresent(Spy.class)) { try { Object instance = fieldReader.read(); if (MockUtil.isMock(instance)) { // A. instance has been spied earlier // B. protect against multiple use of MockitoAnnotations.openMocks() Mockito.reset(instance); } else { // TODO: Add mockMaker option for @Spy annotation (#2740) Object mock = Mockito.mock( instance.getClass(), withSettings() .spiedInstance(instance) .defaultAnswer(Mockito.CALLS_REAL_METHODS) .name(field.getName())); accessor.set(field, fieldOwner, mock); } } catch (Exception e) { throw new MockitoException("Problems initiating spied field " + field.getName(), e); } } return false; } }