chore: add validation for arguments

linux-china · May 21, 2022 · 88d6e08 · 88d6e08
1 parent f4e8c47
commit 88d6e08
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 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
@@ -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/...otlin/com/github/linuxchina/jetbrains/plugins/vitest/VitestWatchedRunnerMarkerProvider.kt b/...otlin/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 == '"'
                 }