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

CORDA-2503: Added installCordaService to UnstartedMockNode to allow tests with custom test-only services (#4655)

Browse Source
This commit is contained in:
Shams Asari
2019-01-28 13:49:01 +00:00
committed by GitHub
parent 631f04e348
commit fc7428a40d
4 changed files with 60 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
testing/node-driver/src/test/kotlin/net/corda/testing/node/MockNetworkTest.kt
View File

@ -1,5 +1,16 @@
package net.corda.testing.node
import co.paralleluniverse.fibers.Suspendable
import net.corda.core.flows.FlowLogic
import net.corda.core.flows.FlowSession
import net.corda.core.flows.InitiatedBy
import net.corda.core.flows.InitiatingFlow
import net.corda.core.identity.Party
import net.corda.core.node.AppServiceHub
import net.corda.core.node.services.CordaService
import net.corda.core.serialization.SingletonSerializeAsToken
import net.corda.core.utilities.getOrThrow
import net.corda.core.utilities.unwrap
import net.corda.testing.core.*
import org.assertj.core.api.Assertions.*
import org.junit.After
@ -44,4 +55,34 @@ class MockNetworkTest {
val ex = assertFailsWith<IllegalStateException> { unstarted.started }
assertThat(ex).hasMessage("Node ID=$NODE_ID is not running")
}
@Test
fun installCordaService() {
val unstarted = mockNetwork.createUnstartedNode()
assertThat(unstarted.installCordaService(TestService::class.java)).isNotNull()
val started = unstarted.start()
started.registerInitiatedFlow(TestResponder::class.java)
val future = started.startFlow(TestInitiator(started.info.singleIdentity()))
mockNetwork.runNetwork()
assertThat(future.getOrThrow()).isEqualTo(TestService::class.java.name)
}
@CordaService
class TestService(services: AppServiceHub) : SingletonSerializeAsToken()
@InitiatingFlow
class TestInitiator(private val party: Party) : FlowLogic<String>() {
@Suspendable
override fun call(): String {
return initiateFlow(party).receive<String>().unwrap { it }
}
}
@InitiatedBy(TestInitiator::class)
class TestResponder(private val otherSide: FlowSession) : FlowLogic<Unit>() {
@Suspendable
override fun call() {
otherSide.send(serviceHub.cordaService(TestService::class.java).javaClass.name)
}
}
}