import random
from unittest import TestCase

from hashtree import HashTree
from . import RedirectedOutput


random.seed(17)


def buildTree(leaves):
	tree = HashTree(len(leaves))
	for l in leaves:
		tree.insertLeaf(l)
	tree.buildTree()
	return tree


class TestMorevna(RedirectedOutput, TestCase):
	def test_batchUpdate(self):
		leaves = [b"a" for i in range(8)]
		t1 = buildTree(leaves)
		keys = list(range(8))

		for i in range(8):
			random.shuffle(keys)
			for k in keys[:i+1]:
				leaves[k] = bytes([random.randrange(256)])
			t2 = buildTree(leaves)
			t1.batchUpdate((k+t1.leafStart, leaves[k]) for k in keys[:i+1])
			self.assertEqual(t1.store, t2.store)