mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 22:24:23 +01:00
5ae210d893
All string generated with urandom(x) were using binascii + os.urandom + encode. In python3 os.urandom has a hex method.
110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
import os
|
|
import binascii
|
|
from pisa.utils.authproxy import AuthServiceProxy, JSONRPCException
|
|
from pisa.conf import BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST, BTC_RPC_PORT
|
|
from pisa.tools import check_txid_format
|
|
|
|
|
|
bitcoin_cli = AuthServiceProxy("http://%s:%s@%s:%d" % (BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST, BTC_RPC_PORT))
|
|
|
|
# Help should always return 0
|
|
assert(bitcoin_cli.help() == 0)
|
|
|
|
# getblockhash should return a blockid (which matches the txid format)
|
|
block_hash = bitcoin_cli.getblockhash(0)
|
|
assert(check_txid_format(block_hash))
|
|
|
|
# Check that the values are within range and of the proper format (all should fail)
|
|
values = [-1, 500, None, '', '111', [], 1.1]
|
|
print("getblockhash fails ({}):".format(len(values)))
|
|
|
|
for v in values:
|
|
try:
|
|
block_hash = bitcoin_cli.getblockhash(v)
|
|
assert False
|
|
except JSONRPCException as e:
|
|
print('\t{}'.format(e))
|
|
|
|
# getblock should return a list of transactions and the height
|
|
block = bitcoin_cli.getblock(block_hash)
|
|
assert(isinstance(block.get('tx'), list))
|
|
assert(len(block.get('tx')) != 0)
|
|
assert(isinstance(block.get('height'), int))
|
|
|
|
# Some fails
|
|
values += ["a"*64, os.urandom(32).hex()]
|
|
print("\ngetblock fails ({}):".format(len(values)))
|
|
|
|
for v in values:
|
|
try:
|
|
block = bitcoin_cli.getblock(v)
|
|
assert False
|
|
except JSONRPCException as e:
|
|
print('\t{}'.format(e))
|
|
|
|
# decoderawtransaction should only return if the given transaction matches a txid format
|
|
coinbase_tx = block.get('tx')[0]
|
|
tx = bitcoin_cli.decoderawtransaction(coinbase_tx)
|
|
assert(isinstance(tx, dict))
|
|
assert(isinstance(tx.get('txid'), str))
|
|
assert(check_txid_format(tx.get('txid')))
|
|
|
|
# Therefore should also work for a random formatted 32-byte hex in our simulation
|
|
random_tx = os.urandom(32).hex()
|
|
tx = bitcoin_cli.decoderawtransaction(random_tx)
|
|
assert(isinstance(tx, dict))
|
|
assert(isinstance(tx.get('txid'), str))
|
|
assert(check_txid_format(tx.get('txid')))
|
|
|
|
# But it should fail for not proper formatted one
|
|
values = [1, None, '', "a"*63, "b"*65, [], os.urandom(31).hex()]
|
|
print("\ndecoderawtransaction fails ({}):".format(len(values)))
|
|
|
|
for v in values:
|
|
try:
|
|
block = bitcoin_cli.decoderawtransaction(v)
|
|
assert False
|
|
except JSONRPCException as e:
|
|
print('\t{}'.format(e))
|
|
|
|
# sendrawtransaction should only allow txids that the simulator has not mined yet
|
|
bitcoin_cli.sendrawtransaction(os.urandom(32).hex())
|
|
|
|
# Any data not matching the txid format or that matches with an already mined transaction should fail
|
|
values += [coinbase_tx]
|
|
|
|
print("\nsendrawtransaction fails ({}):".format(len(values)))
|
|
|
|
for v in values:
|
|
try:
|
|
block = bitcoin_cli.sendrawtransaction(v)
|
|
assert False
|
|
except JSONRPCException as e:
|
|
print('\t{}'.format(e))
|
|
|
|
# getrawtransaction should work for existing transactions, and fail for non-existing ones
|
|
tx = bitcoin_cli.getrawtransaction(coinbase_tx)
|
|
|
|
assert(isinstance(tx, dict))
|
|
assert(isinstance(tx.get('confirmations'), int))
|
|
|
|
print("\nsendrawtransaction fails ({}):".format(len(values)))
|
|
|
|
for v in values:
|
|
try:
|
|
block = bitcoin_cli.sendrawtransaction(v)
|
|
assert False
|
|
except JSONRPCException as e:
|
|
print('\t{}'.format(e))
|
|
|
|
# getblockcount should always return a positive integer
|
|
bc = bitcoin_cli.getblockcount()
|
|
assert (isinstance(bc, int))
|
|
assert (bc >= 0)
|
|
|
|
print("\nAll tests passed!")
|
|
|
|
|
|
|
|
|