mirror of
https://github.com/corda/corda.git
synced 2025-02-18 16:40:55 +00:00
Change tests to reflect changes in code structure.
This commit is contained in:
Katarzyna Streich
parent
e823e11e85
commit
b31598b801
@ -3,12 +3,17 @@ package com.r3corda.core.crypto
|
||||
import com.r3corda.contracts.asset.*
|
||||
import com.r3corda.core.contracts.DOLLARS
|
||||
import com.r3corda.core.contracts.TransactionType
|
||||
import com.r3corda.core.contracts.`issued by`
|
||||
import com.r3corda.core.contracts.`with notary`
|
||||
import com.r3corda.core.serialization.serialize
|
||||
import com.r3corda.core.transactions.MerkleTransaction
|
||||
import com.r3corda.core.transactions.FilterFuns
|
||||
import com.r3corda.core.transactions.buildFilteredTransaction
|
||||
import com.r3corda.core.transactions.getMerkleRoot
|
||||
import com.r3corda.core.transactions.hashConcat
|
||||
import com.r3corda.core.utilities.DUMMY_NOTARY
|
||||
import com.r3corda.testing.ALICE_PUBKEY
|
||||
import com.r3corda.core.utilities.DUMMY_PUBKEY_1
|
||||
import com.r3corda.testing.*
|
||||
|
||||
|
||||
import org.junit.Test
|
||||
import java.util.*
|
||||
@ -17,7 +22,6 @@ import kotlin.test.assertFailsWith
|
||||
import kotlin.test.assertFalse
|
||||
|
||||
//todo
|
||||
//different filtering fun
|
||||
//verification - different root
|
||||
//tests failsWith
|
||||
|
||||
@ -30,107 +34,108 @@ class PartialMerkleTreeTest{
|
||||
nodes.mapTo(hashed, { it.serialize().sha256() })
|
||||
firstTwo = hashed[0].hashConcat(hashed[1])
|
||||
}
|
||||
private fun makeTX() = TransactionType.General.Builder(DUMMY_NOTARY).withItems(1000.DOLLARS.CASH `issued by` DUMMY_CASH_ISSUER `owned by` ALICE_PUBKEY `with notary` DUMMY_NOTARY)
|
||||
private fun makeTX() = TransactionType.General.Builder(DUMMY_NOTARY).withItems(
|
||||
1000.DOLLARS.CASH `issued by` DUMMY_CASH_ISSUER `owned by` ALICE_PUBKEY `with notary` DUMMY_NOTARY)
|
||||
|
||||
val testLedger = ledger {
|
||||
unverifiedTransaction {
|
||||
output("MEGA_CORP cash") {
|
||||
Cash.State(
|
||||
amount = 1000.DOLLARS `issued by` MEGA_CORP.ref(1, 1),
|
||||
owner = MEGA_CORP_PUBKEY
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
transaction {
|
||||
input("MEGA_CORP cash")
|
||||
output("MEGA_CORP cash".output<Cash.State>().copy(owner = DUMMY_PUBKEY_1))
|
||||
command(MEGA_CORP_PUBKEY) { Cash.Commands.Move() }
|
||||
this.verifies()
|
||||
}
|
||||
}
|
||||
|
||||
val testTx = testLedger.interpreter.transactionsToVerify[0]
|
||||
|
||||
//Building full Merkle Tree tests.
|
||||
@Test
|
||||
fun `building not full Merkle tree with 6 nodes`(){
|
||||
assertEquals(6, hashed.size)
|
||||
val mtl = ArrayList<SecureHash>()
|
||||
mtl.addAll(hashed)
|
||||
MerkleTransaction.Companion.buildMerkleTree(mtl, hashed)
|
||||
assertEquals(12, mtl.size)
|
||||
assertEquals(root, mtl.last())
|
||||
val mr = getMerkleRoot(hashed)
|
||||
assertEquals(root, mr)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `building Merkle tree one node`(){
|
||||
val node = 'a'.serialize().sha256()
|
||||
val mtl = mutableListOf<SecureHash>(node)
|
||||
MerkleTransaction.Companion.buildMerkleTree(mtl, listOf(node))
|
||||
assertEquals(1, mtl.size)
|
||||
assertEquals(node, mtl.last())
|
||||
val mr = getMerkleRoot(listOf(node))
|
||||
assertEquals(node, mr)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `building Merkle tree odd number of nodes`(){
|
||||
val odd = hashed.subList(0, 3)
|
||||
val mtl = ArrayList<SecureHash>()
|
||||
val h1 = hashed[0].hashConcat(hashed[1])
|
||||
val h2 = hashed[2].hashConcat(hashed[2])
|
||||
mtl.addAll(odd)
|
||||
MerkleTransaction.Companion.buildMerkleTree(mtl, odd)
|
||||
assertEquals(6, mtl.size)
|
||||
assertEquals(h1, mtl[3])
|
||||
assertEquals(h2, mtl[4])
|
||||
}
|
||||
// @Test
|
||||
// fun `building Merkle tree odd number of nodes`(){
|
||||
// val odd = hashed.subList(0, 3)
|
||||
// val h1 = hashed[0].hashConcat(hashed[1])
|
||||
// val h2 = hashed[2].hashConcat(hashed[2])
|
||||
// val mtl = MerkleTransaction.Companion.buildMerkleTree(odd)
|
||||
// assertEquals(6, mtl.size)
|
||||
// assertEquals(h1, mtl[3])
|
||||
// assertEquals(h2, mtl[4])
|
||||
// }
|
||||
|
||||
@Test
|
||||
fun `building Merkle tree for a transaction`(){
|
||||
val tx = makeTX()
|
||||
tx.addCommand(Cash.Commands.Move(), ALICE_PUBKEY)
|
||||
tx.addCommand(Cash.Commands.Issue(), ALICE_PUBKEY)
|
||||
tx.addCommand(Cash.Commands.Issue(0), ALICE_PUBKEY)
|
||||
tx.addCommand(Cash.Commands.Issue(1), ALICE_PUBKEY)
|
||||
val wtx = tx.toWireTransaction()
|
||||
val mt = MerkleTransaction.buildMerkleTransaction(wtx, {x -> true} )
|
||||
assert(mt.verify())
|
||||
val filterFuns = FilterFuns(
|
||||
filterCommands = { x -> ALICE_PUBKEY in x.signers},
|
||||
filterOutputs = {true},
|
||||
filterInputs = {true})
|
||||
val mt = testTx.buildFilteredTransaction(filterFuns)
|
||||
assert(mt.verify(testTx.id))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `ordering insensitive tree`(){
|
||||
val filterFuns = FilterFuns(filterCommands = { x -> true })
|
||||
val tx1 = makeTX()
|
||||
tx1.addCommand(Cash.Commands.Issue(1), ALICE_PUBKEY)
|
||||
tx1.addCommand(Cash.Commands.Issue(0), ALICE_PUBKEY)
|
||||
val wtx1 = tx1.toWireTransaction()
|
||||
val mt1 = MerkleTransaction.buildMerkleTransaction(wtx1, {x -> true} )
|
||||
|
||||
val tx2 = makeTX()
|
||||
tx2.addCommand(Cash.Commands.Issue(0), ALICE_PUBKEY)
|
||||
tx2.addCommand(Cash.Commands.Issue(1), ALICE_PUBKEY)
|
||||
val wtx2 = tx2.toWireTransaction()
|
||||
val mt2 = MerkleTransaction.buildMerkleTransaction(wtx2, {x -> true} )
|
||||
assert(mt1.merkleRoot == mt2.merkleRoot)
|
||||
assertEquals(wtx1.id, wtx2.id)
|
||||
}
|
||||
|
||||
//Partial Merkle Tree building tests
|
||||
@Test
|
||||
fun `Partial Merkle Tree, only left nodes branch`(){
|
||||
val includeLeaves = listOf(false, false, false, true, false, true)
|
||||
val inclHashes = listOf(hashed[3], hashed[5])
|
||||
val pmt = PartialMerkleTree.build(includeLeaves, hashed)
|
||||
assert(!pmt.includeBranch[2] && !pmt.includeBranch[4]&& !pmt.includeBranch[8])
|
||||
assertEquals(5, pmt.branchHashes.size)
|
||||
assertEquals(10, pmt.includeBranch.size)
|
||||
assert(pmt.verify(inclHashes, root))
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `Partial Merkle Tree, include zero leaves`(){
|
||||
val includeLeaves = listOf(false, false, false, false, false, false)
|
||||
val pmt = PartialMerkleTree.build(includeLeaves, hashed)
|
||||
assertEquals(1, pmt.branchHashes.size)
|
||||
assertEquals(1, pmt.includeBranch.size)
|
||||
assertEquals(root, pmt.branchHashes[0])
|
||||
assert(pmt.verify(emptyList(), root))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `Partial Merkle Tree, include all leaves`(){
|
||||
val includeLeaves = listOf(true, true, true, true, true, true)
|
||||
val pmt = PartialMerkleTree.build(includeLeaves, hashed)
|
||||
assertEquals(6, pmt.branchHashes.size)
|
||||
assertEquals(12, pmt.includeBranch.size)
|
||||
assertEquals(pmt.branchHashes, hashed)
|
||||
}
|
||||
|
||||
//Verification
|
||||
@Test
|
||||
fun `verify Partial Merkle Tree on a simple example`(){
|
||||
val includeLeaves = listOf(false, false, false, true, false, true)
|
||||
val inclHashes = listOf(hashed[3], hashed[5])
|
||||
val pmt = PartialMerkleTree.build(includeLeaves, hashed)
|
||||
assert(pmt.verify(inclHashes, root))
|
||||
assert(pmt.verify(hashed, root))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `verify Partial Merkle Tree - failure`(){
|
||||
fun `verify Partial Merkle Tree - too many leaves failure`(){
|
||||
val includeLeaves = listOf(false, false, false, true, false, true)
|
||||
val inclHashes = listOf(hashed[3], hashed[5], hashed[0])
|
||||
val pmt = PartialMerkleTree.build(includeLeaves, hashed)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user