258 lines
12 KiB
Python
258 lines
12 KiB
Python
|
from binascii import hexlify, unhexlify
|
||
|
|
from twisted.trial import unittest
|
||
|
|
from lbrynet.wallet.script import Template, ParseError, tokenize, push_data
|
||
|
|
from lbrynet.wallet.script import PUSH_SINGLE, PUSH_MANY, OP_HASH160, OP_EQUAL
|
||
|
|
from lbrynet.wallet.script import InputScript, OutputScript
|
||
|
|
from lbrynet.wallet.bcd_data_stream import BCDataStream
|
||
|
|
|
||
|
|
|
||
|
|
def parse(opcodes, source):
|
||
|
|
template = Template('test', opcodes)
|
||
|
|
s = BCDataStream()
|
||
|
|
for t in source:
|
||
|
|
if isinstance(t, bytes):
|
||
|
|
s.write_many(push_data(t))
|
||
|
|
elif isinstance(t, int):
|
||
|
|
s.write_uint8(t)
|
||
|
|
else:
|
||
|
|
raise ValueError()
|
||
|
|
s.reset()
|
||
|
|
return template.parse(tokenize(s))
|
||
|
|
|
||
|
|
|
||
|
|
class TestScriptTemplates(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_push_data(self):
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(PUSH_SINGLE('script_hash'),),
|
||
|
|
(b'abcdef',)
|
||
|
|
), {
|
||
|
|
'script_hash': b'abcdef'
|
||
|
|
}
|
||
|
|
)
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(PUSH_SINGLE('first'), PUSH_SINGLE('last')),
|
||
|
|
(b'Satoshi', b'Nakamoto')
|
||
|
|
), {
|
||
|
|
'first': b'Satoshi',
|
||
|
|
'last': b'Nakamoto'
|
||
|
|
}
|
||
|
|
)
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(OP_HASH160, PUSH_SINGLE('script_hash'), OP_EQUAL),
|
||
|
|
(OP_HASH160, b'abcdef', OP_EQUAL)
|
||
|
|
), {
|
||
|
|
'script_hash': b'abcdef'
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_push_data_many(self):
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(PUSH_MANY('names'),),
|
||
|
|
(b'amit',)
|
||
|
|
), {
|
||
|
|
'names': [b'amit']
|
||
|
|
}
|
||
|
|
)
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(PUSH_MANY('names'),),
|
||
|
|
(b'jeremy', b'amit', b'victor')
|
||
|
|
), {
|
||
|
|
'names': [b'jeremy', b'amit', b'victor']
|
||
|
|
}
|
||
|
|
)
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(OP_HASH160, PUSH_MANY('names'), OP_EQUAL),
|
||
|
|
(OP_HASH160, b'grin', b'jack', OP_EQUAL)
|
||
|
|
), {
|
||
|
|
'names': [b'grin', b'jack']
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_push_data_mixed(self):
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(PUSH_SINGLE('CEO'), PUSH_MANY('Devs'), PUSH_SINGLE(b'CTO'), PUSH_SINGLE(b'State')),
|
||
|
|
(b'jeremy', b'lex', b'amit', b'victor', b'jack', b'grin', b'NH')
|
||
|
|
), {
|
||
|
|
'CEO': b'jeremy',
|
||
|
|
'CTO': b'grin',
|
||
|
|
'Devs': [b'lex', b'amit', b'victor', b'jack'],
|
||
|
|
'State': b'NH'
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_push_data_many_separated(self):
|
||
|
|
self.assertEqual(parse(
|
||
|
|
(PUSH_MANY('Chiefs'), OP_HASH160, PUSH_MANY('Devs')),
|
||
|
|
(b'jeremy', b'grin', OP_HASH160, b'lex', b'jack')
|
||
|
|
), {
|
||
|
|
'Chiefs': [b'jeremy', b'grin'],
|
||
|
|
'Devs': [b'lex', b'jack']
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_push_data_many_not_separated(self):
|
||
|
|
with self.assertRaisesRegexp(ParseError, 'consecutive PUSH_MANY'):
|
||
|
|
parse((PUSH_MANY('Chiefs'), PUSH_MANY('Devs')), (b'jeremy', b'grin', b'lex', b'jack'))
|
||
|
|
|
||
|
|
|
||
|
|
class TestRedeemPubKeyHash(unittest.TestCase):
|
||
|
|
|
||
|
|
def redeem_pubkey_hash(self, sig, pubkey):
|
||
|
|
# this checks that factory function correctly sets up the script
|
||
|
|
src1 = InputScript.redeem_pubkey_hash(unhexlify(sig), unhexlify(pubkey))
|
||
|
|
self.assertEqual(src1.template.name, 'pubkey_hash')
|
||
|
|
self.assertEqual(hexlify(src1.values['signature']), sig)
|
||
|
|
self.assertEqual(hexlify(src1.values['pubkey']), pubkey)
|
||
|
|
# now we test that it will round trip
|
||
|
|
src2 = InputScript(src1.source)
|
||
|
|
self.assertEqual(src2.template.name, 'pubkey_hash')
|
||
|
|
self.assertEqual(hexlify(src2.values['signature']), sig)
|
||
|
|
self.assertEqual(hexlify(src2.values['pubkey']), pubkey)
|
||
|
|
return hexlify(src1.source)
|
||
|
|
|
||
|
|
def test_redeem_pubkey_hash_1(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.redeem_pubkey_hash(
|
||
|
|
b'30450221009dc93f25184a8d483745cd3eceff49727a317c9bfd8be8d3d04517e9cdaf8dd502200e02dc5939cad9562d2b1f303f185957581c4851c98d497af281118825e18a8301',
|
||
|
|
b'025415a06514230521bff3aaface31f6db9d9bbc39bf1ca60a189e78731cfd4e1b'
|
||
|
|
),
|
||
|
|
'4830450221009dc93f25184a8d483745cd3eceff49727a317c9bfd8be8d3d04517e9cdaf8dd502200e02d'
|
||
|
|
'c5939cad9562d2b1f303f185957581c4851c98d497af281118825e18a830121025415a06514230521bff3'
|
||
|
|
'aaface31f6db9d9bbc39bf1ca60a189e78731cfd4e1b'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestRedeemScriptHash(unittest.TestCase):
|
||
|
|
|
||
|
|
def redeem_script_hash(self, sigs, pubkeys):
|
||
|
|
# this checks that factory function correctly sets up the script
|
||
|
|
src1 = InputScript.redeem_script_hash(
|
||
|
|
[unhexlify(sig) for sig in sigs],
|
||
|
|
[unhexlify(pubkey) for pubkey in pubkeys]
|
||
|
|
)
|
||
|
|
subscript1 = src1.values['script']
|
||
|
|
self.assertEqual(src1.template.name, 'script_hash')
|
||
|
|
self.assertEqual([hexlify(v) for v in src1.values['signatures']], sigs)
|
||
|
|
self.assertEqual([hexlify(p) for p in subscript1.values['pubkeys']], pubkeys)
|
||
|
|
self.assertEqual(subscript1.values['signatures_count'], len(sigs))
|
||
|
|
self.assertEqual(subscript1.values['pubkeys_count'], len(pubkeys))
|
||
|
|
# now we test that it will round trip
|
||
|
|
src2 = InputScript(src1.source)
|
||
|
|
subscript2 = src2.values['script']
|
||
|
|
self.assertEqual(src2.template.name, 'script_hash')
|
||
|
|
self.assertEqual([hexlify(v) for v in src2.values['signatures']], sigs)
|
||
|
|
self.assertEqual([hexlify(p) for p in subscript2.values['pubkeys']], pubkeys)
|
||
|
|
self.assertEqual(subscript2.values['signatures_count'], len(sigs))
|
||
|
|
self.assertEqual(subscript2.values['pubkeys_count'], len(pubkeys))
|
||
|
|
return hexlify(src1.source)
|
||
|
|
|
||
|
|
def test_redeem_script_hash_1(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.redeem_script_hash([
|
||
|
|
'3045022100fec82ed82687874f2a29cbdc8334e114af645c45298e85bb1efe69fcf15c617a0220575'
|
||
|
|
'e40399f9ada388d8e522899f4ec3b7256896dd9b02742f6567d960b613f0401',
|
||
|
|
'3044022024890462f731bd1a42a4716797bad94761fc4112e359117e591c07b8520ea33b02201ac68'
|
||
|
|
'9e35c4648e6beff1d42490207ba14027a638a62663b2ee40153299141eb01',
|
||
|
|
'30450221009910823e0142967a73c2d16c1560054d71c0625a385904ba2f1f53e0bc1daa8d02205cd'
|
||
|
|
'70a89c6cf031a8b07d1d5eb0d65d108c4d49c2d403f84fb03ad3dc318777a01'
|
||
|
|
], [
|
||
|
|
'0372ba1fd35e5f1b1437cba0c4ebfc4025b7349366f9f9c7c8c4b03a47bd3f68a4',
|
||
|
|
'03061d250182b2db1ba144167fd8b0ef3fe0fc3a2fa046958f835ffaf0dfdb7692',
|
||
|
|
'02463bfbc1eaec74b5c21c09239ae18dbf6fc07833917df10d0b43e322810cee0c',
|
||
|
|
'02fa6a6455c26fb516cfa85ea8de81dd623a893ffd579ee2a00deb6cdf3633d6bb',
|
||
|
|
'0382910eae483ce4213d79d107bfc78f3d77e2a31ea597be45256171ad0abeaa89'
|
||
|
|
]),
|
||
|
|
'00483045022100fec82ed82687874f2a29cbdc8334e114af645c45298e85bb1efe69fcf15c617a0220575e'
|
||
|
|
'40399f9ada388d8e522899f4ec3b7256896dd9b02742f6567d960b613f0401473044022024890462f731bd'
|
||
|
|
'1a42a4716797bad94761fc4112e359117e591c07b8520ea33b02201ac689e35c4648e6beff1d42490207ba'
|
||
|
|
'14027a638a62663b2ee40153299141eb014830450221009910823e0142967a73c2d16c1560054d71c0625a'
|
||
|
|
'385904ba2f1f53e0bc1daa8d02205cd70a89c6cf031a8b07d1d5eb0d65d108c4d49c2d403f84fb03ad3dc3'
|
||
|
|
'18777a014cad53210372ba1fd35e5f1b1437cba0c4ebfc4025b7349366f9f9c7c8c4b03a47bd3f68a42103'
|
||
|
|
'061d250182b2db1ba144167fd8b0ef3fe0fc3a2fa046958f835ffaf0dfdb76922102463bfbc1eaec74b5c2'
|
||
|
|
'1c09239ae18dbf6fc07833917df10d0b43e322810cee0c2102fa6a6455c26fb516cfa85ea8de81dd623a89'
|
||
|
|
'3ffd579ee2a00deb6cdf3633d6bb210382910eae483ce4213d79d107bfc78f3d77e2a31ea597be45256171'
|
||
|
|
'ad0abeaa8955ae'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestPayPubKeyHash(unittest.TestCase):
|
||
|
|
|
||
|
|
def pay_pubkey_hash(self, pubkey_hash):
|
||
|
|
# this checks that factory function correctly sets up the script
|
||
|
|
src1 = OutputScript.pay_pubkey_hash(unhexlify(pubkey_hash))
|
||
|
|
self.assertEqual(src1.template.name, 'pay_pubkey_hash')
|
||
|
|
self.assertEqual(hexlify(src1.values['pubkey_hash']), pubkey_hash)
|
||
|
|
# now we test that it will round trip
|
||
|
|
src2 = OutputScript(src1.source)
|
||
|
|
self.assertEqual(src2.template.name, 'pay_pubkey_hash')
|
||
|
|
self.assertEqual(hexlify(src2.values['pubkey_hash']), pubkey_hash)
|
||
|
|
return hexlify(src1.source)
|
||
|
|
|
||
|
|
def test_pay_pubkey_hash_1(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.pay_pubkey_hash(b'64d74d12acc93ba1ad495e8d2d0523252d664f4d'),
|
||
|
|
'76a91464d74d12acc93ba1ad495e8d2d0523252d664f4d88ac'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestPayScriptHash(unittest.TestCase):
|
||
|
|
|
||
|
|
def pay_script_hash(self, script_hash):
|
||
|
|
# this checks that factory function correctly sets up the script
|
||
|
|
src1 = OutputScript.pay_script_hash(unhexlify(script_hash))
|
||
|
|
self.assertEqual(src1.template.name, 'pay_script_hash')
|
||
|
|
self.assertEqual(hexlify(src1.values['script_hash']), script_hash)
|
||
|
|
# now we test that it will round trip
|
||
|
|
src2 = OutputScript(src1.source)
|
||
|
|
self.assertEqual(src2.template.name, 'pay_script_hash')
|
||
|
|
self.assertEqual(hexlify(src2.values['script_hash']), script_hash)
|
||
|
|
return hexlify(src1.source)
|
||
|
|
|
||
|
|
def test_pay_pubkey_hash_1(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.pay_script_hash(b'63d65a2ee8c44426d06050cfd71c0f0ff3fc41ac'),
|
||
|
|
'a91463d65a2ee8c44426d06050cfd71c0f0ff3fc41ac87'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestPayClaimNamePubkeyHash(unittest.TestCase):
|
||
|
|
|
||
|
|
def pay_claim_name_pubkey_hash(self, name, claim, pubkey_hash):
|
||
|
|
# this checks that factory function correctly sets up the script
|
||
|
|
src1 = OutputScript.pay_claim_name_pubkey_hash(name, unhexlify(claim), unhexlify(pubkey_hash))
|
||
|
|
self.assertEqual(src1.template.name, 'claim_name+pay_pubkey_hash')
|
||
|
|
self.assertEqual(src1.values['claim_name'], name)
|
||
|
|
self.assertEqual(hexlify(src1.values['claim']), claim)
|
||
|
|
self.assertEqual(hexlify(src1.values['pubkey_hash']), pubkey_hash)
|
||
|
|
# now we test that it will round trip
|
||
|
|
src2 = OutputScript(src1.source)
|
||
|
|
self.assertEqual(src2.template.name, 'claim_name+pay_pubkey_hash')
|
||
|
|
self.assertEqual(src2.values['claim_name'], name)
|
||
|
|
self.assertEqual(hexlify(src2.values['claim']), claim)
|
||
|
|
self.assertEqual(hexlify(src2.values['pubkey_hash']), pubkey_hash)
|
||
|
|
return hexlify(src1.source)
|
||
|
|
|
||
|
|
def test_pay_claim_name_pubkey_hash_1(self):
|
||
|
|
self.assertEqual(
|
||
|
|
self.pay_claim_name_pubkey_hash(
|
||
|
|
# name
|
||
|
|
b'cats',
|
||
|
|
# claim
|
||
|
|
b'080110011a7808011230080410011a084d616361726f6e6922002a003214416c6c20726967687473'
|
||
|
|
b'2072657365727665642e38004a0052005a001a42080110011a30add80aaf02559ba09853636a0658'
|
||
|
|
b'c42b727cb5bb4ba8acedb4b7fe656065a47a31878dbf9912135ddb9e13806cc1479d220a696d6167'
|
||
|
|
b'652f6a7065672a5c080110031a404180cc0fa4d3839ee29cca866baed25fafb43fca1eb3b608ee88'
|
||
|
|
b'9d351d3573d042c7b83e2e643db0d8e062a04e6e9ae6b90540a2f95fe28638d0f18af4361a1c2214'
|
||
|
|
b'f73de93f4299fb32c32f949e02198a8e91101abd',
|
||
|
|
# pub key
|
||
|
|
b'be16e4b0f9bd8f6d47d02b3a887049c36d3b84cb'
|
||
|
|
),
|
||
|
|
'b504636174734cdc080110011a7808011230080410011a084d616361726f6e6922002a003214416c6c207'
|
||
|
|
'269676874732072657365727665642e38004a0052005a001a42080110011a30add80aaf02559ba0985363'
|
||
|
|
'6a0658c42b727cb5bb4ba8acedb4b7fe656065a47a31878dbf9912135ddb9e13806cc1479d220a696d616'
|
||
|
|
'7652f6a7065672a5c080110031a404180cc0fa4d3839ee29cca866baed25fafb43fca1eb3b608ee889d35'
|
||
|
|
'1d3573d042c7b83e2e643db0d8e062a04e6e9ae6b90540a2f95fe28638d0f18af4361a1c2214f73de93f4'
|
||
|
|
'299fb32c32f949e02198a8e91101abd6d7576a914be16e4b0f9bd8f6d47d02b3a887049c36d3b84cb88ac'
|
||
|
|
)
|