diff --git a/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsLanguageAssistant.kt b/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsLanguageAssistant.kt index dd3a3ca9f3..3d3eefe17a 100644 --- a/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsLanguageAssistant.kt +++ b/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsLanguageAssistant.kt @@ -85,11 +85,16 @@ object JsLanguageAssistant : LanguageAssistant() { // TODO: generate tests for all classes, not only the first one // (currently not possible since breaks JsTestGenerator routine) if (memberInfos.isEmpty()) { + val classes = file.statements.filterIsInstance() + if (classes.isEmpty()) return null + memberInfos = generateMemberInfo( e.project!!, emptyList(), - file.statements.filterIsInstance().first() + classes.first() ) + if (memberInfos.isEmpty()) return null + focusedMethodMI = memberInfos.first() } return PsiTargets(