-
Notifications
You must be signed in to change notification settings - Fork 4.6k
/
JUnitTestFramework.java
114 lines (98 loc) · 4.01 KB
/
JUnitTestFramework.java
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*
* Copyright 2010 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.gradle.api.internal.tasks.testing.junit;
import org.gradle.api.Action;
import org.gradle.api.internal.tasks.testing.TestClassProcessor;
import org.gradle.api.internal.tasks.testing.TestFramework;
import org.gradle.api.internal.tasks.testing.WorkerTestClassProcessorFactory;
import org.gradle.api.internal.tasks.testing.detection.ClassFileExtractionManager;
import org.gradle.api.internal.tasks.testing.filter.DefaultTestFilter;
import org.gradle.api.tasks.testing.Test;
import org.gradle.api.tasks.testing.junit.JUnitOptions;
import org.gradle.internal.actor.ActorFactory;
import org.gradle.internal.id.IdGenerator;
import org.gradle.internal.scan.UsedByScanPlugin;
import org.gradle.internal.service.ServiceRegistry;
import org.gradle.internal.time.Clock;
import org.gradle.process.internal.worker.WorkerProcessBuilder;
import java.io.IOException;
import java.io.Serializable;
import java.util.Collections;
import java.util.List;
@UsedByScanPlugin("test-retry")
public class JUnitTestFramework implements TestFramework {
private JUnitOptions options;
private JUnitDetector detector;
private final DefaultTestFilter filter;
public JUnitTestFramework(Test testTask, DefaultTestFilter filter) {
this.filter = filter;
options = new JUnitOptions();
detector = new JUnitDetector(new ClassFileExtractionManager(testTask.getTemporaryDirFactory()));
}
@Override
public WorkerTestClassProcessorFactory getProcessorFactory() {
return new TestClassProcessorFactoryImpl(new JUnitSpec(
options.getIncludeCategories(), options.getExcludeCategories(),
filter.getIncludePatterns(), filter.getExcludePatterns(),
filter.getCommandLineIncludePatterns()));
}
@Override
public Action<WorkerProcessBuilder> getWorkerConfigurationAction() {
return new Action<WorkerProcessBuilder>() {
@Override
public void execute(WorkerProcessBuilder workerProcessBuilder) {
workerProcessBuilder.sharedPackages("junit.framework");
workerProcessBuilder.sharedPackages("junit.extensions");
workerProcessBuilder.sharedPackages("org.junit");
}
};
}
@Override
public List<String> getTestWorkerImplementationModules() {
return Collections.emptyList();
}
@Override
public String getDisplayName() {
return "JUnit 4";
}
@Override
public JUnitOptions getOptions() {
return options;
}
void setOptions(JUnitOptions options) {
this.options = options;
}
@Override
public JUnitDetector getDetector() {
return detector;
}
@Override
public void close() throws IOException {
// Clear expensive state from the test framework to avoid holding on to memory
// This should probably be a part of the test task and managed there.
detector = null;
}
private static class TestClassProcessorFactoryImpl implements WorkerTestClassProcessorFactory, Serializable {
private final JUnitSpec spec;
public TestClassProcessorFactoryImpl(JUnitSpec spec) {
this.spec = spec;
}
@Override
public TestClassProcessor create(ServiceRegistry serviceRegistry) {
return new JUnitTestClassProcessor(spec, serviceRegistry.get(IdGenerator.class), serviceRegistry.get(ActorFactory.class), serviceRegistry.get(Clock.class));
}
}
}