From 67651b9fffa6e6728a70348d0d174c3c0f802bfc Mon Sep 17 00:00:00 2001
From: Martin Braun <martin.braun@ettus.com>
Date: Fri, 9 Oct 2020 11:34:43 +0200
Subject: tests: mock_block: Allow mock blocks to carry MB controllers

---
 host/include/uhd/rfnoc/mock_block.hpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'host/include')

diff --git a/host/include/uhd/rfnoc/mock_block.hpp b/host/include/uhd/rfnoc/mock_block.hpp
index fef8815b7..a456d30a6 100644
--- a/host/include/uhd/rfnoc/mock_block.hpp
+++ b/host/include/uhd/rfnoc/mock_block.hpp
@@ -9,6 +9,7 @@
 #include <uhd/config.hpp>
 #include <uhd/property_tree.hpp>
 #include <uhd/rfnoc/defaults.hpp>
+#include <uhd/rfnoc/mb_controller.hpp>
 #include <uhd/rfnoc/noc_block_base.hpp>
 #include <uhd/rfnoc/register_iface.hpp>
 #include <uhd/types/time_spec.hpp>
@@ -184,6 +185,7 @@ UHD_API mock_block_container get_mock_block(const noc_id_t noc_id,
     const uhd::device_addr_t& args                     = uhd::device_addr_t(),
     const size_t mtu                                   = 8000,
     const device_type_t device_id                      = ANY_DEVICE,
-    std::shared_ptr<mock_reg_iface_t> client_reg_iface = nullptr);
+    std::shared_ptr<mock_reg_iface_t> client_reg_iface = nullptr,
+    mb_controller::sptr mbc                            = nullptr);
 
 }}; // namespace uhd::rfnoc
-- 
cgit v1.2.3