sf-daq/sf_daq_buffer
0
0
Fork 0
mirror of https://github.com/paulscherrerinstitute/sf_daq_buffer.git synced 2026-05-03 13:14:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add simple transfer test

Browse Source
This commit is contained in:
babic_a
2020-04-02 18:02:45 +02:00
parent 378cf77754
commit dbf33bd8e2
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core-writer/test/test_ZmqRecvModule.cpp
+14 -7
View File
@@ -13,12 +13,13 @@ void generate_stream(size_t n_messages)
zmq::socket_t socket(context, ZMQ_PUSH);
socket.bind("tcp://127.0.0.1:11000");
for (size_t i=0; i<n_messages; i++) {
// TODO: Implement the actual sending.
// Header: {"frame": 0, "shape": [16], "type": "uint8"}
// Data: bytes of 16x uint8
}
string header = "{\"frame\": 0, \"shape\": [1,16], \"type\": \"uint8\"}";
unique_ptr<char[]> buffer(make_unique<char[]>(16));
for (size_t i=0; i<n_messages; i++) {
socket.send(header.c_str(), header.length(), ZMQ_SNDMORE);
socket.send(buffer.get(), 16);
}
}
TEST(ZmqRecvModule, basic_interaction)
@@ -27,7 +28,7 @@ TEST(ZmqRecvModule, basic_interaction)
ZmqRecvModule zmq_recv_module(ring_buffer, {});
uint8_t n_receivers = 4;
zmq_recv_module.start_recv("tcp://127.0.0.1:10000", n_receivers);
zmq_recv_module.start_recv("tcp://127.0.0.1:11000", n_receivers);
zmq_recv_module.start_writing();
zmq_recv_module.stop_writing();
@@ -37,7 +38,7 @@ TEST(ZmqRecvModule, basic_interaction)
TEST(ZmqRecvModule, simple_recv)
{
size_t n_msg = 10;
size_t n_msg = 9;
thread sender(generate_stream, n_msg);
RingBuffer ring_buffer(n_msg);
@@ -47,6 +48,8 @@ TEST(ZmqRecvModule, simple_recv)
zmq_recv_module.start_recv("tcp://127.0.0.1:11000", 4);
sender.join();
this_thread::sleep_for(chrono::milliseconds(100));
zmq_recv_module.stop_recv();
for (size_t i=0;i<n_msg;i++) {
@@ -56,5 +59,9 @@ TEST(ZmqRecvModule, simple_recv)
ASSERT_TRUE(data.second != nullptr);
}
// no more messages in the buffer.
auto data = ring_buffer.read();
ASSERT_TRUE(data.first == nullptr);
ASSERT_TRUE(ring_buffer.is_empty());
}