@@ -20,6 +20,8 @@ import com.intellij.psi.PsiElement
2020import com.jetbrains.nodejs.run.NodeJsRunConfiguration
2121import java.nio.file.Paths
2222
23+ fun JSCallExpression.isTest () = this .methodExpression?.text.equals(" test" )
24+
2325class AvaJavaScriptTestRunnerRunConfigurationGenerator : AnAction () {
2426 private fun writeError (text : String ) {
2527 val notification =
@@ -33,19 +35,18 @@ class AvaJavaScriptTestRunnerRunConfigurationGenerator : AnAction() {
3335 }
3436
3537 private fun getTestName (element : PsiElement ? ): String? {
36- if (element == null ) {
38+ if (element == null || element !is JSCallExpression ) {
3739 return null
3840 }
39- if (element is JSCallExpression ) {
40- val jsCallExpression: JSCallExpression = element
41- val method: JSExpression ? = jsCallExpression.methodExpression
42- if (method?.text.equals(" test" )) {
43- val arguments: Array <JSExpression > = jsCallExpression.arguments
44- if (arguments.isNotEmpty()) {
45- if (arguments[0 ] is JSLiteralExpression ) {
46- val expression: JSLiteralExpression = arguments[0 ] as JSLiteralExpression
47- return expression.stringValue
48- }
41+
42+ val jsCallExpression: JSCallExpression = element
43+
44+ if (jsCallExpression.isTest()) {
45+ val arguments: Array <JSExpression > = jsCallExpression.arguments
46+ if (arguments.isNotEmpty()) {
47+ if (arguments[0 ] is JSLiteralExpression ) {
48+ val expression: JSLiteralExpression = arguments[0 ] as JSLiteralExpression
49+ return expression.stringValue
4950 }
5051 return null
5152 }
0 commit comments