mirror of
https://github.com/paulscherrerinstitute/sf_daq_buffer.git
synced 2026-04-21 02:44:35 +02:00
99 lines
2.6 KiB
C++
99 lines
2.6 KiB
C++
#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);
|
|
} |