Add: Map bencmark

Author: sergeypospelov

utbot-framework-test/src/benchmarks/kotlin/org/utbot/benchmarks/BenchmarksTest.kt

@@ -1,6 +1,5 @@
 package org.utbot.benchmarks
 
-import org.junit.jupiter.api.RepeatedTest
 import org.openjdk.jmh.annotations.Benchmark
 import org.openjdk.jmh.annotations.Fork
 import org.openjdk.jmh.annotations.Measurement
@@ -10,26 +9,37 @@ import org.openjdk.jmh.annotations.State
 import org.openjdk.jmh.annotations.Threads
 import org.openjdk.jmh.annotations.Warmup
 import org.utbot.common.FileUtil
-import org.utbot.framework.plugin.api.TestCaseGenerator
+import org.utbot.examples.strings11.StringConcat
+import org.utbot.framework.UtSettings
+import org.utbot.framework.plugin.api.MockStrategyApi
+import org.utbot.framework.plugin.api.util.UtContext
+import org.utbot.framework.plugin.api.util.executableId
+import org.utbot.framework.plugin.api.util.withUtContext
 import org.utbot.framework.plugin.services.JdkInfoDefaultProvider
 import org.utbot.framework.util.SootUtils
 import org.utbot.tests.infrastructure.TestSpecificTestCaseGenerator
-import org.utbot.tests.infrastructure.UtValueTestCaseChecker
-import org.utbot.tests.infrastructure.ignoreExecutionsNumber
 import java.util.concurrent.TimeUnit
 
 @State(Scope.Benchmark)
 @Fork(1)
 @Threads(1)
-@Warmup(iterations = 3)
-@Measurement(iterations = 5, timeUnit = TimeUnit.SECONDS)
-class BenchmarksTest : UtValueTestCaseChecker(
-    Benchmarks::class,
-) {
-    lateinit var testCaseGenerator: TestCaseGenerator
+@Warmup(iterations = 2)
+@Measurement(iterations = 4, timeUnit = TimeUnit.SECONDS)
+class BenchmarksTest {
+    lateinit var testCaseGenerator: TestSpecificTestCaseGenerator
+    lateinit var context: UtContext
+
     @Setup
     fun setUp() {
+        UtSettings.utBotGenerationTimeoutInMillis = 200_000
+        UtSettings.useFuzzing = false
+        UtSettings.useConcreteExecution = false
+
         val testClass = Benchmarks::class.java
+
+        context = UtContext(testClass.classLoader)
+        UtContext.setUtContext(context)
+
         SootUtils.runSoot(Benchmarks::class.java, true, JdkInfoDefaultProvider().info)
         testCaseGenerator =  TestSpecificTestCaseGenerator(
             FileUtil.locateClassPath(testClass)!!.toPath(),
@@ -41,22 +51,52 @@ class BenchmarksTest : UtValueTestCaseChecker(
 
     @Benchmark
     fun benchmarkLongMethod() {
-        check(
-            Benchmarks::longMethod,
-            ignoreExecutionsNumber,
-            { x, y, r -> r == 1 },
-            { x, y, r -> r == 2 },
-        )
+        withUtContext(context) {
+            testCaseGenerator.generate(
+                Benchmarks::longMethod.executableId,
+                MockStrategyApi.NO_MOCKS
+            )
+        }
     }
 
 
     @Benchmark
     fun benchmarkCycle() {
-        check(
-            Benchmarks::cycle,
-            ignoreExecutionsNumber,
-            { _, r -> r == 1 },
-            { _, r -> r == 2 },
-        )
+        withUtContext(context) {
+            testCaseGenerator.generate(
+                Benchmarks::cycle.executableId,
+                MockStrategyApi.NO_MOCKS
+            )
+        }
+    }
+
+    @Benchmark
+    fun benchmarkExceptionInToString() {
+        withUtContext(context) {
+            testCaseGenerator.generate(
+                StringConcat::exceptionInToString.executableId,
+                MockStrategyApi.NO_MOCKS
+            )
+        }
+    }
+
+    @Benchmark
+    fun benchmarkPutAndGet() {
+        withUtContext(context) {
+            testCaseGenerator.generate(
+                Benchmarks::mapPutAndGet.executableId,
+                MockStrategyApi.NO_MOCKS
+            )
+        }
+    }
+
+    @Benchmark
+    fun benchmarkPutInMapFromParameters() {
+        withUtContext(context) {
+            testCaseGenerator.generate(
+                Benchmarks::putInMapFromParameters.executableId,
+                MockStrategyApi.NO_MOCKS
+            )
+        }
     }
 }

utbot-framework-test/src/main/java/org/utbot/benchmarks/Benchmarks.java

@@ -1,5 +1,8 @@
 package org.utbot.benchmarks;
 
+import java.util.HashMap;
+import java.util.Map;
+
 interface Interface {
     default int foo(int x) {
         return x + bar();
@@ -697,4 +700,21 @@ public int interfaceCallMethod(int x) {
     public int bar() {
         return 2;
     }
+
+    Integer mapPutAndGet() {
+        Map<Long, Integer> values = new HashMap<>();
+
+        values.put(1L, 2);
+        values.put(1L, 3);
+
+        return values.get(1L);
+    }
+
+    @SuppressWarnings("OverwrittenKey")
+    Integer putInMapFromParameters(Map<Long, Integer> values) {
+        values.put(1L, 2);
+        values.put(1L, 3);
+
+        return values.get(1L);
+    }
 }