From b6f778a9867acefc616aaa4ff59f967e8dcd6601 Mon Sep 17 00:00:00 2001 From: Anna Kozlova Date: Thu, 6 Mar 2014 17:23:25 +0100 Subject: [PATCH] disable test navigation outside test roots --- .../intellij/testAssistant/TestDataLineMarkerProvider.java | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/plugins/IdeaTestAssistant/src/com/intellij/testAssistant/TestDataLineMarkerProvider.java b/plugins/IdeaTestAssistant/src/com/intellij/testAssistant/TestDataLineMarkerProvider.java index 73776916edf3..ce3b5a39a06a 100644 --- a/plugins/IdeaTestAssistant/src/com/intellij/testAssistant/TestDataLineMarkerProvider.java +++ b/plugins/IdeaTestAssistant/src/com/intellij/testAssistant/TestDataLineMarkerProvider.java @@ -29,6 +29,7 @@ import com.intellij.openapi.roots.ProjectRootManager; import com.intellij.openapi.vfs.VfsUtil; import com.intellij.openapi.vfs.VirtualFile; import com.intellij.psi.*; +import com.intellij.psi.util.PsiUtilCore; import com.intellij.util.PlatformIcons; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; @@ -51,6 +52,10 @@ public class TestDataLineMarkerProvider implements LineMarkerProvider { if (ApplicationManager.getApplication().isUnitTestMode()) { return null; } + final VirtualFile file = PsiUtilCore.getVirtualFile(element); + if (file == null || !ProjectFileIndex.SERVICE.getInstance(element.getProject()).isInTestSourceContent(file)) { + return null; + } if (element instanceof PsiMethod) { final PsiMethod method = (PsiMethod)element; if (isTestMethod(method)) {