def testSimpleReadWrite(nodesLeaves, tempdir):
nodes, leaves = nodesLeaves
fhs = FileHashStore(tempdir)
for leaf in leaves:
fhs.writeLeaf(leaf)
for i, leaf in enumerate(leaves):
assert leaf == fhs.readLeaf(i + 1)
for node in nodes:
fhs.writeNode(node)
for i, node in enumerate(nodes):
assert node[2] == fhs.readNode(i + 1)
lvs = fhs.readLeafs(1, len(leaves))
for i, l in enumerate(lvs):
assert leaves[i] == l
nds = fhs.readNodes(1, len(nodes))
for i, n in enumerate(nds):
assert nodes[i][2] == n
# Check that hash store can be closed and re-opened and the contents remain same
leaf_count = fhs.leafCount
node_count = fhs.nodeCount
fhs.close()
reopened_hash_store = FileHashStore(tempdir)
assert reopened_hash_store.leafCount == leaf_count
assert reopened_hash_store.nodeCount == node_count
def writtenFhs(tempdir, nodes, leaves):
fhs = FileHashStore(tempdir)
for leaf in leaves:
fhs.writeLeaf(leaf)
for node in nodes:
fhs.writeNode(node)
return fhs
def testIncorrectWrites(tempdir):
fhs = FileHashStore(tempdir, leafSize=50, nodeSize=50)
with pytest.raises(ValueError):
fhs.writeLeaf(b"less than 50")
with pytest.raises(ValueError):
fhs.writeNode((8, 1, b"also less than 50"))
with pytest.raises(ValueError):
fhs.writeLeaf(b"more than 50" + b'1' * 50)
with pytest.raises(ValueError):
fhs.writeNode((4, 1, b"also more than 50" + b'1' * 50))
def testIncorrectWrites(tempdir):
fhs = FileHashStore(tempdir, leafSize=50, nodeSize=50)
with pytest.raises(ValueError):
fhs.writeLeaf(b"less than 50")
with pytest.raises(ValueError):
fhs.writeNode((8, 1, b"also less than 50"))
with pytest.raises(ValueError):
fhs.writeLeaf(b"more than 50" + b'1'*50)
with pytest.raises(ValueError):
fhs.writeNode((4, 1, b"also more than 50" + b'1'*50))
def testConsistencyVerificationOnStartupCase1(tempdir):
"""
One more node was added to nodes file
"""
fhs = FileHashStore(tempdir)
tree = CompactMerkleTree(hashStore=fhs)
ledger = Ledger(tree=tree, dataDir=tempdir)
tranzNum = 10
for d in range(tranzNum):
ledger.add(str(d).encode())
ledger.stop()
# Writing one more node without adding of it to leaf and transaction logs
badNode = (None, None, ('X' * 32))
fhs.writeNode(badNode)
with pytest.raises(ConsistencyVerificationFailed):
tree = CompactMerkleTree(hashStore=fhs)
ledger = NoTransactionRecoveryLedger(tree=tree, dataDir=tempdir)
ledger.recoverTreeFromHashStore()
ledger.stop()
def testConsistencyVerificationOnStartupCase1(tempdir):
"""
One more node was added to nodes file
"""
fhs = FileHashStore(tempdir)
tree = CompactMerkleTree(hashStore=fhs)
ledger = Ledger(tree=tree, dataDir=tempdir)
tranzNum = 10
for d in range(tranzNum):
ledger.add(str(d).encode())
ledger.stop()
# Writing one more node without adding of it to leaf and transaction logs
badNode = (None, None, ('X' * 32))
fhs.writeNode(badNode)
with pytest.raises(ConsistencyVerificationFailed):
tree = CompactMerkleTree(hashStore=fhs)
ledger = NoTransactionRecoveryLedger(tree=tree, dataDir=tempdir)
ledger.recoverTreeFromHashStore()
ledger.stop()
def testSimpleReadWrite(nodesLeaves, tempdir):
nodes, leaves = nodesLeaves
fhs = FileHashStore(tempdir)
for leaf in leaves:
fhs.writeLeaf(leaf)
for i, leaf in enumerate(leaves):
assert leaf == fhs.readLeaf(i + 1)
for node in nodes:
fhs.writeNode(node)
for i, node in enumerate(nodes):
assert node[2] == fhs.readNode(i + 1)
lvs = fhs.readLeafs(1, len(leaves))
for i, l in enumerate(lvs):
assert leaves[i] == l
nds = fhs.readNodes(1, len(nodes))
for i, n in enumerate(nds):
assert nodes[i][2] == n