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);
}