TCPSocket_TEST.cpp

Go to the documentation of this file.

#include <cppunit/extensions/HelperMacros.h>
#include "../TCPSocket.h"
 
#include <cstring>
 
static const char send_msg[] = "The quick brown fox jumps over the lazy dog";
static char recv_msg[sizeof(send_msg)];
static const int len = sizeof(send_msg);
 
class TCPSocket_TEST : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE( TCPSocket_TEST );
    CPPUNIT_TEST( testSocketHandle );
    CPPUNIT_TEST( testPeerStatus );
    CPPUNIT_TEST_SUITE_END();
 
private:
    NET::TCPSocket* server_socket;
    NET::TCPSocket* client_socket;
 
public:
    void setUp()
    {
        server_socket = new NET::TCPSocket();
        client_socket = new NET::TCPSocket();
    }
 
    void tearDown()
    {
        delete server_socket;
        delete client_socket;
    }
 
    void testSocketHandle()
    {
        server_socket->bind( "127.0.0.1", 47777);
        server_socket->listen();
        client_socket->connect( "127.0.0.1", 47777);
 
        NET::TCPSocket::Handle tmp_handle;
        tmp_handle = server_socket->accept();
        NET::TCPSocket::Handle handle;
        NET::TCPSocket::Handle& ref_handle = handle;
        handle = ref_handle;
        CPPUNIT_ASSERT( !handle );
        CPPUNIT_ASSERT( tmp_handle );
        handle = tmp_handle;
        CPPUNIT_ASSERT( handle );
        CPPUNIT_ASSERT( !tmp_handle );
        NET::TCPSocket session_socket(handle);
        CPPUNIT_ASSERT( !handle );
        CPPUNIT_ASSERT_THROW( NET::TCPSocket bad_socket(handle), NET::SocketException );
        client_socket->disconnect();
    }
 
    void testPeerStatus()
    {
        int ret;
        server_socket->bind( "127.0.0.1", 47777);
        server_socket->listen();
        CPPUNIT_ASSERT( !server_socket->peerDisconnected() );
 
        client_socket->connect( "127.0.0.1", 47777);
        NET::TCPSocket::Handle handle = server_socket->accept();
        CPPUNIT_ASSERT( handle );
        NET::TCPSocket session_socket(handle);
        CPPUNIT_ASSERT( !handle );
 
        session_socket.send( send_msg, len);
        ret = client_socket->receive( recv_msg, len);
        CPPUNIT_ASSERT_EQUAL( len, ret );
        CPPUNIT_ASSERT( !session_socket.peerDisconnected() );
        CPPUNIT_ASSERT( std::memcmp( send_msg, recv_msg, len) == 0 );
 
        client_socket->disconnect();
        ret = session_socket.send( send_msg, len);
        CPPUNIT_ASSERT_EQUAL( -1, ret );
        CPPUNIT_ASSERT( session_socket.peerDisconnected() );
    }
};
 
CPPUNIT_TEST_SUITE_REGISTRATION( TCPSocket_TEST );