All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.fitbur.mockito.internal.configuration.injection.SpyOnInjectedFieldsHandler Maven / Gradle / Ivy

    There is a newer version: 1.0.0
    Show newest version
    /*
 * Copyright (c) 2007 Mockito contributors
 * This program is made available under the terms of the MIT License.
 */

package com.fitbur.mockito.internal.configuration.injection;

import com.fitbur.mockito.Mockito;
import com.fitbur.mockito.Spy;
import com.fitbur.mockito.exceptions.base.MockitoException;
import com.fitbur.mockito.internal.util.MockUtil;
import com.fitbur.mockito.internal.util.reflection.FieldReader;
import com.fitbur.mockito.internal.util.reflection.FieldSetter;

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

import static com.fitbur.mockito.Mockito.withSettings;
import static com.fitbur.mockito.internal.util.reflection.FieldSetter.setField;

/**
 * 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 {

    @Override
    protected boolean processInjection(Field field, Object fieldOwner, Set mockCandidates) {
        FieldReader fieldReader = new FieldReader(fieldOwner, field);

        // TODO refoctor : code duplicated in SpyAnnotationEngine
        if(!fieldReader.isNull() && field.isAnnotationPresent(Spy.class)) {
            try {
                Object instance = fieldReader.read();
                if (new MockUtil().isMock(instance)) {
                    // A. instance has been spied earlier
                    // B. protect against multiple use of MockitoAnnotations.initMocks()
                    Mockito.reset(instance);
                } else {
                    Object mock = Mockito.mock(instance.getClass(), withSettings()
					    .spiedInstance(instance)
					    .defaultAnswer(Mockito.CALLS_REAL_METHODS)
					    .name(field.getName()));
					setField(fieldOwner, field, mock);
                }
            } catch (Exception e) {
                throw new MockitoException("Problems initiating spied field " + field.getName(), e);
            }
        }

        return false;
    }
}
    
    
    
    

    




    
    
    © 2015 - 2024 Weber Informatics LLC | Privacy Policy