/*
* Copyright (c) 2007 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockito.internal.handler;
import static org.mockito.internal.util.Primitives.defaultValue;
import java.util.List;
import org.mockito.internal.InternalMockHandler;
import org.mockito.internal.stubbing.InvocationContainer;
import org.mockito.invocation.Invocation;
import org.mockito.mock.MockCreationSettings;
import org.mockito.stubbing.Answer;
/**
* Protects the results from delegate MockHandler. Makes sure the results are valid.
*
* by Szczepan Faber, created at: 5/22/12
*/
class NullResultGuardian<T> implements InternalMockHandler<T> {
private final InternalMockHandler<T> delegate;
public NullResultGuardian(InternalMockHandler<T> delegate) {
this.delegate = delegate;
}
@Override
public Object handle(Invocation invocation) throws Throwable {
Object result = delegate.handle(invocation);
Class<?> returnType = invocation.getMethod().getReturnType();
if(result == null && returnType.isPrimitive()) {
//primitive values cannot be null
return defaultValue(returnType);
}
return result;
}
@Override
public MockCreationSettings<T> getMockSettings() {
return delegate.getMockSettings();
}
@Override
public void setAnswersForStubbing(List<Answer<?>> answers) {
delegate.setAnswersForStubbing(answers);
}
@Override
public InvocationContainer getInvocationContainer() {
return delegate.getInvocationContainer();
}
}