7b4690103a4bacaaaba804500bae08d6f34992c2
[idea/community.git] / java / java-tests / testSrc / com / intellij / refactoring / IntroduceConstantTest.java
1 package com.intellij.refactoring;
2
3 import com.intellij.JavaTestUtil;
4 import com.intellij.openapi.projectRoots.Sdk;
5 import com.intellij.openapi.projectRoots.impl.JavaSdkImpl;
6 import com.intellij.psi.PsiClass;
7 import com.intellij.psi.PsiJavaFile;
8 import com.intellij.psi.PsiLocalVariable;
9 import com.intellij.psi.util.PsiTreeUtil;
10 import com.intellij.testFramework.LightCodeInsightTestCase;
11 import junit.framework.Assert;
12 import org.jetbrains.annotations.NonNls;
13
14 /**
15  * @author ven
16  */
17 public class IntroduceConstantTest extends LightCodeInsightTestCase {
18   @NonNls private static final String BASE_PATH = "/refactoring/introduceConstant/";
19
20   @Override
21   protected String getTestDataPath() {
22     return JavaTestUtil.getJavaTestDataPath();
23   }
24
25   public void testInNonNls() throws Exception {
26     doTest(false);
27   }
28
29   private void doTest(boolean makeEnumConstant) throws Exception {
30     configureByFile(BASE_PATH + getTestName(false) + ".java");
31     convertLocal(makeEnumConstant);
32     checkResultByFile(BASE_PATH + getTestName(false) + "_after.java");
33   }
34
35   public void testEnumConstant() throws Exception {
36     doTest(true);    
37   }
38
39   private static void convertLocal(final boolean makeEnumConstant) {
40     PsiLocalVariable local = PsiTreeUtil.getParentOfType(getFile().findElementAt(getEditor().getCaretModel().getOffset()), PsiLocalVariable.class);
41     new MockLocalToFieldHandler(getProject(), true, makeEnumConstant).convertLocalToField(local, getEditor());
42   }
43
44   public void testPartialStringLiteral() throws Exception {
45     configureByFile(BASE_PATH + getTestName(false) + ".java");
46     new MockIntroduceConstantHandler(null).invoke(getProject(), getEditor(), getFile(), null);
47     checkResultByFile(BASE_PATH + getTestName(false) + "_after.java");
48   }
49
50   public void testPartialStringLiteralQualified() throws Exception {
51     configureByFile(BASE_PATH + getTestName(false) + ".java");
52     final PsiClass psiClass = ((PsiJavaFile)getFile()).getClasses()[0];
53     Assert.assertNotNull(psiClass);
54     final PsiClass targetClass = psiClass.findInnerClassByName("D", false);
55     Assert.assertNotNull(targetClass);
56     new MockIntroduceConstantHandler(targetClass).invoke(getProject(), getEditor(), getFile(), null);
57     checkResultByFile(BASE_PATH + getTestName(false) + "_after.java");
58   }
59
60   protected Sdk getProjectJDK() {
61     return JavaSdkImpl.getMockJdk15("java 1.5");
62   }
63 }