Make core-buffer folder with all common stuff used in the buffer

core-buffer/test/CMakeLists.txt

@@ -0,0 +1,5 @@
+add_executable(core-buffer_tests test_main.cpp)
+
+target_link_libraries(core-buffer_tests
+        core-buffer
+        gtest)

core-buffer/test/mock/udp.hpp

@@ -0,0 +1,16 @@
+#ifndef MOCK_UDP_H
+#define MOCK_UDP_H
+
+const int MOCK_UDP_PORT(13000);
+
+sockaddr_in get_server_address(uint16_t udp_port)
+{
+    sockaddr_in server_address = {0};
+    server_address.sin_family = AF_INET;
+    server_address.sin_addr.s_addr = INADDR_ANY;
+    server_address.sin_port = htons(udp_port);
+
+    return server_address;
+}
+
+#endif

core-buffer/test/test_UdpReceiver.cpp

@@ -0,0 +1,99 @@
+#include <netinet/in.h>
+#include <jungfrau.hpp>
+#include "gtest/gtest.h"
+#include "UdpReceiver.hpp"
+#include "mock/udp.hpp"
+
+
+TEST(UdpReceiver, simple_recv)
+{
+    uint16_t udp_port = MOCK_UDP_PORT;
+
+    auto send_socket_fd = socket(AF_INET,SOCK_DGRAM,0);
+    ASSERT_TRUE(send_socket_fd >= 0);
+
+    UdpReceiver udp_receiver;
+    udp_receiver.bind(udp_port);
+
+    jungfrau_packet send_udp_buffer;
+    send_udp_buffer.packetnum = 91;
+    send_udp_buffer.framenum = 92;
+    send_udp_buffer.bunchid = 93;
+    send_udp_buffer.debug = 94;
+
+    auto server_address = get_server_address(udp_port);
+    ::sendto(
+            send_socket_fd,
+            &send_udp_buffer,
+            JUNGFRAU_BYTES_PER_PACKET,
+            0,
+            (sockaddr*) &server_address,
+            sizeof(server_address));
+
+    jungfrau_packet recv_udp_buffer;
+    ASSERT_TRUE(udp_receiver.receive(
+            &recv_udp_buffer, JUNGFRAU_BYTES_PER_PACKET));
+
+    EXPECT_EQ(send_udp_buffer.packetnum, recv_udp_buffer.packetnum);
+    EXPECT_EQ(send_udp_buffer.framenum, recv_udp_buffer.framenum);
+    EXPECT_EQ(send_udp_buffer.bunchid, recv_udp_buffer.bunchid);
+    EXPECT_EQ(send_udp_buffer.debug, recv_udp_buffer.debug);
+
+    ASSERT_FALSE(udp_receiver.receive(
+            &recv_udp_buffer, JUNGFRAU_BYTES_PER_PACKET));
+
+    udp_receiver.close();
+    ::close(send_socket_fd);
+}
+
+TEST(UdpReceiver, false_recv)
+{
+    uint16_t udp_port = MOCK_UDP_PORT;
+
+    auto send_socket_fd = socket(AF_INET,SOCK_DGRAM,0);
+    ASSERT_TRUE(send_socket_fd >= 0);
+
+    UdpReceiver udp_receiver;
+    udp_receiver.bind(udp_port);
+
+    jungfrau_packet send_udp_buffer;
+    jungfrau_packet recv_udp_buffer;
+
+    auto server_address = get_server_address(udp_port);
+
+    ::sendto(
+            send_socket_fd,
+            &send_udp_buffer,
+            JUNGFRAU_BYTES_PER_PACKET-1,
+            0,
+            (sockaddr*) &server_address,
+            sizeof(server_address));
+
+    ASSERT_FALSE(udp_receiver.receive(
+            &recv_udp_buffer, JUNGFRAU_BYTES_PER_PACKET));
+
+    ::sendto(
+            send_socket_fd,
+            &send_udp_buffer,
+            JUNGFRAU_BYTES_PER_PACKET,
+            0,
+            (sockaddr*) &server_address,
+            sizeof(server_address));
+
+    ASSERT_TRUE(udp_receiver.receive(
+            &recv_udp_buffer, JUNGFRAU_BYTES_PER_PACKET));
+
+    ::sendto(
+            send_socket_fd,
+            &send_udp_buffer,
+            JUNGFRAU_BYTES_PER_PACKET-1,
+            0,
+            (sockaddr*) &server_address,
+            sizeof(server_address));
+
+    ASSERT_TRUE(udp_receiver.receive(
+            &recv_udp_buffer, JUNGFRAU_BYTES_PER_PACKET-1));
+
+    udp_receiver.close();
+    ::close(send_socket_fd);
+}

core-buffer/test/test_UdpRecvModule.cpp

@@ -0,0 +1,91 @@
+#include "gtest/gtest.h"
+#include "UdpRecvModule.hpp"
+#include "jungfrau.hpp"
+#include "mock/udp.hpp"
+
+using namespace std;
+
+TEST(UdpRecvModule, basic_interaction)
+{
+    uint16_t udp_port(12000);
+
+    RingBuffer<UdpFrameMetadata> ring_buffer(10);
+    UdpRecvModule udp_recv_module(ring_buffer);
+
+    udp_recv_module.start_recv(udp_port, JUNGFRAU_DATA_BYTES_PER_FRAME);
+
+    EXPECT_THROW(
+            udp_recv_module.start_recv(udp_port, JUNGFRAU_BYTES_PER_PACKET),
+            runtime_error);
+
+    // Stop should never throw an exception.
+    udp_recv_module.stop_recv();
+    EXPECT_NO_THROW(udp_recv_module.stop_recv());
+}
+
+TEST(UdpRecvModule, simple_recv)
+{
+    uint16_t udp_port(MOCK_UDP_PORT);
+    size_t n_msg(128);
+
+    RingBuffer<UdpFrameMetadata> ring_buffer(10);
+    UdpRecvModule udp_recv_module(ring_buffer);
+
+    udp_recv_module.start_recv(udp_port, JUNGFRAU_DATA_BYTES_PER_FRAME);
+
+    this_thread::sleep_for(chrono::milliseconds(100));
+
+    // The first slot should be already reserved in the ring buffer.
+    ASSERT_FALSE(ring_buffer.is_empty());
+
+    auto send_socket_fd = socket(AF_INET,SOCK_DGRAM,0);
+    ASSERT_TRUE(send_socket_fd >= 0);
+
+    auto server_address = get_server_address(udp_port);
+
+    jungfrau_packet send_udp_buffer;
+    send_udp_buffer.bunchid = 100;
+    send_udp_buffer.debug = 1000;
+
+    send_udp_buffer.framenum = 1;
+    for (size_t i=0; i<n_msg; i++) {
+        send_udp_buffer.packetnum = i;
+
+        ::sendto(
+                send_socket_fd,
+                &send_udp_buffer,
+                JUNGFRAU_BYTES_PER_PACKET,
+                0,
+                (sockaddr*) &server_address,
+                sizeof(server_address));
+    }
+
+    ASSERT_FALSE(ring_buffer.is_empty());
+    auto result = ring_buffer.read();
+    // The slot should be reserved, but not yet committed.
+    // Only with next frame packet commit.
+    ASSERT_TRUE(result.first == nullptr);
+
+    // When packet from new frame is received, the previous frame should be
+    // committed to the ring buffer.
+    send_udp_buffer.framenum = 2;
+    for (size_t i=0; i<128; i++){
+        send_udp_buffer.packetnum = i;
+
+        ::sendto(
+                send_socket_fd,
+                &send_udp_buffer,
+                JUNGFRAU_BYTES_PER_PACKET,
+                0,
+                (sockaddr*) &server_address,
+                sizeof(server_address));
+    }
+
+    this_thread::sleep_for(chrono::milliseconds(100));
+
+    ASSERT_FALSE(ring_buffer.is_empty());
+    auto result2 = ring_buffer.read();
+    ASSERT_TRUE(result2.first != nullptr);
+
+    ::close(send_socket_fd);
+}

core-buffer/test/test_buffer_utils.cpp

@@ -0,0 +1,47 @@
+#include "gtest/gtest.h"
+#include "buffer_utils.hpp"
+
+using namespace std;
+
+TEST(BufferUtils, get_filename)
+{
+    auto expected_file = "/root/device-1/12300000/12345000.h5";
+
+    auto root_folder = "/root";
+    auto device_name = "device-1";
+
+    auto result = get_filename(
+            root_folder,
+            device_name,
+            12345000);
+
+    ASSERT_EQ(result, expected_file);
+
+    auto result2 = get_filename(
+            root_folder,
+            device_name,
+            12345999);
+
+    ASSERT_EQ(result2, expected_file);
+
+    auto result3 = get_filename(
+            root_folder,
+            device_name,
+            12346000);
+
+    ASSERT_NE(result3, expected_file);
+
+    auto result4 = get_filename(
+            root_folder,
+            device_name,
+            12344999);
+
+    ASSERT_NE(result4, expected_file);
+}
+
+TEST(BufferUtils, get_file_frame_index)
+{
+    ASSERT_EQ(get_file_frame_index(12345000), 0);
+    ASSERT_EQ(get_file_frame_index(12345543), 543);
+    ASSERT_EQ(get_file_frame_index(12345999), 999);
+}

core-buffer/test/test_main.cpp

@@ -0,0 +1,11 @@
+#include "gtest/gtest.h"
+#include "test_UdpReceiver.cpp"
+#include "test_UdpRecvModule.cpp"
+#include "test_buffer_utils.cpp"
+
+using namespace std;
+
+int main(int argc, char **argv) {
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}