MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-04-19 04:51:24 +07:00
Code Issues Packages Projects Releases Wiki Activity

PY-40061: Don't create tests in python site-packages

Browse Source 
GitOrigin-RevId: ab8dc1dc5b6fe784a90806458368db8707adff3c
This commit is contained in:
Ilya.Kazakevich
2021-09-17 03:02:11 +03:00
committed by intellij-monorepo-bot
parent cd64f48ac4
commit 61953969e4
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python/src/com/jetbrains/python/codeInsight/testIntegration/PyTestCreationModel.kt
View File

@@ -2,6 +2,8 @@
package com.jetbrains.python.codeInsight.testIntegration
import com.intellij.openapi.module.ModuleUtil
import com.intellij.openapi.roots.OrderRootType
import com.intellij.openapi.vfs.VfsUtil
import com.intellij.openapi.vfs.VirtualFile
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiNamedElement
@@ -11,6 +13,7 @@ import com.jetbrains.python.codeInsight.testIntegration.PyTestCreationModel.Comp
import com.jetbrains.python.psi.PyClass
import com.jetbrains.python.psi.PyFile
import com.jetbrains.python.psi.PyFunction
import com.jetbrains.python.sdk.pythonSdk
import com.jetbrains.python.testing.PythonUnitTestDetectorsBasedOnSettings
/**
@@ -77,7 +80,13 @@ class PyTestCreationModel(var fileName: String,
private fun getTestFolder(element: PsiElement): VirtualFile =
ModuleUtil.findModuleForPsiElement(element)?.let { module ->
FilenameIndex.getVirtualFilesByName( "tests", module.moduleContentScope).firstOrNull()
// No need to create tests in site-packages (aka classes root)
val roots = module.pythonSdk?.rootProvider?.getFiles(OrderRootType.CLASSES) ?: return@let null
return@let FilenameIndex.getVirtualFilesByName("tests", module.moduleContentScope).firstOrNull { possibleRoot ->
possibleRoot.isDirectory && roots.none {
VfsUtil.isAncestor(it, possibleRoot, false)
}
}
} ?: element.containingFile.containingDirectory.virtualFile
}