Re-add lost fix (#6652)

This commit is contained in:
Nikolett Nagy 2020-08-20 09:23:03 +01:00 committed by GitHub
parent a6b2a3159d
commit ab21833390
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -6,6 +6,7 @@ import net.corda.core.contracts.Contract
import net.corda.core.contracts.ContractState import net.corda.core.contracts.ContractState
import net.corda.core.contracts.StateRef import net.corda.core.contracts.StateRef
import net.corda.core.identity.AbstractParty import net.corda.core.identity.AbstractParty
import net.corda.core.node.ServiceHub
import net.corda.core.node.services.Vault import net.corda.core.node.services.Vault
import net.corda.core.node.services.queryBy import net.corda.core.node.services.queryBy
import net.corda.core.node.services.vault.DEFAULT_PAGE_SIZE import net.corda.core.node.services.vault.DEFAULT_PAGE_SIZE
@ -22,7 +23,9 @@ import net.corda.core.transactions.TransactionBuilder
import net.corda.testing.core.ALICE_NAME import net.corda.testing.core.ALICE_NAME
import net.corda.testing.node.MockNetwork import net.corda.testing.node.MockNetwork
import net.corda.testing.node.MockNetworkParameters import net.corda.testing.node.MockNetworkParameters
import net.corda.testing.node.StartedMockNode
import net.corda.testing.node.internal.cordappsForPackages import net.corda.testing.node.internal.cordappsForPackages
import org.junit.AfterClass
import org.junit.BeforeClass import org.junit.BeforeClass
import org.junit.Test import org.junit.Test
import javax.persistence.Column import javax.persistence.Column
@ -33,24 +36,29 @@ import kotlin.test.assertEquals
class VaultQueryJoinTest { class VaultQueryJoinTest {
companion object { companion object {
private val mockNetwork = MockNetwork( private var mockNetwork: MockNetwork? = null
MockNetworkParameters( private lateinit var aliceNode: StartedMockNode
cordappsForAllNodes = cordappsForPackages( private lateinit var notaryNode: StartedMockNode
listOf( private lateinit var serviceHubHandle: ServiceHub
"net.corda.node.services.vault"
)
)
)
)
private val aliceNode = mockNetwork.createPartyNode(ALICE_NAME)
private val notaryNode = mockNetwork.defaultNotaryNode
private val serviceHubHandle = aliceNode.services
private val createdStateRefs = mutableListOf<StateRef>() private val createdStateRefs = mutableListOf<StateRef>()
private const val numObjectsInLedger = DEFAULT_PAGE_SIZE + 1 private const val numObjectsInLedger = DEFAULT_PAGE_SIZE + 1
@BeforeClass @BeforeClass
@JvmStatic @JvmStatic
fun setup() { fun setup() {
mockNetwork = MockNetwork(
MockNetworkParameters(
cordappsForAllNodes = cordappsForPackages(
listOf(
"net.corda.node.services.vault"
)
)
)
).also { mockNetwork ->
aliceNode = mockNetwork.createPartyNode(ALICE_NAME)
notaryNode = mockNetwork.defaultNotaryNode
serviceHubHandle = aliceNode.services
}
repeat(numObjectsInLedger) { index -> repeat(numObjectsInLedger) { index ->
createdStateRefs.add(addSimpleObjectToLedger(DummyData(index))) createdStateRefs.add(addSimpleObjectToLedger(DummyData(index)))
} }
@ -58,6 +66,13 @@ class VaultQueryJoinTest {
System.setProperty("net.corda.vault.query.disable.corda3879", "false"); System.setProperty("net.corda.vault.query.disable.corda3879", "false");
} }
@AfterClass
@JvmStatic
fun teardown() {
mockNetwork?.stopNodes()
mockNetwork = null
}
private fun addSimpleObjectToLedger(dummyObject: DummyData): StateRef { private fun addSimpleObjectToLedger(dummyObject: DummyData): StateRef {
val tx = TransactionBuilder(notaryNode.info.legalIdentities.first()) val tx = TransactionBuilder(notaryNode.info.legalIdentities.first())
tx.addOutputState( tx.addOutputState(