mirror of
https://github.com/paulscherrerinstitute/sf_daq_buffer.git
synced 2026-06-07 16:48:41 +02:00
Add basic test for writer receiver
This commit is contained in:
@@ -0,0 +1,69 @@
|
||||
#include <gtest/gtest.h>
|
||||
#include "WriterZmqReceiver.hpp"
|
||||
#include "bitshuffle/bitshuffle.h"
|
||||
#include "zmq.h"
|
||||
|
||||
TEST(WriterZmqReceiver, basic_test)
|
||||
{
|
||||
size_t n_modules = 4;
|
||||
uint64_t pulse_id = 12345;
|
||||
|
||||
auto ctx = zmq_ctx_new();
|
||||
zmq_ctx_set (ctx, ZMQ_IO_THREADS, 1);
|
||||
|
||||
void* sockets[n_modules];
|
||||
for (size_t i = 0; i < n_modules; i++) {
|
||||
sockets[i] = zmq_socket(ctx, ZMQ_PUSH);
|
||||
|
||||
int linger = 0;
|
||||
if (zmq_setsockopt(sockets[i], ZMQ_LINGER, &linger,
|
||||
sizeof(linger)) != 0) {
|
||||
throw runtime_error(zmq_strerror(errno));
|
||||
}
|
||||
|
||||
stringstream ipc_addr;
|
||||
ipc_addr << REPLAY_STREAM_IPC_URL << i;
|
||||
const auto ipc = ipc_addr.str();
|
||||
|
||||
if (zmq_bind(sockets[i], ipc.c_str()) != 0) {
|
||||
throw runtime_error(zmq_strerror(errno));
|
||||
}
|
||||
}
|
||||
this_thread::sleep_for(chrono::milliseconds(100));
|
||||
|
||||
WriterZmqReceiver receiver(ctx, REPLAY_STREAM_IPC_URL, n_modules);
|
||||
this_thread::sleep_for(chrono::milliseconds(100));
|
||||
|
||||
size_t compressed_frame_size = 5000;
|
||||
auto frame_buffer = make_unique<char[]>(compressed_frame_size);
|
||||
|
||||
ImageMetadata image_metadata;
|
||||
auto compress_size = bshuf_compress_lz4_bound(
|
||||
MODULE_N_PIXELS, PIXEL_N_BYTES, MODULE_N_PIXELS);
|
||||
auto image_buffer = make_unique<char[]>(compress_size * n_modules);
|
||||
|
||||
for (size_t i = 0; i < n_modules; i++) {
|
||||
|
||||
CompressedModuleFrame frame_metadata;
|
||||
frame_metadata.module_frame.pulse_id = pulse_id;
|
||||
frame_metadata.module_frame.frame_index = pulse_id+100;
|
||||
frame_metadata.module_frame.n_received_packets = 128;
|
||||
frame_metadata.module_frame.daq_rec = 4;
|
||||
|
||||
frame_metadata.is_frame_present = 1;
|
||||
frame_metadata.compressed_size = compressed_frame_size;
|
||||
|
||||
zmq_send(sockets[i],
|
||||
&frame_metadata,
|
||||
sizeof(CompressedModuleFrame),
|
||||
ZMQ_SNDMORE);
|
||||
|
||||
zmq_send(sockets[i],
|
||||
(char*)(frame_buffer.get()),
|
||||
compressed_frame_size,
|
||||
0);
|
||||
}
|
||||
|
||||
receiver.get_next_image(pulse_id, &image_metadata, image_buffer.get());
|
||||
EXPECT_EQ(pulse_id, image_metadata.pulse_id);
|
||||
}
|
||||
Reference in New Issue
Block a user