PageRenderTime 26ms CodeModel.GetById 6ms RepoModel.GetById 0ms app.codeStats 0ms

/src/test/rpc_wallet_tests.cpp

https://gitlab.com/Ltaimao/bitcoin
C++ | 183 lines | 105 code | 34 blank | 44 comment | 8 complexity | 2eef683b4dcc37ac77cdd20a2c3581fb MD5 | raw file
    

  1. // Copyright (c) 2013-2014 The Bitcoin Core developers
    2. 

  2. // Distributed under the MIT/X11 software license, see the accompanying
    3. 

  3. // file COPYING or http://www.opensource.org/licenses/mit-license.php.
    4. 

  4. 

  5. #include "rpcserver.h"
    6. 

  6. #include "rpcclient.h"
    7. 

  7. 

  8. #include "base58.h"
    9. 

  9. #include "wallet.h"
    10. 

  10. 

  11. #include <boost/algorithm/string.hpp>
    12. 

  12. #include <boost/test/unit_test.hpp>
    13. 

  13. 

  14. using namespace std;
    15. 

  15. using namespace json_spirit;
    16. 

  16. 

  17. extern Array createArgs(int nRequired, const char* address1 = NULL, const char* address2 = NULL);
    18. 

  18. extern Value CallRPC(string args);
    19. 

  19. 

  20. extern CWallet* pwalletMain;
    21. 

  21. 

  22. BOOST_AUTO_TEST_SUITE(rpc_wallet_tests)
    23. 

  23. 

  24. BOOST_AUTO_TEST_CASE(rpc_addmultisig)
    25. 

  25. {
    26. 

  26.     LOCK(pwalletMain->cs_wallet);
    27. 

  27. 

  28.     rpcfn_type addmultisig = tableRPC["addmultisigaddress"]->actor;
    29. 

  29. 

  30.     // old, 65-byte-long:
    31. 

  31.     const char address1Hex[] = "0434e3e09f49ea168c5bbf53f877ff4206923858aab7c7e1df25bc263978107c95e35065a27ef6f1b27222db0ec97e0e895eaca603d3ee0d4c060ce3d8a00286c8";
    32. 

  32.     // new, compressed:
    33. 

  33.     const char address2Hex[] = "0388c2037017c62240b6b72ac1a2a5f94da790596ebd06177c8572752922165cb4";
    34. 

  34. 

  35.     Value v;
    36. 

  36.     CBitcoinAddress address;
    37. 

  37.     BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex), false));
    38. 

  38.     address.SetString(v.get_str());
    39. 

  39.     BOOST_CHECK(address.IsValid() && address.IsScript());
    40. 

  40. 

  41.     BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex, address2Hex), false));
    42. 

  42.     address.SetString(v.get_str());
    43. 

  43.     BOOST_CHECK(address.IsValid() && address.IsScript());
    44. 

  44. 

  45.     BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(2, address1Hex, address2Hex), false));
    46. 

  46.     address.SetString(v.get_str());
    47. 

  47.     BOOST_CHECK(address.IsValid() && address.IsScript());
    48. 

  48. 

  49.     BOOST_CHECK_THROW(addmultisig(createArgs(0), false), runtime_error);
    50. 

  50.     BOOST_CHECK_THROW(addmultisig(createArgs(1), false), runtime_error);
    51. 

  51.     BOOST_CHECK_THROW(addmultisig(createArgs(2, address1Hex), false), runtime_error);
    52. 

  52. 

  53.     BOOST_CHECK_THROW(addmultisig(createArgs(1, ""), false), runtime_error);
    54. 

  54.     BOOST_CHECK_THROW(addmultisig(createArgs(1, "NotAValidPubkey"), false), runtime_error);
    55. 

  55. 

  56.     string short1(address1Hex, address1Hex + sizeof(address1Hex) - 2); // last byte missing
    57. 

  57.     BOOST_CHECK_THROW(addmultisig(createArgs(2, short1.c_str()), false), runtime_error);
    58. 

  58. 

  59.     string short2(address1Hex + 1, address1Hex + sizeof(address1Hex)); // first byte missing
    60. 

  60.     BOOST_CHECK_THROW(addmultisig(createArgs(2, short2.c_str()), false), runtime_error);
    61. 

  61. }
    62. 

  62. 

  63. BOOST_AUTO_TEST_CASE(rpc_wallet)
    64. 

  64. {
    65. 

  65.     // Test RPC calls for various wallet statistics
    66. 

  66.     Value r;
    67. 

  67. 

  68.     LOCK2(cs_main, pwalletMain->cs_wallet);
    69. 

  69. 

  70.     CPubKey demoPubkey = pwalletMain->GenerateNewKey();
    71. 

  71.     CBitcoinAddress demoAddress = CBitcoinAddress(CTxDestination(demoPubkey.GetID()));
    72. 

  72.     Value retValue;
    73. 

  73.     string strAccount = "walletDemoAccount";
    74. 

  74.     string strPurpose = "receive";
    75. 

  75.     BOOST_CHECK_NO_THROW({ /*Initialize Wallet with an account */
    76. 

  76.         CWalletDB walletdb(pwalletMain->strWalletFile);
    77. 

  77.         CAccount account;
    78. 

  78.         account.vchPubKey = demoPubkey;
    79. 

  79.         pwalletMain->SetAddressBook(account.vchPubKey.GetID(), strAccount, strPurpose);
    80. 

  80.         walletdb.WriteAccount(strAccount, account);
    81. 

  81.     });
    82. 

  82. 

  83.     CPubKey setaccountDemoPubkey = pwalletMain->GenerateNewKey();
    84. 

  84.     CBitcoinAddress setaccountDemoAddress = CBitcoinAddress(CTxDestination(setaccountDemoPubkey.GetID()));
    85. 

  85. 

  86.     /*********************************
    87. 

  87.      * 			setaccount
    88. 

  88.      *********************************/
    89. 

  89.     BOOST_CHECK_NO_THROW(CallRPC("setaccount " + setaccountDemoAddress.ToString() + " nullaccount"));
    90. 

  90.     /* 1D1ZrZNe3JUo7ZycKEYQQiQAWd9y54F4XZ is not owned by the test wallet. */
    91. 

  91.     BOOST_CHECK_THROW(CallRPC("setaccount 1D1ZrZNe3JUo7ZycKEYQQiQAWd9y54F4XZ nullaccount"), runtime_error);
    92. 

  92.     BOOST_CHECK_THROW(CallRPC("setaccount"), runtime_error);
    93. 

  93.     /* 1D1ZrZNe3JUo7ZycKEYQQiQAWd9y54F4X (33 chars) is an illegal address (should be 34 chars) */
    94. 

  94.     BOOST_CHECK_THROW(CallRPC("setaccount 1D1ZrZNe3JUo7ZycKEYQQiQAWd9y54F4X nullaccount"), runtime_error);
    95. 

  95. 

  96.     /*********************************
    97. 

  97.      * 			listunspent
    98. 

  98.      *********************************/
    99. 

  99.     BOOST_CHECK_NO_THROW(CallRPC("listunspent"));
    100. 

  100.     BOOST_CHECK_THROW(CallRPC("listunspent string"), runtime_error);
    101. 

  101.     BOOST_CHECK_THROW(CallRPC("listunspent 0 string"), runtime_error);
    102. 

  102.     BOOST_CHECK_THROW(CallRPC("listunspent 0 1 not_array"), runtime_error);
    103. 

  103.     BOOST_CHECK_THROW(CallRPC("listunspent 0 1 [] extra"), runtime_error);
    104. 

  104.     BOOST_CHECK_NO_THROW(r = CallRPC("listunspent 0 1 []"));
    105. 

  105.     BOOST_CHECK(r.get_array().empty());
    106. 

  106. 

  107.     /*********************************
    108. 

  108.      * 		listreceivedbyaddress
    109. 

  109.      *********************************/
    110. 

  110.     BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaddress"));
    111. 

  111.     BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaddress 0"));
    112. 

  112.     BOOST_CHECK_THROW(CallRPC("listreceivedbyaddress not_int"), runtime_error);
    113. 

  113.     BOOST_CHECK_THROW(CallRPC("listreceivedbyaddress 0 not_bool"), runtime_error);
    114. 

  114.     BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaddress 0 true"));
    115. 

  115.     BOOST_CHECK_THROW(CallRPC("listreceivedbyaddress 0 true extra"), runtime_error);
    116. 

  116. 

  117.     /*********************************
    118. 

  118.      * 		listreceivedbyaccount
    119. 

  119.      *********************************/
    120. 

  120.     BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaccount"));
    121. 

  121.     BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaccount 0"));
    122. 

  122.     BOOST_CHECK_THROW(CallRPC("listreceivedbyaccount not_int"), runtime_error);
    123. 

  123.     BOOST_CHECK_THROW(CallRPC("listreceivedbyaccount 0 not_bool"), runtime_error);
    124. 

  124.     BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaccount 0 true"));
    125. 

  125.     BOOST_CHECK_THROW(CallRPC("listreceivedbyaccount 0 true extra"), runtime_error);
    126. 

  126. 

  127.     /*********************************
    128. 

  128.      * 		getrawchangeaddress
    129. 

  129.      *********************************/
    130. 

  130.     BOOST_CHECK_NO_THROW(CallRPC("getrawchangeaddress"));
    131. 

  131. 

  132.     /*********************************
    133. 

  133.      * 		getnewaddress
    134. 

  134.      *********************************/
    135. 

  135.     BOOST_CHECK_NO_THROW(CallRPC("getnewaddress"));
    136. 

  136.     BOOST_CHECK_NO_THROW(CallRPC("getnewaddress getnewaddress_demoaccount"));
    137. 

  137. 

  138.     /*********************************
    139. 

  139.      * 		getaccountaddress
    140. 

  140.      *********************************/
    141. 

  141.     BOOST_CHECK_NO_THROW(CallRPC("getaccountaddress \"\""));
    142. 

  142.     BOOST_CHECK_NO_THROW(CallRPC("getaccountaddress accountThatDoesntExists")); // Should generate a new account
    143. 

  143.     BOOST_CHECK_NO_THROW(retValue = CallRPC("getaccountaddress " + strAccount));
    144. 

  144.     BOOST_CHECK(CBitcoinAddress(retValue.get_str()).Get() == demoAddress.Get());
    145. 

  145. 

  146.     /*********************************
    147. 

  147.      * 			getaccount
    148. 

  148.      *********************************/
    149. 

  149.     BOOST_CHECK_THROW(CallRPC("getaccount"), runtime_error);
    150. 

  150.     BOOST_CHECK_NO_THROW(CallRPC("getaccount " + demoAddress.ToString()));
    151. 

  151. 

  152.     /*********************************
    153. 

  153.      * 	signmessage + verifymessage
    154. 

  154.      *********************************/
    155. 

  155.     BOOST_CHECK_NO_THROW(retValue = CallRPC("signmessage " + demoAddress.ToString() + " mymessage"));
    156. 

  156.     BOOST_CHECK_THROW(CallRPC("signmessage"), runtime_error);
    157. 

  157.     /* Should throw error because this address is not loaded in the wallet */
    158. 

  158.     BOOST_CHECK_THROW(CallRPC("signmessage 1QFqqMUD55ZV3PJEJZtaKCsQmjLT6JkjvJ mymessage"), runtime_error);
    159. 

  159. 

  160.     /* missing arguments */
    161. 

  161.     BOOST_CHECK_THROW(CallRPC("verifymessage " + demoAddress.ToString()), runtime_error);
    162. 

  162.     BOOST_CHECK_THROW(CallRPC("verifymessage " + demoAddress.ToString() + " " + retValue.get_str()), runtime_error);
    163. 

  163.     /* Illegal address */
    164. 

  164.     BOOST_CHECK_THROW(CallRPC("verifymessage 1D1ZrZNe3JUo7ZycKEYQQiQAWd9y54F4X " + retValue.get_str() + " mymessage"), runtime_error);
    165. 

  165.     /* wrong address */
    166. 

  166.     BOOST_CHECK(CallRPC("verifymessage 1D1ZrZNe3JUo7ZycKEYQQiQAWd9y54F4XZ " + retValue.get_str() + " mymessage").get_bool() == false);
    167. 

  167.     /* Correct address and signature but wrong message */
    168. 

  168.     BOOST_CHECK(CallRPC("verifymessage " + demoAddress.ToString() + " " + retValue.get_str() + " wrongmessage").get_bool() == false);
    169. 

  169.     /* Correct address, message and signature*/
    170. 

  170.     BOOST_CHECK(CallRPC("verifymessage " + demoAddress.ToString() + " " + retValue.get_str() + " mymessage").get_bool() == true);
    171. 

  171. 

  172.     /*********************************
    173. 

  173.      * 		getaddressesbyaccount
    174. 

  174.      *********************************/
    175. 

  175.     BOOST_CHECK_THROW(CallRPC("getaddressesbyaccount"), runtime_error);
    176. 

  176.     BOOST_CHECK_NO_THROW(retValue = CallRPC("getaddressesbyaccount " + strAccount));
    177. 

  177.     Array arr = retValue.get_array();
    178. 

  178.     BOOST_CHECK(arr.size() > 0);
    179. 

  179.     BOOST_CHECK(CBitcoinAddress(arr[0].get_str()).Get() == demoAddress.Get());
    180. 

  180. }
    181. 

  181. 

  182. 

  183. BOOST_AUTO_TEST_SUITE_END()
    184. 

  184.