string_plugin_router_test.cpp

Go to the documentation of this file.

#include "../../src/plugin_router/string_plugin_router.h"

#include <fruit/fruit.h>
#include <gtest/gtest.h>

#include <algorithm>
#include <boost/array.hpp>
#include <boost/asio.hpp>
#include <functional>
#include <iostream>
#include <iterator>
#include <memory>
#include <unordered_map>

#include "../../src/socket/udp_socket.h"
#include "std_msgs/UInt8MultiArray.h"

using boost::asio::ip::udp;
using fruit::Annotated;
using fruit::Component;
using fruit::createComponent;
using fruit::Injector;

class MockStringSocket : public Socket {
 private:
  std::shared_ptr<std::function<void(std::vector<uint8_t>)>> callback_;
  int counter = 0;

 public:
  // デフォルトコンストラクタは削除
  MockStringSocket() = delete;
  INJECT(MockStringSocket(
      ASSISTED(udp::endpoint) target_socket,
      ASSISTED(std::shared_ptr<std::function<void(std::vector<uint8_t>)>>)
          callback))
      : callback_(callback) {}

  virtual void Start() override {
    std::string first_message = "first message";
    std::vector<uint8_t> first_data(first_message.begin(), first_message.end());
    (*callback_)(first_data);
  }

  virtual void SendData(std::vector<uint8_t> data) override {
    std::string message(data.begin(), data.end());

    counter += 1;
    switch (counter) {
      case 1:
        ASSERT_STREQ(message.c_str(), "first message");
        break;
      case 2:
        ASSERT_STREQ(message.c_str(), "second message");
        break;
      case 3:
        ASSERT_STREQ(message.c_str(), "third message");
        break;
    }
  }
};

// Mockを入れるもの
Component<SocketFactory> getMockStringUdpSourceComponent() {
  return createComponent().bind<Socket, MockStringSocket>();
}

Component<Annotated<StringAnnotation, PluginRouterFactory>>
getMockStringPluginRouterComponent() {
  return createComponent()
      .replace(getUdpSocketComponent)
      .with(getMockStringUdpSourceComponent)
      .install(getStringPluginRouterComponent);
}

// XmlRpcValueが不正なケース
TEST(TestSuite, string_plugin_try_start_with_invalid_xml) {
  std::shared_ptr<rapidjson::Document> config(new rapidjson::Document);
  config->Parse("{}");

  // objectを作成し、受信スレッドを開始
  Injector<Annotated<StringAnnotation, PluginRouterFactory>> injector(
      getMockStringPluginRouterComponent);
  PluginRouterFactory pluginRouterFactory =
      injector.get<fruit::Annotated<StringAnnotation, PluginRouterFactory>>();
  // データは送信しないのでportは何でも良い
  auto source = pluginRouterFactory(config, udp::endpoint(udp::v4(), 0));
  auto result = source->TryStart();
  // TryStartに失敗して、config間違いのエラーメッセージを受け取る
  ASSERT_FALSE(result.is_success);
  ASSERT_STREQ(result.error_message, "invalid config parameters");
}

// pluginが見つからないケース
TEST(TestSuite, string_plugin_try_start_not_found_plugin) {
  std::shared_ptr<rapidjson::Document> config(new rapidjson::Document);
  config->Parse(
      "[{\"plugin_name\":\"string_loopback::StringLoopback\",\"param\":"
      "\"Parameter\"},{"
      "\"plugin_name\":\"not_found_plugin::NotFoundPlugin\"}]");
  ROS_INFO("This test produces a ROS_ERROR message because of a loading error in the plugin, but the behavior is as expected.");

  // objectを作成し、受信スレッドを開始
  Injector<Annotated<StringAnnotation, PluginRouterFactory>> injector(
      getMockStringPluginRouterComponent);
  PluginRouterFactory pluginRouterFactory =
      injector.get<fruit::Annotated<StringAnnotation, PluginRouterFactory>>();
  // データは送信しないのでportは何でも良い
  auto source = pluginRouterFactory(config, udp::endpoint(udp::v4(), 0));
  auto result = source->TryStart();
  // TryStartに失敗して、pluginがない旨のメッセージを受け取る
  ASSERT_FALSE(result.is_success);
  std::string error_message = result.error_message;
  ASSERT_EQ(error_message.rfind("Failed to load"), 0);
}

// Loopback Pluginを使うケース
TEST(TestSuite, string_plugin_try_start_with_loopback_plugin) {
  std::shared_ptr<rapidjson::Document> config(new rapidjson::Document);
  config->Parse(
      "[{\"plugin_name\":\"string_loopback::StringLoopback\",\"param\":"
      "\"Parameter\"}]");

  // objectを作成し、受信スレッドを開始
  Injector<Annotated<StringAnnotation, PluginRouterFactory>> injector(
      getMockStringPluginRouterComponent);
  PluginRouterFactory pluginRouterFactory =
      injector.get<fruit::Annotated<StringAnnotation, PluginRouterFactory>>();
  // データは送信しないのでportは何でも良い
  auto source = pluginRouterFactory(config, udp::endpoint(udp::v4(), 0));
  auto result = source->TryStart();
  // TryStartに成功する
  // データの送信と評価はMockStringSocket内でやっている
  ASSERT_TRUE(result.is_success);
}