main.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include <unistd.h>
 
#include <fstream>
#include <iostream>
#include <thread>
 
#include "data_stream_assert.h"
#include "PGenericSocketManager.h"
 
 
typedef PGenericSocketManager<std::string, PEmptyBackend, PMockBackend> SocketManager;
typedef SocketManager::Backend Backend;
typedef SocketManager::Mock Mock;
 
 
 
void threadSendData(PSocketMode::PSocketMode mode, size_t nbMessage, const std::string & address, size_t port){
        SocketManager manager(mode);    //Mock mode
        data_stream_assert(manager.addServerSocket("Alice",
                Backend::server(address, port),
                Mock::server(address, port, true, "./")));
        data_stream_assert(manager.isConnected("Alice"));
        for(size_t i(0lu); i < nbMessage; ++i){
                data_stream_assert(manager.sendData("Alice", i));
                std::cout << "message n°"<<i<<", Send value " << i << std::endl;
        }
}
 
 
void threadRecievedData(PSocketMode::PSocketMode mode, const std::string & address, size_t nbMessage, size_t port){
        SocketManager manager(mode);    //Mock mode
        data_stream_assert(manager.addClientSocket("Bob",
                Backend::client(address, port),
                Mock::client(address, port, false, "./")));
        size_t i(0lu);
        while(i < nbMessage){
                size_t value(0lu);
                if(mode == PSocketMode::NO_MOCK){
                        data_stream_assert(!manager.recvData("Bob", value));
                }else{
                        data_stream_assert(manager.recvData("Bob", value));
                }
                ++i;
        }
}
 
 
void testMockSocket(PSocketMode::PSocketMode mode){
        size_t nbMessage(10lu), port(3390lu);
        std::string hostName("localhost");
        threadSendData(mode, nbMessage, hostName, port);
        threadRecievedData(mode, hostName, nbMessage, port);
}
 
 
 
void threadSendMessage(PSocketMode::PSocketMode mode, size_t nbMessage, const std::string & address, size_t port){
        SocketManager manager(mode);    //Mock mode
        data_stream_assert(manager.addServerSocket("Alice",
                Backend::server(address, port),
                Mock::server(address, port, true, "./testMessage")));
        data_stream_assert(manager.isConnected("Alice"));
        for(size_t i(0lu); i < nbMessage; ++i){
                Backend::Message msg;
                msg.push_back(i + 1);
                msg.push_back(i + 2);
                msg.push_back(i + 3);
                
                data_stream_assert(manager.sendMsg("Alice", msg));
                std::cout << "message n°"<<i<<", Send message " << i << std::endl;
        }
}
 
 
void threadRecievedMessage(PSocketMode::PSocketMode mode, const std::string & address, size_t nbMessage, size_t port){
        SocketManager manager(mode);    //Mock mode
        data_stream_assert(manager.addClientSocket("Bob",
                Backend::client(address, port),
                Mock::client(address, port, false, "./testMessage")));
        size_t i(0lu);
        while(i < nbMessage){
                Backend::Message msg;
                if(mode == PSocketMode::NO_MOCK){
                        data_stream_assert(!manager.recvMsg("Bob", msg));
                }else{
                        data_stream_assert(manager.recvMsg("Bob", msg));
                        data_stream_assert(Backend::msgSize(msg) == 3lu);
                        
                        DataStreamIter data = Backend::msgData(msg);
                        data_stream_assert(data[0] == (char)i + 1);
                        data_stream_assert(data[1] == (char)i + 2);
                        data_stream_assert(data[2] == (char)i + 3);
                }
                ++i;
        }
}
 
 
void testMockSocketMessage(PSocketMode::PSocketMode mode){
        size_t nbMessage(10lu), port(3390lu);
        std::string hostName("localhost");
        threadSendMessage(mode, nbMessage, hostName, port);
        threadRecievedMessage(mode, hostName, nbMessage, port);
}
 
int main(int argc, char **argv){
        testMockSocket(PSocketMode::NO_MOCK);
        testMockSocket(PSocketMode::MOCK);
        
        testMockSocketMessage(PSocketMode::NO_MOCK);
        testMockSocketMessage(PSocketMode::MOCK);
        return 0;
}