python/testSrc/com/jetbrains/python/refactoring/classes/PyClassRefactoringTest.java

/*
 * Copyright 2000-2013 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.jetbrains.python.refactoring.classes;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.command.CommandProcessor;
import com.intellij.openapi.project.Project;
import com.intellij.refactoring.BaseRefactoringProcessor;
import com.jetbrains.python.fixtures.PyTestCase;
import com.jetbrains.python.psi.PyClass;
import com.jetbrains.python.psi.PyElement;
import com.jetbrains.python.psi.PyFunction;
import com.jetbrains.python.psi.PyTargetExpression;
import com.jetbrains.python.psi.stubs.PyClassNameIndex;
import com.jetbrains.python.refactoring.classes.membersManager.MembersManager;
import com.jetbrains.python.refactoring.classes.membersManager.PyMemberInfo;
import org.hamcrest.Matchers;
import org.jetbrains.annotations.NotNull;
import org.junit.Assert;

import java.util.Collection;

/**
 * @author Dennis.Ushakov
 */
public abstract class PyClassRefactoringTest extends PyTestCase {
  @NotNull
  private final String myRefactoringName;

  /**
   * @param refactoringName name of the refactoring. It will be used as folder name for tests
   */
  protected PyClassRefactoringTest(@NotNull final String refactoringName) {
    myRefactoringName = refactoringName;
  }

  /**
   * Finds memberInfo by class name and member name.
   *
   * @param clazzName  name of class
   * @param memberName name of member (See {@link #findMember(String, String)} for naming protocol)
   * @return member info
   * @see #findMember(String, String)
   */
  @NotNull
  protected PyMemberInfo<PyElement> findMemberInfo(@NotNull final String clazzName, @NotNull final String memberName) {
    final PyClass clazz = findClass(clazzName);
    return MembersManager.findMember(clazz, findMember(clazzName, memberName));
  }

  /**
   * @param className  class where member should be found
   * @param memberName member that starts with dot (<code>.</code>) is treated as method.
   *                   member that starts with dash (<code>#</code>) is treated as attribute.
   *                   It is treated parent class otherwise
   * @return member or null if not found
   */
  @NotNull
  protected PyElement findMember(@NotNull final String className, @NotNull String memberName) {
    final PyElement result;
    //TODO: Get rid of this chain of copy pastes
    if (memberName.contains(".")) {
      result = findMethod(className, memberName.substring(1));
    }
    else if (memberName.contains("#")) {
      result = findField(className, memberName.substring(1));
    }
    else {
      result = findClass(memberName);
    }
    Assert.assertNotNull(String.format("No member %s found in class %s", memberName, className), result);
    return result;
  }

  private PyElement findField(final String className, final String memberName) {
    final PyClass aClass = findClass(className);
    final PyTargetExpression attribute = aClass.findClassAttribute(memberName, false);
    if (attribute != null) {
      return attribute;
    }
    return aClass.findInstanceAttribute(memberName, false);
  }

  private PyFunction findMethod(final String className, final String name) {
    final PyClass clazz = findClass(className);
    return clazz.findMethodByName(name, false);
  }

  protected PyClass findClass(final String name) {
    final Project project = myFixture.getProject();
    final Collection<PyClass> classes = PyClassNameIndex.find(name, project, false);
    Assert.assertThat(String.format("Expected one class named %s", name), classes, Matchers.hasSize(1));
    return classes.iterator().next();
  }


  protected void moveViaProcessor(@NotNull Project project, @NotNull final BaseRefactoringProcessor processor) {
    CommandProcessor.getInstance().executeCommand(project, new Runnable() {
      @Override
      public void run() {
        ApplicationManager.getApplication().runWriteAction(new Runnable() {
          @Override
          public void run() {
            processor.run();
          }
        });
      }
    }, null, null);
  }

  /**
   * Adds several files to project from folder {@link #myRefactoringName} with extension <pre>py</pre>.
   * Call it <strong>before</strong> refactoring.
   * After refactoring use {@link #checkMultiFile(String...)} to make sure refactoring is ok.
   *
   * @param fileNamesNoExtensions file (module) names to add with out of extensions
   * @see #checkMultiFile(String...)
   */
  protected void configureMultiFile(@NotNull final String... fileNamesNoExtensions) {
    final String baseName = getMultiFileBaseName() + "/";

    for (final String fileNameNoExtension : fileNamesNoExtensions) {
      final String fileNameBefore = String.format("%s.py", fileNameNoExtension);
      myFixture.copyFileToProject(baseName + fileNameBefore, fileNameBefore);
    }
  }

  /**
   * Checks files <strong>after</strong> refactoring. See {@link #configureMultiFile(String...)} for more info.
   *
   * @param fileNamesNoExtensions file names to check with out of extension
   * @see #configureMultiFile(String...)
   */
  protected void checkMultiFile(@NotNull final String... fileNamesNoExtensions) {
    for (final String fileNameNoExtension : fileNamesNoExtensions) {
      final String fileNameAfter = String.format("%s.after.py", fileNameNoExtension);
      final String fileNameBefore = String.format("%s.py", fileNameNoExtension);
      myFixture.checkResultByFile(fileNameBefore, "/" + getMultiFileBaseName() + "/" + fileNameAfter, true);
    }
  }

  /**
   * @return folder name with {@link #myRefactoringName} and test name added
   */
  @NotNull
  protected String getMultiFileBaseName() {
    return "refactoring/" + myRefactoringName + "/" + getTestName(true);
  }
}