view/connection/TestSocketSet.cpp

/*!
  \file
  \brief SocketSet のテスト

  \author Satofumi KAMIMURA

  $Id$
*/

#include "TestSocketSet.h"
#include "SocketSet.h"
#include "TcpipCtrl.h"
#include "TcpipAccept.h"
#include "ThreadCreator.h"

CPPUNIT_TEST_SUITE_REGISTRATION(SocketSetTest);
CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(SocketSetTest, "SocketSetTest");

using namespace beego;


enum { Port = 9763 };

static int connectThread(void* args) {
  TcpipCtrl* con = static_cast<TcpipCtrl*>(args);

  con->connect("localhost", Port);
  return con->isConnected();
}


/*!
  \brief 受信確認のテスト
*/
void SocketSetTest::readCheckTest(void) {
  TcpipAccept server;
  bool ret = server.activate(Port);
  CPPUNIT_ASSERT_EQUAL(true, ret);

  SocketSet socket_set(1);

  // 接続処理
  TcpipCtrl con;
  ThreadCreator thread(connectThread, &con);
  thread.run();
  thread.stop();
  SDL_Delay(10);

  // 接続待ちポートの起動
  TcpipCtrl* server_con = server.accept(10, &socket_set);
  CPPUNIT_ASSERT(server_con);
  CPPUNIT_ASSERT_EQUAL(true, con.isConnected());
  CPPUNIT_ASSERT_EQUAL(true, server_con->isConnected());

  // １文字送信、受信
  int n = con.send("A", 1);
  CPPUNIT_ASSERT_EQUAL(1, n);

  char ch;
  for (int i = 0; i < 10; ++i) {
    n = socket_set.check(10);
    if (n > 0) {
      break;
    }
  }
  CPPUNIT_ASSERT_EQUAL(1, n);

  for (int i = 0; i < 10; ++i) {
    n = server_con->recv(&ch, 1, 10);
    if (n > 0) {
      break;
    }
  }
  if (n <= 0) {
    //fprintf(stderr, " %d: %s, %s\n", n, server_con->what(), con.what());
  }
  CPPUNIT_ASSERT_EQUAL(1, n);
  CPPUNIT_ASSERT_EQUAL('A', ch);

  // n 文字送信、受信
  n = con.send("ABC", 3);
  CPPUNIT_ASSERT_EQUAL(3, n);

  n = socket_set.check(50);
  CPPUNIT_ASSERT_EQUAL(1, n);

  char buffer[] = { '\0', '\0', '\0', '\0' };
  n = server_con->recv(buffer, 3, 50);
  CPPUNIT_ASSERT_EQUAL(3, n);
  CPPUNIT_ASSERT_EQUAL(std::string("ABC"), std::string(buffer));

  // 受信タイムアウトのテスト
  n = socket_set.check(50);
  CPPUNIT_ASSERT_EQUAL(0, n);

  // 切断の検出
  con.disconnect();
  CPPUNIT_ASSERT_EQUAL(false, con.isConnected());

  n = socket_set.check(0);
  CPPUNIT_ASSERT_EQUAL(1, n);

  n = server_con->recv(&ch, 1, 0);
  CPPUNIT_ASSERT(n < 0);
  CPPUNIT_ASSERT_EQUAL(false, server_con->isConnected());
}