view/connection/TestTcpipCtrl.cpp

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

  \author Satofumi KAMIMURA

  $Id$
*/

#include "TestTcpipCtrl.h"
#include "TcpipCtrl.h"
#include "TcpipAccept.h"
#include "ThreadCreator.h"

CPPUNIT_TEST_SUITE_REGISTRATION(TcpipCtrlTest);
CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(TcpipCtrlTest, "TcpipCtrlTest");

using namespace beego;


/*!
  \brief オブジェクトの生成テスト
*/
void TcpipCtrlTest::createTest(void) {
  TcpipCtrl client;
  CPPUNIT_ASSERT_EQUAL(false, client.isConnected());
}


enum { Port = 9763 };

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

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

/*!
  \brief 接続テスト
*/
void TcpipCtrlTest::connectionTest(void) {

  TcpipAccept server;
  bool ret = server.activate(Port);
  CPPUNIT_ASSERT_EQUAL(true, ret);

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

  // 接続待ちポートの起動
  TcpipCtrl* server_con = server.accept(50, NULL);

  CPPUNIT_ASSERT(server_con != NULL);
  CPPUNIT_ASSERT_EQUAL(true, con.isConnected());
  CPPUNIT_ASSERT_EQUAL(true, server_con->isConnected());

  con.disconnect();
  CPPUNIT_ASSERT_EQUAL(false, con.isConnected());

  // 受信後に切断が認識されるため、以下のテストは保留
  //CPPUNIT_ASSERT_EQUAL(false, server_con->isConnected());
}


/*!
  \brief 送受信テスト
*/
void TcpipCtrlTest::readWriteTest(void) {

  TcpipAccept server;
  bool ret = server.activate(Port);
  CPPUNIT_ASSERT_EQUAL(true, ret);

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

  // 接続待ちポートの起動
  TcpipCtrl* server_con = server.accept(50, NULL);
  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 = server_con->recv(&ch, 1, 10);
    if (n > 0) {
      break;
    }
  }
  CPPUNIT_ASSERT_EQUAL(1, n);
  CPPUNIT_ASSERT_EQUAL('A', ch);

  // n 文字送信、受信
  n = con.send("ABC", 3);
  CPPUNIT_ASSERT_EQUAL(3, 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 = server_con->recv(&ch, 1, 0);
  CPPUNIT_ASSERT_EQUAL(0, n);

  // サーバからの送信した場合の受信タイムアウトテスト
  n = con.recv(&ch, 1, 0);
  CPPUNIT_ASSERT_EQUAL(0, n);

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

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


void TcpipCtrlTest::disconnectTest(void) {
  TcpipAccept server;
  bool ret = server.activate(Port);
  CPPUNIT_ASSERT_EQUAL(true, ret);

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

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

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

  // 切断確認
  con.disconnect();
  CPPUNIT_ASSERT_EQUAL(false, con.isConnected());
  CPPUNIT_ASSERT_EQUAL(true, server_con->isConnected());

  for (int i = 0; i < 10; ++i) {
    n = server_con->size(10);
    if (n > 0) {
      break;
    }
  }

  // 受信データがあれば、切断状態にならないのを確認
  CPPUNIT_ASSERT_EQUAL(3, server_con->size(10));

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