org.mockito.internal.handler.InvocationNotifierHandler 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.handler;
import static org.mockito.internal.exceptions.Reporter.invocationListenerThrewException;
import java.util.List;
import org.mockito.invocation.Invocation;
import org.mockito.invocation.InvocationContainer;
import org.mockito.invocation.MockHandler;
import org.mockito.listeners.InvocationListener;
import org.mockito.mock.MockCreationSettings;
/**
* Handler, that call all listeners wanted for this mock, before delegating it
* to the parameterized handler.
*/
class InvocationNotifierHandler implements MockHandler {
private final List invocationListeners;
private final MockHandler mockHandler;
public InvocationNotifierHandler(MockHandler mockHandler, MockCreationSettings settings) {
this.mockHandler = mockHandler;
this.invocationListeners = settings.getInvocationListeners();
}
@Override
public Object handle(Invocation invocation) throws Throwable {
try {
Object returnedValue = mockHandler.handle(invocation);
notifyMethodCall(invocation, returnedValue);
return returnedValue;
} catch (Throwable t) {
notifyMethodCallException(invocation, t);
throw t;
}
}
private void notifyMethodCall(Invocation invocation, Object returnValue) {
for (InvocationListener listener : invocationListeners) {
try {
listener.reportInvocation(
new NotifiedMethodInvocationReport(invocation, returnValue));
} catch (Throwable listenerThrowable) {
throw invocationListenerThrewException(listener, listenerThrowable);
}
}
}
private void notifyMethodCallException(Invocation invocation, Throwable exception) {
for (InvocationListener listener : invocationListeners) {
try {
listener.reportInvocation(
new NotifiedMethodInvocationReport(invocation, exception));
} catch (Throwable listenerThrowable) {
throw invocationListenerThrewException(listener, listenerThrowable);
}
}
}
@Override
public MockCreationSettings getMockSettings() {
return mockHandler.getMockSettings();
}
@Override
public InvocationContainer getInvocationContainer() {
return mockHandler.getInvocationContainer();
}
}