From 88d6e08cd6fac12e64ca78f3ff711fdd331ed598 Mon Sep 17 00:00:00 2001 From: linux_china Date: Sat, 21 May 2022 15:50:52 -0700 Subject: [PATCH] chore: add validation for arguments --- .../jetbrains/plugins/vitest/VitestRunnerMarkerProvider.kt | 2 +- .../plugins/vitest/VitestWatchedRunnerMarkerProvider.kt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestRunnerMarkerProvider.kt b/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestRunnerMarkerProvider.kt index 870ffbf..b10ba36 100644 --- a/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestRunnerMarkerProvider.kt +++ b/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestRunnerMarkerProvider.kt @@ -22,7 +22,7 @@ class VitestRunnerMarkerProvider : VitestBaseRunLineMarkerProvider() { override fun getLineMarkerInfo(psiElement: PsiElement): LineMarkerInfo<*>? { if (psiElement is JSCallExpression) { - if (isVitestTestMethod(psiElement)) { + if (isVitestTestMethod(psiElement) && psiElement.arguments.isNotEmpty()) { val testMethod = psiElement.firstChild.text val testName = psiElement.arguments[0].text.trim { it == '\'' || it == '"' diff --git a/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestWatchedRunnerMarkerProvider.kt b/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestWatchedRunnerMarkerProvider.kt index 1c42b68..6c435fe 100644 --- a/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestWatchedRunnerMarkerProvider.kt +++ b/src/main/kotlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestWatchedRunnerMarkerProvider.kt @@ -19,7 +19,7 @@ class VitestWatchedRunnerMarkerProvider : VitestBaseRunLineMarkerProvider() { override fun getLineMarkerInfo(psiElement: PsiElement): LineMarkerInfo<*>? { if (psiElement is JSCallExpression) { - if (isVitestTestMethod(psiElement)) { + if (isVitestTestMethod(psiElement) && psiElement.arguments.isNotEmpty()) { val testName = psiElement.arguments[0].text.trim { it == '\'' || it == '"' }