src/test/rpc_tests.cpp

   1 #include <boost/test/unit_test.hpp>
   2 #include <boost/foreach.hpp>
   3
   4 #include "base58.h"
   5 #include "util.h"
   6 #include "json/json_spirit_reader_template.h"
   7 #include "json/json_spirit_writer_template.h"
   8 #include "json/json_spirit_utils.h"
   9
  10 using namespace std;
  11 using namespace json_spirit;
  12
  13 typedef Value(*rpcfn_type)(const Array& params, bool fHelp);
  14 extern map<string, rpcfn_type> mapCallTable;
  15
  16 BOOST_AUTO_TEST_SUITE(rpc_tests)
  17
  18 static Array
  19 createArgs(int nRequired, const char* address1=NULL, const char* address2=NULL)
  20 {
  21     Array result;
  22     result.push_back(nRequired);
  23     Array addresses;
  24     if (address1) addresses.push_back(address1);
  25     if (address2) addresses.push_back(address1);
  26     result.push_back(addresses);
  27     return result;
  28 }
  29
  30 // This can be removed this when addmultisigaddress is enabled on main net:
  31 struct TestNetFixture
  32 {
  33     TestNetFixture() { fTestNet = true; }
  34     ~TestNetFixture() { fTestNet = false; }
  35 };
  36
  37 BOOST_FIXTURE_TEST_CASE(rpc_addmultisig, TestNetFixture)
  38 {
  39     rpcfn_type addmultisig = mapCallTable["addmultisigaddress"];
  40
  41     // old, 65-byte-long:
  42     const char* address1Hex = "0434e3e09f49ea168c5bbf53f877ff4206923858aab7c7e1df25bc263978107c95e35065a27ef6f1b27222db0ec97e0e895eaca603d3ee0d4c060ce3d8a00286c8";
  43     // new, compressed:
  44     const char* address2Hex = "0388c2037017c62240b6b72ac1a2a5f94da790596ebd06177c8572752922165cb4";
  45
  46     Value v;
  47     CBitcoinAddress address;
  48     BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex), false));
  49     address.SetString(v.get_str());
  50     BOOST_CHECK(address.IsValid() && address.IsScript());
  51
  52     BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex, address2Hex), false));
  53     address.SetString(v.get_str());
  54     BOOST_CHECK(address.IsValid() && address.IsScript());
  55
  56     BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(2, address1Hex, address2Hex), false));
  57     address.SetString(v.get_str());
  58     BOOST_CHECK(address.IsValid() && address.IsScript());
  59
  60     BOOST_CHECK_THROW(addmultisig(createArgs(0), false), runtime_error);
  61     BOOST_CHECK_THROW(addmultisig(createArgs(1), false), runtime_error);
  62     BOOST_CHECK_THROW(addmultisig(createArgs(2, address1Hex), false), runtime_error);
  63
  64     BOOST_CHECK_THROW(addmultisig(createArgs(1, ""), false), runtime_error);
  65     BOOST_CHECK_THROW(addmultisig(createArgs(1, "NotAValidPubkey"), false), runtime_error);
  66
  67     string short1(address1Hex, address1Hex+sizeof(address1Hex)-2); // last byte missing
  68     BOOST_CHECK_THROW(addmultisig(createArgs(2, short1.c_str()), false), runtime_error);
  69
  70     string short2(address1Hex+2, address1Hex+sizeof(address1Hex)); // first byte missing
  71     BOOST_CHECK_THROW(addmultisig(createArgs(2, short2.c_str()), false), runtime_error);
  72 }
  73
  74 BOOST_AUTO_TEST_SUITE_END()