Skip to content

Commit 54af0a3

Browse files
committed
Updated UtModelVisitor for new reference models
1 parent 7ad4b90 commit 54af0a3

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

utbot-framework/src/main/kotlin/org/utbot/framework/fields/ExecutionStateAnalyzer.kt

+5-5
Original file line numberDiff line numberDiff line change
@@ -160,23 +160,23 @@ private class FieldStateVisitor : UtModelVisitor<FieldData>() {
160160
)
161161
}
162162

163-
override fun visit(element: UtClassRefModel, data: FieldData) {
163+
override fun visit(element: UtNullModel, data: FieldData) {
164164
recordFieldState(data, element)
165165
}
166166

167-
override fun visit(element: UtEnumConstantModel, data: FieldData) {
167+
override fun visit(element: UtPrimitiveModel, data: FieldData) {
168168
recordFieldState(data, element)
169169
}
170170

171-
override fun visit(element: UtNullModel, data: FieldData) {
171+
override fun visit(element: UtVoidModel, data: FieldData) {
172172
recordFieldState(data, element)
173173
}
174174

175-
override fun visit(element: UtPrimitiveModel, data: FieldData) {
175+
override fun visit(element: UtClassRefModel, data: FieldData) {
176176
recordFieldState(data, element)
177177
}
178178

179-
override fun visit(element: UtVoidModel, data: FieldData) {
179+
override fun visit(element: UtEnumConstantModel, data: FieldData) {
180180
recordFieldState(data, element)
181181
}
182182

utbot-framework/src/main/kotlin/org/utbot/framework/util/UtModelVisitor.kt

+4-2
Original file line numberDiff line numberDiff line change
@@ -21,21 +21,23 @@ abstract class UtModelVisitor<D> {
2121

2222
abstract fun visit(element: UtModel, data: D)
2323

24-
abstract fun visit(element: UtClassRefModel, data: D)
25-
abstract fun visit(element: UtEnumConstantModel, data: D)
2624
abstract fun visit(element: UtNullModel, data: D)
2725
abstract fun visit(element: UtPrimitiveModel, data: D)
2826
abstract fun visit(element: UtVoidModel, data: D)
2927

3028
open fun visit(element: UtReferenceModel, data: D) {
3129
if (!canTraverseReferenceModel(element)) return
3230
when (element) {
31+
is UtClassRefModel -> visit(element, data)
32+
is UtEnumConstantModel -> visit(element, data)
3333
is UtArrayModel -> visit(element, data)
3434
is UtAssembleModel -> visit(element, data)
3535
is UtCompositeModel -> visit(element, data)
3636
}
3737
}
3838

39+
abstract fun visit(element: UtClassRefModel, data: D)
40+
abstract fun visit(element: UtEnumConstantModel, data: D)
3941
protected abstract fun visit(element: UtArrayModel, data: D)
4042
protected abstract fun visit(element: UtAssembleModel, data: D)
4143
protected abstract fun visit(element: UtCompositeModel, data: D)

0 commit comments

Comments
 (0)