A unit test to check that enum ids in recursive models are consistent

Author: dtim

utbot-fuzzers/src/test/java/org/utbot/framework/plugin/api/samples/InnerClassWithEnums.java

@@ -0,0 +1,29 @@
+package org.utbot.framework.plugin.api.samples;
+
+import org.jetbrains.annotations.NotNull;
+
+public class InnerClassWithEnums {
+    private SampleEnum a;
+    private SampleEnum b;
+
+    public InnerClassWithEnums(SampleEnum a, SampleEnum b) {
+        this.a = a;
+        this.b = b;
+    }
+
+    public SampleEnum getA() {
+        return a;
+    }
+
+    public void setA(SampleEnum a) {
+        this.a = a;
+    }
+
+    public SampleEnum getB() {
+        return b;
+    }
+
+    public void setB(SampleEnum b) {
+        this.b = b;
+    }
+}

utbot-fuzzers/src/test/java/org/utbot/framework/plugin/api/samples/OuterClassWithEnums.java

@@ -0,0 +1,37 @@
+package org.utbot.framework.plugin.api.samples;
+
+public class OuterClassWithEnums {
+    private SampleEnum value;
+    private final InnerClassWithEnums left;
+    private final InnerClassWithEnums right;
+
+    public OuterClassWithEnums(SampleEnum value, InnerClassWithEnums left, InnerClassWithEnums right) {
+        this.value = value;
+        this.left = left;
+        this.right = right;
+    }
+
+    public void setValue(SampleEnum value) {
+        this.value = value;
+    }
+
+    public SampleEnum getA() {
+        if (value == SampleEnum.LEFT && left != null) {
+            return left.getA();
+        } else if (value == SampleEnum.RIGHT && right != null) {
+            return right.getA();
+        } else {
+            return null;
+        }
+    }
+
+    public SampleEnum getB() {
+        if (value == SampleEnum.LEFT && left != null) {
+            return left.getB();
+        } else if (value == SampleEnum.RIGHT && right != null) {
+            return right.getB();
+        } else {
+            return null;
+        }
+    }
+}

utbot-fuzzers/src/test/java/org/utbot/framework/plugin/api/samples/SampleEnum.java

@@ -0,0 +1,6 @@
+package org.utbot.framework.plugin.api.samples;
+
+public enum SampleEnum {
+    LEFT,
+    RIGHT
+}

utbot-fuzzers/src/test/kotlin/org/utbot/framework/plugin/api/ModelProviderTest.kt

@@ -24,7 +24,9 @@ import org.utbot.fuzzer.providers.StringConstantModelProvider
 import org.junit.jupiter.api.Assertions.*
 import org.junit.jupiter.api.Test
 import org.utbot.framework.plugin.api.samples.FieldSetterClass
+import org.utbot.framework.plugin.api.samples.OuterClassWithEnums
 import org.utbot.framework.plugin.api.samples.PackagePrivateFieldAndClass
+import org.utbot.framework.plugin.api.samples.SampleEnum
 import org.utbot.framework.plugin.api.util.primitiveByWrapper
 import org.utbot.framework.plugin.api.util.primitiveWrappers
 import org.utbot.framework.plugin.api.util.voidWrapperClassId
@@ -523,6 +525,44 @@ class ModelProviderTest {
         }
     }
 
+    @Test
+    fun `test that enum models in a recursive object model are consistent`() {
+        withUtContext(UtContext(this::class.java.classLoader)) {
+            val idGenerator = ReferencePreservingIntIdGenerator()
+            val expectedIds = SampleEnum.values().associateWith { idGenerator.getOrCreateIdForValue(it) }
+
+            val result = collect(
+                ObjectModelProvider(idGenerator),
+                parameters = listOf(OuterClassWithEnums::class.java.id)
+            )
+
+            assertEquals(1, result.size)
+            val models = result[0] ?: fail("Inconsistent result")
+
+            for (model in models) {
+                val outerModel = (model as? UtAssembleModel)
+                    ?.finalInstantiationModel as? UtExecutableCallModel
+                    ?: fail("No final instantiation model found for the outer class")
+                for (param in outerModel.params) {
+                    when (param) {
+                        is UtEnumConstantModel -> {
+                            assertEquals(expectedIds[param.value], param.id)
+                        }
+                        is UtAssembleModel -> {
+                            for (enumParam in (param.finalInstantiationModel as UtExecutableCallModel).params) {
+                                enumParam as UtEnumConstantModel
+                                assertEquals(expectedIds[enumParam.value], enumParam.id)
+                            }
+                        }
+                        else -> {
+                            fail("Unexpected parameter model: $param")
+                        }
+                    }
+                }
+            }
+        }
+    }
+
     private fun collect(
         modelProvider: ModelProvider,
         name: String = "testMethod",