Improve caching

Author: EgorkaKulikov

utbot-instrumentation/src/main/kotlin/org/utbot/instrumentation/instrumentation/execution/SpringUtExecutionInstrumentation.kt

@@ -51,20 +51,19 @@ class SpringUtExecutionInstrumentation(
     ): UtConcreteExecutionResult {
         RepositoryInteraction.recordedInteractions.clear()
 
-        val beansToReset: Set<NamedBean> =
-            if (clazz in relatedBeansCache) {
-                relatedBeansCache.getValue(clazz)
-            } else {
-                val relatedBeans = springContext.getBeansOfType(clazz)
-                val accessibleTypes = relatedBeans.flatMap { typesAnalyzer.collectAccessibleTypes(it.bean) }
-
-                accessibleTypes
-                    .flatMap { springContext.getBeansOfType(it) }
-                    .toSet()
-            }
+        val beansToReset: Set<NamedBean> = relatedBeansCache.getOrPut(clazz) {
+            val relatedBeans = springContext.getBeansOfType(clazz)
+            val accessibleTypes = relatedBeans.flatMap { typesAnalyzer.collectAccessibleTypes(it.bean) }
 
-        beansToReset.forEach { springContext.resetBean(it.beanName) }
+            val beansToReset = accessibleTypes
+                .flatMap { springContext.getBeansOfType(it) }
+                .toSet()
 
+            relatedBeansCache[clazz] = beansToReset
+            beansToReset
+        }
+
+        beansToReset.forEach { springContext.resetBean(it.beanName) }
         val jdbcTemplate = getBean("jdbcTemplate")
 
         val entityNames = beansToReset