136 lines
3.9 KiB
Python
Executable file
136 lines
3.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
"""
|
|
test/unit_tests_d/ut_tx_deserialize: TX deserialization unit test for the MMGen suite
|
|
"""
|
|
|
|
import os,json
|
|
|
|
from mmgen.common import *
|
|
from ..include.common import *
|
|
from mmgen.protocol import init_proto
|
|
from mmgen.tx import MMGenTX,DeserializedTX
|
|
from mmgen.rpc import rpc_init
|
|
from mmgen.daemon import CoinDaemon
|
|
|
|
class unit_test(object):
|
|
|
|
def _get_core_repo_root(self):
|
|
self.core_repo_root = os.getenv('CORE_REPO_ROOT')
|
|
if not self.core_repo_root:
|
|
die(1,'The environmental variable CORE_REPO_ROOT must be set before running this test')
|
|
|
|
def run_test(self,name,ut):
|
|
|
|
async def test_tx(tx_proto,tx_hex,desc,n):
|
|
|
|
def has_nonstandard_outputs(outputs):
|
|
for o in outputs:
|
|
t = o['scriptPubKey']['type']
|
|
if t in ('nonstandard','pubkey','nulldata'):
|
|
return True
|
|
return False
|
|
|
|
rpc = await rpc_init(proto=tx_proto)
|
|
d = await rpc.call('decoderawtransaction',tx_hex)
|
|
|
|
if has_nonstandard_outputs(d['vout']): return False
|
|
|
|
dt = DeserializedTX(tx_proto,tx_hex)
|
|
|
|
if opt.verbose:
|
|
Msg('\n====================================================')
|
|
Msg_r('.' if opt.quiet else '{:>3}) {}\n'.format(n,desc))
|
|
if opt.verbose:
|
|
Pmsg(d)
|
|
Msg('----------------------------------------------------')
|
|
Pmsg(dt)
|
|
|
|
# metadata
|
|
assert dt['txid'] == d['txid'],'TXID does not match'
|
|
assert dt['lock_time'] == d['locktime'],'Locktime does not match'
|
|
assert dt['version'] == d['version'],'Version does not match'
|
|
|
|
# inputs
|
|
a,b = d['vin'],dt['txins']
|
|
for i in range(len(a)):
|
|
assert a[i]['txid'] == b[i]['txid'],'TxID of input {} does not match'.format(i)
|
|
assert a[i]['vout'] == b[i]['vout'],'vout of input {} does not match'.format(i)
|
|
assert a[i]['sequence'] == int(b[i]['nSeq'],16),(
|
|
'nSeq of input {} does not match'.format(i))
|
|
if 'txinwitness' in a[i]:
|
|
assert a[i]['txinwitness'] == b[i]['witness'],(
|
|
'witness of input {} does not match'.format(i))
|
|
|
|
# outputs
|
|
a,b = d['vout'],dt['txouts']
|
|
for i in range(len(a)):
|
|
A = a[i]['scriptPubKey']['addresses'][0]
|
|
B = b[i]['address']
|
|
fs = 'address of output {} does not match\nA: {}\nB: {}'
|
|
assert A == B, fs.format(i,A,B)
|
|
|
|
A = a[i]['value']
|
|
B = b[i]['amount']
|
|
fs = 'value of output {} does not match\nA: {}\nB: {}'
|
|
assert A == B, fs.format(i,A,B)
|
|
|
|
A = a[i]['scriptPubKey']['hex']
|
|
B = b[i]['scriptPubKey']
|
|
fs = 'scriptPubKey of output {} does not match\nA: {}\nB: {}'
|
|
assert A == B, fs.format(i,A,B)
|
|
|
|
return True
|
|
|
|
def print_info(fn,extra_desc):
|
|
if opt.names:
|
|
Msg_r('{} {} ({}){}'.format(
|
|
purple('Testing'),
|
|
cyan(name),
|
|
extra_desc,
|
|
'' if opt.quiet else '\n'))
|
|
else:
|
|
Msg_r(f'Testing {extra_desc} transactions from {fn!r}')
|
|
if not opt.quiet:
|
|
Msg('')
|
|
|
|
async def test_core_vectors():
|
|
self._get_core_repo_root()
|
|
fn_b = 'src/test/data/tx_valid.json'
|
|
fn = os.path.join(self.core_repo_root,fn_b)
|
|
data = json.loads(open(fn).read())
|
|
print_info(fn_b,'Core test vector')
|
|
n = 1
|
|
for e in data:
|
|
if type(e[0]) == list:
|
|
await test_tx(
|
|
tx_proto = init_proto('btc'),
|
|
tx_hex = e[1],
|
|
desc = desc,
|
|
n = n )
|
|
n += 1
|
|
else:
|
|
desc = e[0]
|
|
Msg('OK')
|
|
|
|
async def test_mmgen_txs():
|
|
fns = ( ('btc',False,'test/ref/0B8D5A[15.31789,14,tl=1320969600].rawtx'),
|
|
('btc',True,'test/ref/0C7115[15.86255,14,tl=1320969600].testnet.rawtx'),
|
|
# ('bch',False,'test/ref/460D4D-BCH[10.19764,tl=1320969600].rawtx')
|
|
)
|
|
print_info('test/ref/*rawtx','MMGen reference')
|
|
for n,(coin,testnet,fn) in enumerate(fns):
|
|
tx = MMGenTX.Unsigned(filename=fn)
|
|
await test_tx(
|
|
tx_proto = tx.proto,
|
|
tx_hex = tx.hex,
|
|
desc = fn,
|
|
n = n+1 )
|
|
Msg('OK')
|
|
|
|
start_test_daemons('btc',remove_datadir=True)
|
|
start_test_daemons('btc_tn')
|
|
run_session(test_core_vectors())
|
|
run_session(test_mmgen_txs())
|
|
stop_test_daemons('btc','btc_tn')
|
|
|
|
return True
|