tests: device3_test: add graph impl test

5 files changed, 197 insertions, 0 deletions

diff --git a/host/tests/common/CMakeLists.txt b/host/tests/common/CMakeLists.txt
new file mode 100644
index 000000000..91a70dfae
--- /dev/null
+++ b/host/tests/common/CMakeLists.txt
@@ -0,0 +1,15 @@
+#
+# Copyright 2018 Ettus Research, a National Instruments Company
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+#
+
+########################################################################
+# Build uhd_test static lib
+########################################################################
+INCLUDE_DIRECTORIES("${CMAKE_SOURCE_DIR}/lib/include")
+ADD_LIBRARY(uhd_test ${CMAKE_CURRENT_SOURCE_DIR}/mock_ctrl_iface_impl.cpp
+                     ${CMAKE_CURRENT_SOURCE_DIR}/mock_zero_copy.cpp
+                     ${CMAKE_SOURCE_DIR}/lib/rfnoc/graph_impl.cpp
+                     ${CMAKE_SOURCE_DIR}/lib/rfnoc/async_msg_handler.cpp
+)
\ No newline at end of file
diff --git a/host/tests/common/mock_ctrl_iface_impl.cpp b/host/tests/common/mock_ctrl_iface_impl.cpp
new file mode 100644
index 000000000..24a938e4f
--- /dev/null
+++ b/host/tests/common/mock_ctrl_iface_impl.cpp
@@ -0,0 +1,35 @@
+//
+// Copyright 2018 Ettus Research, a National Instruments Company
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+
+#include "mock_ctrl_iface_impl.hpp"
+static const uint64_t TEST_NOC_ID = 0xAAAABBBBCCCCDDDD;
+
+uint64_t mock_ctrl_iface_impl::send_cmd_pkt(
+        const size_t addr,
+        const size_t data,
+        const bool readback,
+        const uint64_t timestamp
+) {
+    if (not readback) {
+        std::cout << str(boost::format("[MOCK] poke to addr: %016X, data == %016X") % addr % data) << std::endl;
+    } else {
+        std::cout << str(boost::format("[MOCK] peek64 to addr: %016X") % data) << std::endl;
+        switch (data) {
+            case uhd::rfnoc::SR_READBACK_REG_ID:
+                return TEST_NOC_ID;
+            case uhd::rfnoc::SR_READBACK_REG_FIFOSIZE:
+                return 0x0000000000010000;
+            case uhd::rfnoc::SR_READBACK_REG_USER:
+                return 0x0123456789ABCDEF;
+            case uhd::rfnoc::SR_READBACK_COMPAT:
+                return uhd::rfnoc::NOC_SHELL_COMPAT_MAJOR << 32 |
+                       uhd::rfnoc::NOC_SHELL_COMPAT_MINOR;
+            default:
+                return 0;
+        }
+    }
+    return 0;
+}
\ No newline at end of file
diff --git a/host/tests/common/mock_ctrl_iface_impl.hpp b/host/tests/common/mock_ctrl_iface_impl.hpp
new file mode 100644
index 000000000..8d2aafed6
--- /dev/null
+++ b/host/tests/common/mock_ctrl_iface_impl.hpp
@@ -0,0 +1,30 @@
+//
+// Copyright 2018 Ettus Research, a National Instruments Company
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+
+
+#ifndef INCLUDED_MOCK_CTRL_IFACE_IMPL_HPP
+#define INCLUDED_MOCK_CTRL_IFACE_IMPL_HPP
+
+#include <uhd/rfnoc/constants.hpp>
+#include <uhdlib/rfnoc/ctrl_iface.hpp>
+#include <boost/thread/mutex.hpp>
+#include <boost/thread/thread.hpp>
+#include <boost/format.hpp>
+#include <boost/bind.hpp>
+#include <boost/make_shared.hpp>
+#include <queue>
+
+class mock_ctrl_iface_impl : public uhd::rfnoc::ctrl_iface
+{
+
+    uint64_t send_cmd_pkt(
+            const size_t addr,
+            const size_t data,
+            const bool readback=false,
+            const uint64_t timestamp=0
+    );
+};
+#endif /* INCLUDED_MOCK_CTRL_IFACE_IMPL_HPP */
\ No newline at end of file
diff --git a/host/tests/common/mock_zero_copy.cpp b/host/tests/common/mock_zero_copy.cpp
new file mode 100644
index 000000000..b1d384ec7
--- /dev/null
+++ b/host/tests/common/mock_zero_copy.cpp
@@ -0,0 +1,29 @@
+//
+// Copyright 2018 Ettus Research, a National Instruments Company
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+
+
+#include "mock_zero_copy.hpp"
+#include <boost/shared_ptr.hpp>
+
+uhd::transport::managed_recv_buffer::sptr mock_zero_copy::get_recv_buff(double) {
+    if (_rx_mems.empty()) {
+        return uhd::transport::managed_recv_buffer::sptr(); // timeout
+    }
+    _mrbs.push_back(boost::make_shared<mock_mrb>());
+    uhd::transport::managed_recv_buffer::sptr mrb =
+        _mrbs.back()->get_new(_rx_mems.front(), _rx_lens.front());
+    _rx_mems.pop_front();
+    _rx_lens.pop_front();
+    return mrb;
+}
+
+uhd::transport::managed_send_buffer::sptr mock_zero_copy::get_send_buff(double) {
+    _msbs.push_back(boost::make_shared<mock_msb>());
+    _tx_mems.push_back(
+        boost::shared_array<uint8_t>(new uint8_t[SEND_BUFF_SIZE]));
+    _tx_lens.push_back(SEND_BUFF_SIZE);
+    return _msbs.back()->get_new(_tx_mems.back(), &_tx_lens.back());
+}
diff --git a/host/tests/common/mock_zero_copy.hpp b/host/tests/common/mock_zero_copy.hpp
new file mode 100644
index 000000000..838975129
--- /dev/null
+++ b/host/tests/common/mock_zero_copy.hpp
@@ -0,0 +1,88 @@
+//
+// Copyright 2018 Ettus Research, a National Instruments Company
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+
+#ifndef INCLUDED_MOCK_XPORT_HPP
+#define INCLUDED_MOCK_XPORT_HPP
+
+#include <uhdlib/rfnoc/xports.hpp>
+#include <uhd/transport/chdr.hpp>
+#include <uhd/transport/vrt_if_packet.hpp>
+#include <uhd/transport/zero_copy.hpp>
+#include <uhd/types/endianness.hpp>
+#include <uhd/types/sid.hpp>
+#include <uhd/utils/byteswap.hpp>
+#include <boost/make_shared.hpp>
+#include <boost/shared_ptr.hpp>
+#include <boost/shared_array.hpp>
+#include <list>
+#include <vector>
+/***********************************************************************
+ * Transport mockups
+ **********************************************************************/
+/*! A single transport class that implements send() and recv()
+ *
+ * Tx and Rx are separate. We can access the other end of the FIFOs from
+ * this class.
+ */
+static constexpr size_t SEND_BUFF_SIZE = 1024;
+static constexpr size_t RECV_BUFF_SIZE = 1024;
+
+/***********************************************************************
+ * Dummy managed buffers for testing
+ **********************************************************************/
+class mock_msb : public uhd::transport::managed_send_buffer {
+    public:
+    void release(void) { /* nop */
+    }
+
+    sptr get_new(boost::shared_array<uint8_t> mem, size_t* len) {
+        _mem = mem;
+        return make(this, mem.get(), *len);
+    }
+
+    private:
+    boost::shared_array<uint8_t> _mem;
+};
+
+
+class mock_mrb : public uhd::transport::managed_recv_buffer {
+    public:
+    void release(void) { /* nop */
+    }
+
+    sptr get_new(boost::shared_array<uint8_t> mem, size_t len) {
+        _mem = mem;
+        return make(this, _mem.get(), len);
+    }
+
+    private:
+    boost::shared_array<uint8_t> _mem;
+};
+
+class mock_zero_copy : public uhd::transport::zero_copy_if {
+    public:
+    typedef boost::shared_ptr<mock_zero_copy> sptr;
+    uhd::transport::managed_recv_buffer::sptr get_recv_buff(double);
+
+    uhd::transport::managed_send_buffer::sptr get_send_buff(double);
+
+    size_t get_num_recv_frames(void) const { return 1; }
+    size_t get_num_send_frames(void) const { return 1; }
+    size_t get_recv_frame_size(void) const { return RECV_BUFF_SIZE; }
+    size_t get_send_frame_size(void) const { return SEND_BUFF_SIZE; }
+
+    private:
+    std::list<boost::shared_array<uint8_t>> _tx_mems;
+    std::list<size_t> _tx_lens;
+
+    std::list<boost::shared_array<uint8_t>> _rx_mems;
+    std::list<size_t> _rx_lens;
+
+    std::vector<boost::shared_ptr<mock_msb>> _msbs;
+    std::vector<boost::shared_ptr<mock_mrb>> _mrbs;
+};
+
+#endif /*INCLUDED_MOCK_XPORT_HPP*/
\ No newline at end of file