fprime/Drv/Udp/test/ut/UdpTester.hpp

// ======================================================================
// \title  TcpClient/test/ut/Tester.hpp
// \author mstarch
// \brief  hpp file for ByteStreamDriverModel test harness implementation class
//
// \copyright
// Copyright 2009-2015, by the California Institute of Technology.
// ALL RIGHTS RESERVED.  United States Government Sponsorship
// acknowledged.
//
// ======================================================================

#ifndef TESTER_HPP
#define TESTER_HPP

#include "UdpGTestBase.hpp"
#include "Drv/Udp/UdpComponentImpl.hpp"
#include "Drv/Ip/TcpServerSocket.hpp"

#define SEND_DATA_BUFFER_SIZE 1024

namespace Drv {

  class UdpTester :
    public UdpGTestBase
  {
      // Maximum size of histories storing events, telemetry, and port outputs
      static const U32 MAX_HISTORY_SIZE = 1000;
      // Instance ID supplied to the component instance under test
      static const FwEnumStoreType TEST_INSTANCE_ID = 0;
      // Queue depth supplied to component instance under test
      static const FwSizeType TEST_INSTANCE_QUEUE_DEPTH = 100;

      // ----------------------------------------------------------------------
      // Construction and destruction
      // ----------------------------------------------------------------------

    public:

      //! Construct object UdpTester
      //!
      UdpTester();

      void initSetup();

      //! Destroy object UdpTester
      //!
      ~UdpTester();

    public:

      // ----------------------------------------------------------------------
      // Tests
      // ----------------------------------------------------------------------

      //! Test basic messaging
      //!
      void test_basic_messaging();

      //! Test basic reconnection behavior
      //!
      void test_multiple_messaging();

      //! Test receive via thread
      //!
      void test_receive_thread();

      //! Test advanced (duration) reconnect
      //!
      void test_advanced_reconnect();

      // Helpers
      void test_with_loop(U32 iterations, bool recv_thread=false);

      bool wait_on_change(Drv::IpSocket &socket, bool open, U32 iterations);

    private:

      // ----------------------------------------------------------------------
      // Handler overrides for typed from ports
      // ----------------------------------------------------------------------

      //! Handler for from_recv
      //!
      void from_recv_handler(
          const FwIndexType portNum, /*!< The port number*/
          Fw::Buffer &recvBuffer,
          const ByteStreamStatus &recvStatus
      ) override;

      //! Handler for from_allocate
      //!
      Fw::Buffer from_allocate_handler(
          const FwIndexType portNum, /*!< The port number*/
          U32 size
      ) override;

    private:

      // ----------------------------------------------------------------------
      // Helper methods
      // ----------------------------------------------------------------------

      //! Connect ports
      //!
      void connectPorts();

      //! Initialize components
      //!
      void initComponents();

    private:

      // ----------------------------------------------------------------------
      // Variables
      // ----------------------------------------------------------------------

      //! The component under test
      //!
      UdpComponentImpl component;
      Fw::Buffer m_data_buffer;
      Fw::Buffer m_data_buffer2;
      U8 m_data_storage[SEND_DATA_BUFFER_SIZE];
      std::atomic<bool> m_spinner;
  };

} // end namespace Drv

#endif