test_udpSocket.cpp

Go to the documentation of this file.

 
 
#include <gtest/gtest.h>
 
#include <chrono>
#include <cstdint>
#include <string>
#include <vector>
 
#include "profinet/udpSocket.h"
#include "profinet/util.h"
 
namespace profinet
{
 
namespace
{
 
std::uint16_t GetLocalPort(const UdpSocket& socket)
{
  return socket.LocalEndpoint().port;
}
 
} // namespace
 
TEST(UdpSocketTests, RecvFromReturnsPayloadAndSourceEndpoint)
{
  UdpSocket receiver(0, std::chrono::duration<double>(1.0));
 
  const std::uint16_t receiverPort = GetLocalPort(receiver);
  ASSERT_NE(receiverPort, 0U);
 
  UdpSocket sender("127.0.0.1", receiverPort, std::chrono::duration<double>(1.0));
 
  const std::vector<std::uint8_t> payload{0x01, 0x02, 0x03, 0x04};
 
  sender.Send(payload);
 
  const UdpDatagram datagram = receiver.RecvFrom(128);
 
  EXPECT_EQ(datagram.data, payload);
  EXPECT_EQ(datagram.source.address, "127.0.0.1");
  EXPECT_NE(datagram.source.port, 0U);
}
 
TEST(UdpSocketTests, SendToSendsToSuppliedEndpoint)
{
  UdpSocket receiver(0, std::chrono::duration<double>(1.0));
 
  const std::uint16_t receiverPort = GetLocalPort(receiver);
  ASSERT_NE(receiverPort, 0U);
 
  UdpSocket sender("127.0.0.1", receiverPort, std::chrono::duration<double>(1.0));
 
  const std::vector<std::uint8_t> payload{0x10, 0x20, 0x30, 0x40};
 
  sender.Send(payload);
 
  const UdpDatagram received = receiver.RecvFrom(128);
 
  ASSERT_EQ(received.data, payload);
 
  const std::vector<std::uint8_t> response{0xAA, 0xBB, 0xCC};
 
  receiver.SendTo(response, received.source);
 
  const std::vector<std::uint8_t> echoed = sender.Recv(128);
 
  EXPECT_EQ(echoed, response);
}
 
TEST(UdpSocketTests, RecvFromReturnsEmptyDatagramOnTimeout)
{
  UdpSocket receiver(0, std::chrono::duration<double>(0.02));
 
  const UdpDatagram datagram = receiver.RecvFrom(128);
 
  EXPECT_TRUE(datagram.data.empty());
  EXPECT_TRUE(datagram.source.address.empty());
  EXPECT_EQ(datagram.source.port, 0U);
}
 
TEST(UdpSocketTests, MoveConstructionTransfersSocketOwnership)
{
  UdpSocket original(0, std::chrono::duration<double>(1.0));
 
  const UdpEndpoint endpoint = original.LocalEndpoint();
 
  UdpSocket moved(std::move(original));
 
  EXPECT_EQ(moved.LocalEndpoint(), endpoint);
}
 
TEST(UdpSocketTests, MoveAssignmentTransfersSocketOwnership)
{
  UdpSocket source(0, std::chrono::duration<double>(1.0));
  const UdpEndpoint sourceEndpoint = source.LocalEndpoint();
 
  UdpSocket destination(0, std::chrono::duration<double>(1.0));
 
  destination = std::move(source);
 
  EXPECT_EQ(destination.LocalEndpoint(), sourceEndpoint);
}
 
} // namespace profinet