From 2329f7ad2fc6e769d55f3c200ee7f7c9782ada88 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 8 Apr 2020 16:02:03 +0200
Subject: [PATCH] Make stream generation a generic mock

---
 core-writer/test/mock/stream.cpp        | 20 ++++++++++++++++++++
 core-writer/test/test_ZmqRecvModule.cpp | 21 ++-------------------
 2 files changed, 22 insertions(+), 19 deletions(-)
 create mode 100644 core-writer/test/mock/stream.cpp

diff --git a/core-writer/test/mock/stream.cpp b/core-writer/test/mock/stream.cpp
new file mode 100644
index 0000000..a7b00d9
--- /dev/null
+++ b/core-writer/test/mock/stream.cpp
@@ -0,0 +1,20 @@
+#include <zmq.hpp>
+
+void generate_stream(size_t n_messages)
+{
+    zmq::context_t context(1);
+    zmq::socket_t socket(context, ZMQ_PUSH);
+    socket.bind("tcp://127.0.0.1:11000");
+
+    string header = "{\"frame\": 0, \"shape\": [1,16], \"type\": \"uint8\"}";
+    zmq::const_buffer header_msg(header.c_str(), header.length());
+
+    size_t buffer_size = 16;
+    char buffer[buffer_size];
+    zmq::const_buffer buffer_msg(buffer, buffer_size);
+
+    for (size_t i=0; i<n_messages; i++) {
+        socket.send(header_msg, zmq::send_flags::sndmore);
+        socket.send(buffer_msg);
+    }
+}
\ No newline at end of file
diff --git a/core-writer/test/test_ZmqRecvModule.cpp b/core-writer/test/test_ZmqRecvModule.cpp
index 52b87a7..f4d99f7 100644
--- a/core-writer/test/test_ZmqRecvModule.cpp
+++ b/core-writer/test/test_ZmqRecvModule.cpp
@@ -5,27 +5,10 @@
 #include <string>
 #include "RingBuffer.hpp"
 
+#include "mock/stream.cpp"
+
 using namespace std;
 
-void generate_stream(size_t n_messages)
-{
-    zmq::context_t context(1);
-    zmq::socket_t socket(context, ZMQ_PUSH);
-    socket.bind("tcp://127.0.0.1:11000");
-
-    string header = "{\"frame\": 0, \"shape\": [1,16], \"type\": \"uint8\"}";
-    zmq::const_buffer header_msg(header.c_str(), header.length());
-
-    size_t buffer_size = 16;
-    char buffer[buffer_size];
-    zmq::const_buffer buffer_msg(buffer, buffer_size);
-
-    for (size_t i=0; i<n_messages; i++) {
-        socket.send(header_msg, zmq::send_flags::sndmore);
-        socket.send(buffer_msg);
-    }
-}
-
 TEST(ZmqRecvModule, basic_interaction)
 {
     RingBuffer ring_buffer(10);