/********************************************************************************
 *    Copyright (C) 2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#include <gtest/gtest.h>
#include <FairMQChannel.h>
#include <FairMQParts.h>
#include <FairMQLogger.h>
#include <FairMQTransportFactory.h>
#include <fairmq/Tools.h>
#include <fairmq/ProgOptions.h>

#include <algorithm>
#include <memory>
#include <sstream>
#include <string>
#include <thread>

namespace
{

using namespace std;

void CheckOldOptionInterface(FairMQChannel& channel, const string& option, const string& transport)
{
    int value = 500;
    channel.GetSocket().SetOption(option, &value, sizeof(value));
    value = 0;
    size_t valueSize = sizeof(value);
    channel.GetSocket().GetOption(option, &value, &valueSize);
    if (transport == "nanomsg" && (option == "snd-hwm" || option == "rcv-hwm")) {
        ASSERT_EQ(value, -1);
    } else {
        ASSERT_EQ(value, 500);
    }
}

void RunOptionsTest(const string& transport)
{
    size_t session{fair::mq::tools::UuidHash()};

    fair::mq::ProgOptions config;
    config.SetProperty<string>("session", to_string(session));
    auto factory = FairMQTransportFactory::CreateTransportFactory(transport, fair::mq::tools::Uuid(), &config);
    FairMQChannel channel("Push", "push", factory);

    CheckOldOptionInterface(channel, "linger", transport);
    CheckOldOptionInterface(channel, "snd-hwm", transport);
    CheckOldOptionInterface(channel, "rcv-hwm", transport);
    CheckOldOptionInterface(channel, "snd-size", transport);
    CheckOldOptionInterface(channel, "rcv-size", transport);

    channel.GetSocket().SetLinger(300);
    ASSERT_EQ(channel.GetSocket().GetLinger(), 300);

    channel.GetSocket().SetSndBufSize(500);
    if (transport == "nanomsg") { // nanomsg doesn't use this option and the getter always returns -1
        ASSERT_EQ(channel.GetSocket().GetSndBufSize(), -1);
    } else {
        ASSERT_EQ(channel.GetSocket().GetSndBufSize(), 500);
    }

    channel.GetSocket().SetRcvBufSize(500);
    if (transport == "nanomsg") { // nanomsg doesn't use this option and the getter always returns -1
        ASSERT_EQ(channel.GetSocket().GetRcvBufSize(), -1);
    } else {
        ASSERT_EQ(channel.GetSocket().GetRcvBufSize(), 500);
    }

    channel.GetSocket().SetSndKernelSize(8000);
    ASSERT_EQ(channel.GetSocket().GetSndKernelSize(), 8000);

    channel.GetSocket().SetRcvKernelSize(8000);
    ASSERT_EQ(channel.GetSocket().GetRcvKernelSize(), 8000);
}

TEST(Options, zeromq)
{
    RunOptionsTest("zeromq");
}

TEST(Options, shmem)
{
    RunOptionsTest("shmem");
}

#ifdef BUILD_NANOMSG_TRANSPORT
TEST(Options, nanomsg)
{
    RunOptionsTest("nanomsg");
}
#endif /* BUILD_NANOMSG_TRANSPORT */

} // namespace