ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-02-18 16:40:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add simple partial tree building and verification tests (also on transactions).

Browse Source
This commit is contained in:
Katarzyna Streich 2016-10-11 18:09:12 +01:00 committed by Katarzyna Streich
parent 930f6bbb24
commit 0c573c76e2
1 changed files with 79 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

116
core/src/test/kotlin/com/r3corda/core/crypto/PartialMerkleTreeTest.kt
View File

@ -1,32 +1,36 @@
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.`with notary`
import com.r3corda.core.serialization.serialize
import com.r3corda.core.transactions.MerkleTransaction
import com.r3corda.core.transactions.hashConcat
import com.r3corda.core.utilities.DUMMY_NOTARY
import com.r3corda.testing.ALICE_PUBKEY
import org.junit.Test
import java.util.*
import kotlin.test.assertEquals
import kotlin.test.assertFailsWith
import kotlin.test.assertFalse
//todo
//transaction tests
//different filtering fun
//verification - different root
//tests failsWith
//verification tests
//ordering insensitive tests
//filter commands tests
class PartialMerkleTreeTest{
val nodes = "abcdef"
val hashed: MutableList<SecureHash> = ArrayList()
val includeBranch: MutableList<Boolean> = ArrayList()
val branchHashes: MutableList<SecureHash> = ArrayList()
val root = SecureHash.Companion.parse("F6D8FB3720114F8D040D64F633B0D9178EB09A55AA7D62FAE1A070D1BF561051")
var firstTwo:SecureHash
init {
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)
@Test
fun `building not full Merkle tree with 6 nodes`(){
@ -44,54 +48,92 @@ class PartialMerkleTreeTest{
val mtl = mutableListOf<SecureHash>(node)
MerkleTransaction.Companion.buildMerkleTree(mtl, listOf(node))
assertEquals(1, mtl.size)
val root = SecureHash.Companion.parse("08CB04523D895A51CDB818655668E7D6809A1FB8A8861845125FAAA86608F607")
assertEquals(root, mtl.last())
assertEquals(node, mtl.last())
}
//todo with tx
// //create wire transaction
// @Test
// fun `building Merkle tree for a transaction`(){
//
// }
//
//Partial Merkle Tree tests
@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 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())
}
@Test
fun `ordering insensitive tree`(){
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)
}
//Partial Merkle Tree building tests
@Test
fun `Partial Merkle Tree, only left nodes branch`(){
val includeLeaves = listOf(false, false, false, true, false, true)
PartialMerkleTree.Companion.whichNodesInBranch(3, 0, includeLeaves, hashed, includeBranch, branchHashes)
assert(!includeBranch[2] && !includeBranch[4]&& !includeBranch[8])
assertEquals(5, branchHashes.size)
assertEquals(10, includeBranch.size)
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)
}
@Test
fun `Partial Merkle Tree, include zero leaves`(){
val includeLeaves = listOf(false, false, false, false, false, false)
PartialMerkleTree.Companion.whichNodesInBranch(3, 0, includeLeaves, hashed, includeBranch, branchHashes)
assertEquals(1, branchHashes.size)
assertEquals(1, includeBranch.size)
assertEquals(root, branchHashes[0])
val pmt = PartialMerkleTree.build(includeLeaves, hashed)
assertEquals(1, pmt.branchHashes.size)
assertEquals(1, pmt.includeBranch.size)
assertEquals(root, pmt.branchHashes[0])
}
@Test
fun `Partial Merkle Tree, include all leaves`(){
val includeLeaves = listOf(true, true, true, true, true, true)
PartialMerkleTree.Companion.whichNodesInBranch(3, 0, includeLeaves, hashed, includeBranch, branchHashes)
assertEquals(6, branchHashes.size)
assertEquals(12, includeBranch.size)
assertEquals(branchHashes, hashed)
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))
}
@Test
fun `tree hash root`(){
val s = PartialMerkleTree.Companion.treeHash(0, 3, hashed)
assertEquals(root, s)
}
@Test
fun `tree hash first two`(){
val s = PartialMerkleTree.Companion.treeHash(0, 1, hashed)
assertEquals(firstTwo, s)
fun `verify Partial Merkle Tree - 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)
assertFalse(pmt.verify(inclHashes, root))
}
}