hyperpy/merkle-tree-stream
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Move naming to Generator (reflect reality)

Browse Source 
Closes https://github.com/datpy/merkle-tree-stream/issues/4.
This commit is contained in:
Luke Murphy
2019-10-08 22:26:24 +02:00
parent 71526e64aa
commit a293d4bada
5 changed files with 44 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
test/test_tree.py
View File

@ -4,53 +4,53 @@ import hashlib
import pytest
from merkle_tree_stream import MerkleTreeIterator, MerkleTreeNode
from merkle_tree_stream import MerkleTreeGenerator, MerkleTreeNode
def test_hashes(leaf, parent):
merkle_iter = MerkleTreeIterator(leaf=leaf, parent=parent)
merkle = MerkleTreeGenerator(leaf=leaf, parent=parent)
merkle_iter.write(b'a')
merkle_iter.write(b'b')
merkle.write(b'a')
merkle.write(b'b')
expected_count = 2 + 1 # nodes plus parent
assert len(merkle_iter) == expected_count
assert len(merkle) == expected_count
assert next(merkle_iter) == MerkleTreeNode(
assert next(merkle) == MerkleTreeNode(
index=0, parent=1, hash=hashlib.sha256(b'a').digest(), size=1, data=b'a'
)
assert next(merkle_iter) == MerkleTreeNode(
assert next(merkle) == MerkleTreeNode(
index=2, parent=1, hash=hashlib.sha256(b'b').digest(), size=1, data=b'b'
)
hashed = hashlib.sha256(b'a')
hashed.update(b'b')
assert next(merkle_iter) == MerkleTreeNode(
assert next(merkle) == MerkleTreeNode(
index=1, parent=3, hash=hashed.digest(), size=2, data=b''
)
with pytest.raises(StopIteration):
next(merkle_iter)
next(merkle)
def test_single_root(leaf, parent):
merkle_iter = MerkleTreeIterator(leaf=leaf, parent=parent)
merkle = MerkleTreeGenerator(leaf=leaf, parent=parent)
merkle_iter.write(b'a')
merkle_iter.write(b'b')
merkle_iter.write(b'c')
merkle_iter.write(b'd')
merkle.write(b'a')
merkle.write(b'b')
merkle.write(b'c')
merkle.write(b'd')
assert len(merkle_iter.roots) == 1
assert len(merkle.roots) == 1
def multiple_roots(leaf, parent):
merkle_iter = MerkleTreeIterator(leaf=leaf, parent=parent)
merkle = MerkleTreeGenerator(leaf=leaf, parent=parent)
merkle_iter.write(b'a')
merkle_iter.write(b'b')
merkle_iter.write(b'c')
merkle.write(b'a')
merkle.write(b'b')
merkle.write(b'c')
assert len(merkle_iter.roots) > 1
assert len(merkle.roots) > 1