Add is_block_segregated and is_tx_segregated, test.

Author: evoskuil

include/bitcoin/database/impl/query/archive_read.ipp

@@ -56,6 +56,20 @@ inline bool CLASS::is_coinbase(const tx_link& link) const NOEXCEPT
     return store_.tx.get(link, tx) && tx.coinbase;
 }
 
+TEMPLATE
+inline bool CLASS::is_tx_segregated(const tx_link& link) const NOEXCEPT
+{
+    size_t light{}, heavy{};
+    return get_tx_sizes(light, heavy, link) && heavy != light;
+}
+
+TEMPLATE
+inline bool CLASS::is_block_segregated(const header_link& link) const NOEXCEPT
+{
+    size_t light{}, heavy{};
+    return get_block_sizes(light, heavy, link) && heavy != light;
+}
+
 TEMPLATE
 inline bool CLASS::is_milestone(const header_link& link) const NOEXCEPT
 {

include/bitcoin/database/query.hpp

@@ -326,6 +326,8 @@ class query
     inline bool is_block(const hash_digest& key) const NOEXCEPT;
     inline bool is_tx(const hash_digest& key) const NOEXCEPT;
     inline bool is_coinbase(const tx_link& link) const NOEXCEPT;
+    inline bool is_tx_segregated(const tx_link& link) const NOEXCEPT;
+    inline bool is_block_segregated(const header_link& link) const NOEXCEPT;
     inline bool is_milestone(const header_link& link) const NOEXCEPT;
     inline bool is_associated(const header_link& link) const NOEXCEPT;
     inline bool is_confirmable(const header_link& link) const NOEXCEPT;

test/query/archive_read.cpp

@@ -28,6 +28,8 @@ static_assert(is_same_type<database::context::flag_t::integer, decltype(system::
 // nop event handler.
 const auto events_handler = [](auto, auto) {};
 
+// is_coinbase
+
 BOOST_AUTO_TEST_CASE(query_archive_read__is_coinbase__coinbase__true)
 {
     settings settings{};
@@ -63,6 +65,40 @@ BOOST_AUTO_TEST_CASE(query_archive_read__is_coinbase__non_coinbase__false)
     BOOST_REQUIRE(!query.is_coinbase(42));
 }
 
+// is_tx_segregated
+
+BOOST_AUTO_TEST_CASE(query_archive_read__is_tx_segregated__always__expected)
+{
+    settings settings{};
+    settings.path = TEST_DIRECTORY;
+    test::chunk_store store{ settings };
+    test::query_accessor query{ store };
+    BOOST_REQUIRE_EQUAL(store.create(events_handler), error::success);
+    BOOST_REQUIRE(query.initialize(test::genesis));
+    BOOST_REQUIRE(query.set(test::block1a, context{}, false, false));
+    BOOST_REQUIRE(!query.is_tx_segregated(0));
+    BOOST_REQUIRE( query.is_tx_segregated(1));
+    BOOST_REQUIRE(!query.is_tx_segregated(2));
+}
+
+// is_block_segregated
+
+BOOST_AUTO_TEST_CASE(query_archive_read__is_block_segregated__always__expected)
+{
+    settings settings{};
+    settings.path = TEST_DIRECTORY;
+    test::chunk_store store{ settings };
+    test::query_accessor query{ store };
+    BOOST_REQUIRE_EQUAL(store.create(events_handler), error::success);
+    BOOST_REQUIRE(query.initialize(test::genesis));
+    BOOST_REQUIRE(query.set(test::block1a, context{}, false, false));
+    BOOST_REQUIRE(!query.is_block_segregated(0));
+    BOOST_REQUIRE( query.is_block_segregated(1));
+    BOOST_REQUIRE(!query.is_block_segregated(2));
+}
+
+// is_milestone
+
 BOOST_AUTO_TEST_CASE(query_archive_read__is_milestone__genesis__false)
 {
     settings settings{};