rpc_test_utils.h

// Copyright 2018, Beeri 15.  All rights reserved.
// Author: Roman Gershman (romange@gmail.com)
//
#pragma once

#include "base/gtest.h"
#include "util/asio/io_context_pool.h"
#include "util/rpc/rpc_connection.h"

namespace util {
class ReconnectableSocket;
class AcceptServer;
class FiberSyncSocket;

namespace rpc {

template <typename Src, typename Dest>
void Copy(const Src& src, Dest* dest) {
  dest->resize(src.size());
  std::copy(src.begin(), src.end(), dest->begin());
}

class TestBridge final : public ConnectionBridge {
  bool clear_;

 public:
  TestBridge(bool clear) : clear_(clear) {
  }

  ~TestBridge();

  void Join() final;

  // header and letter are input/output parameters.
  // HandleEnvelope reads first the input and if everything is parsed fine, it sends
  // back another header, letter pair.
  void HandleEnvelope(uint64_t rpc_id, Envelope* envelope, EnvelopeWriter writer) final;
};

class TestInterface final : public ServiceInterface {
  bool clear_ = false;

 public:
  void set_clear(bool c) {
    clear_ = c;
  }
  ConnectionBridge* CreateConnectionBridge() override {
    return new TestBridge{clear_};
  }
};

class ServerTest : public testing::Test {
 public:
  ServerTest();

 protected:
  static void SetUpTestCase() {
  }

  static void TearDownTestCase() {
  }

  void SetUp() override;

  void TearDown() override;

  std::unique_ptr<TestInterface> service_;
  std::unique_ptr<AcceptServer> server_;
  std::unique_ptr<IoContextPool> pool_;
  std::unique_ptr<FiberSyncSocket> sock2_;

  ::boost::system::error_code ec_;
  uint16_t port_ = 0;
};

}  // namespace rpc
}  // namespace util