Tests for address.get_balance().

Author: evoskuil

test/protocols/electrum/electrum.cpp

@@ -22,7 +22,8 @@
 #include <future>
 #include <boost/format.hpp>
 
-electrum_setup_fixture::electrum_setup_fixture(const initializer& setup)
+electrum_setup_fixture::electrum_setup_fixture(const initializer& setup,
+    bool address_index)
   : config_
     {
       system::chain::selection::mainnet,
@@ -33,6 +34,9 @@ electrum_setup_fixture::electrum_setup_fixture(const initializer& setup)
     {
         [&]() NOEXCEPT -> const database::settings&
         {
+            if (!address_index)
+                config_.database.address_buckets = 0;
+
             config_.database.path = TEST_DIRECTORY;
             return config_.database;
         }()

test/protocols/electrum/electrum.hpp

@@ -29,7 +29,8 @@ struct electrum_setup_fixture
     DELETE_COPY_MOVE(electrum_setup_fixture);
 
     using initializer = std::function<bool(test::query_t&)>;
-    explicit electrum_setup_fixture(const initializer& setup);
+    explicit electrum_setup_fixture(const initializer& setup,
+        bool address_index=true);
     ~electrum_setup_fixture();
 
     boost::json::value get(const std::string& request);
@@ -72,4 +73,16 @@ struct electrum_three_block_confirmed_address_setup_fixture
     }
 };
 
+struct electrum_disabled_address_index_setup_fixture
+  : electrum_setup_fixture
+{
+    inline electrum_disabled_address_index_setup_fixture()
+      : electrum_setup_fixture([](test::query_t& query)
+        {
+            return test::setup_three_block_confirmed_address_store(query);
+        }, false)
+    {
+    }
+};
+
 #endif

test/protocols/electrum/electrum_addresses.cpp

@@ -19,9 +19,96 @@
 #include "../../test.hpp"
 #include "electrum.hpp"
 
-BOOST_FIXTURE_TEST_SUITE(electrum_tests, electrum_three_block_confirmed_address_setup_fixture)
+using namespace system;
+static const code not_found{ server::error::not_found };
+static const code wrong_version{ server::error::wrong_version };
+static const code not_implemented{ server::error::not_implemented };
+static const code invalid_argument{ server::error::invalid_argument };
+
+BOOST_FIXTURE_TEST_SUITE(electrum_disabled_address_tests, electrum_disabled_address_index_setup_fixture)
+
+BOOST_AUTO_TEST_CASE(electrum__blockchain_address_get_balance__no_address_index__not_implemented)
+{
+    BOOST_REQUIRE(!query_.address_enabled());
+    BOOST_REQUIRE(handshake(electrum::version::v1_0));
+
+    const auto request = R"({"id":901,"method":"blockchain.address.get_balance","params":["%1%"]})" "\n";
+    const auto response = get((boost::format(request) % "1JqDybm2nWTENrHvMyafbSXXtTk5Uv5QAn").str());
+    REQUIRE_NO_THROW_TRUE(response.at("error").as_object().at("code").is_int64());
+    BOOST_REQUIRE_EQUAL(response.at("error").as_object().at("code").as_int64(), not_implemented.value());
+}
+
+BOOST_AUTO_TEST_SUITE_END()
+
+BOOST_FIXTURE_TEST_SUITE(electrum_tests, electrum_ten_block_setup_fixture)
 
 // blockchain.address.get_balance
+
+BOOST_AUTO_TEST_CASE(electrum__blockchain_address_get_balance__missing_arguments__dropped)
+{
+    BOOST_REQUIRE(query_.address_enabled());
+    BOOST_REQUIRE(handshake(electrum::version::v1_0));
+
+    const auto response = get(R"({"id":901,"method":"blockchain.address.get_balance","params":[]})" "\n");
+    REQUIRE_NO_THROW_TRUE(response.at("dropped").as_bool());
+}
+
+BOOST_AUTO_TEST_CASE(electrum__blockchain_address_get_balance__empty_v1_3__wrong_version)
+{
+    BOOST_REQUIRE(handshake(electrum::version::v1_3));
+
+    const auto response = get(R"({"id":902,"method":"blockchain.address.get_balance","params":[""]})" "\n");
+    REQUIRE_NO_THROW_TRUE(response.at("error").as_object().at("code").is_int64());
+    BOOST_REQUIRE_EQUAL(response.at("error").as_object().at("code").as_int64(), wrong_version.value());
+}
+
+BOOST_AUTO_TEST_CASE(electrum__blockchain_address_get_balance__invalid_address__invalid_argument)
+{
+    BOOST_REQUIRE(handshake(electrum::version::v1_1));
+
+    const auto response = get(R"({"id":903,"method":"blockchain.address.get_balance","params":["invalid"]})" "\n");
+    REQUIRE_NO_THROW_TRUE(response.at("error").as_object().at("code").is_int64());
+    BOOST_REQUIRE_EQUAL(response.at("error").as_object().at("code").as_int64(), invalid_argument.value());
+}
+
+BOOST_AUTO_TEST_CASE(electrum__blockchain_address_get_balance__not_found_address__zero)
+{
+    BOOST_REQUIRE(handshake(electrum::version::v1_0));
+
+    const auto request = R"({"id":904,"method":"blockchain.address.get_balance","params":["%1%"]})" "\n";
+    const auto response = get((boost::format(request) % "1JqDybm2nWTENrHvMyafbSXXtTk5Uv5QAn").str());
+    REQUIRE_NO_THROW_TRUE(response.at("result").is_object());
+
+    const auto& result = response.at("result").as_object();
+    REQUIRE_NO_THROW_TRUE(result.at("confirmed").is_int64());
+    REQUIRE_NO_THROW_TRUE(result.at("unconfirmed").is_int64());
+    BOOST_REQUIRE_EQUAL(result.at("confirmed").as_int64(), 0x00);
+    BOOST_REQUIRE_EQUAL(result.at("unconfirmed").as_int64(), 0x00);
+}
+
+BOOST_AUTO_TEST_CASE(electrum__blockchain_address_get_balance__confirmed_and_unconfirmed_address__expected)
+{
+    BOOST_REQUIRE(handshake(electrum::version::v1_0));
+
+    // Add one confirmed p2sh/p2kh block and one unconfirmed.
+    BOOST_REQUIRE(query_.set(test::bogus_block10, database::context{ 0, 10, 0 }, false, false));
+    BOOST_REQUIRE(query_.set(test::bogus_block11, database::context{ 0, 11, 0 }, false, false));
+    BOOST_REQUIRE(query_.push_confirmed(query_.to_header(test::bogus_block10.hash()), true));
+
+    const auto request = R"({"id":905,"method":"blockchain.address.get_balance","params":["%1%"]})" "\n";
+    const auto response = get((boost::format(request) % "1BaMPFdqMUQ46BV8iRcwbVfsam57oBLMM").str());
+    REQUIRE_NO_THROW_TRUE(response.at("result").is_object());
+
+    const auto& result = response.at("result").as_object();
+    REQUIRE_NO_THROW_TRUE(result.at("confirmed").is_int64());
+    REQUIRE_NO_THROW_TRUE(result.at("unconfirmed").is_int64());
+    BOOST_REQUIRE_EQUAL(result.at("confirmed").as_int64(), 0x09); // bogus_block11
+
+    // Currently unconfirmed ALWAYS returns zero (no graph to order conflicts).
+    BOOST_REQUIRE_EQUAL(result.at("unconfirmed").as_int64(), 0x00);
+    ////BOOST_REQUIRE_EQUAL(result.at("unconfirmed").as_int64(), 0x10 + 0x11 + 0x12); // bogus_block11
+}
+
 // blockchain.address.get_history
 // blockchain.address.get_mempool
 // blockchain.address.list_unspent