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
|
package com.intellij.structuralsearch;
import com.intellij.codeInsight.daemon.quickFix.LightQuickFixTestCase;
import com.intellij.lang.Language;
import com.intellij.openapi.fileTypes.FileType;
import com.intellij.openapi.fileTypes.StdFileTypes;
import com.intellij.openapi.roots.LanguageLevelProjectExtension;
import com.intellij.openapi.util.io.FileUtilRt;
import com.intellij.openapi.vfs.CharsetToolkit;
import com.intellij.pom.java.LanguageLevel;
import com.intellij.structuralsearch.impl.matcher.MatcherImpl;
import com.intellij.structuralsearch.impl.matcher.MatcherImplUtil;
import java.io.File;
import java.io.IOException;
import java.util.List;
abstract class StructuralSearchTestCase extends LightQuickFixTestCase {
protected MatchOptions options;
protected Matcher testMatcher;
@Override
protected void setUp() throws Exception {
super.setUp();
StructuralSearchUtil.ourUseUniversalMatchingAlgorithm = false;
testMatcher = new Matcher(getProject());
options = new MatchOptions();
options.setLooseMatching(true);
options.setRecursiveSearch(true);
LanguageLevelProjectExtension.getInstance(getProject()).setLanguageLevel(LanguageLevel.JDK_1_5);
}
@Override
protected void tearDown() throws Exception {
testMatcher = null;
options = null;
super.tearDown();
}
protected int findMatchesCount(String in, String pattern, boolean filePattern, FileType fileType) {
return findMatches(in,pattern,filePattern, fileType).size();
}
protected List<MatchResult> findMatches(String in,
String pattern,
boolean filePattern,
FileType patternFileType,
Language patternLanugage,
FileType sourceFileType,
String sourceExtension,
boolean physicalSourceFile) {
return findMatches(in, pattern, filePattern, patternFileType, patternLanugage, sourceFileType, sourceExtension, physicalSourceFile,
true);
}
protected List<MatchResult> findMatches(String in,
String pattern,
boolean filePattern,
FileType patternFileType,
Language patternLanugage,
FileType sourceFileType,
String sourceExtension,
boolean physicalSourceFile,
boolean transform) {
options.clearVariableConstraints();
options.setSearchPattern(pattern);
if (transform) {
MatcherImplUtil.transform(options);
}
pattern = options.getSearchPattern();
options.setFileType(patternFileType);
options.setDialect(patternLanugage);
MatcherImpl.validate(getProject(), options);
return testMatcher.testFindMatches(in, pattern, options, filePattern, sourceFileType, sourceExtension, physicalSourceFile);
}
protected List<MatchResult> findMatches(String in, String pattern, boolean filePattern, FileType patternFileType) {
return findMatches(in, pattern, filePattern, patternFileType, null, patternFileType, null, false);
}
protected int findMatchesCount(String in, String pattern, boolean filePattern) {
return findMatchesCount(in, pattern,filePattern, StdFileTypes.JAVA);
}
protected int findMatchesCount(String in, String pattern) {
return findMatchesCount(in,pattern,false);
}
protected List<MatchResult> findMatches(String in, String pattern) {
return findMatches(in,pattern,false, StdFileTypes.JAVA);
}
protected String loadFile(String fileName) throws IOException {
return FileUtilRt.loadFile(new File(getTestDataPath() + fileName), CharsetToolkit.UTF8, true);
}
}
|
