Fix bug in FeatureProcessor (#224)

* Fix bug in FeatureProcessor

* Set testCodeGeneration to false

* Add comments

Co-authored-by: Victor Samoilov <samoilov.victor@huawei.com>

Author: Atos1337

utbot-analytics/src/main/kotlin/org/utbot/features/FeatureProcessorWithStatesRepetition.kt

@@ -52,7 +52,7 @@ class FeatureProcessorWithStatesRepetition(
         val states = mutableListOf<Pair<Int, Long>>()
         var newCoverage = 0
 
-        generateSequence(executionState) { currentState ->
+        generateSequence(executionState) { it.parent }.forEach { currentState ->
             val stateHashCode = currentState.hashCode()
 
             if (currentState.features.isEmpty()) {
@@ -68,8 +68,6 @@ class FeatureProcessorWithStatesRepetition(
                     newCoverage++
                 }
             }
-
-            currentState.parent
         }
 
         generatedTestCases++

utbot-analytics/src/test/java/org/utbot/features/OnePath.java

@@ -0,0 +1,10 @@
+package org.utbot.features;
+
+/**
+ * Class that have one execution path
+ */
+public class OnePath {
+    int onePath() {
+        return 1;
+    }
+}

utbot-analytics/src/test/kotlin/org/utbot/features/FeatureProcessorWithRepetitionTest.kt

@@ -1,15 +1,38 @@
 package org.utbot.features
 
+import org.junit.jupiter.api.AfterAll
 import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.BeforeAll
 import org.junit.jupiter.api.Test
+import org.utbot.analytics.EngineAnalyticsContext
+import org.utbot.examples.AbstractTestCaseGeneratorTest
+import org.utbot.examples.eq
+import org.utbot.examples.withFeaturePath
+import java.io.File
+import java.io.FileInputStream
 
-class FeatureProcessorWithRepetitionTest {
+class FeatureProcessorWithRepetitionTest : AbstractTestCaseGeneratorTest(OnePath::class, false) {
     companion object {
+        const val featureDir = "src/test/resources/features"
         fun reward(coverage: Double, time: Double) = RewardEstimator.reward(coverage, time)
+
+        @JvmStatic
+        @BeforeAll
+        fun beforeAll() {
+            File(featureDir).mkdir()
+            EngineAnalyticsContext.featureProcessorFactory = FeatureProcessorWithStatesRepetitionFactory()
+            EngineAnalyticsContext.featureExtractorFactory = FeatureExtractorFactoryImpl()
+        }
+
+        @JvmStatic
+        @AfterAll
+        fun afterAll() {
+            File(featureDir).deleteRecursively()
+        }
     }
 
     @Test
-    fun testDumpFeatures() {
+    fun testCalculateRewards() {
         val statesToInt = mapOf(
             "a0" to 1,
             "c0" to 2,
@@ -70,4 +93,19 @@ class FeatureProcessorWithRepetitionTest {
             Assertions.assertEquals(it.value, rewards[statesToInt[it.key]])
         }
     }
+
+    /**
+     * Test, that we correctly add test cases and dump them into file
+     */
+    @Test
+    fun addTestCaseTest() {
+        withFeaturePath(featureDir) {
+            check(
+                OnePath::onePath,
+                eq(1)
+            )
+
+            Assertions.assertTrue(FileInputStream("$featureDir/0.csv").bufferedReader().readLines().size > 1)
+        }
+    }
 }

utbot-framework/src/test/kotlin/org/utbot/examples/AbstractTestCaseGeneratorTest.kt

@@ -2811,3 +2811,18 @@ inline fun <reified T> withPathSelectorType(pathSelectorType: PathSelectorType,
         UtSettings.pathSelectorType = prev
     }
 }
+
+inline fun <reified T> withFeaturePath(featurePath: String, block: () -> T): T {
+    val prevFeaturePath = UtSettings.featurePath
+    val prevEnableFeatureProcess = UtSettings.enableFeatureProcess
+
+    UtSettings.featurePath = featurePath
+    UtSettings.enableFeatureProcess = true
+
+    try {
+        return block()
+    } finally {
+        UtSettings.featurePath = prevFeaturePath
+        UtSettings.enableFeatureProcess = prevEnableFeatureProcess
+    }
+}