Generate mockito with privileges

Markoutte · Markoutte · commit 8866b984192d · 2022-08-09T09:10:53.000+03:00
diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/concrete/MockValueConstructor.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/concrete/MockValueConstructor.kt
@@ -45,6 +45,8 @@ import org.mockito.Mockito
 import org.mockito.stubbing.Answer
 import org.objectweb.asm.Type
 import org.utbot.common.withAccessibility
+import java.security.AccessController
+import java.security.PrivilegedAction
 
 /**
  * Constructs values (including mocks) from models.
@@ -224,7 +226,7 @@ class MockValueConstructor(
     }
 
     private fun generateMockitoMock(clazz: Class<*>, mocks: Map<ExecutableId, List<UtModel>>): Any {
-        return Mockito.mock(clazz, generateMockitoAnswer(mocks))
+        return AccessController.doPrivileged(PrivilegedAction { Mockito.mock(clazz, generateMockitoAnswer(mocks)) })
     }
 
     private fun computeConcreteValuesForMethods(

Original file line number	Diff line number	Diff line change
`@@ -45,6 +45,8 @@ import org.mockito.Mockito`
`45`	`45`	`import org.mockito.stubbing.Answer`
`46`	`46`	`import org.objectweb.asm.Type`
`47`	`47`	`import org.utbot.common.withAccessibility`
	`48`	`+import java.security.AccessController`
	`49`	`+import java.security.PrivilegedAction`
`48`	`50`
`49`	`51`	`/**`
`50`	`52`	`* Constructs values (including mocks) from models.`
`@@ -224,7 +226,7 @@ class MockValueConstructor(`
`224`	`226`	`}`
`225`	`227`
`226`	`228`	`private fun generateMockitoMock(clazz: Class<*>, mocks: Map<ExecutableId, List<UtModel>>): Any {`
`227`		`- return Mockito.mock(clazz, generateMockitoAnswer(mocks))`
	`229`	`+ return AccessController.doPrivileged(PrivilegedAction { Mockito.mock(clazz, generateMockitoAnswer(mocks)) })`
`228`	`230`	`}`
`229`	`231`
`230`	`232`	`private fun computeConcreteValuesForMethods(`