Add assertNotNull for actual in parameterized test generation

Author: sofurihafe

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/Domain.kt

@@ -216,6 +216,8 @@ sealed class TestFramework(
 
     val assertNull by lazy { assertionId("assertNull", objectClassId) }
 
+    val assertNotNull by lazy { assertionId("assertNotNull", objectClassId) }
+
     val assertFalse by lazy { assertionId("assertFalse", booleanClassId) }
 
     val assertTrue by lazy { assertionId("assertTrue", booleanClassId) }

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgMethodConstructor.kt

@@ -1217,7 +1217,10 @@ internal class CgMethodConstructor(val context: CgContext) : CgContextOwner by c
                     else -> ifStatement(
                         CgEqualTo(expected, nullLiteral()),
                         trueBranch = { +testFrameworkManager.assertions[testFramework.assertNull](actual).toStatement() },
-                        falseBranch = { generateDeepEqualsAssertion(expected, actual) }
+                        falseBranch = {
+                            +testFrameworkManager.assertions[testFrameworkManager.assertNotNull](actual).toStatement()
+                            generateDeepEqualsAssertion(expected, actual)
+                        }
                     )
                 }
             }
@@ -1349,10 +1352,7 @@ internal class CgMethodConstructor(val context: CgContext) : CgContextOwner by c
     fun createParameterizedTestMethod(testSet: CgMethodTestSet, dataProviderMethodName: String): CgTestMethod {
         //TODO: orientation on generic execution may be misleading, but what is the alternative?
         //may be a heuristic to select a model with minimal number of internal nulls should be used
-        val genericExecution = testSet.executions
-            .firstOrNull { it.result is UtExecutionSuccess && (it.result as UtExecutionSuccess).model !is UtNullModel }
-            ?: testSet.executions
-                .firstOrNull { it.result is UtExecutionSuccess } ?: testSet.executions.first()
+        val genericExecution = chooseGenericExecution(testSet.executions)
 
         val statics = genericExecution.stateBefore.statics
 
@@ -1420,6 +1420,13 @@ internal class CgMethodConstructor(val context: CgContext) : CgContextOwner by c
         }
     }
 
+    private fun chooseGenericExecution(executions: List<UtExecution>): UtExecution {
+        return executions
+            .firstOrNull { it.result is UtExecutionSuccess && (it.result as UtExecutionSuccess).model !is UtNullModel }
+            ?: executions
+                .firstOrNull { it.result is UtExecutionSuccess } ?: executions.first()
+    }
+
     private fun createParameterDeclarations(
         testSet: CgMethodTestSet,
         genericExecution: UtExecution,

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/TestFrameworkManager.kt

@@ -60,6 +60,7 @@ internal abstract class TestFrameworkManager(val context: CgContext)
     val assertDoubleEquals = context.testFramework.assertDoubleEquals
 
     val assertNull = context.testFramework.assertNull
+    val assertNotNull = context.testFramework.assertNotNull
     val assertTrue = context.testFramework.assertTrue
     val assertFalse = context.testFramework.assertFalse