azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Go to Test Action: Make available in the entire class text range

Browse Source 
#KT-10757 Fixed
This commit is contained in:
Alexey Sedunov
2016-01-28 16:13:00 +03:00
parent 3e2036b6ec
commit e386a6b0e9
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/kotlin/idea/testIntegration/KotlinTestCreator.kt
+14 -2
View File
@@ -20,13 +20,25 @@ import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project
import com.intellij.psi.PsiFile
import com.intellij.testIntegration.TestCreator
import org.jetbrains.kotlin.psi.KtClassOrObject
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.psi.KtNamedDeclaration
import org.jetbrains.kotlin.psi.psiUtil.parents
class KotlinTestCreator : TestCreator {
private fun getTarget(editor: Editor, file: PsiFile): KtNamedDeclaration? {
return file.findElementAt(editor.caretModel.offset)
?.parents
?.firstOrNull { it is KtClassOrObject || it is KtNamedDeclaration && it.parent is KtFile } as? KtNamedDeclaration
}
override fun isAvailable(project: Project, editor: Editor, file: PsiFile): Boolean {
return KotlinCreateTestIntention().isAvailable(project, editor, file)
val declaration = getTarget(editor, file) ?: return false
return KotlinCreateTestIntention().applicabilityRange(declaration) != null
}
override fun createTest(project: Project, editor: Editor, file: PsiFile) {
KotlinCreateTestIntention().invoke(project, editor, file)
val declaration = getTarget(editor, file) ?: return
KotlinCreateTestIntention().applyTo(declaration, editor)
}
}