Add consistency proof checking to testcase1

Fix consistency proof checking when first size is power of 2

1 files changed, 1 insertions, 1 deletions

diff --git a/tools/fetchallcerts.py b/tools/fetchallcerts.py
index 39ffd64..398c563 100755
--- a/tools/fetchallcerts.py
+++ b/tools/fetchallcerts.py
@@ -170,7 +170,7 @@ print "calculated root hash", base64.b16encode(calculated_root_hash)
 
 if oldsth and oldsth["tree_size"] > 0 and oldsth["tree_size"] != tree_size:
     consistency_proof = [base64.decodestring(entry) for entry in get_consistency_proof(args.baseurl, oldsth["tree_size"], tree_size)]
-    (old_treehead, new_treehead) = verify_consistency_proof(consistency_proof, oldsth["tree_size"], tree_size)
+    (old_treehead, new_treehead) = verify_consistency_proof(consistency_proof, oldsth["tree_size"], tree_size, base64.b64decode(oldsth["sha256_root_hash"]))
     assert old_treehead == base64.b64decode(oldsth["sha256_root_hash"])
     assert new_treehead == base64.b64decode(sth["sha256_root_hash"])