From 689c25be0bbed963359dddf15f84ef5b73d4f1d2 Mon Sep 17 00:00:00 2001 From: Yury Kamenev Date: Thu, 2 Mar 2023 13:14:39 +0300 Subject: [PATCH] Use deep equality for mock results --- .../org/utbot/framework/codegen/tree/CgMethodConstructor.kt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgMethodConstructor.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgMethodConstructor.kt index 6a4fe6e14f..fca06f2837 100644 --- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgMethodConstructor.kt +++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgMethodConstructor.kt @@ -577,6 +577,7 @@ open class CgMethodConstructor(val context: CgContext) : CgContextOwner by conte val modelWithField = ModelWithField(expectedModel, expectedModelField) if (modelWithField in visitedModels) return + @Suppress("NAME_SHADOWING") var expected = expected if (expected == null) { require(!needExpectedDeclaration(expectedModel)) @@ -754,7 +755,8 @@ open class CgMethodConstructor(val context: CgContext) : CgContextOwner by conte return } - if (expected.hasNotParametrizedCustomEquals()) { + // We can use overridden equals if we have one, but not for mocks. + if (expected.hasNotParametrizedCustomEquals() && !expectedModel.isMock) { // We rely on already existing equals currentBlock += CgSingleLineComment("${expected.type.canonicalName} has overridden equals method") currentBlock += assertions[assertEquals](expected, actual).toStatement()