init field in setup: use setup method template(IDEA-86135)

This commit is contained in:
Anna Kozlova
2012-05-18 10:17:08 +04:00
parent eaf77e0975
commit fb7ddf8340
5 changed files with 12 additions and 4 deletions
@@ -15,11 +15,14 @@
*/
package com.intellij.testIntegration;
import com.intellij.ide.fileTemplates.FileTemplate;
import com.intellij.ide.fileTemplates.FileTemplateDescriptor;
import com.intellij.ide.fileTemplates.FileTemplateManager;
import com.intellij.lang.Language;
import com.intellij.lang.java.JavaLanguage;
import com.intellij.openapi.module.Module;
import com.intellij.openapi.roots.ProjectRootManager;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.*;
import com.intellij.psi.search.GlobalSearchScope;
@@ -104,4 +107,9 @@ public abstract class JavaTestFramework implements TestFramework {
public FileTemplateDescriptor getParametersMethodFileTemplateDescriptor() {
return null;
}
protected PsiMethod createSetUpPatternMethod(JVMElementFactory factory) {
final FileTemplate template = FileTemplateManager.getInstance().getCodeTemplate(getSetUpMethodFileTemplateDescriptor().getFileName());
return factory.createMethodFromText(StringUtil.replace(template.getText(), "${BODY}", ""), null);
}
}