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 }