blob: e44dfe46d6f3ef5c44187d91c02076662a438d9a (
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
package com.intellij.coverage;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.module.Module;
import com.intellij.openapi.module.ModuleManager;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.roots.CompilerModuleExtension;
import com.intellij.openapi.vfs.VfsUtilCore;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.rt.coverage.util.classFinder.ClassFinder;
import com.intellij.rt.coverage.util.classFinder.ClassPathEntry;
import com.intellij.util.lang.UrlClassLoader;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.regex.Pattern;
/**
* @author anna
*/
class IdeaClassFinder extends ClassFinder {
private static final Logger LOG = Logger.getInstance("#" + IdeaClassFinder.class.getName());
private final Project myProject;
private final CoverageSuitesBundle myCurrentSuite;
public IdeaClassFinder(Project project, CoverageSuitesBundle currentSuite) {
super(obtainPatternsFromSuite(currentSuite), new ArrayList());
myProject = project;
myCurrentSuite = currentSuite;
}
private static List<Pattern> obtainPatternsFromSuite(CoverageSuitesBundle currentSuiteBundle) {
final List<Pattern> includePatterns = new ArrayList<Pattern>();
for (CoverageSuite currentSuite : currentSuiteBundle.getSuites()) {
for (String pattern : ((JavaCoverageSuite)currentSuite).getFilteredPackageNames()) {
includePatterns.add(Pattern.compile(pattern + ".*"));
}
for (String pattern : ((JavaCoverageSuite)currentSuite).getFilteredClassNames()) {
includePatterns.add(Pattern.compile(pattern));
}
}
return includePatterns;
}
@Override
protected Collection getClassPathEntries() {
final Collection entries = super.getClassPathEntries();
final Module[] modules = ModuleManager.getInstance(myProject).getModules();
for (Module module : modules) {
final CompilerModuleExtension extension = CompilerModuleExtension.getInstance(module);
if (extension != null) {
final VirtualFile outputFile = extension.getCompilerOutputPath();
try {
if (outputFile != null) {
final URL outputURL = VfsUtilCore.virtualToIoFile(outputFile).toURI().toURL();
entries.add(new ClassPathEntry(outputFile.getPath(), UrlClassLoader.build().urls(outputURL).get()));
}
if (myCurrentSuite.isTrackTestFolders()) {
final VirtualFile testOutput = extension.getCompilerOutputPathForTests();
if (testOutput != null) {
final URL testOutputURL = VfsUtilCore.virtualToIoFile(testOutput).toURI().toURL();
entries.add(new ClassPathEntry(testOutput.getPath(), UrlClassLoader.build().urls(testOutputURL).get()));
}
}
}
catch (MalformedURLException e1) {
LOG.error(e1);
}
}
}
return entries;
}
}
|
