socket_communicator_test.cc

/*!
 *  Copyright (c) 2019 by Contributors
 * \file msg_queue.cc
 * \brief Message queue for DGL distributed training.
 */
#include <gtest/gtest.h>
#include <string.h>
#include <unistd.h>
#include <string>

#include "../src/graph/network/socket_communicator.h"

using std::string;
using dgl::network::SocketSender;
using dgl::network::SocketReceiver;

TEST(SocketCommunicatorTest, SendAndRecv) {
  int pid = fork();
  const char * msg = "0123456789";
  ASSERT_GE(pid, 0);
  if (pid > 0) {  // parent: server
    char serbuff[10];
    memset(serbuff, '\0', 10);
    SocketReceiver receiver;
    receiver.Wait("127.0.0.1", 2049, 1, 500);
    receiver.Recv(serbuff, 10);
    ASSERT_EQ(string("0123456789"), string(serbuff, 10));
    receiver.Finalize();
  } else {  // child: client
    sleep(1);
    SocketSender sender;
    sender.AddReceiver("127.0.0.1", 2049, 0);
    sender.Connect();
    sender.Send(msg, 10, 0);
    sender.Finalize();
  }
}