mirror of
https://github.com/nasa/fprime.git
synced 2025-12-10 00:44:37 -06:00
6915c93c10
* Update IP stack source to use SizeType instead of I/U32 * Update UTs reflecting U32/I32 change to SizeType * Static cast sizes per review * Fix commented out thing
116 lines
3.5 KiB
C++
116 lines
3.5 KiB
C++
// ======================================================================
|
|
// \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 "Drv/Ip/TcpServerSocket.hpp"
|
|
#include "Drv/TcpClient/TcpClientComponentImpl.hpp"
|
|
#include "TcpClientGTestBase.hpp"
|
|
|
|
#define SEND_DATA_BUFFER_SIZE 1024
|
|
|
|
namespace Drv {
|
|
|
|
class TcpClientTester : public TcpClientGTestBase {
|
|
// Maximum size of histories storing events, telemetry, and port outputs
|
|
static const FwSizeType 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 TcpClientTester
|
|
//!
|
|
TcpClientTester();
|
|
|
|
//! Destroy object TcpClientTester
|
|
//!
|
|
~TcpClientTester();
|
|
|
|
public:
|
|
// ----------------------------------------------------------------------
|
|
// Tests
|
|
// ----------------------------------------------------------------------
|
|
|
|
void setup_helper(Drv::TcpServerSocket& server, Drv::SocketDescriptor& server_fd, bool recv_thread, bool reconnect);
|
|
|
|
void test_basic_messaging();
|
|
|
|
void test_multiple_messaging();
|
|
|
|
void test_receive_thread();
|
|
|
|
void test_advanced_reconnect();
|
|
|
|
void test_no_automatic_send_connection();
|
|
|
|
void test_no_automatic_recv_connection();
|
|
|
|
void test_with_loop(U32 iterations, bool recv_thread = false);
|
|
|
|
void test_buffer_deallocation();
|
|
|
|
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& ByteStreamStatus) override;
|
|
|
|
//! Handler for from_allocate
|
|
//!
|
|
Fw::Buffer from_allocate_handler(const FwIndexType portNum, /*!< The port number*/
|
|
FwSizeType size) override;
|
|
|
|
private:
|
|
// ----------------------------------------------------------------------
|
|
// Helper methods
|
|
// ----------------------------------------------------------------------
|
|
|
|
bool wait_on_change(Drv::IpSocket& socket, bool open, U32 iterations);
|
|
|
|
//! Connect ports
|
|
//!
|
|
void connectPorts();
|
|
|
|
//! Initialize components
|
|
//!
|
|
void initComponents();
|
|
|
|
private:
|
|
// ----------------------------------------------------------------------
|
|
// Variables
|
|
// ----------------------------------------------------------------------
|
|
|
|
//! The component under test
|
|
//!
|
|
TcpClientComponentImpl 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
|