test/functional/receivedby.py

   1 #!/usr/bin/env python3
   2 # Copyright (c) 2014-2016 The Bitcoin Core developers
   3 # Distributed under the MIT software license, see the accompanying
   4 # file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5 """Test the listreceivedbyaddress RPC."""
   6
   7 from test_framework.test_framework import BitcoinTestFramework
   8 from test_framework.util import *
   9
  10 def get_sub_array_from_array(object_array, to_match):
  11     '''
  12         Finds and returns a sub array from an array of arrays.
  13         to_match should be a unique idetifier of a sub array
  14     '''
  15     for item in object_array:
  16         all_match = True
  17         for key,value in to_match.items():
  18             if item[key] != value:
  19                 all_match = False
  20         if not all_match:
  21             continue
  22         return item
  23     return []
  24
  25 class ReceivedByTest(BitcoinTestFramework):
  26     def set_test_params(self):
  27         self.num_nodes = 2
  28         self.enable_mocktime()
  29
  30     def run_test(self):
  31         '''
  32         listreceivedbyaddress Test
  33         '''
  34         # Send from node 0 to 1
  35         addr = self.nodes[1].getnewaddress()
  36         txid = self.nodes[0].sendtoaddress(addr, 0.1)
  37         self.sync_all()
  38
  39         #Check not listed in listreceivedbyaddress because has 0 confirmations
  40         assert_array_result(self.nodes[1].listreceivedbyaddress(),
  41                            {"address":addr},
  42                            { },
  43                            True)
  44         #Bury Tx under 10 block so it will be returned by listreceivedbyaddress
  45         self.nodes[1].generate(10)
  46         self.sync_all()
  47         assert_array_result(self.nodes[1].listreceivedbyaddress(),
  48                            {"address":addr},
  49                            {"address":addr, "account":"", "amount":Decimal("0.1"), "confirmations":10, "txids":[txid,]})
  50         #With min confidence < 10
  51         assert_array_result(self.nodes[1].listreceivedbyaddress(5),
  52                            {"address":addr},
  53                            {"address":addr, "account":"", "amount":Decimal("0.1"), "confirmations":10, "txids":[txid,]})
  54         #With min confidence > 10, should not find Tx
  55         assert_array_result(self.nodes[1].listreceivedbyaddress(11),{"address":addr},{ },True)
  56
  57         #Empty Tx
  58         addr = self.nodes[1].getnewaddress()
  59         assert_array_result(self.nodes[1].listreceivedbyaddress(0,True),
  60                            {"address":addr},
  61                            {"address":addr, "account":"", "amount":0, "confirmations":0, "txids":[]})
  62
  63         '''
  64             getreceivedbyaddress Test
  65         '''
  66         # Send from node 0 to 1
  67         addr = self.nodes[1].getnewaddress()
  68         txid = self.nodes[0].sendtoaddress(addr, 0.1)
  69         self.sync_all()
  70
  71         #Check balance is 0 because of 0 confirmations
  72         balance = self.nodes[1].getreceivedbyaddress(addr)
  73         if balance != Decimal("0.0"):
  74             raise AssertionError("Wrong balance returned by getreceivedbyaddress, %0.2f"%(balance))
  75
  76         #Check balance is 0.1
  77         balance = self.nodes[1].getreceivedbyaddress(addr,0)
  78         if balance != Decimal("0.1"):
  79             raise AssertionError("Wrong balance returned by getreceivedbyaddress, %0.2f"%(balance))
  80
  81         #Bury Tx under 10 block so it will be returned by the default getreceivedbyaddress
  82         self.nodes[1].generate(10)
  83         self.sync_all()
  84         balance = self.nodes[1].getreceivedbyaddress(addr)
  85         if balance != Decimal("0.1"):
  86             raise AssertionError("Wrong balance returned by getreceivedbyaddress, %0.2f"%(balance))
  87
  88         '''
  89             listreceivedbyaccount + getreceivedbyaccount Test
  90         '''
  91         #set pre-state
  92         addrArr = self.nodes[1].getnewaddress()
  93         account = self.nodes[1].getaccount(addrArr)
  94         received_by_account_json = get_sub_array_from_array(self.nodes[1].listreceivedbyaccount(),{"account":account})
  95         if len(received_by_account_json) == 0:
  96             raise AssertionError("No accounts found in node")
  97         balance_by_account = self.nodes[1].getreceivedbyaccount(account)
  98
  99         txid = self.nodes[0].sendtoaddress(addr, 0.1)
 100         self.sync_all()
 101
 102         # listreceivedbyaccount should return received_by_account_json because of 0 confirmations
 103         assert_array_result(self.nodes[1].listreceivedbyaccount(),
 104                            {"account":account},
 105                            received_by_account_json)
 106
 107         # getreceivedbyaddress should return same balance because of 0 confirmations
 108         balance = self.nodes[1].getreceivedbyaccount(account)
 109         if balance != balance_by_account:
 110             raise AssertionError("Wrong balance returned by getreceivedbyaccount, %0.2f"%(balance))
 111
 112         self.nodes[1].generate(10)
 113         self.sync_all()
 114         # listreceivedbyaccount should return updated account balance
 115         assert_array_result(self.nodes[1].listreceivedbyaccount(),
 116                            {"account":account},
 117                            {"account":received_by_account_json["account"], "amount":(received_by_account_json["amount"] + Decimal("0.1"))})
 118
 119         # getreceivedbyaddress should return updates balance
 120         balance = self.nodes[1].getreceivedbyaccount(account)
 121         if balance != balance_by_account + Decimal("0.1"):
 122             raise AssertionError("Wrong balance returned by getreceivedbyaccount, %0.2f"%(balance))
 123
 124         #Create a new account named "mynewaccount" that has a 0 balance
 125         self.nodes[1].getaccountaddress("mynewaccount")
 126         received_by_account_json = get_sub_array_from_array(self.nodes[1].listreceivedbyaccount(0,True),{"account":"mynewaccount"})
 127         if len(received_by_account_json) == 0:
 128             raise AssertionError("No accounts found in node")
 129
 130         # Test includeempty of listreceivedbyaccount
 131         if received_by_account_json["amount"] != Decimal("0.0"):
 132             raise AssertionError("Wrong balance returned by listreceivedbyaccount, %0.2f"%(received_by_account_json["amount"]))
 133
 134         # Test getreceivedbyaccount for 0 amount accounts
 135         balance = self.nodes[1].getreceivedbyaccount("mynewaccount")
 136         if balance != Decimal("0.0"):
 137             raise AssertionError("Wrong balance returned by getreceivedbyaccount, %0.2f"%(balance))
 138
 139 if __name__ == '__main__':
 140     ReceivedByTest().main()