ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-04-27 06:19:54 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge pull request #6852 from corda/dan/ENT-6092-port-to-os

Browse Source 
ENT-6092 Add flush to `MockServices.withEntityManager` (#4180)
This commit is contained in:
Rick Parker 2021-01-29 17:06:12 +00:00 committed by GitHub
commit ade302be73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
testing/node-driver/src/main/kotlin/net/corda/testing/node/MockServices.kt
View File

@ -251,11 +251,15 @@ open class MockServices private constructor(
override fun jdbcSession(): Connection = persistence.createSession() override fun jdbcSession(): Connection = persistence.createSession()
override fun <T : Any?> withEntityManager(block: EntityManager.() -> T): T { override fun <T : Any?> withEntityManager(block: EntityManager.() -> T): T {
return block(contextTransaction.entityManager) return contextTransaction.entityManager.run {
block(this).also { flush () }
}
} }
override fun withEntityManager(block: Consumer<EntityManager>) { override fun withEntityManager(block: Consumer<EntityManager>) {
return block.accept(contextTransaction.entityManager) return contextTransaction.entityManager.run {
block.accept(this).also { flush () }
}
} }
} }
} }