fixed PY-9074 Test runner starts all tests if cursor is at the very end of the test method

This commit is contained in:
Ekaterina Tuzova
2014-11-06 18:42:55 +03:00
parent 600a043dc5
commit d4d2f12431
@@ -33,10 +33,7 @@ import com.intellij.openapi.roots.ProjectRootManager;
import com.intellij.openapi.util.Ref;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiDirectory;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiFileSystemItem;
import com.intellij.psi.*;
import com.intellij.psi.util.PsiTreeUtil;
import com.jetbrains.python.PythonModuleTypeBase;
import com.jetbrains.python.facet.PythonFacetSettings;
@@ -116,7 +113,10 @@ abstract public class PythonTestConfigurationProducer extends RunConfigurationPr
if (context == null) return false;
final Location location = context.getLocation();
if (location == null || !isAvailable(location)) return false;
final PsiElement element = location.getPsiElement();
PsiElement element = location.getPsiElement();
if (element instanceof PsiWhiteSpace) {
element = PyUtil.findNonWhitespaceAtOffset(element.getContainingFile(), element.getTextOffset());
}
if (PythonUnitTestRunnableScriptFilter.isIfNameMain(location)) return false;
final Module module = location.getModule();