restart_request_server.cpp

Go to the documentation of this file.

/*
 * Copyright (C) 2014 Pavel Kirienko <pavel.kirienko@gmail.com>
 */
 
#include <gtest/gtest.h>
#include <uavcan/protocol/restart_request_server.hpp>
#include "helpers.hpp"
 
 
struct RestartHandler : public uavcan::IRestartRequestHandler
{
    bool accept;
 
    bool handleRestartRequest(uavcan::NodeID request_source)
    {
        std::cout << "Restart request from " << int(request_source.get()) << " will be "
                  << (accept ? "accepted" : "rejected") << std::endl;
        return accept;
    }
};
 
 
TEST(RestartRequestServer, Basic)
{
    InterlinkedTestNodesWithSysClock nodes;
 
    uavcan::RestartRequestServer rrs(nodes.a);
 
    ServiceClientWithCollector<uavcan::protocol::RestartNode> rrs_cln(nodes.b);
 
    uavcan::GlobalDataTypeRegistry::instance().reset();
    uavcan::DefaultDataTypeRegistrator<uavcan::protocol::RestartNode> _reg1;
 
    ASSERT_LE(0, rrs.start());
 
    uavcan::protocol::RestartNode::Request request;
    request.magic_number = uavcan::protocol::RestartNode::Request::MAGIC_NUMBER;
 
    /*
     * Rejected - handler was not set
     */
    ASSERT_LE(0, rrs_cln.call(1, request));
    nodes.spinBoth(uavcan::MonotonicDuration::fromMSec(10));
 
    ASSERT_TRUE(rrs_cln.collector.result.get());
    ASSERT_TRUE(rrs_cln.collector.result->isSuccessful());
    ASSERT_FALSE(rrs_cln.collector.result->getResponse().ok);
 
    /*
     * Accepted
     */
    RestartHandler handler;
    handler.accept = true;
    rrs.setHandler(&handler);
 
    ASSERT_LE(0, rrs_cln.call(1, request));
    nodes.spinBoth(uavcan::MonotonicDuration::fromMSec(10));
 
    ASSERT_TRUE(rrs_cln.collector.result->isSuccessful());
    ASSERT_TRUE(rrs_cln.collector.result->getResponse().ok);
 
    /*
     * Rejected by handler
     */
    handler.accept = false;
 
    ASSERT_LE(0, rrs_cln.call(1, request));
    nodes.spinBoth(uavcan::MonotonicDuration::fromMSec(10));
 
    ASSERT_TRUE(rrs_cln.collector.result->isSuccessful());
    ASSERT_FALSE(rrs_cln.collector.result->getResponse().ok);
 
    /*
     * Rejected because of invalid magic number
     */
    handler.accept = true;
 
    ASSERT_LE(0, rrs_cln.call(1, uavcan::protocol::RestartNode::Request()));
    nodes.spinBoth(uavcan::MonotonicDuration::fromMSec(10));
 
    ASSERT_TRUE(rrs_cln.collector.result->isSuccessful());
    ASSERT_FALSE(rrs_cln.collector.result->getResponse().ok);
}