117 lines
4.1 KiB
Python
117 lines
4.1 KiB
Python
|
# Copyright (c) 2018, John L. Jegutanis
|
||
|
|
#
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
# See the file "LICENCE" for information about the copyright
|
||
|
|
# and warranty status of this software.
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
from binascii import unhexlify
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from electrumx.lib.coins import Coin, Namecoin
|
||
|
|
from electrumx.lib.hash import hash_to_hex_str
|
||
|
|
from electrumx.lib.script import OpCodes, Script
|
||
|
|
|
||
|
|
TRANSACTION_DIR = os.path.join(
|
||
|
|
os.path.dirname(os.path.realpath(__file__)), 'transactions')
|
||
|
|
|
||
|
|
# Find testcases
|
||
|
|
transactions_bitcoin = []
|
||
|
|
transactions_alts = []
|
||
|
|
|
||
|
|
for name in os.listdir(TRANSACTION_DIR):
|
||
|
|
name_parts = name.split("_")
|
||
|
|
tx_list = transactions_bitcoin if name_parts[0] == "bitcoin" else transactions_alts
|
||
|
|
try:
|
||
|
|
coinFound = Coin.lookup_coin_class(name_parts[0], name_parts[1])
|
||
|
|
with open(os.path.join(TRANSACTION_DIR, name)) as f:
|
||
|
|
tx_list.append((coinFound, json.load(f)))
|
||
|
|
except Exception as e:
|
||
|
|
tx_list.append(pytest.fail(name))
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(params=transactions_bitcoin)
|
||
|
|
def transaction_details_bitcoin(request):
|
||
|
|
return request.param
|
||
|
|
|
||
|
|
|
||
|
|
def test_transaction_bitcoin(transaction_details_bitcoin):
|
||
|
|
coin, tx_info = transaction_details_bitcoin
|
||
|
|
|
||
|
|
raw_tx = unhexlify(tx_info['hex'])
|
||
|
|
tx, vsize = coin.DESERIALIZER(raw_tx, 0).read_tx_and_vsize()
|
||
|
|
tx_hash = tx.txid_rev
|
||
|
|
assert tx_info['txid'] == hash_to_hex_str(tx_hash)
|
||
|
|
assert tx_info['hash'] == hash_to_hex_str(tx.wtxid_rev)
|
||
|
|
|
||
|
|
assert tx_info['version'] == tx.version
|
||
|
|
assert tx_info['vsize'] == vsize
|
||
|
|
|
||
|
|
vin = tx_info['vin']
|
||
|
|
for i in range(len(vin)):
|
||
|
|
assert vin[i]['txid'] == hash_to_hex_str(tx.inputs[i].prev_txid_rev)
|
||
|
|
assert vin[i]['vout'] == tx.inputs[i].prev_idx
|
||
|
|
if "txinwitness" in vin[i] or (hasattr(tx, "witness") and tx.witness[i]):
|
||
|
|
assert vin[i]["txinwitness"] == [x.hex() for x in tx.witness[i]]
|
||
|
|
|
||
|
|
vout = tx_info['vout']
|
||
|
|
for i in range(len(vout)):
|
||
|
|
# value pk_script
|
||
|
|
assert vout[i]['value'] == tx.outputs[i].value / 10**8
|
||
|
|
spk = vout[i]['scriptPubKey']
|
||
|
|
tx_pks = tx.outputs[i].pk_script
|
||
|
|
assert spk['hex'] == tx_pks.hex()
|
||
|
|
|
||
|
|
|
||
|
|
##########
|
||
|
|
# Non-Bitcoin stuff goes below this line.
|
||
|
|
|
||
|
|
@pytest.fixture(params=transactions_alts)
|
||
|
|
def transaction_details_altcoin(request):
|
||
|
|
return request.param
|
||
|
|
|
||
|
|
|
||
|
|
def test_transaction_alts(transaction_details_altcoin):
|
||
|
|
coin, tx_info = transaction_details_altcoin
|
||
|
|
|
||
|
|
raw_tx = unhexlify(tx_info['hex'])
|
||
|
|
tx = coin.DESERIALIZER(raw_tx, 0).read_tx()
|
||
|
|
tx_hash = tx.txid_rev
|
||
|
|
assert tx_info['txid'] == hash_to_hex_str(tx_hash)
|
||
|
|
if expected_wtxid := tx_info.get('hash'):
|
||
|
|
assert expected_wtxid == hash_to_hex_str(tx.wtxid_rev)
|
||
|
|
|
||
|
|
vin = tx_info['vin']
|
||
|
|
for i in range(len(vin)):
|
||
|
|
assert vin[i]['txid'] == hash_to_hex_str(tx.inputs[i].prev_txid_rev)
|
||
|
|
assert vin[i]['vout'] == tx.inputs[i].prev_idx
|
||
|
|
|
||
|
|
vout = tx_info['vout']
|
||
|
|
for i in range(len(vout)):
|
||
|
|
# value pk_script
|
||
|
|
assert vout[i]['value'] == tx.outputs[i].value
|
||
|
|
spk = vout[i]['scriptPubKey']
|
||
|
|
tx_pks = tx.outputs[i].pk_script
|
||
|
|
assert spk['hex'] == tx_pks.hex()
|
||
|
|
if "addresses" in spk:
|
||
|
|
assert len(spk["addresses"]) == 1
|
||
|
|
address = spk["addresses"][0]
|
||
|
|
else:
|
||
|
|
address = spk["address"]
|
||
|
|
assert coin.address_to_hashX(address) == coin.hashX_from_script(tx_pks)
|
||
|
|
if issubclass(coin, Namecoin):
|
||
|
|
if "nameOp" not in spk or "name" not in spk["nameOp"]:
|
||
|
|
assert coin.name_hashX_from_script(tx_pks) is None
|
||
|
|
else:
|
||
|
|
OP_NAME_UPDATE = OpCodes.OP_3
|
||
|
|
normalized_name_op_script = bytearray()
|
||
|
|
normalized_name_op_script.append(OP_NAME_UPDATE)
|
||
|
|
normalized_name_op_script.extend(Script.push_data(spk["nameOp"]["name"].encode("ascii")))
|
||
|
|
normalized_name_op_script.extend(Script.push_data(bytes([])))
|
||
|
|
normalized_name_op_script.append(OpCodes.OP_2DROP)
|
||
|
|
normalized_name_op_script.append(OpCodes.OP_DROP)
|
||
|
|
normalized_name_op_script.append(OpCodes.OP_RETURN)
|
||
|
|
assert coin.name_hashX_from_script(tx_pks) == Coin.hashX_from_script(normalized_name_op_script)
|