platform/networkstrate/service_network_test.cpp

/*
 * Copyright (c) 2019-2022 ExpoLab, UC Davis
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use,
 * copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 *
 */

#include "platform/networkstrate/service_network.h"

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

#include <future>
#include <thread>

#include "platform/common/network/tcp_socket.h"
#include "platform/networkstrate/mock_service_interface.h"

namespace resdb {

using ::testing::Invoke;
using ::testing::Return;

ResDBConfig GenerateDBConfig() {
  std::vector<ReplicaInfo> replicas;
  ReplicaInfo self_info;
  self_info.set_ip("127.0.0.1");
  self_info.set_port(1234);
  ReplicaInfo dest_info;
  dest_info.set_ip("127.0.0.1");
  dest_info.set_port(1235);
  replicas.push_back(dest_info);

  return ResDBConfig(replicas, self_info, KeyInfo(), CertificateInfo());
}

void SendData(const std::string& data) {
  TcpSocket client_socket;
  int ret = client_socket.Connect("127.0.0.1", 1234);
  ASSERT_EQ(ret, 0);
  ret = client_socket.Send(data);
  ASSERT_EQ(ret, 0);
}

TEST(ServiceNetworkTest, RecvData) {
  std::promise<bool> init;
  std::future<bool> init_done = init.get_future();

  std::promise<bool> recv;
  std::future<bool> recv_done = recv.get_future();

  std::unique_ptr<MockServiceInterface> service =
      std::make_unique<MockServiceInterface>();
  bool finished = false;
  EXPECT_CALL(*service, IsRunning).WillRepeatedly(Invoke([&]() {
    return !finished;
  }));

  EXPECT_CALL(*service, Process)
      .WillOnce(Invoke([&](std::unique_ptr<Context>,
                           std::unique_ptr<DataInfo> request_info) {
        EXPECT_EQ(
            std::string((char*)request_info->buff, request_info->data_len),
            "test");
        recv.set_value(true);
        return 0;
      }));

  std::thread svr_thead2 = std::thread([&]() {
    ServiceNetwork server(GenerateDBConfig(), std::move(service));
    init.set_value(true);
    server.Run();
  });
  init_done.get();
  SendData("test");
  recv_done.get();
  finished = true;
  svr_thead2.join();
}

TEST(ServiceNetworkTest, RunningDone) {
  std::unique_ptr<MockServiceInterface> service =
      std::make_unique<MockServiceInterface>();
  bool finished = false;
  EXPECT_CALL(*service, IsRunning).WillRepeatedly(Invoke([&]() {
    return !finished;
  }));

  EXPECT_CALL(*service, Process).Times(0);

  std::promise<bool> init;
  std::future<bool> init_done = init.get_future();

  std::thread svr_thead2 = std::thread([&]() {
    ServiceNetwork server(GenerateDBConfig(), std::move(service));
    init.set_value(true);
    server.Run();
  });
  init_done.get();
  finished = true;
  svr_thead2.join();
}

}  // namespace resdb