GH-1039 add ability to broadcast a block from test_control_plugin

AntelopeIO · Nov 22, 2024 · 6fe2a17 · 6fe2a17
1 parent 20609f9
commit 6fe2a17
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/plugins/net_plugin/include/eosio/net_plugin/net_plugin.hpp b/plugins/net_plugin/include/eosio/net_plugin/net_plugin.hpp
@@ -94,6 +94,9 @@ namespace eosio {
         void register_increment_failed_p2p_connections(std::function<void()>&&);
         void register_increment_dropped_trxs(std::function<void()>&&);
 
+        // for testing
+        void broadcast_block(const signed_block_ptr& b, const block_id_type& id);
+
       private:
         std::shared_ptr<class net_plugin_impl> my;
    };

diff --git a/plugins/net_plugin/net_plugin.cpp b/plugins/net_plugin/net_plugin.cpp
@@ -4459,6 +4459,11 @@ namespace eosio {
       my->increment_dropped_trxs = std::move(fun);
    }
 
+   void net_plugin::broadcast_block(const signed_block_ptr& b, const block_id_type& id) {
+      fc_dlog(logger, "broadcasting block ${n} ${id}", ("n", b->block_num())("id", id));
+      my->dispatcher.bcast_block(b, id);
+   }
+
    //----------------------------------------------------------------------------
 
    size_t connections_manager::number_connections() const {

diff --git a/plugins/test_control_plugin/CMakeLists.txt b/plugins/test_control_plugin/CMakeLists.txt
@@ -4,6 +4,6 @@ add_library( test_control_plugin
              test_control_plugin.cpp
              ${HEADERS} )
 
-target_link_libraries( test_control_plugin producer_plugin chain_plugin http_client_plugin appbase eosio_chain )
+target_link_libraries( test_control_plugin producer_plugin chain_plugin net_plugin http_client_plugin appbase eosio_chain )
 target_include_directories( test_control_plugin
                             PUBLIC "${CMAKE_CURRENT_SOURCE_DIR}/include" )
diff --git a/plugins/test_control_plugin/include/eosio/test_control_plugin/test_control_plugin.hpp b/plugins/test_control_plugin/include/eosio/test_control_plugin/test_control_plugin.hpp
@@ -1,6 +1,7 @@
 #pragma once
 #include <eosio/chain/application.hpp>
 #include <eosio/chain_plugin/chain_plugin.hpp>
+#include <eosio/net_plugin/net_plugin.hpp>
 #include <fc/variant.hpp>
 #include <memory>
 
@@ -53,7 +54,7 @@ class read_write {
 
 class test_control_plugin : public plugin<test_control_plugin> {
 public:
-   APPBASE_PLUGIN_REQUIRES((chain_plugin))
+   APPBASE_PLUGIN_REQUIRES((chain_plugin)(net_plugin))
 
    test_control_plugin();
    test_control_plugin(const test_control_plugin&) = delete;