From d4d2f124310798bbabdcc353569035106ead92a8 Mon Sep 17 00:00:00 2001
From: Ekaterina Tuzova <Ekaterina.Tuzova@jetbrains.com>
Date: Thu, 6 Nov 2014 18:42:55 +0300
Subject: [PATCH] fixed PY-9074 Test runner starts all tests if cursor is at
 the very end of the test method

---
 .../testing/PythonTestConfigurationProducer.java       | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java b/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java
index 4b3085cf5d5d..06433551fa6f 100644
--- a/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java
+++ b/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java
@@ -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();