Refactors test structure

test/simulator/bitcoin_sim_tests.py

@@ -0,0 +1,109 @@
+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, binascii.hexlify(os.urandom(32)).decode()]
+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 = binascii.hexlify(os.urandom(32)).decode()
+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, [], binascii.hexlify(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(binascii.hexlify(os.urandom(32)).decode())
+
+# 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!")
+
+
+
+