mirror of
https://github.com/corda/corda.git
synced 2025-06-22 17:09:00 +00:00
CORDA-716 Make asContextEnv available to smoketesting (#2039)
This commit is contained in:
@ -8,6 +8,7 @@ import net.corda.node.serialization.KryoServerSerializationScheme
|
||||
import net.corda.nodeapi.internal.serialization.*
|
||||
import net.corda.nodeapi.internal.serialization.amqp.AMQPClientSerializationScheme
|
||||
import net.corda.nodeapi.internal.serialization.amqp.AMQPServerSerializationScheme
|
||||
import net.corda.testing.common.internal.asContextEnv
|
||||
import org.junit.rules.TestRule
|
||||
import org.junit.runner.Description
|
||||
import org.junit.runners.model.Statement
|
||||
@ -32,16 +33,6 @@ fun <T> withTestSerialization(inheritable: Boolean = false, callable: (Serializa
|
||||
return createTestSerializationEnv().asContextEnv(inheritable, callable)
|
||||
}
|
||||
|
||||
private fun <T> SerializationEnvironment.asContextEnv(inheritable: Boolean, callable: (SerializationEnvironment) -> T): T {
|
||||
val property = if (inheritable) _inheritableContextSerializationEnv else _contextSerializationEnv
|
||||
property.set(this)
|
||||
try {
|
||||
return callable(this)
|
||||
} finally {
|
||||
property.set(null)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* For example your test class uses [SerializationEnvironmentRule] but you want to turn it off for one method.
|
||||
* Use sparingly, ideally a test class shouldn't mix serialization init mechanisms.
|
||||
|
Reference in New Issue
Block a user