@@ -20,18 +20,18 @@ data class TestClassModel(
20
20
val class2methodTestSets = testSets.groupBy { it.executableId.classId }
21
21
22
22
val classesWithMethodsUnderTest = testSets
23
- .distinctBy { it.executableId.classId }
24
23
.map { it.executableId.classId }
24
+ .distinct()
25
25
26
26
// For each class stores list of its "direct" nested classes
27
- val class2nestedClasses = mutableMapOf<ClassId , MutableList <ClassId >>()
27
+ val class2nestedClasses = mutableMapOf<ClassId , MutableSet <ClassId >>()
28
28
29
29
for (classId in classesWithMethodsUnderTest) {
30
30
var currentClass = classId
31
31
var enclosingClass = currentClass.enclosingClass
32
32
// while we haven't reached the top of nested class hierarchy or the main class under test
33
33
while (enclosingClass != null && currentClass != classUnderTest) {
34
- class2nestedClasses.getOrPut(enclosingClass) { mutableListOf () } + = currentClass
34
+ class2nestedClasses.getOrPut(enclosingClass) { mutableSetOf () } + = currentClass
35
35
currentClass = enclosingClass
36
36
enclosingClass = enclosingClass.enclosingClass
37
37
}
@@ -42,7 +42,7 @@ data class TestClassModel(
42
42
private fun constructRecursively (
43
43
clazz : ClassId ,
44
44
class2methodTestSets : Map <ClassId , List <CgMethodTestSet >>,
45
- class2nestedClasses : Map <ClassId , List <ClassId >>
45
+ class2nestedClasses : Map <ClassId , Set <ClassId >>
46
46
): TestClassModel {
47
47
val currentNestedClasses = class2nestedClasses.getOrDefault(clazz, listOf ())
48
48
val currentMethodTestSets = class2methodTestSets.getOrDefault(clazz, listOf ())
0 commit comments