ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-03-22 12:05:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

ENT-6092 Add flush to MockServices.withEntityManager (#4180)

Browse Source
This commit is contained in:
Dan Newton 2021-01-29 15:29:33 +00:00 committed by Dan Newton
parent d214f5ecbf
commit 682de39d4c
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 <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>) {
return block.accept(contextTransaction.entityManager)
return contextTransaction.entityManager.run {
block.accept(this).also { flush () }
}
}
}
}