TestHelper.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac

    This file is part of Leosac.

    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Leosac is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once

#include "FakeGPIO.hpp"
#include "core/CoreUtils.hpp"
#include "core/MessageBus.hpp"
#include "core/config/ConfigChecker.hpp"
#include "gtest/gtest.h"
#include <boost/property_tree/ptree.hpp>
#include <zmqpp/zmqpp.hpp>

namespace Leosac
{
namespace Test
{
namespace Helper
{
template <typename ModuleType>
bool test_run_module(zmqpp::context *ctx, zmqpp::socket *pipe,
                     const boost::property_tree::ptree &cfg)
{
    {
        CoreUtilsPtr u = std::make_shared<CoreUtils>(
            nullptr, nullptr, std::make_shared<ConfigChecker>(), false);
        ModuleType module(*ctx, pipe, cfg, u);

        pipe->send(zmqpp::signal::ok);
        module.run();
    }
    return true;
}

bool bus_read_extract(zmqpp::message *)
{
    return true;
}

template <typename... Content>
bool bus_read_extract(zmqpp::message *m, const char *first_arg, Content... content);

template <typename T, typename... Content>
typename std::enable_if<!std::is_same<const char *, T>::value, bool>::type
bus_read_extract(zmqpp::message *m, T first_arg, Content... content)
{
    T value;
    *m >> value;
    if (value != first_arg)
        return false;

    return bus_read_extract(m, content...);
}

template <typename... Content>
bool bus_read_extract(zmqpp::message *m, const char *first_arg, Content... content)
{
    std::string value;
    *m >> value;

    if (strcmp(value.c_str(), first_arg) != 0)
        return false;
    return bus_read_extract(m, content...);
}

template <typename... Content>
bool bus_read(zmqpp::socket &sub, Content... content)
{
    zmqpp::message msg;

    if (!sub.receive(msg))
        return false;
    return bus_read_extract(&msg, content...);
}

class TestHelper : public ::testing::Test
{
  private:
    virtual bool run_module(zmqpp::socket *pipe) = 0;

  public:
    TestHelper()
        : ctx_()
        , bus_(ctx_)
        , bus_sub_(ctx_, zmqpp::socket_type::sub)
        , bus_push_(ctx_, zmqpp::socket_type::push)
        , module_actor_(nullptr)
    {
        bus_sub_.connect("inproc://zmq-bus-pub");
        bus_push_.connect("inproc://zmq-bus-pull");
    }

    virtual ~TestHelper()
    {
    }

    virtual void SetUp() override final
    {
        module_actor_ = std::unique_ptr<zmqpp::actor>(new zmqpp::actor(
            std::bind(&TestHelper::run_module, this, std::placeholders::_1)));
    }

    zmqpp::context ctx_;

    MessageBus bus_;

    zmqpp::socket bus_sub_;

    zmqpp::socket bus_push_;

    std::unique_ptr<zmqpp::actor> module_actor_;
};
}
}
}