blob: d60e36062d8133d7b68db7673c68ecc06334a800 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
package org.junit.internal.requests;
import org.junit.internal.builders.AllDefaultPossibilitiesBuilder;
import org.junit.internal.builders.SuiteMethodBuilder;
import org.junit.runner.Runner;
import org.junit.runners.model.RunnerBuilder;
public class ClassRequest extends MemoizingRequest {
/*
* We have to use the f prefix, because IntelliJ's JUnit4IdeaTestRunner uses
* reflection to access this field. See
* https://github.com/junit-team/junit4/issues/960
*/
private final Class<?> fTestClass;
private final boolean canUseSuiteMethod;
public ClassRequest(Class<?> testClass, boolean canUseSuiteMethod) {
this.fTestClass = testClass;
this.canUseSuiteMethod = canUseSuiteMethod;
}
public ClassRequest(Class<?> testClass) {
this(testClass, true);
}
@Override
protected Runner createRunner() {
return new CustomAllDefaultPossibilitiesBuilder().safeRunnerForClass(fTestClass);
}
private class CustomAllDefaultPossibilitiesBuilder extends AllDefaultPossibilitiesBuilder {
@Override
protected RunnerBuilder suiteMethodBuilder() {
return new CustomSuiteMethodBuilder();
}
}
/*
* Customization of {@link SuiteMethodBuilder} that prevents use of the
* suite method when creating a runner for fTestClass when canUseSuiteMethod
* is false.
*/
private class CustomSuiteMethodBuilder extends SuiteMethodBuilder {
@Override
public Runner runnerForClass(Class<?> testClass) throws Throwable {
if (testClass == fTestClass && !canUseSuiteMethod) {
return null;
}
return super.runnerForClass(testClass);
}
}
}
|