File tree Expand file tree Collapse file tree 1 file changed +12
-2
lines changed
utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js Expand file tree Collapse file tree 1 file changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -78,10 +78,20 @@ object JsLanguageAssistant : LanguageAssistant() {
7878 file = file,
7979 )
8080 }
81- val memberInfos = generateMemberInfo(e.project!! , file.statements.filterIsInstance<JSFunction >())
82- val focusedMethodMI = memberInfos.find { member ->
81+ var memberInfos = generateMemberInfo(e.project!! , file.statements.filterIsInstance<JSFunction >())
82+ var focusedMethodMI = memberInfos.find { member ->
8383 member.member?.name == focusedMethod?.name
8484 }
85+ // TODO: generate tests for all classes, not only the first one
86+ // (currently not possible since breaks JsTestGenerator routine)
87+ if (memberInfos.isEmpty()) {
88+ memberInfos = generateMemberInfo(
89+ e.project!! ,
90+ emptyList(),
91+ file.statements.filterIsInstance<ES6Class >().first()
92+ )
93+ focusedMethodMI = memberInfos.first()
94+ }
8595 return PsiTargets (
8696 methods = memberInfos,
8797 focusedMethod = focusedMethodMI,
You can’t perform that action at this time.
0 commit comments