Skip to content

Commit a720808

Browse files
rudolf101zishkaz
andauthored
Fix empty member list for generate test action (#1906)
* Added methods of the first class to targets in UI if no top-level funcs were found Co-authored-by: Sergey Loktev <loktevsergay.2003@gmail.com>
1 parent 1db6355 commit a720808

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsLanguageAssistant.kt

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)