43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
import unittest
|
|
from binascii import hexlify, unhexlify
|
|
|
|
from lbrynet.wallet.claim_proofs import get_hash_for_outpoint, verify_proof
|
|
from lbryschema.hashing import double_sha256
|
|
|
|
|
|
class ClaimProofsTestCase(unittest.TestCase):
|
|
def test_verify_proof(self):
|
|
claim1_name = 97 # 'a'
|
|
claim1_txid = 'bd9fa7ffd57d810d4ce14de76beea29d847b8ac34e8e536802534ecb1ca43b68'
|
|
claim1_outpoint = 0
|
|
claim1_height = 10
|
|
claim1_node_hash = get_hash_for_outpoint(
|
|
unhexlify(claim1_txid)[::-1], claim1_outpoint, claim1_height)
|
|
|
|
claim2_name = 98 # 'b'
|
|
claim2_txid = 'ad9fa7ffd57d810d4ce14de76beea29d847b8ac34e8e536802534ecb1ca43b68'
|
|
claim2_outpoint = 1
|
|
claim2_height = 5
|
|
claim2_node_hash = get_hash_for_outpoint(
|
|
unhexlify(claim2_txid)[::-1], claim2_outpoint, claim2_height)
|
|
to_hash1 = claim1_node_hash
|
|
hash1 = double_sha256(to_hash1)
|
|
to_hash2 = bytes((claim1_name,)) + hash1 + bytes((claim2_name,)) + claim2_node_hash
|
|
|
|
root_hash = double_sha256(to_hash2)
|
|
|
|
proof = {
|
|
'last takeover height': claim1_height, 'txhash': claim1_txid, 'nOut': claim1_outpoint,
|
|
'nodes': [
|
|
{'children': [
|
|
{'character': 97},
|
|
{
|
|
'character': 98,
|
|
'nodeHash': hexlify(claim2_node_hash[::-1])
|
|
}
|
|
]},
|
|
{'children': []},
|
|
]
|
|
}
|
|
out = verify_proof(proof, hexlify(root_hash[::-1]), 'a')
|
|
self.assertEqual(out, True)
|