assertSame is used on each depth of deepEquals

UnitTestBot · Oct 4, 2023 · 883af21 · 883af21
1 parent 6626808
commit 883af21
Showing 1 changed file with 6 additions and 6 deletions.
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
@@ -619,6 +619,11 @@ open class CgMethodConstructor(val context: CgContext) : CgContextOwner by conte
         visitedModels += modelWithField
 
         with(testFrameworkManager) {
+            if (expectedModel.isMockModel()) {
+                currentBlock += assertions[assertSame](expected, actual).toStatement()
+                return
+            }
+
             if (depth >= DEEP_EQUALS_MAX_DEPTH) {
                 currentBlock += CgSingleLineComment("Current deep equals depth exceeds max depth $DEEP_EQUALS_MAX_DEPTH")
                 currentBlock += getDeepEqualsAssertion(expected, actual).toStatement()
@@ -1226,12 +1231,7 @@ open class CgMethodConstructor(val context: CgContext) : CgContextOwner by conte
         if (!successfullyConstructedCustomAssert) {
             val expectedVariable = variableConstructor.getOrCreateVariable(expected, expectedVariableName)
             if (emptyLineIfNeeded) emptyLineIfNeeded()
-
-            if (expected.isMockModel()) {
-                testFrameworkManager.assertSame(expectedVariable, actual)
-            } else {
-                assertEquality(expectedVariable, actual)
-            }
+            assertEquality(expectedVariable, actual)
         }
     }