class ArmoryDTiabTest(TiabTest):
def armoryDTiabTestCallback(self, action, args):
if action == REFRESH_ACTION:
for wltID in args:
print wltID
if wltID in self.wltIDs:
self.numberOfWalletsScanned += 1
def setUp(self):
self.verifyBlockHeight()
# Load the primary file from the test net in a box
self.fileA = os.path.join(self.tiab.tiabDirectory, 'tiab', 'armory', \
'armory_%s_.wallet' % FIRST_WLT_NAME)
self.wltA = PyBtcWallet().readWalletFile(self.fileA)
self.fileB = os.path.join(self.tiab.tiabDirectory, 'tiab', 'armory', \
'armory_%s_.wallet' % SECOND_WLT_NAME)
self.wltB = PyBtcWallet().readWalletFile(self.fileB)
self.fileC = os.path.join(self.tiab.tiabDirectory, 'tiab', 'armory', \
'armory_%s_.wallet' % THIRD_WLT_NAME)
self.wltC = PyBtcWallet().readWalletFile(self.fileC)
self.jsonServer = Armory_Json_Rpc_Server(self.wltA, \
inWltMap={SECOND_WLT_NAME : self.wltB, \
THIRD_WLT_NAME : self.wltC}, \
armoryHomeDir=os.path.join(self.tiab.tiabDirectory, \
'tiab','armory'))
self.wltIDs = [self.wltA.uniqueIDB58, self.wltB.uniqueIDB58, self.wltC.uniqueIDB58]
#register a callback
TheBDM.registerCppNotification(self.armoryDTiabTestCallback)
#flag to check on wallet scan status
self.numberOfWalletsScanned = 0
self.wltA.registerWallet()
time.sleep(0.5)
self.wltB.registerWallet()
time.sleep(0.5)
self.wltC.registerWallet()
time.sleep(0.5)
#wait on scan for 20sec then raise if the scan hasn't finished yet
i = 0
while self.numberOfWalletsScanned < 3:
time.sleep(0.5)
i += 1
if i >= 40:
raise RuntimeError("self.numberOfWalletsScanned = %d" % self.numberOfWalletsScanned)
def tearDown(self):
TheBDM.unregisterCppNotification(self.armoryDTiabTestCallback)
self.wltA.unregisterWallet()
self.wltB.unregisterWallet()
self.wltC.unregisterWallet()
self.resetWalletFiles()
def testActiveWallet(self):
self.jsonServer.jsonrpc_setactivewallet(FIRST_WLT_NAME)
result = self.jsonServer.jsonrpc_getactivewallet()
self.assertEqual(result, FIRST_WLT_NAME)
bogusResult = self.jsonServer.jsonrpc_setactivewallet('bogus wallet name')
self.assertTrue('does not exist' in bogusResult)
# Tests all of the address meta data functions at once
def testAddressMetaData(self):
testInput = {TIAB_WLT_1_ADDR_1:
{'chain': 5,
'index': 2},
TIAB_WLT_1_ADDR_2:
{'CrazyField': 'what',
'1': 1,
'2': 2}}
self.jsonServer.jsonrpc_setaddressmetadata(testInput)
testOutput1=self.jsonServer.jsonrpc_getaddressmetadata()
self.assertEqual(testOutput1, testInput)
self.jsonServer.jsonrpc_clearaddressmetadata()
testOutput2=self.jsonServer.jsonrpc_getaddressmetadata()
self.assertEqual(testOutput2, {})
def testListloadedwallets(self):
result = self.jsonServer.jsonrpc_listloadedwallets()
self.assertEqual(len(result.keys()), 3)
self.assertTrue(FIRST_WLT_NAME in result.values())
self.assertTrue(SECOND_WLT_NAME in result.values())
self.assertTrue(THIRD_WLT_NAME in result.values())
def getPrivateKey(self, address):
hash160 = addrStr_to_hash160(address)[1]
if self.wltA.isLocked:
self.wltA.unlock(securePassphrase=SecureBinaryData(PASSPHRASE1),
tempKeyLifetime=1000000)
return self.wltA.addrMap[hash160].binPrivKey32_Plain.toBinStr()
# Test Create lockbox and list loaded lockbox at the same time.
# Also test set and get active lockbox.
def testLockboxMethods(self):
self.assertEqual(self.jsonServer.jsonrpc_getactivelockbox(), None)
addrFromFirstWlt = self.wltA.peekNextUnusedAddr().getPubKey().toHexStr()
addrFromSecondWlt = self.wltB.peekNextUnusedAddr().getPubKey().toHexStr()
addrFromThirdWlt = self.wltC.peekNextUnusedAddr().getPubKey().toHexStr()
# This test should succeed.
actualResult1 = self.jsonServer.jsonrpc_createlockbox(2, 3, \
addrFromFirstWlt, \
SECOND_WLT_NAME, \
THIRD_WLT_NAME)
self.assertTrue(TWO_OF_THREE_LB_NAME in actualResult1.values())
# This test should fail because the first createlockbox() used the 2nd &
# 3rd addresses.
actualResult2 = self.jsonServer.jsonrpc_createlockbox(2, 3, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt)
self.assertTrue(TWO_OF_THREE_LB_NAME in actualResult2['Error'])
# This test should succeed.
actualResult3 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromSecondWlt, \
addrFromThirdWlt)
self.assertEqual(TWO_OF_TWO_LB_NAME, actualResult3['id'])
listResult3 = self.jsonServer.jsonrpc_listloadedlockboxes()
self.assertEqual(len(listResult3.keys()), 2)
self.assertTrue(TWO_OF_TWO_LB_NAME in listResult3.values())
# This test should fail because of a bad wallet name.
actualResult4 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_WLT_NAME)
self.assertTrue(BAD_WLT_NAME in actualResult4['Error'])
# This test should fail because of a malformed, uncompressed public key.
actualResult5 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_PK_UNCOMP_1)
self.assertTrue(BAD_PK_UNCOMP_1 in actualResult5['Error'])
# This test should fail because of a malformed, compressed public key.
actualResult6 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_PK_COMP_1)
self.assertTrue(BAD_PK_COMP_1 in actualResult6['Error'])
# This test should fail due to a compressed public key being used.
foundUncompAddr = False
actualResult7 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
GOOD_PK_COMP_1)
self.assertTrue(GOOD_PK_COMP_1 in actualResult7['Error'])
# This test should fail due to a malformed public key (incorrect length).
actualResult8 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_PK)
self.assertTrue(BAD_PK in actualResult8['Error'])
# These tests should fail for various reasons related to the # of inputs.
actualResult09 = self.jsonServer.jsonrpc_createlockbox(3, 2, \
addrFromFirstWlt, \
addrFromThirdWlt)
self.assertTrue(ERRSTR09 in actualResult09['Error'])
actualResult10 = self.jsonServer.jsonrpc_createlockbox(8, 8, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt, \
FIRST_WLT_NAME, \
SECOND_WLT_NAME, \
THIRD_WLT_NAME, \
GOOD_PK_UNCOMP_1, \
GOOD_PK_UNCOMP_2)
self.assertTrue(ERRSTR10 in actualResult10['Error'])
actualResult11 = self.jsonServer.jsonrpc_createlockbox(1, 8, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt, \
FIRST_WLT_NAME, \
SECOND_WLT_NAME, \
THIRD_WLT_NAME, \
GOOD_PK_UNCOMP_1, \
GOOD_PK_UNCOMP_2)
self.assertTrue(ERRSTR11 in actualResult11['Error'])
actualResult12 = self.jsonServer.jsonrpc_createlockbox(1, 4)
self.assertTrue(ERRSTR12 in actualResult12['Error'])
actualResult13 = self.jsonServer.jsonrpc_createlockbox(1, 2, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt)
self.assertTrue(ERRSTR13 in actualResult13['Error'])
actualResult14 = self.jsonServer.jsonrpc_createlockbox(1, 2, \
addrFromFirstWlt)
self.assertTrue(ERRSTR14 in actualResult14['Error'])
# These tests should succeed.
self.jsonServer.jsonrpc_setactivelockbox(TWO_OF_TWO_LB_NAME)
self.assertEqual(self.jsonServer.jsonrpc_getactivelockbox(), \
TWO_OF_TWO_LB_NAME)
self.jsonServer.jsonrpc_setactivelockbox(TWO_OF_THREE_LB_NAME)
self.assertEqual(self.jsonServer.jsonrpc_getactivelockbox(), \
TWO_OF_THREE_LB_NAME)
def testVerifysignature(self):
clearSignMessage = ASv1CS(self.getPrivateKey(TIAB_WLT_1_ADDR_1), \
TEST_MESSAGE)
inMsg = '\"' + clearSignMessage + '\"'
result = self.jsonServer.jsonrpc_verifysignature(inMsg)
self.assertEqual(result['message'], TEST_MESSAGE)
self.assertEqual(result['address'], TIAB_WLT_1_ADDR_1)
# Requires Supernode
@SkipTest
def testReceivedfromsigner(self):
clearSignMessage2 = ASv1CS(self.getPrivateKey(TIAB_WLT_1_ADDR_3), \
TEST_MESSAGE)
inMsg2 = '\"' + clearSignMessage2 + '\"'
result2 = self.jsonServer.jsonrpc_getreceivedfromsigner(inMsg2)
self.assertEqual(result2['message'], TEST_MESSAGE)
self.assertEqual(result2['amount'], 0)
# Requires Supernode
@SkipTest
def testReceivedfromaddress(self):
result = self.jsonServer.jsonrpc_getreceivedfromaddress(TIAB_WLT_3_ADDR_3)
self.assertEqual(result, 0)
result = self.jsonServer.jsonrpc_getreceivedfromaddress(TIAB_WLT_1_ADDR_2)
self.assertEqual(result, EXPECTED_RECEIVED_FROM_TIAB_WLT_1_ADDR_2)
def testGettransaction(self):
tx = self.jsonServer.jsonrpc_gettransaction('db0ee46beff3a61f38bfc563f92c11449ed57c3d7d5cd5aafbe0114e5a9ceee4')
self.assertEqual(tx, {'category': 'send', 'inputs': [{'fromtxid': '04b865ecf5fca3a56f6ce73a571a09a668f4b7aa5a7547a5f51fae08eadcdbb5',
'ismine': True, 'fromtxindex': 1, 'value': 1000.0, 'address': 'mtZ2d1jFZ9YNp3Ku5Fb2u8Tfu3RgimBHAD'}],
'direction': 'send', 'fee': 0.0001, 'totalinputs': 1000.0, 'outputs':
[{'address': 'mpXd2u8fPVYdL1Nf9bZ4EFnqhkNyghGLxL', 'value': 20.0, 'ismine': False},
{'address': 'mgLjhTCUtbeLPP9fDkBnG8oztWgJaXZQjn', 'value': 979.9999, 'ismine': True}],
'txid': 'db0ee46beff3a61f38bfc563f92c11449ed57c3d7d5cd5aafbe0114e5a9ceee4', 'confirmations': 10,
'orderinblock': 1, 'mainbranch': True, 'numtxin': 1, 'numtxout': 2,
'netdiff': -20.0001, 'infomissing': False})
def testGetblock(self):
block = self.jsonServer.jsonrpc_getblock('0000000064a1ad1f15981a713a6ef08fd98f69854c781dc7b8789cc5f678e01f')
# This method is broken in a couple ways.
# For no just verify that raw transaction is correct
self.assertEqual(block['rawheader'], '02000000d8778a50d43d3e02c4c20bdd0ed97077a3c4bef3e86ce58975f6f43a00000000d25912cfc67228748494d421512c7a6cc31668fa82b72265261558802a89f4c2e0350153ffff001d10bcc285',)
def testGetinfo(self):
info = self.jsonServer.jsonrpc_getarmorydinfo()
self.assertEqual(info['blocks'], TOP_TIAB_BLOCK)
self.assertEqual(info['bdmstate'], BDM_BLOCKCHAIN_READY)
self.assertEqual(info['walletversionstr'], '1.35')
self.assertEqual(info['difficulty'], 1.0)
self.assertEqual(info['balance'], FIRST_WLT_BALANCE)
def testListtransactions(self):
#takes a history page count now, not an amount of transactions to return
txList = self.jsonServer.jsonrpc_listtransactions(0)
self.assertTrue(len(txList)>10)
self.assertEqual(txList[0], {'blockhash': '0000000064a1ad1f15981a713a6ef08fd98f69854c781dc7b8789cc5f678e01f',
'blockindex': 1, 'confirmations': 31, 'address': 'mtZ2d1jFZ9YNp3Ku5Fb2u8Tfu3RgimBHAD',
'category': 'receive', 'account': '',
'txid': '04b865ecf5fca3a56f6ce73a571a09a668f4b7aa5a7547a5f51fae08eadcdbb5',
'blocktime': 1392588256, 'amount': 1000.0, 'timereceived': 1392588256, 'time': 1392588256})
def testGetledgersimple(self):
ledger = self.jsonServer.jsonrpc_getledgersimple(FIRST_WLT_NAME)
self.assertTrue(len(ledger)>4)
amountList = [row['amount'] for row in ledger]
expectedAmountList = [20.0, 6.0, 10.0, 0.8, 30.0]
self.assertEqual(amountList[:5], expectedAmountList)
def testGetledger(self):
ledger = self.jsonServer.jsonrpc_getledger(FIRST_WLT_NAME)
self.assertTrue(len(ledger)>6)
amountList = [row['amount'] for row in ledger]
#expectedAmountList = [1000.0, 20.0, 30.0, 0.8, 10.0, 6.0, 20.0]
expectedAmountList = [20.0, 6.0, 10.0, 0.8, 30.0, 20.0, 1000.0]
self.assertEqual(amountList, expectedAmountList)
self.assertEqual(ledger[0]['direction'], 'receive')
self.assertEqual(len(ledger[0]['recipme']), 1)
self.assertEqual(ledger[0]['recipme'][0]['amount'], 20.0)
self.assertEqual(len(ledger[0]['recipother']), 1)
self.assertEqual(ledger[0]['recipother'][0]['amount'], 0.8998)
self.assertEqual(len(ledger[0]['senderme']), 0)
self.assertEqual(ledger[0]['senderother'][0]['amount'], 11.8999)
self.assertEqual(ledger[5]['direction'], 'send')
self.assertEqual(len(ledger[5]['senderother']), 0)
self.assertEqual(len(ledger[5]['senderme']), 1)
self.assertEqual(ledger[5]['senderme'][0]['amount'], 1000.0)
def testSendtoaddress(self):
# Send 1 BTC
serializedUnsignedTx = \
self.jsonServer.jsonrpc_createustxtoaddress(TIAB_WLT_3_ADDR_3, \
BTC_TO_SEND)
unsignedTx = UnsignedTransaction().unserializeAscii(serializedUnsignedTx)
# Should have 2 txouts to TIAB_WLT_3_ADDR_3 and the change
self.assertEqual(len(unsignedTx.decorTxOuts), 2)
foundTxOut = False
for txout in unsignedTx.decorTxOuts:
if script_to_addrStr(txout.binScript) == TIAB_WLT_3_ADDR_3:
self.assertEqual(txout.value, JSONtoAmount(BTC_TO_SEND))
foundTxOut = True
self.assertTrue(foundTxOut)
# Test two paths through signing method and make sure they are equal
# Wallets in the TIAB start out unencrypted
f = open(TX_FILENAME, 'w')
f.write(serializedUnsignedTx)
f.close()
serializedSignedTxUnencrypted = \
self.jsonServer.jsonrpc_signasciitransaction(TX_FILENAME)
self.jsonServer.jsonrpc_encryptwallet(PASSPHRASE1)
self.jsonServer.jsonrpc_walletpassphrase(PASSPHRASE1)
serializedSignedTxEncrypted = \
self.jsonServer.jsonrpc_signasciitransaction(TX_FILENAME)
# Other tests expect wallet to be unencrypted
self.wltA.unlock(securePassphrase=SecureBinaryData(PASSPHRASE1),
tempKeyLifetime=1000000)
self.wltA.changeWalletEncryption()
signedTxUnencrypted = UnsignedTransaction().unserializeAscii(serializedSignedTxUnencrypted)
signedTxEncrypted = UnsignedTransaction().unserializeAscii(serializedSignedTxEncrypted)
# check number of outputs 1 Btc goes to a single output and the other goes to change
self.assertEqual(len(signedTxUnencrypted.decorTxOuts), 2)
self.assertEqual(len(signedTxEncrypted.decorTxOuts), 2)
self.assertEqual(signedTxUnencrypted.asciiID, signedTxEncrypted.asciiID)
self.assertTrue(JSONtoAmount(BTC_TO_SEND) in
[signedTxEncrypted.decorTxOuts[0].value,
signedTxEncrypted.decorTxOuts[1].value])
self.assertTrue(JSONtoAmount(BTC_TO_SEND) in
[signedTxUnencrypted.decorTxOuts[0].value,
signedTxUnencrypted.decorTxOuts[1].value])
f = open(TX_FILENAME, 'w')
f.write(signedTxEncrypted.serializeAscii())
f.close()
txHexToBroadcast = self.jsonServer.jsonrpc_gethextxtobroadcast(TX_FILENAME)
finalPyTx = PyTx().unserialize(hex_to_binary(txHexToBroadcast))
self.assertEqual(len(finalPyTx.outputs), 2)
self.assertTrue(JSONtoAmount(BTC_TO_SEND) in
[finalPyTx.outputs[0].value,
finalPyTx.outputs[1].value])
def testSendmany(self):
# Send 1 BTC
serializedUnsignedTx = \
self.jsonServer.jsonrpc_createustxformany(None, ','.join([TIAB_WLT_3_ADDR_2, str(BTC_TO_SEND)]), \
','.join([TIAB_WLT_3_ADDR_3, str(BTC_TO_SEND)]))
unsignedTx = UnsignedTransaction().unserializeAscii(serializedUnsignedTx)
# Should have 2 txouts to TIAB_WLT_3_ADDR_3 and the change
self.assertEqual(len(unsignedTx.decorTxOuts), 3)
txOutsFound = 0
for txout in unsignedTx.decorTxOuts:
if script_to_addrStr(txout.binScript) in [TIAB_WLT_3_ADDR_2, TIAB_WLT_3_ADDR_3]:
self.assertEqual(txout.value, JSONtoAmount(BTC_TO_SEND))
txOutsFound += 1
self.assertEqual(txOutsFound, 2)
def testListUnspent(self):
actualResult = self.jsonServer.jsonrpc_listunspent()
tx1Found = False
tx5Found = False
for i in range(len(actualResult)):
if actualResult[i]['txid'] == EXPECTED_UNSPENT_TX1_HEX:
self.assertEqual(actualResult[i]['amount'], \
EXPECTED_UNSPENT_TX1_BAL)
self.assertEqual(actualResult[i]['confirmations'], \
EXPECTED_UNSPENT_TX1_CONF)
self.assertEqual(actualResult[i]['priority'], \
EXPECTED_UNSPENT_TX1_PRI)
tx1Found = True
elif actualResult[i]['txid'] == EXPECTED_UNSPENT_TX5_HEX:
self.assertEqual(actualResult[i]['amount'], \
EXPECTED_UNSPENT_TX5_BAL)
self.assertEqual(actualResult[i]['confirmations'], \
EXPECTED_UNSPENT_TX5_CONF)
self.assertEqual(actualResult[i]['priority'], \
EXPECTED_UNSPENT_TX5_PRI)
tx5Found = True
self.assertTrue(tx1Found)
self.assertTrue(tx5Found)
def testListAddrUnspent(self):
totStr = '%s,%s' % (TIAB_WLT_1_ADDR_3, TIAB_WLT_1_ADDR_8)
actualResult = self.jsonServer.jsonrpc_listaddrunspent(totStr)
self.assertEqual(actualResult['addrbalance'][TIAB_WLT_1_ADDR_3], \
TIAB_WLT_1_PK_UTXO_BAL_3)
self.assertEqual(actualResult['addrbalance'][TIAB_WLT_1_ADDR_8], \
TIAB_WLT_1_PK_UTXO_BAL_8)
# NB: Ideally, the TAB asserts would be against addresses with multiple
# UTXOs. As is, this test case works but could be better.
self.assertEqual(actualResult['totalbalance'], TIAB_WLT_1_PK_UTXO_BAL_3 + TIAB_WLT_1_PK_UTXO_BAL_8 )
self.assertEqual(actualResult['numutxo'], 2)
def testGetNewAddress(self):
actualResult = self.jsonServer.jsonrpc_getnewaddress()
self.assertEqual(actualResult, EXPECTED_TIAB_NEXT_ADDR)
def testGetBalance(self):
balances = {'spendable' : FIRST_WLT_BALANCE, \
'spend' : FIRST_WLT_BALANCE, \
'unconf' : 0, \
'unconfirmed' : 0, \
'total' : FIRST_WLT_BALANCE, \
'ultimate' : FIRST_WLT_BALANCE, \
'unspent' : FIRST_WLT_BALANCE, \
'full' : FIRST_WLT_BALANCE}
for balanceType in balances.keys():
result = self.jsonServer.jsonrpc_getbalance(balanceType)
self.assertEqual(result,
AmountToJSON(self.wltA.getBalance(balanceType)))
class ArmoryDTiabTest(TiabTest):
def armoryDTiabTestCallback(self, action, args):
if action == REFRESH_ACTION:
for wltID in args:
print wltID
if wltID in self.wltIDs:
self.numberOfWalletsScanned += 1
def setUp(self):
self.verifyBlockHeight()
# Load the primary file from the test net in a box
self.fileA = os.path.join(self.tiab.tiabDirectory, 'tiab', 'armory', \
'armory_%s_.wallet' % FIRST_WLT_NAME)
self.wltA = PyBtcWallet().readWalletFile(self.fileA)
self.fileB = os.path.join(self.tiab.tiabDirectory, 'tiab', 'armory', \
'armory_%s_.wallet' % SECOND_WLT_NAME)
self.wltB = PyBtcWallet().readWalletFile(self.fileB)
self.fileC = os.path.join(self.tiab.tiabDirectory, 'tiab', 'armory', \
'armory_%s_.wallet' % THIRD_WLT_NAME)
self.wltC = PyBtcWallet().readWalletFile(self.fileC)
self.jsonServer = Armory_Json_Rpc_Server(self.wltA, \
inWltMap={SECOND_WLT_NAME : self.wltB, \
THIRD_WLT_NAME : self.wltC}, \
armoryHomeDir=os.path.join(self.tiab.tiabDirectory, \
'tiab','armory'))
self.wltIDs = [
self.wltA.uniqueIDB58, self.wltB.uniqueIDB58, self.wltC.uniqueIDB58
]
#register a callback
TheBDM.registerCppNotification(self.armoryDTiabTestCallback)
#flag to check on wallet scan status
self.numberOfWalletsScanned = 0
self.wltA.registerWallet()
time.sleep(0.5)
self.wltB.registerWallet()
time.sleep(0.5)
self.wltC.registerWallet()
time.sleep(0.5)
#wait on scan for 20sec then raise if the scan hasn't finished yet
i = 0
while self.numberOfWalletsScanned < 3:
time.sleep(0.5)
i += 1
if i >= 40:
raise RuntimeError("self.numberOfWalletsScanned = %d" %
self.numberOfWalletsScanned)
def tearDown(self):
TheBDM.unregisterCppNotification(self.armoryDTiabTestCallback)
self.wltA.unregisterWallet()
self.wltB.unregisterWallet()
self.wltC.unregisterWallet()
self.resetWalletFiles()
def testActiveWallet(self):
self.jsonServer.jsonrpc_setactivewallet(FIRST_WLT_NAME)
result = self.jsonServer.jsonrpc_getactivewallet()
self.assertEqual(result, FIRST_WLT_NAME)
bogusResult = self.jsonServer.jsonrpc_setactivewallet(
'bogus wallet name')
self.assertTrue('does not exist' in bogusResult)
# Tests all of the address meta data functions at once
def testAddressMetaData(self):
testInput = {
TIAB_WLT_1_ADDR_1: {
'chain': 5,
'index': 2
},
TIAB_WLT_1_ADDR_2: {
'CrazyField': 'what',
'1': 1,
'2': 2
}
}
self.jsonServer.jsonrpc_setaddressmetadata(testInput)
testOutput1 = self.jsonServer.jsonrpc_getaddressmetadata()
self.assertEqual(testOutput1, testInput)
self.jsonServer.jsonrpc_clearaddressmetadata()
testOutput2 = self.jsonServer.jsonrpc_getaddressmetadata()
self.assertEqual(testOutput2, {})
def testListloadedwallets(self):
result = self.jsonServer.jsonrpc_listloadedwallets()
self.assertEqual(len(result.keys()), 3)
self.assertTrue(FIRST_WLT_NAME in result.values())
self.assertTrue(SECOND_WLT_NAME in result.values())
self.assertTrue(THIRD_WLT_NAME in result.values())
def getPrivateKey(self, address):
hash160 = addrStr_to_hash160(address)[1]
if self.wltA.isLocked:
self.wltA.unlock(securePassphrase=SecureBinaryData(PASSPHRASE1),
tempKeyLifetime=1000000)
return self.wltA.addrMap[hash160].binPrivKey32_Plain.toBinStr()
# Test Create lockbox and list loaded lockbox at the same time.
# Also test set and get active lockbox.
def testLockboxMethods(self):
self.assertEqual(self.jsonServer.jsonrpc_getactivelockbox(), None)
addrFromFirstWlt = self.wltA.peekNextUnusedAddr().getPubKey().toHexStr(
)
addrFromSecondWlt = self.wltB.peekNextUnusedAddr().getPubKey(
).toHexStr()
addrFromThirdWlt = self.wltC.peekNextUnusedAddr().getPubKey().toHexStr(
)
# This test should succeed.
actualResult1 = self.jsonServer.jsonrpc_createlockbox(2, 3, \
addrFromFirstWlt, \
SECOND_WLT_NAME, \
THIRD_WLT_NAME)
self.assertTrue(TWO_OF_THREE_LB_NAME in actualResult1.values())
# This test should fail because the first createlockbox() used the 2nd &
# 3rd addresses.
actualResult2 = self.jsonServer.jsonrpc_createlockbox(2, 3, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt)
self.assertTrue(TWO_OF_THREE_LB_NAME in actualResult2['Error'])
# This test should succeed.
actualResult3 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromSecondWlt, \
addrFromThirdWlt)
self.assertEqual(TWO_OF_TWO_LB_NAME, actualResult3['id'])
listResult3 = self.jsonServer.jsonrpc_listloadedlockboxes()
self.assertEqual(len(listResult3.keys()), 2)
self.assertTrue(TWO_OF_TWO_LB_NAME in listResult3.values())
# This test should fail because of a bad wallet name.
actualResult4 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_WLT_NAME)
self.assertTrue(BAD_WLT_NAME in actualResult4['Error'])
# This test should fail because of a malformed, uncompressed public key.
actualResult5 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_PK_UNCOMP_1)
self.assertTrue(BAD_PK_UNCOMP_1 in actualResult5['Error'])
# This test should fail because of a malformed, compressed public key.
actualResult6 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_PK_COMP_1)
self.assertTrue(BAD_PK_COMP_1 in actualResult6['Error'])
# This test should fail due to a compressed public key being used.
foundUncompAddr = False
actualResult7 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
GOOD_PK_COMP_1)
self.assertTrue(GOOD_PK_COMP_1 in actualResult7['Error'])
# This test should fail due to a malformed public key (incorrect length).
actualResult8 = self.jsonServer.jsonrpc_createlockbox(2, 2, \
addrFromFirstWlt, \
BAD_PK)
self.assertTrue(BAD_PK in actualResult8['Error'])
# These tests should fail for various reasons related to the # of inputs.
actualResult09 = self.jsonServer.jsonrpc_createlockbox(3, 2, \
addrFromFirstWlt, \
addrFromThirdWlt)
self.assertTrue(ERRSTR09 in actualResult09['Error'])
actualResult10 = self.jsonServer.jsonrpc_createlockbox(8, 8, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt, \
FIRST_WLT_NAME, \
SECOND_WLT_NAME, \
THIRD_WLT_NAME, \
GOOD_PK_UNCOMP_1, \
GOOD_PK_UNCOMP_2)
self.assertTrue(ERRSTR10 in actualResult10['Error'])
actualResult11 = self.jsonServer.jsonrpc_createlockbox(1, 8, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt, \
FIRST_WLT_NAME, \
SECOND_WLT_NAME, \
THIRD_WLT_NAME, \
GOOD_PK_UNCOMP_1, \
GOOD_PK_UNCOMP_2)
self.assertTrue(ERRSTR11 in actualResult11['Error'])
actualResult12 = self.jsonServer.jsonrpc_createlockbox(1, 4)
self.assertTrue(ERRSTR12 in actualResult12['Error'])
actualResult13 = self.jsonServer.jsonrpc_createlockbox(1, 2, \
addrFromFirstWlt, \
addrFromSecondWlt, \
addrFromThirdWlt)
self.assertTrue(ERRSTR13 in actualResult13['Error'])
actualResult14 = self.jsonServer.jsonrpc_createlockbox(1, 2, \
addrFromFirstWlt)
self.assertTrue(ERRSTR14 in actualResult14['Error'])
# These tests should succeed.
self.jsonServer.jsonrpc_setactivelockbox(TWO_OF_TWO_LB_NAME)
self.assertEqual(self.jsonServer.jsonrpc_getactivelockbox(), \
TWO_OF_TWO_LB_NAME)
self.jsonServer.jsonrpc_setactivelockbox(TWO_OF_THREE_LB_NAME)
self.assertEqual(self.jsonServer.jsonrpc_getactivelockbox(), \
TWO_OF_THREE_LB_NAME)
def testVerifysignature(self):
clearSignMessage = ASv1CS(self.getPrivateKey(TIAB_WLT_1_ADDR_1), \
TEST_MESSAGE)
inMsg = '\"' + clearSignMessage + '\"'
result = self.jsonServer.jsonrpc_verifysignature(inMsg)
self.assertEqual(result['message'], TEST_MESSAGE)
self.assertEqual(result['address'], TIAB_WLT_1_ADDR_1)
# Requires Supernode
@SkipTest
def testReceivedfromsigner(self):
clearSignMessage2 = ASv1CS(self.getPrivateKey(TIAB_WLT_1_ADDR_3), \
TEST_MESSAGE)
inMsg2 = '\"' + clearSignMessage2 + '\"'
result2 = self.jsonServer.jsonrpc_getreceivedfromsigner(inMsg2)
self.assertEqual(result2['message'], TEST_MESSAGE)
self.assertEqual(result2['amount'], 0)
# Requires Supernode
@SkipTest
def testReceivedfromaddress(self):
result = self.jsonServer.jsonrpc_getreceivedfromaddress(
TIAB_WLT_3_ADDR_3)
self.assertEqual(result, 0)
result = self.jsonServer.jsonrpc_getreceivedfromaddress(
TIAB_WLT_1_ADDR_2)
self.assertEqual(result, EXPECTED_RECEIVED_FROM_TIAB_WLT_1_ADDR_2)
def testGettransaction(self):
tx = self.jsonServer.jsonrpc_gettransaction(
'db0ee46beff3a61f38bfc563f92c11449ed57c3d7d5cd5aafbe0114e5a9ceee4')
self.assertEqual(
tx, {
'category':
'send',
'inputs': [{
'fromtxid':
'04b865ecf5fca3a56f6ce73a571a09a668f4b7aa5a7547a5f51fae08eadcdbb5',
'ismine': True,
'fromtxindex': 1,
'value': 1000.0,
'address': 'mtZ2d1jFZ9YNp3Ku5Fb2u8Tfu3RgimBHAD'
}],
'direction':
'send',
'fee':
0.0001,
'totalinputs':
1000.0,
'outputs': [{
'address': 'mpXd2u8fPVYdL1Nf9bZ4EFnqhkNyghGLxL',
'value': 20.0,
'ismine': False
}, {
'address': 'mgLjhTCUtbeLPP9fDkBnG8oztWgJaXZQjn',
'value': 979.9999,
'ismine': True
}],
'txid':
'db0ee46beff3a61f38bfc563f92c11449ed57c3d7d5cd5aafbe0114e5a9ceee4',
'confirmations':
10,
'orderinblock':
1,
'mainbranch':
True,
'numtxin':
1,
'numtxout':
2,
'netdiff':
-20.0001,
'infomissing':
False
})
def testGetblock(self):
block = self.jsonServer.jsonrpc_getblock(
'0000000064a1ad1f15981a713a6ef08fd98f69854c781dc7b8789cc5f678e01f')
# This method is broken in a couple ways.
# For no just verify that raw transaction is correct
self.assertEqual(
block['rawheader'],
'02000000d8778a50d43d3e02c4c20bdd0ed97077a3c4bef3e86ce58975f6f43a00000000d25912cfc67228748494d421512c7a6cc31668fa82b72265261558802a89f4c2e0350153ffff001d10bcc285',
)
def testGetinfo(self):
info = self.jsonServer.jsonrpc_getarmorydinfo()
self.assertEqual(info['blocks'], TOP_TIAB_BLOCK)
self.assertEqual(info['bdmstate'], BDM_BLOCKCHAIN_READY)
self.assertEqual(info['walletversionstr'], '1.35')
self.assertEqual(info['difficulty'], 1.0)
self.assertEqual(info['balance'], FIRST_WLT_BALANCE)
def testListtransactions(self):
#takes a history page count now, not an amount of transactions to return
txList = self.jsonServer.jsonrpc_listtransactions(0)
self.assertTrue(len(txList) > 10)
self.assertEqual(
txList[0], {
'blockhash':
'0000000064a1ad1f15981a713a6ef08fd98f69854c781dc7b8789cc5f678e01f',
'blockindex': 1,
'confirmations': 31,
'address': 'mtZ2d1jFZ9YNp3Ku5Fb2u8Tfu3RgimBHAD',
'category': 'receive',
'account': '',
'txid':
'04b865ecf5fca3a56f6ce73a571a09a668f4b7aa5a7547a5f51fae08eadcdbb5',
'blocktime': 1392588256,
'amount': 1000.0,
'timereceived': 1392588256,
'time': 1392588256
})
def testGetledgersimple(self):
ledger = self.jsonServer.jsonrpc_getledgersimple(FIRST_WLT_NAME)
self.assertTrue(len(ledger) > 4)
amountList = [row['amount'] for row in ledger]
expectedAmountList = [20.0, 6.0, 10.0, 0.8, 30.0]
self.assertEqual(amountList[:5], expectedAmountList)
def testGetledger(self):
ledger = self.jsonServer.jsonrpc_getledger(FIRST_WLT_NAME)
self.assertTrue(len(ledger) > 6)
amountList = [row['amount'] for row in ledger]
#expectedAmountList = [1000.0, 20.0, 30.0, 0.8, 10.0, 6.0, 20.0]
expectedAmountList = [20.0, 6.0, 10.0, 0.8, 30.0, 20.0, 1000.0]
self.assertEqual(amountList, expectedAmountList)
self.assertEqual(ledger[0]['direction'], 'receive')
self.assertEqual(len(ledger[0]['recipme']), 1)
self.assertEqual(ledger[0]['recipme'][0]['amount'], 20.0)
self.assertEqual(len(ledger[0]['recipother']), 1)
self.assertEqual(ledger[0]['recipother'][0]['amount'], 0.8998)
self.assertEqual(len(ledger[0]['senderme']), 0)
self.assertEqual(ledger[0]['senderother'][0]['amount'], 11.8999)
self.assertEqual(ledger[5]['direction'], 'send')
self.assertEqual(len(ledger[5]['senderother']), 0)
self.assertEqual(len(ledger[5]['senderme']), 1)
self.assertEqual(ledger[5]['senderme'][0]['amount'], 1000.0)
def testSendtoaddress(self):
# Send 1 BTC
serializedUnsignedTx = \
self.jsonServer.jsonrpc_createustxtoaddress(TIAB_WLT_3_ADDR_3, \
BTC_TO_SEND)
unsignedTx = UnsignedTransaction().unserializeAscii(
serializedUnsignedTx)
# Should have 2 txouts to TIAB_WLT_3_ADDR_3 and the change
self.assertEqual(len(unsignedTx.decorTxOuts), 2)
foundTxOut = False
for txout in unsignedTx.decorTxOuts:
if script_to_addrStr(txout.binScript) == TIAB_WLT_3_ADDR_3:
self.assertEqual(txout.value, JSONtoAmount(BTC_TO_SEND))
foundTxOut = True
self.assertTrue(foundTxOut)
# Test two paths through signing method and make sure they are equal
# Wallets in the TIAB start out unencrypted
f = open(TX_FILENAME, 'w')
f.write(serializedUnsignedTx)
f.close()
serializedSignedTxUnencrypted = \
self.jsonServer.jsonrpc_signasciitransaction(TX_FILENAME)
self.jsonServer.jsonrpc_encryptwallet(PASSPHRASE1)
self.jsonServer.jsonrpc_walletpassphrase(PASSPHRASE1)
serializedSignedTxEncrypted = \
self.jsonServer.jsonrpc_signasciitransaction(TX_FILENAME)
# Other tests expect wallet to be unencrypted
self.wltA.unlock(securePassphrase=SecureBinaryData(PASSPHRASE1),
tempKeyLifetime=1000000)
self.wltA.changeWalletEncryption()
signedTxUnencrypted = UnsignedTransaction().unserializeAscii(
serializedSignedTxUnencrypted)
signedTxEncrypted = UnsignedTransaction().unserializeAscii(
serializedSignedTxEncrypted)
# check number of outputs 1 Btc goes to a single output and the other goes to change
self.assertEqual(len(signedTxUnencrypted.decorTxOuts), 2)
self.assertEqual(len(signedTxEncrypted.decorTxOuts), 2)
self.assertEqual(signedTxUnencrypted.asciiID,
signedTxEncrypted.asciiID)
self.assertTrue(
JSONtoAmount(BTC_TO_SEND) in [
signedTxEncrypted.decorTxOuts[0].value,
signedTxEncrypted.decorTxOuts[1].value
])
self.assertTrue(
JSONtoAmount(BTC_TO_SEND) in [
signedTxUnencrypted.decorTxOuts[0].value,
signedTxUnencrypted.decorTxOuts[1].value
])
f = open(TX_FILENAME, 'w')
f.write(signedTxEncrypted.serializeAscii())
f.close()
txHexToBroadcast = self.jsonServer.jsonrpc_gethextxtobroadcast(
TX_FILENAME)
finalPyTx = PyTx().unserialize(hex_to_binary(txHexToBroadcast))
self.assertEqual(len(finalPyTx.outputs), 2)
self.assertTrue(
JSONtoAmount(BTC_TO_SEND) in
[finalPyTx.outputs[0].value, finalPyTx.outputs[1].value])
def testSendmany(self):
# Send 1 BTC
serializedUnsignedTx = \
self.jsonServer.jsonrpc_createustxformany(None, ','.join([TIAB_WLT_3_ADDR_2, str(BTC_TO_SEND)]), \
','.join([TIAB_WLT_3_ADDR_3, str(BTC_TO_SEND)]))
unsignedTx = UnsignedTransaction().unserializeAscii(
serializedUnsignedTx)
# Should have 2 txouts to TIAB_WLT_3_ADDR_3 and the change
self.assertEqual(len(unsignedTx.decorTxOuts), 3)
txOutsFound = 0
for txout in unsignedTx.decorTxOuts:
if script_to_addrStr(
txout.binScript) in [TIAB_WLT_3_ADDR_2, TIAB_WLT_3_ADDR_3]:
self.assertEqual(txout.value, JSONtoAmount(BTC_TO_SEND))
txOutsFound += 1
self.assertEqual(txOutsFound, 2)
def testListUnspent(self):
actualResult = self.jsonServer.jsonrpc_listunspent()
tx1Found = False
tx5Found = False
for i in range(len(actualResult)):
if actualResult[i]['txid'] == EXPECTED_UNSPENT_TX1_HEX:
self.assertEqual(actualResult[i]['amount'], \
EXPECTED_UNSPENT_TX1_BAL)
self.assertEqual(actualResult[i]['confirmations'], \
EXPECTED_UNSPENT_TX1_CONF)
self.assertEqual(actualResult[i]['priority'], \
EXPECTED_UNSPENT_TX1_PRI)
tx1Found = True
elif actualResult[i]['txid'] == EXPECTED_UNSPENT_TX5_HEX:
self.assertEqual(actualResult[i]['amount'], \
EXPECTED_UNSPENT_TX5_BAL)
self.assertEqual(actualResult[i]['confirmations'], \
EXPECTED_UNSPENT_TX5_CONF)
self.assertEqual(actualResult[i]['priority'], \
EXPECTED_UNSPENT_TX5_PRI)
tx5Found = True
self.assertTrue(tx1Found)
self.assertTrue(tx5Found)
def testListAddrUnspent(self):
totStr = '%s,%s' % (TIAB_WLT_1_ADDR_3, TIAB_WLT_1_ADDR_8)
actualResult = self.jsonServer.jsonrpc_listaddrunspent(totStr)
self.assertEqual(actualResult['addrbalance'][TIAB_WLT_1_ADDR_3], \
TIAB_WLT_1_PK_UTXO_BAL_3)
self.assertEqual(actualResult['addrbalance'][TIAB_WLT_1_ADDR_8], \
TIAB_WLT_1_PK_UTXO_BAL_8)
# NB: Ideally, the TAB asserts would be against addresses with multiple
# UTXOs. As is, this test case works but could be better.
self.assertEqual(actualResult['totalbalance'],
TIAB_WLT_1_PK_UTXO_BAL_3 + TIAB_WLT_1_PK_UTXO_BAL_8)
self.assertEqual(actualResult['numutxo'], 2)
def testGetNewAddress(self):
actualResult = self.jsonServer.jsonrpc_getnewaddress()
self.assertEqual(actualResult, EXPECTED_TIAB_NEXT_ADDR)
def testGetBalance(self):
balances = {'spendable' : FIRST_WLT_BALANCE, \
'spend' : FIRST_WLT_BALANCE, \
'unconf' : 0, \
'unconfirmed' : 0, \
'total' : FIRST_WLT_BALANCE, \
'ultimate' : FIRST_WLT_BALANCE, \
'unspent' : FIRST_WLT_BALANCE, \
'full' : FIRST_WLT_BALANCE}
for balanceType in balances.keys():
result = self.jsonServer.jsonrpc_getbalance(balanceType)
self.assertEqual(result,
AmountToJSON(self.wltA.getBalance(balanceType)))
