Skip to content

Commit 84bbb1f

Browse files
committed
Small refactoring
Uncomment TODO comment
1 parent 58761b9 commit 84bbb1f

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

utbot-framework-api/src/main/kotlin/org/utbot/framework/plugin/api/util/SpringModelUtils.kt

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,7 @@ object SpringModelUtils {
364364
pathVariableClassId,
365365
requestParamClassId,
366366
requestHeaderClassId,
367-
// cookieValueClassId, // TODO uncomment when #2542 is fixed
367+
cookieValueClassId,
368368
requestAttributesClassId,
369369
sessionAttributesClassId,
370370
modelAttributesClassId,
@@ -451,15 +451,11 @@ object SpringModelUtils {
451451
val headersContentModel = createHeadersContentModel(methodId, arguments, idGenerator)
452452
requestBuilderModel = addHeadersToRequestBuilderModel(headersContentModel, requestBuilderModel, idGenerator)
453453

454-
val cookieClassId = cookieClassIds.singleOrNull()
455-
if(cookieClassId is ClassId) {
454+
cookieClassIds.singleOrNull()?.let { cookieClassId ->
456455
val cookieValuesModel = createCookieValuesModel(cookieClassId, methodId, arguments, idGenerator)
457456
requestBuilderModel =
458457
addCookiesToRequestBuilderModel(cookieClassId, cookieValuesModel, requestBuilderModel, idGenerator)
459-
}
460-
else{
461-
logger.warn { "Cookie library not found" }
462-
}
458+
} ?: logger.warn { "Cookie library not found" }
463459

464460
val requestAttributes = collectArgumentsWithAnnotationModels(methodId, requestAttributesClassId, arguments)
465461
requestBuilderModel =

0 commit comments

Comments
 (0)