ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2024-12-24 07:06:44 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Made NodeUnloadHandlerTests a unit test by using mock node (#3572)

Browse Source
This commit is contained in:
Shams Asari 2018-07-12 15:22:31 +01:00 committed by GitHub
parent 7f42a9b48c
commit ab08ce21f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
node/src/integration-test/kotlin/net/corda/node/NodeUnloadHandlerTests.kt → node/src/test/kotlin/net/corda/node/internal/NodeUnloadHandlerTests.kt
View File

@ -1,13 +1,12 @@
package net.corda.node
package net.corda.node.internal
import net.corda.core.internal.packageName
import net.corda.core.node.ServiceHub
import net.corda.core.node.services.CordaService
import net.corda.core.serialization.SingletonSerializeAsToken
import net.corda.core.utilities.contextLogger
import net.corda.core.utilities.getOrThrow
import net.corda.testing.core.DUMMY_BANK_A_NAME
import net.corda.testing.driver.DriverParameters
import net.corda.testing.driver.driver
import net.corda.testing.node.internal.InternalMockNetwork
import org.junit.After
import org.junit.Assert.assertTrue
import org.junit.Test
import java.util.concurrent.CountDownLatch
@ -15,16 +14,23 @@ import java.util.concurrent.TimeUnit
class NodeUnloadHandlerTests {
companion object {
val latch = CountDownLatch(1)
val registerLatch = CountDownLatch(1)
val shutdownLatch = CountDownLatch(1)
}
private val mockNet = InternalMockNetwork(cordappPackages = listOf(javaClass.packageName), notarySpecs = emptyList())
@After
fun cleanUp() {
mockNet.stopNodes()
}
@Test
fun `should be able to register run on stop lambda`() {
driver(DriverParameters(startNodesInProcess = true, extraCordappPackagesToScan = listOf("net.corda.node"), notarySpecs = emptyList())) {
startNode(providedName = DUMMY_BANK_A_NAME).getOrThrow()
// just want to fall off the end of this for the mo...
}
assertTrue("Timed out waiting for AbstractNode to invoke the test service shutdown callback", latch.await(30, TimeUnit.SECONDS))
val node = mockNet.createNode()
registerLatch.await() // Make sure the handler is registered on node start up
node.dispose()
assertTrue("Timed out waiting for AbstractNode to invoke the test service shutdown callback", shutdownLatch.await(30, TimeUnit.SECONDS))
}
@Suppress("unused")
@ -36,11 +42,12 @@ class NodeUnloadHandlerTests {
init {
serviceHub.registerUnloadHandler(this::shutdown)
registerLatch.countDown()
}
private fun shutdown() {
log.info("shutting down")
latch.countDown()
shutdownLatch.countDown()
}
}
}