org.mockito.internal.runners.DefaultInternalRunner 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) 2018 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockito.internal.runners;
import org.junit.runner.Description;
import org.junit.runner.manipulation.Filter;
import org.junit.runner.manipulation.NoTestsRemainException;
import org.junit.runner.notification.Failure;
import org.junit.runner.notification.RunListener;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.BlockJUnit4ClassRunner;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.InitializationError;
import org.junit.runners.model.Statement;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.internal.junit.DefaultTestFinishedEvent;
import org.mockito.internal.junit.MockitoTestListener;
import org.mockito.internal.util.Supplier;
public class DefaultInternalRunner implements InternalRunner {
private final BlockJUnit4ClassRunner runner;
public DefaultInternalRunner(
Class testClass, final Supplier listenerSupplier)
throws InitializationError {
runner =
new BlockJUnit4ClassRunner(testClass) {
public Object target;
private MockitoTestListener mockitoTestListener;
@Override
protected Statement withBefores(
FrameworkMethod method, final Object target, Statement statement) {
this.target = target;
final Statement base = super.withBefores(method, target, statement);
return new Statement() {
@Override
public void evaluate() throws Throwable {
AutoCloseable closeable;
if (mockitoTestListener == null) {
// get new test listener and add it to the framework
mockitoTestListener = listenerSupplier.get();
Mockito.framework().addListener(mockitoTestListener);
// init annotated mocks before tests
closeable = MockitoAnnotations.openMocks(target);
} else {
closeable = null;
}
try {
base.evaluate();
} finally {
if (closeable != null) {
closeable.close();
}
}
}
};
}
public void run(final RunNotifier notifier) {
RunListener listener =
new RunListener() {
Throwable failure;
@Override
public void testFailure(Failure failure) throws Exception {
this.failure = failure.getException();
}
@Override
public void testFinished(Description description)
throws Exception {
try {
if (mockitoTestListener != null) {
Mockito.framework()
.removeListener(mockitoTestListener);
mockitoTestListener.testFinished(
new DefaultTestFinishedEvent(
target,
description.getMethodName(),
failure));
mockitoTestListener = null;
}
Mockito.validateMockitoUsage();
} catch (Throwable t) {
// In order to produce clean exception to the user we
// need to fire test failure with the right description
// Otherwise JUnit framework will report failure with
// some generic test name
notifier.fireTestFailure(new Failure(description, t));
}
}
};
notifier.addListener(listener);
super.run(notifier);
}
};
}
@Override
public void run(final RunNotifier notifier) {
runner.run(notifier);
}
@Override
public Description getDescription() {
return runner.getDescription();
}
@Override
public void filter(Filter filter) throws NoTestsRemainException {
runner.filter(filter);
}
}