From b1cde5eea5743af5843225fec690cca3de4cddbd Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 15 Apr 2020 10:42:39 +0200
Subject: [PATCH] Add simple recv test to ZmqReceiver

---
 core-writer/test/test_ZmqReceiver.cpp | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/core-writer/test/test_ZmqReceiver.cpp b/core-writer/test/test_ZmqReceiver.cpp
index e72f455..039526f 100644
--- a/core-writer/test/test_ZmqReceiver.cpp
+++ b/core-writer/test/test_ZmqReceiver.cpp
@@ -1,5 +1,7 @@
+#include <thread>
 #include "gtest/gtest.h"
 #include "ZmqReceiver.hpp"
+#include "mock/stream.hpp"
 
 using namespace std;
 namespace pt = boost::property_tree;
@@ -146,4 +148,23 @@ TEST(ZmqReceiver, read_json_header)
 
   auto module_number = reinterpret_cast<uint64_t*>(metadata->header_values.at("module_number").get());
   ASSERT_TRUE(module_number[0] == 0);
+}
+
+TEST(ZmqReceiver, simple_recv)
+{
+    size_t n_msg = 10;
+
+    thread sender(generate_stream, n_msg);
+    RingBuffer<FrameMetadata> ring_buffer(n_msg);
+
+    ZmqReceiver receiver({});
+    receiver.connect(MOCK_STREAM_ADDRESS, 100);
+
+    this_thread::sleep_for(chrono::milliseconds(100));
+
+    for (size_t i=0; i<n_msg; i++) {
+        EXPECT_TRUE(receiver.receive().first != nullptr);
+    }
+
+    sender.join();
 }
\ No newline at end of file