def test_trim_tree(self):
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hash3 = UInt256(data=binascii.unhexlify(b'cc' * 32))
hashes = [hash1, hash2, hash3]
m = MerkleTree(hashes)
flags = bytearray.fromhex('0000')
m.Trim(flags)
self.assertEqual(None, m.Root.LeftChild)
self.assertEqual(None, m.Root.RightChild)
def test_trim_node3(self):
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hashes = [hash1, hash2]
m = MerkleTree(hashes)
depth = 1
flags = bytearray.fromhex('0000')
m._TrimNode(m.Root, 1, depth, flags)
self.assertNotEqual(None, m.Root.LeftChild)
self.assertNotEqual(None, m.Root.RightChild)
def test_to_hash_array(self):
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hash3 = UInt256(data=binascii.unhexlify(b'cc' * 32))
hash4 = UInt256(data=binascii.unhexlify(b'dd' * 32))
hash5 = UInt256(data=binascii.unhexlify(b'ee' * 32))
hashes = [hash1, hash2, hash3, hash4, hash5]
m = MerkleTree(hashes)
hash_array = m.ToHashArray()
for i, h in enumerate(hashes):
self.assertEqual(h.ToBytes(), hash_array[i].ToBytes())
def test_trim_node1(self):
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hash3 = UInt256(data=binascii.unhexlify(b'cc' * 32))
hashes = [hash1, hash2, hash3]
m = MerkleTree(hashes)
depth = 2
flags = bytearray.fromhex(
'11110000'
) # 1 byte left node , 1 byte right node 00=delete, non-00 is keep
m._TrimNode(m.Root, 1, depth, flags)
self.assertEqual(None, m.Root.LeftChild)
self.assertEqual(None, m.Root.RightChild)
def test_block_deserialize(self):
block = Helper.AsSerializableWithType(self.rawblock_hex,
'neo.Core.Block.Block')
self.assertEqual(self.rb_prev, block.PrevHash.ToBytes())
self.assertEqual(self.rb_merlke, block.MerkleRoot.ToBytes())
self.assertEqual(self.rb_ts, block.Timestamp)
self.assertEqual(self.rb_h, block.Index)
self.assertEqual(self.rb_nonce, block.ConsensusData)
self.assertEqual(self.rconsenusdata, block.ConsensusData)
# self.assertEqual(self.rb_hash, block.HashToString())
tx = block.Transactions[0]
self.assertEqual(tx.Nonce, self.rblock_tx_nonce)
self.assertEqual(len(tx.inputs), 0)
self.assertEqual(len(tx.outputs), 0)
self.assertEqual(len(tx.Attributes), 0)
self.assertEqual(type(tx.scripts), list)
rawdata = block.RawData()
compair_data = self.rawblock[:len(rawdata)]
self.assertEqual(rawdata, compair_data)
out = block.Hash.ToBytes()
self.assertEqual(out, self.rb_hash)
root = MerkleTree.ComputeRoot([tx.Hash for tx in block.Transactions])
self.assertEqual(root, block.MerkleRoot)
def test_compute_root_multiple_hashes(self):
expected_hash = Helper.bin_dbl_sha256(
binascii.unhexlify(b'aa' * 32 + b'bb' * 32))
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hashes = [hash1, hash2]
root = MerkleTree.ComputeRoot(hashes)
self.assertEqual(expected_hash, root.ToArray())
def test_trim_node2(self):
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hash3 = UInt256(data=binascii.unhexlify(b'cc' * 32))
hash4 = UInt256(data=binascii.unhexlify(b'dd' * 32))
hash5 = UInt256(data=binascii.unhexlify(b'ee' * 32))
hash6 = UInt256(data=binascii.unhexlify(b'11' * 32))
hash7 = UInt256(data=binascii.unhexlify(b'22' * 32))
hash8 = UInt256(data=binascii.unhexlify(b'33' * 32))
hash9 = UInt256(data=binascii.unhexlify(b'44' * 32))
hashes = [
hash1, hash2, hash3, hash4, hash5, hash6, hash7, hash8, hash9
]
m = MerkleTree(hashes)
depth = 3
flags = bytearray.fromhex('111100000000')
m._TrimNode(m.Root, 1, depth, flags)
self.assertEqual(None, m.Root.LeftChild)
self.assertEqual(None, m.Root.RightChild)
def test_node_methods(self):
hash1 = UInt256(data=binascii.unhexlify(b'aa' * 32))
hash2 = UInt256(data=binascii.unhexlify(b'bb' * 32))
hashes = [hash1, hash2]
m = MerkleTree(hashes)
self.assertEqual(True, m.Root.IsRoot())
self.assertEqual(False, m.Root.IsLeaf())
self.assertEqual(False, m.Root.LeftChild.IsRoot())
self.assertEqual(True, m.Root.LeftChild.IsLeaf())
# I have no acceptable test vector
m.Root.LeftChild.Size()
def Deserialize(self, reader):
"""
Deserialize full object.
Args:
reader (neo.IO.BinaryReader):
"""
super(Block, self).Deserialize(reader)
self.Transactions = []
byt = reader.ReadVarInt()
transaction_length = byt
if transaction_length < 1:
raise Exception('Invalid format')
for i in range(0, transaction_length):
tx = Transaction.DeserializeFrom(reader)
self.Transactions.append(tx)
if MerkleTree.ComputeRoot([tx.Hash for tx in self.Transactions]) != self.MerkleRoot:
raise Exception("Merkle Root Mismatch")
def test_block_two(self):
hexdata = binascii.unhexlify(self.b2raw)
block = Helper.AsSerializableWithType(hexdata, 'neo.Core.Block.Block')
self.assertEqual(block.Index, self.b2height)
self.assertEqual(block.ConsensusData, self.b2nonce)
self.assertEqual(block.Timestamp, self.b2timestamp)
self.assertEqual(block.PrevHash.ToBytes(), self.b2prev_hash)
self.assertEqual(block.Hash.ToString(), self.b2hash)
next_consensus_address = Crypto.ToAddress(block.NextConsensus)
self.assertEqual(next_consensus_address, self.b2nextconsensus)
witness = block.Script
ins = binascii.hexlify(witness.InvocationScript)
vns = binascii.hexlify(witness.VerificationScript)
self.assertEqual(ins, self.b2invocation)
self.assertEqual(vns, self.b2verification)
self.assertEqual(len(block.Transactions), self.b2tx_len)
tx = block.Transactions[0]
self.assertEqual(tx.inputs, self.b2tx_vin)
self.assertEqual(tx.outputs, self.b2tx_vout)
self.assertEqual(tx.Nonce, self.b2tx_nonce)
txhash = tx.Hash.ToBytes()
self.assertEqual(txhash, self.b2tx_id)
root = MerkleTree.ComputeRoot([tx.Hash for tx in block.Transactions])
self.assertEqual(root, block.MerkleRoot)
def RebuildMerkleRoot(self):
"""Rebuild the merkle root of the block"""
logger.debug("Rebuilding merkle root!")
if self.Transactions is not None and len(self.Transactions) > 0:
self.MerkleRoot = MerkleTree.ComputeRoot(
[tx.Hash for tx in self.Transactions])
def test_build_no_leaves(self):
with self.assertRaises(Exception) as context:
MerkleTree([]).__Build([])
self.assertTrue("Leaves must have length" in str(context.exception))
def test_computer_root_no_input(self):
with self.assertRaises(Exception) as context:
MerkleTree.ComputeRoot([])
self.assertTrue("Hashes must have length" in str(context.exception))
def test_compute_root_single_hash(self):
data = binascii.unhexlify(b'aa' * 32)
hash1 = UInt256(data=data)
root = MerkleTree.ComputeRoot([hash1])
self.assertEqual(data, root.ToArray())
