org.mockito.internal.stubbing.BaseStubbing Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockito-core Show documentation
Show all versions of mockito-core Show documentation
Mockito mock objects library core API and implementation
/*
* Copyright (c) 2007 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockito.internal.stubbing;
import static org.mockito.internal.exceptions.Reporter.notAnException;
import static org.mockito.internal.progress.ThreadSafeMockingProgress.mockingProgress;
import org.mockito.internal.stubbing.answers.CallsRealMethods;
import org.mockito.internal.stubbing.answers.Returns;
import org.mockito.internal.stubbing.answers.ThrowsException;
import org.mockito.internal.stubbing.answers.ThrowsExceptionForClassType;
import org.mockito.stubbing.Answer;
import org.mockito.stubbing.OngoingStubbing;
public abstract class BaseStubbing implements OngoingStubbing {
// Keep strong ref to mock preventing premature garbage collection when using 'One-liner stubs'.
// See #1541.
private final Object strongMockRef;
BaseStubbing(Object mock) {
this.strongMockRef = mock;
}
@Override
public OngoingStubbing then(Answer answer) {
return thenAnswer(answer);
}
@Override
public OngoingStubbing thenReturn(T value) {
return thenAnswer(new Returns(value));
}
@Override
public OngoingStubbing thenReturn(T value, T... values) {
OngoingStubbing stubbing = thenReturn(value);
if (values == null) {
// For no good reason we're configuring null answer here
// This has been like that since forever, so let's keep it for compatibility (unless
// users complain)
return stubbing.thenReturn(null);
}
for (T v : values) {
stubbing = stubbing.thenReturn(v);
}
return stubbing;
}
private OngoingStubbing thenThrow(Throwable throwable) {
return thenAnswer(new ThrowsException(throwable));
}
@Override
public OngoingStubbing thenThrow(Throwable... throwables) {
if (throwables == null) {
return thenThrow((Throwable) null);
}
OngoingStubbing stubbing = null;
for (Throwable t : throwables) {
if (stubbing == null) {
stubbing = thenThrow(t);
} else {
stubbing = stubbing.thenThrow(t);
}
}
return stubbing;
}
@Override
public OngoingStubbing thenThrow(Class throwableType) {
if (throwableType == null) {
mockingProgress().reset();
throw notAnException();
}
return thenAnswer(new ThrowsExceptionForClassType(throwableType));
}
@Override
public OngoingStubbing thenThrow(
Class toBeThrown, Class... nextToBeThrown) {
if (nextToBeThrown == null) {
return thenThrow((Class) null);
}
OngoingStubbing stubbing = thenThrow(toBeThrown);
for (Class t : nextToBeThrown) {
stubbing = stubbing.thenThrow(t);
}
return stubbing;
}
@Override
public OngoingStubbing thenCallRealMethod() {
return thenAnswer(new CallsRealMethods());
}
@Override
@SuppressWarnings("unchecked")
public M getMock() {
return (M) this.strongMockRef;
}
}