/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             * 
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *  
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
/**
 * runTestSub.cxx
 *
 * @since 2015-09-05
 * @author A. Rybalchenko
 */

#include "FairMQLogger.h"
#include "FairMQTestSub.h"

#include <string>
#include <chrono>
#include <thread>

int main(int argc, char** argv)
{
    reinit_logger(false);
    SET_LOG_CONSOLE_LEVEL(WARN);

    FairMQTestSub testSub;
    testSub.CatchSignals();

    std::string transport;
    if (argc != 2)
    {
        LOG(ERROR) << "Transport for the test not specified!";
        return 1;
    }
    transport = argv[1];

    if (transport == "zeromq" || transport == "nanomsg")
    {
        testSub.SetTransport(transport);
    }
    else
    {
        LOG(ERROR) << "Incorrect transport requested! Expected 'zeromq' or 'nanomsg', found: " << transport;
        return 1;
    }

    testSub.SetProperty(FairMQTestSub::Id, "testSub_" + std::to_string(getpid()));

    FairMQChannel controlChannel("push", "connect", "tcp://127.0.0.1:5555");
    if (transport == "nanomsg")
    {
        controlChannel.UpdateAddress("tcp://127.0.0.1:5755");
    }
    controlChannel.UpdateRateLogging(0);
    testSub.fChannels["control"].push_back(controlChannel);

    FairMQChannel subChannel("sub", "connect", "tcp://127.0.0.1:5556");
    if (transport == "nanomsg")
    {
        subChannel.UpdateAddress("tcp://127.0.0.1:5756");
    }
    subChannel.UpdateRateLogging(0);
    testSub.fChannels["data"].push_back(subChannel);

    testSub.ChangeState("INIT_DEVICE");
    testSub.WaitForEndOfState("INIT_DEVICE");

    testSub.ChangeState("INIT_TASK");
    testSub.WaitForEndOfState("INIT_TASK");

    testSub.ChangeState("RUN");
    testSub.WaitForEndOfState("RUN");

    // nanomsg does not implement the LINGER option. Give the sockets some time before their queues are terminated
    if (transport == "nanomsg")
    {
        std::this_thread::sleep_for(std::chrono::seconds(1));
    }

    testSub.ChangeState("RESET_TASK");
    testSub.WaitForEndOfState("RESET_TASK");

    testSub.ChangeState("RESET_DEVICE");
    testSub.WaitForEndOfState("RESET_DEVICE");

    testSub.ChangeState("END");

    return 0;
}