Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Improved JUnit test to correctly validate tokenizable behaviour.
  • Loading branch information
josecoll committed Jun 27, 2017
commit e2014b01dacdf6205ee28754fe3c5c271130c7d0
Original file line number Diff line number Diff line change
Expand Up @@ -598,8 +598,16 @@ class FlowFrameworkTests {
}

@Test
fun `access vault query service within a flow`() {
val result = node2.services.startFlow(VaultQueryFlow()).resultFuture
fun `verify vault query service is tokenizable by force checkpointing within a flow`() {
val ptx = TransactionBuilder(notary = notary1.info.notaryIdentity)
ptx.addOutputState(DummyState())
val stx = node1.services.signInitialTransaction(ptx)

node1.registerFlowFactory(VaultQueryFlow::class) {
WaitingFlows.Committer(it)
}.map { it.stateMachine }
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You don't need to do the map

val result = node2.services.startFlow(VaultQueryFlow(stx, node1.info.legalIdentity)).resultFuture

mockNet.runNetwork()
assertThat(result.getOrThrow()).isEmpty()
}
Expand Down Expand Up @@ -870,10 +878,16 @@ class FlowFrameworkTests {
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'd take this opportunity to delete this test

}

private class VaultQueryFlow : FlowLogic<List<StateAndRef<ContractState>>>() {
@InitiatingFlow
private class VaultQueryFlow(val stx: SignedTransaction, val otherParty: Party) : FlowLogic<List<StateAndRef<ContractState>>>() {
@Suspendable
override fun call(): List<StateAndRef<ContractState>> {
return serviceHub.vaultQueryService.queryBy<ContractState>().states
send(otherParty, stx)
// hold onto reference here to force checkpoint of vaultQueryService and thus
// prove it is registered as a tokenizableService in the node
val vaultQuerySvc = serviceHub.vaultQueryService
waitForLedgerCommit(stx.id)
return vaultQuerySvc.queryBy<ContractState>().states
}
}

Expand Down