187 lines
8.5 KiB
C++
187 lines
8.5 KiB
C++
// Copyright (c) 2013-2014 The Bitcoin Core developers
|
|
// Copyright (c) 2017-2019 The PIVX developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include "rpc/server.h"
|
|
#include "rpc/client.h"
|
|
|
|
#include "base58.h"
|
|
#include "wallet/wallet.h"
|
|
|
|
#include "test/test_agrarian.h"
|
|
|
|
#include <boost/algorithm/string.hpp>
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include <univalue.h>
|
|
|
|
using namespace std;
|
|
|
|
extern UniValue createArgs(int nRequired, const char* address1 = NULL, const char* address2 = NULL);
|
|
extern UniValue CallRPC(string args);
|
|
|
|
extern CWallet* pwalletMain;
|
|
|
|
BOOST_FIXTURE_TEST_SUITE(rpc_wallet_tests, TestingSetup)
|
|
|
|
BOOST_AUTO_TEST_CASE(rpc_addmultisig)
|
|
{
|
|
LOCK(pwalletMain->cs_wallet);
|
|
|
|
rpcfn_type addmultisig = tableRPC["addmultisigaddress"]->actor;
|
|
|
|
// old, 65-byte-long:
|
|
const char address1Hex[] = "041431A18C7039660CD9E3612A2A47DC53B69CB38EA4AD743B7DF8245FD0438F8E7270415F1085B9DC4D7DA367C69F1245E27EE5552A481D6854184C80F0BB8456";
|
|
// new, compressed:
|
|
const char address2Hex[] = "029BBEFF390CE736BD396AF43B52A1C14ED52C086B1E5585C15931F68725772BAC";
|
|
|
|
UniValue v;
|
|
CBitcoinAddress address;
|
|
BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex), false));
|
|
address.SetString(v.get_str());
|
|
BOOST_CHECK(address.IsValid() && address.IsScript());
|
|
|
|
BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex, address2Hex), false));
|
|
address.SetString(v.get_str());
|
|
BOOST_CHECK(address.IsValid() && address.IsScript());
|
|
|
|
BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(2, address1Hex, address2Hex), false));
|
|
address.SetString(v.get_str());
|
|
BOOST_CHECK(address.IsValid() && address.IsScript());
|
|
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(0), false), runtime_error);
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(1), false), runtime_error);
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(2, address1Hex), false), runtime_error);
|
|
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(1, ""), false), runtime_error);
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(1, "NotAValidPubkey"), false), runtime_error);
|
|
|
|
string short1(address1Hex, address1Hex + sizeof(address1Hex) - 2); // last byte missing
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(2, short1.c_str()), false), runtime_error);
|
|
|
|
string short2(address1Hex + 1, address1Hex + sizeof(address1Hex)); // first byte missing
|
|
BOOST_CHECK_THROW(addmultisig(createArgs(2, short2.c_str()), false), runtime_error);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(rpc_wallet)
|
|
{
|
|
// Test RPC calls for various wallet statistics
|
|
UniValue r;
|
|
|
|
LOCK2(cs_main, pwalletMain->cs_wallet);
|
|
|
|
CPubKey demoPubkey = pwalletMain->GenerateNewKey();
|
|
CBitcoinAddress demoAddress = CBitcoinAddress(CTxDestination(demoPubkey.GetID()));
|
|
UniValue retValue;
|
|
string strAccount = "walletDemoAccount";
|
|
string strPurpose = "receive";
|
|
BOOST_CHECK_NO_THROW({ /*Initialize Wallet with an account */
|
|
CWalletDB walletdb(pwalletMain->strWalletFile);
|
|
CAccount account;
|
|
account.vchPubKey = demoPubkey;
|
|
pwalletMain->SetAddressBook(account.vchPubKey.GetID(), strAccount, strPurpose);
|
|
walletdb.WriteAccount(strAccount, account);
|
|
});
|
|
|
|
CPubKey setaccountDemoPubkey = pwalletMain->GenerateNewKey();
|
|
CBitcoinAddress setaccountDemoAddress = CBitcoinAddress(CTxDestination(setaccountDemoPubkey.GetID()));
|
|
|
|
/*********************************
|
|
* setaccount
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("setaccount " + setaccountDemoAddress.ToString() + " nullaccount"));
|
|
/* D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5R is not owned by the test wallet. */
|
|
BOOST_CHECK_THROW(CallRPC("setaccount D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5R nullaccount"), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("setaccount"), runtime_error);
|
|
/* D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5 (33 chars) is an illegal address (should be 34 chars) */
|
|
BOOST_CHECK_THROW(CallRPC("setaccount D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5 nullaccount"), runtime_error);
|
|
|
|
/*********************************
|
|
* listunspent
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("listunspent"));
|
|
BOOST_CHECK_THROW(CallRPC("listunspent string"), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("listunspent 0 string"), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("listunspent 0 1 not_array"), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("listunspent 0 1 [] extra"), runtime_error);
|
|
BOOST_CHECK_NO_THROW(r = CallRPC("listunspent 0 1 []"));
|
|
BOOST_CHECK(r.get_array().empty());
|
|
|
|
/*********************************
|
|
* listreceivedbyaddress
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaddress"));
|
|
BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaddress 0"));
|
|
BOOST_CHECK_THROW(CallRPC("listreceivedbyaddress not_int"), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("listreceivedbyaddress 0 not_bool"), runtime_error);
|
|
BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaddress 0 true"));
|
|
BOOST_CHECK_THROW(CallRPC("listreceivedbyaddress 0 true extra"), runtime_error);
|
|
|
|
/*********************************
|
|
* listreceivedbyaccount
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaccount"));
|
|
BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaccount 0"));
|
|
BOOST_CHECK_THROW(CallRPC("listreceivedbyaccount not_int"), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("listreceivedbyaccount 0 not_bool"), runtime_error);
|
|
BOOST_CHECK_NO_THROW(CallRPC("listreceivedbyaccount 0 true"));
|
|
BOOST_CHECK_THROW(CallRPC("listreceivedbyaccount 0 true extra"), runtime_error);
|
|
|
|
/*********************************
|
|
* getrawchangeaddress
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("getrawchangeaddress"));
|
|
|
|
/*********************************
|
|
* getnewaddress
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("getnewaddress"));
|
|
BOOST_CHECK_NO_THROW(CallRPC("getnewaddress getnewaddress_demoaccount"));
|
|
|
|
/*********************************
|
|
* getaccountaddress
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(CallRPC("getaccountaddress \"\""));
|
|
BOOST_CHECK_NO_THROW(CallRPC("getaccountaddress accountThatDoesntExists")); // Should generate a new account
|
|
BOOST_CHECK_NO_THROW(retValue = CallRPC("getaccountaddress " + strAccount));
|
|
BOOST_CHECK(CBitcoinAddress(retValue.get_str()).Get() == demoAddress.Get());
|
|
|
|
/*********************************
|
|
* getaccount
|
|
*********************************/
|
|
BOOST_CHECK_THROW(CallRPC("getaccount"), runtime_error);
|
|
BOOST_CHECK_NO_THROW(CallRPC("getaccount " + demoAddress.ToString()));
|
|
|
|
/*********************************
|
|
* signmessage + verifymessage
|
|
*********************************/
|
|
BOOST_CHECK_NO_THROW(retValue = CallRPC("signmessage " + demoAddress.ToString() + " mymessage"));
|
|
BOOST_CHECK_THROW(CallRPC("signmessage"), runtime_error);
|
|
/* Should throw error because this address is not loaded in the wallet */
|
|
BOOST_CHECK_THROW(CallRPC("signmessage D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5R mymessage"), runtime_error);
|
|
|
|
/* missing arguments */
|
|
BOOST_CHECK_THROW(CallRPC("verifymessage " + demoAddress.ToString()), runtime_error);
|
|
BOOST_CHECK_THROW(CallRPC("verifymessage " + demoAddress.ToString() + " " + retValue.get_str()), runtime_error);
|
|
/* Illegal address */
|
|
BOOST_CHECK_THROW(CallRPC("verifymessage D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5 " + retValue.get_str() + " mymessage"), runtime_error);
|
|
/* wrong address */
|
|
BOOST_CHECK(CallRPC("verifymessage D8w12Vu3WVhn543dgrUUf9uYu6HLwnPm5R " + retValue.get_str() + " mymessage").get_bool() == false);
|
|
/* Correct address and signature but wrong message */
|
|
BOOST_CHECK(CallRPC("verifymessage " + demoAddress.ToString() + " " + retValue.get_str() + " wrongmessage").get_bool() == false);
|
|
/* Correct address, message and signature*/
|
|
BOOST_CHECK(CallRPC("verifymessage " + demoAddress.ToString() + " " + retValue.get_str() + " mymessage").get_bool() == true);
|
|
|
|
/*********************************
|
|
* getaddressesbyaccount
|
|
*********************************/
|
|
BOOST_CHECK_THROW(CallRPC("getaddressesbyaccount"), runtime_error);
|
|
BOOST_CHECK_NO_THROW(retValue = CallRPC("getaddressesbyaccount " + strAccount));
|
|
UniValue arr = retValue.get_array();
|
|
BOOST_CHECK(arr.size() > 0);
|
|
BOOST_CHECK(CBitcoinAddress(arr[0].get_str()).Get() == demoAddress.Get());
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|