Avoid fully qualified names rendering in mockedConstruction #2175 (#2176)

Author: EgorkaKulikov

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/services/framework/MockFrameworkManager.kt

@@ -48,6 +48,7 @@ import org.utbot.framework.codegen.tree.CgStatementConstructor
 import org.utbot.framework.codegen.tree.CgStatementConstructorImpl
 import org.utbot.framework.codegen.tree.CgVariableConstructor
 import org.utbot.framework.codegen.tree.hasAmbiguousOverloadsOf
+import org.utbot.framework.codegen.tree.importIfNeeded
 import org.utbot.framework.codegen.util.isAccessibleFrom
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.ConstructorId
@@ -287,6 +288,10 @@ private class MockitoStaticMocker(context: CgContext, private val mocker: Object
             nameGenerator.variableName(MOCKED_CONSTRUCTION_NAME),
             mockConstructionInitializer.mockConstructionCall
         )
+
+        importIfNeeded(MockitoStaticMocking.mockedConstructionClassId)
+        importIfNeeded(classId)
+
         resources += mockedConstructionDeclaration
         +CgAssignment(mockedConstructionDeclaration.variable, mockConstructionInitializer.mockConstructionCall)
         mockedStaticConstructions += classId
@@ -351,6 +356,9 @@ private class MockitoStaticMocker(context: CgContext, private val mocker: Object
             ).also {
                 resources += it
                 +CgAssignment(it.variable, classMockStaticCall)
+
+                importIfNeeded(MockitoStaticMocking.mockedStaticClassId)
+                importIfNeeded(classId)
             }
         }
 
@@ -367,10 +375,6 @@ private class MockitoStaticMocker(context: CgContext, private val mocker: Object
                 isMock = true
             )
         )
-        val contextParameter = variableConstructor.declareParameter(
-            mockedConstructionContextClassId,
-            nameGenerator.variableName("context")
-        )
 
         val mockAnswerStatements = mutableMapOf<Int, List<CgStatement>>()
 
@@ -425,12 +429,20 @@ private class MockitoStaticMocker(context: CgContext, private val mocker: Object
             listOf(switchCase, CgStatementExecutableCall(mockClassCounter[atomicIntegerGetAndIncrement]()))
         }
 
+        val contextParameter = variableConstructor.declareParameter(
+            mockedConstructionContextClassId,
+            nameGenerator.variableName("context")
+        )
+
         val answersBlock = CgAnonymousFunction(
             voidClassId,
             listOf(mockParameter, contextParameter).map { CgParameterDeclaration(it, isVararg = false) },
             mockConstructionBody
         )
 
+        importIfNeeded(mockedConstructionContextClassId)
+        importIfNeeded(classId)
+
         return MockConstructionBlock(
             mockitoClassId[MockitoStaticMocking.mockConstructionMethodId](clazz, answersBlock),
             !atMostOneBranchOrAllEmpty