android/testSrc/org/jetbrains/android/refactoring/AndroidInlineStyleTest.java

package org.jetbrains.android.refactoring;

import com.intellij.openapi.actionSystem.Presentation;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiElement;
import com.intellij.refactoring.actions.InlineAction;
import com.intellij.refactoring.util.CommonRefactoringUtil;
import com.intellij.testFramework.fixtures.IdeaProjectTestFixture;
import com.intellij.testFramework.fixtures.TestFixtureBuilder;
import com.intellij.util.IncorrectOperationException;
import com.intellij.util.containers.MultiMap;
import org.jetbrains.android.AndroidTestCase;
import org.jetbrains.annotations.NotNull;

import java.util.List;

/**
 * @author Eugene.Kudelevsky
 */
public class AndroidInlineStyleTest extends AndroidTestCase {
  private static final String BASE_PATH = "refactoring/inlineStyle/";

  public void test1() {
    doTest(true);
  }

  public void test2() {
    doTest(true);
  }

  public void test3() {
    doTest(true);
  }

  public void test4() {
    doTest(true);
  }

  public void test5() {
    doTestDisabled();
  }

  public void test6() {
    doTest(true);
  }

  public void test7() {
    doTest(false);
  }

  public void test8() {
    doTestDisabled();
  }

  /*
  Test disabled. This test relied on
     <Button style="@style/" ... />
  returning a valid android reference for the style attribute (because the inline
  style refactoring calls AndroidDomUtil#getAndroidResourceReference to see
  whether the refactoring should be enabled, which now returns null.
  However, it's arguably even more correct for the refactoring to not be enabled
  for this element than for it to be enabled and to fail with an error message.

  public void test9() {
    doTestErrorMessageShown(true, true, false);
  }
  */

  public void test10() {
    doTestDisabled();
  }

  public void test11() {
    doTestErrorMessageShown(true, true, false);
  }

  public void test12() {
    doTestErrorMessageShown(true, true, true);
  }

  public void test13() {
    doTestErrorMessageShown(true, true, true);
  }

  public void test14() {
    final String libStylesPath = getContentRootPath("lib") + "/res/values/styles.xml";
    final String stylesLocalPath = BASE_PATH + getTestName(true) + "_styles.xml";
    myFixture.copyFileToProject(stylesLocalPath, libStylesPath);
    doTestErrorMessageShown(true, true, true);
    myFixture.checkResultByFile(libStylesPath, stylesLocalPath, true);
  }

  public void test15() {
    final String libModuleDir = getContentRootPath("lib");
    final String libStylesPath = libModuleDir + "/res/values/styles.xml";
    final String appStylePath = "/res/values/styles.xml";
    final String appLayoutLocalPath = BASE_PATH + getTestName(true) + "_1.xml";
    final String stylesLocalPath = BASE_PATH + getTestName(true) + "_styles.xml";
    final String appLayoutPath = "res/layout/layout.xml";

    myFixture.copyFileToProject(stylesLocalPath, libStylesPath);
    myFixture.copyFileToProject(stylesLocalPath, appStylePath);
    myFixture.copyFileToProject(appLayoutLocalPath, appLayoutPath);

    final MultiMap<PsiElement,String> conflicts = performAction(false, libModuleDir + "/res/layout").getSecond().getConflicts();
    assertEquals(1, conflicts.keySet().size());
    assertEquals(1, conflicts.values().size());

    myFixture.checkResultByFile(BASE_PATH + getTestName(true) + "_after.xml");
    myFixture.checkResultByFile(libStylesPath, BASE_PATH + getTestName(true) + "_styles_after.xml", true);
    myFixture.checkResultByFile(appStylePath, stylesLocalPath, true);
    myFixture.checkResultByFile(appLayoutPath, appLayoutLocalPath, true);
  }

  public void test35() throws Exception {
    final String libModuleDir = getContentRootPath("lib");
    final String libStylesPath = libModuleDir + "/res/values/styles.xml";
    final String appStylePath = "/res/values/styles.xml";
    final String appLayoutLocalPath = BASE_PATH + getTestName(true) + "_1.xml";
    final String stylesLocalPath = BASE_PATH + getTestName(true) + "_styles.xml";
    final String appLayoutPath = "res/layout/layout.xml";

    myFixture.copyFileToProject(stylesLocalPath, libStylesPath);
    myFixture.copyFileToProject(stylesLocalPath, appStylePath);
    myFixture.copyFileToProject(appLayoutLocalPath, appLayoutPath);
    doTest(true, libModuleDir + "/res/layout");
  }

  public void test16() {
    final String libModuleDir = getContentRootPath("lib");
    final String libStylesPath = libModuleDir + "/res/values/styles.xml";
    final String appLayoutPath = "res/layout/layout.xml";

    myFixture.copyFileToProject(BASE_PATH + getTestName(true) + "_styles.xml", libStylesPath);
    myFixture.copyFileToProject(BASE_PATH + getTestName(true) + "_1.xml", appLayoutPath);

    final String testName = getTestName(true);
    final VirtualFile f = myFixture
      .copyFileToProject(BASE_PATH + testName + ".xml", libModuleDir + "/res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    doInlineStyleReferenceAction(false, true);

    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    myFixture.checkResultByFile(appLayoutPath, BASE_PATH + getTestName(true) + "_1_after.xml", true);
    myFixture.checkResultByFile(libStylesPath, BASE_PATH + getTestName(true) + "_styles_after.xml", true);
  }

  public void test17() {
    final String testName = getTestName(true);
    myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    myFixture.copyFileToProject(BASE_PATH + testName + "_1.xml", "res/layout/test1.xml");
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    doCommonInlineAction(true);
    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    myFixture.checkResultByFile("res/layout/test1.xml", BASE_PATH + testName + "_1.xml", true);
    myFixture.checkResultByFile("res/values/styles.xml", BASE_PATH + testName + "_styles.xml", true);
  }

  private AndroidInlineTestConfig doCommonInlineAction(boolean inlineThisOnly) {
    final AndroidInlineTestConfig config = new AndroidInlineTestConfig(inlineThisOnly);
    AndroidInlineStyleHandler.setTestConfig(config);
    try {
      final Presentation p = myFixture.testAction(new InlineAction());
      assertTrue(p.isEnabled());
      assertTrue(p.isVisible());
    }
    finally {
      AndroidInlineStyleHandler.setTestConfig(null);
    }
    return config;
  }

  public void test18() {
    final String testName = getTestName(true);
    final String stylesPath = BASE_PATH + testName + "_styles.xml";
    myFixture.copyFileToProject(stylesPath, "res/values/styles.xml");
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    try {
      doCommonInlineAction(false);
      fail();
    }
    catch (CommonRefactoringUtil.RefactoringErrorHintException e) {
    }
    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    myFixture.checkResultByFile("res/values/styles.xml", stylesPath, true);
  }

  public void test19() {
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/layout/test.xml");
    myFixture.copyFileToProject(BASE_PATH + testName + "_1.xml", "res/layout/test1.xml");
    myFixture.configureFromExistingVirtualFile(f);
    doCommonInlineAction(false);
    myFixture.checkResultByFile(BASE_PATH + testName + "_styles_after.xml", true);
    myFixture.checkResultByFile("res/layout/test.xml", BASE_PATH + testName + "_after.xml", true);
    myFixture.checkResultByFile("res/layout/test1.xml", BASE_PATH + testName + "_1_after.xml", true);
  }

  public void test20() {
    final boolean thisOnly = false;
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    final String layoutPath = BASE_PATH + testName + ".xml";
    myFixture.copyFileToProject(layoutPath, "res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    try {
      doCommonInlineAction(thisOnly);
      fail();
    }
    catch (CommonRefactoringUtil.RefactoringErrorHintException e) {
    }
    myFixture.checkResultByFile(BASE_PATH + testName + "_styles_after.xml", true);
    myFixture.checkResultByFile("res/layout/test.xml", layoutPath, true);
  }

  public void test21() {
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    final String layoutPath = BASE_PATH + testName + ".xml";
    myFixture.copyFileToProject(layoutPath, "res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    try {
      doCommonInlineAction(false);
      fail();
    }
    catch (IncorrectOperationException e) {
      assertTrue(e.getMessage().length() > 0);
    }
  }

  public void test22() {
    final String testName = getTestName(true);
    myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    myFixture.copyFileToProject(BASE_PATH + "MyActivity.java", "src/p1/p2/MyActivity.java");
    myFixture.copyFileToProject("R.java", "gen/p1/p2/R.java");
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    final AndroidInlineTestConfig testConfig = new AndroidInlineTestConfig(false);
    myFixture.testAction(new AndroidInlineStyleReferenceAction(testConfig));
    final MultiMap<PsiElement, String> conflicts = testConfig.getConflicts();
    assertEquals(1, conflicts.keySet().size());
    assertEquals(1, conflicts.values().size());
    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    myFixture.checkResultByFile("src/p1/p2/MyActivity.java", BASE_PATH + "MyActivity.java", true);
    myFixture.checkResultByFile("res/values/styles.xml", BASE_PATH + testName + "_styles_after.xml", true);
  }

  public void test23() {
    final String testName = getTestName(true);
    myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    myFixture.copyFileToProject(BASE_PATH + testName + "_1.xml", "res/layout/test1.xml");
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/layout/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    doCommonInlineAction(false);
    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    myFixture.checkResultByFile("res/layout/test1.xml", BASE_PATH + testName + "_1_after.xml", true);
    myFixture.checkResultByFile("res/values/styles.xml", BASE_PATH + testName + "_styles_after.xml", true);
  }

  public void test24() {
    doTestCommonInlineInValues(true);
  }

  public void test25() {
    doTestInlineActionInValues(true, true, true);
  }

  public void test26() {
    doTestInlineActionInValues(false, true, true);
  }

  public void test27() {
    doTestInlineActionInValues(false, false, false);
  }

  public void test28() {
    doTestCommonInlineInValues(false);
  }

  public void test29() {
    doTestCommonInlineInValues(false);
  }

  public void test30() {
    doTestCommonInlineConflicts();
  }

  public void test31() {
    doTestCommonInlineConflicts();
  }

  public void test32() {
    doTestCommonInlineConflicts();
  }

  private void doTestCommonInlineConflicts() {
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + getTestName(true) + ".xml", "res/values/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    final MultiMap<PsiElement, String> conflicts = doCommonInlineAction(false).getConflicts();
    assertEquals(1, conflicts.keySet().size());
    assertEquals(1, conflicts.values().size());
    myFixture.checkResultByFile(BASE_PATH + getTestName(true) + "_after.xml");
  }

  public void test33() {
    doTestCommonInlineInValues(true);
  }

  public void test34() {
    doTestDisabled();
  }

  private void doTestCommonInlineErrorHintInValues(boolean thisOnly) {
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/values/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    try {
      doCommonInlineAction(thisOnly);
      fail();
    }
    catch (IncorrectOperationException e) {
      assertTrue(e.getMessage().length() > 0);
    }
  }

  private void doTestInlineActionInValues(boolean thisOnly, boolean enabled, boolean checkResults) {
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/values/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    doInlineStyleReferenceAction(thisOnly, enabled);
    if (checkResults) {
      myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    }
  }

  private void doTestCommonInlineInValues(boolean thisOnly) {
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml", "res/values/test.xml");
    myFixture.configureFromExistingVirtualFile(f);
    doCommonInlineAction(thisOnly);
    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
  }

  private void doTest(boolean inlineThisOnly) {
    doTest(inlineThisOnly, "res/layout");
  }

  private void doTest(boolean inlineThisOnly, String dirToCopy) {
    final String testName = getTestName(true);
    myFixture.copyFileToProject(BASE_PATH + testName + "_styles.xml", "res/values/styles.xml");
    final VirtualFile f = myFixture
      .copyFileToProject(BASE_PATH + testName + ".xml", dirToCopy + "/test" + testName + "_" + Boolean.toString(inlineThisOnly) + ".xml");
    myFixture.configureFromExistingVirtualFile(f);
    doInlineStyleReferenceAction(inlineThisOnly, true);
    myFixture.checkResultByFile(BASE_PATH + testName + "_after.xml");
    myFixture.checkResultByFile("res/values/styles.xml", BASE_PATH + testName + (
      inlineThisOnly ? "_styles.xml" : "_styles_after.xml"), true);
  }

  private void doInlineStyleReferenceAction(boolean inlineThisOnly, boolean enabled) {
    final Presentation p =
      myFixture.testAction(new AndroidInlineStyleReferenceAction(new AndroidInlineTestConfig(inlineThisOnly)));
    assertEquals(enabled, p.isEnabled());
    assertTrue(p.isVisible());
  }

  private void doTestErrorMessageShown(boolean testInlineThis, boolean testInlineAll, boolean copyStyles, String dirToCopy) {
    if (copyStyles) {
      myFixture.copyFileToProject(BASE_PATH + getTestName(true) + "_styles.xml",
                                  "res/values/styles.xml");
    }
    if (testInlineThis) {
      doTestErrorMessageShown(true, copyStyles, dirToCopy);
    }

    if (testInlineAll) {
      doTestErrorMessageShown(false, copyStyles, dirToCopy);
    }
  }

  private void doTestErrorMessageShown(boolean testInlineThis, boolean testInlineAll, boolean copyStyles) {
    doTestErrorMessageShown(testInlineThis, testInlineAll, copyStyles, "res/layout");
  }

  private void doTestErrorMessageShown(boolean inlineThisOnly, boolean checkStylesXml, String dirToCopy) {
    try {
      performAction(inlineThisOnly, dirToCopy);
      fail();
    }
    catch (IncorrectOperationException e) {
      assertTrue(e.getMessage().length() > 0);
      final String testName = getTestName(true);
      myFixture.checkResultByFile(BASE_PATH + testName + ".xml");
      if (checkStylesXml) {
        myFixture.checkResultByFile("res/values/styles.xml",
                                    BASE_PATH + testName + "_styles.xml", true);
      }
    }
  }

  private void doTestDisabled() {
    final Presentation presentation = performAction(true, "res/layout").getFirst();
    assertFalse(presentation.isEnabled());
    assertTrue(presentation.isVisible());
  }

  private Pair<Presentation, AndroidInlineTestConfig> performAction(boolean inlineThisOnly, String dirToCopy) {
    final String testName = getTestName(true);
    final VirtualFile f = myFixture.copyFileToProject(BASE_PATH + testName + ".xml",
                                                      dirToCopy + "/test" + testName + "_" +
                                                      Boolean.toString(inlineThisOnly) + ".xml");
    myFixture.configureFromExistingVirtualFile(f);
    final AndroidInlineTestConfig config = new AndroidInlineTestConfig(inlineThisOnly);
    final Presentation presentation = myFixture.testAction(new AndroidInlineStyleReferenceAction(config));
    return Pair.create(presentation, config);
  }

  @Override
  protected void configureAdditionalModules(@NotNull TestFixtureBuilder<IdeaProjectTestFixture> projectBuilder,
                                            @NotNull List<MyAdditionalModuleData> modules) {
    final String testName = getTestName(true);
    if (testName.equals("14") || testName.equals("15") || testName.equals("16") || testName.equals("35")) {
      addModuleWithAndroidFacet(projectBuilder, modules, "lib", true);
    }
  }
}