Make findEqualValueOfWellKnownType catch exceptions #2554 (#2555)

UnitTestBot · Aug 25, 2023 · de1712a · de1712a
1 parent 4a75abc
commit de1712a
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/...in/org/utbot/instrumentation/instrumentation/execution/constructors/UtModelConstructor.kt b/...in/org/utbot/instrumentation/instrumentation/execution/constructors/UtModelConstructor.kt
@@ -348,12 +348,14 @@ class UtModelConstructor(
                 ?: constructCompositeModel(value, remainingDepth)
         }
 
-    private fun findEqualValueOfWellKnownType(value: Any): Pair<Any, ClassId>? = when (value) {
-        is List<*> -> ArrayList(value) to listClassId
-        is Set<*> -> LinkedHashSet(value) to setClassId
-        is Map<*, *> -> LinkedHashMap(value) to mapClassId
-        else -> null
-    }
+    private fun findEqualValueOfWellKnownType(value: Any): Pair<Any, ClassId>? = runCatching {
+        when (value) {
+            is List<*> -> ArrayList(value) to listClassId
+            is Set<*> -> LinkedHashSet(value) to setClassId
+            is Map<*, *> -> LinkedHashMap(value) to mapClassId
+            else -> null
+        }
+    }.getOrNull()
 
     /**
      * Constructs custom UtModel but does it only for predefined list of classes.