Extract mock method

Author: CaelmBleidd

usvm-ts/src/main/kotlin/org/usvm/machine/interpreter/TsInterpreter.kt

@@ -21,6 +21,7 @@ import org.jacodb.ets.base.EtsThrowStmt
 import org.jacodb.ets.base.EtsType
 import org.jacodb.ets.base.EtsValue
 import org.jacodb.ets.model.EtsMethod
+import org.jacodb.ets.model.EtsMethodSignature
 import org.jacodb.ets.utils.callExpr
 import org.jacodb.ets.utils.getDeclaredLocals
 import org.usvm.StepResult
@@ -164,25 +165,7 @@ class TsInterpreter(
             }
 
             if (!tsOptions.interproceduralAnalysis && methodResult == TsMethodResult.NoCall) {
-                scope.doWithState {
-                    with(ctx) {
-                        val resultSort = typeToSort(it.method.returnType)
-                        val resultValue = when (resultSort) {
-                            is UAddressSort -> makeSymbolicRefUntyped()
-                            is TsUnresolvedSort -> {
-                                mkFakeValue(
-                                    scope,
-                                    makeSymbolicPrimitive(ctx.boolSort),
-                                    makeSymbolicPrimitive(ctx.fp64Sort),
-                                    makeSymbolicRefUntyped()
-                                )
-                            }
-
-                            else -> makeSymbolicPrimitive(resultSort)
-                        }
-                        this@doWithState.methodResult = TsMethodResult.Success.MockedCall(it.method, resultValue)
-                    }
-                }
+                mockMethodCall(scope, it.method)
                 return
             }
 
@@ -305,25 +288,7 @@ class TsInterpreter(
         }
 
         // intraprocedural analysis
-        scope.doWithState {
-            with(ctx) {
-                val resultSort = typeToSort(stmt.expr.method.returnType)
-                val resultValue = when (resultSort) {
-                    is UAddressSort -> makeSymbolicRefUntyped()
-                    is TsUnresolvedSort -> {
-                        mkFakeValue(
-                            scope,
-                            makeSymbolicPrimitive(ctx.boolSort),
-                            makeSymbolicPrimitive(ctx.fp64Sort),
-                            makeSymbolicRefUntyped()
-                        )
-                    }
-
-                    else -> makeSymbolicPrimitive(resultSort)
-                }
-                methodResult = TsMethodResult.Success.MockedCall(stmt.expr.method, resultValue)
-            }
-        }
+        mockMethodCall(scope, stmt.method.signature)
     }
 
     private fun visitThrowStmt(scope: TsStepScope, stmt: EtsThrowStmt) {
@@ -417,6 +382,28 @@ class TsInterpreter(
         return state
     }
 
+    private fun mockMethodCall(scope: TsStepScope, method: EtsMethodSignature) {
+        scope.doWithState {
+            with(ctx) {
+                val resultSort = typeToSort(method.returnType)
+                val resultValue = when (resultSort) {
+                    is UAddressSort -> makeSymbolicRefUntyped()
+                    is TsUnresolvedSort -> {
+                        mkFakeValue(
+                            scope,
+                            makeSymbolicPrimitive(ctx.boolSort),
+                            makeSymbolicPrimitive(ctx.fp64Sort),
+                            makeSymbolicRefUntyped()
+                        )
+                    }
+
+                    else -> makeSymbolicPrimitive(resultSort)
+                }
+                methodResult = TsMethodResult.Success.MockedCall(method, resultValue)
+            }
+        }
+    }
+
     // TODO: expand with interpreter implementation
     private val EtsStmt.nextStmt: EtsStmt?
         get() = applicationGraph.successors(this).firstOrNull()