App/testsuite/src/TestPeerHandshake.cpp

   1 /*
   2  * This file is part of TouristP2PImpl
   3  *
   4  * Copyright (C) 2007,2008 NUST Institute of Information Technology
   5  * Author: Faisal Khan <faisal.khan at [niit.edu.pk | cern.ch ]>
   6  *
   7  * TouristP2PImpl is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * TouristP2PImpl is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with TouristP2PImpl; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20 */
  21 #include "TestPeerHandshake.h"
  22 #include "Tourist/App/Application.h"
  23 #include "CppUnit/TestCaller.h"
  24 #include "CppUnit/TestSuite.h"
  25 #include "Poco/Thread.h"
  26
  27 using Tourist::App::Application;
  28 using Poco::Thread;
  29
  30 using namespace Tourist::App;
  31
  32 TestPeerHandshake::TestPeerHandshake(const std::string& name): CppUnit::TestCase(name)
  33 {
  34 }
  35
  36 TestPeerHandshake::~TestPeerHandshake()
  37 {
  38 }
  39
  40 void TestPeerHandshake::setUp()
  41 {
  42 }
  43
  44 void TestPeerHandshake::tearDown()
  45 {
  46 }
  47
  48 void TestPeerHandshake::testHandshake()
  49 {
  50   //create Application-1, configure the transport
  51   //create Application-2, configure transport with different port
  52   Config config_1;
  53   config_1.setTCPEnable(true);
  54   config_1.setTCPPort(5000);
  55   Application app1(config_1);
  56
  57   Config config_2;
  58   config_2.setTCPEnable(true);
  59   config_2.setTCPPort(5010);
  60   Application app2(config_2);
  61
  62   assert (app1.init() == 0);
  63   assert (app2.init() == 0);
  64
  65   Thread t1, t2;
  66   t1.start(app1);
  67   t2.start(app2);
  68
  69   Thread::sleep(20);
  70   //connect to remote-node of application-1 with application-2
  71   RemoteNode *rn = NULL;
  72   vector<TransportInfo> transportInfo;
  73   transportInfo = app1.getEnabledTransports();
  74
  75   assert (transportInfo.size() != 0);
  76
  77   //TODO: test it for all the transports.
  78   TransportInfo tInfo = transportInfo.at(0);
  79   int status = app2.connectionTo(tInfo.host, tInfo.port, tInfo.type, &rn, 10);
  80   assert (status == 0);
  81
  82   assert (rn != NULL);
  83   assert (rn -> getHost() == tInfo.host);
  84   assert (rn -> getPort() == tInfo.port);
  85   assert (rn -> getProtocol() == tInfo.type);
  86   assert (rn -> getPipe() != NULL);
  87
  88   app1.stopApp();
  89   app2.stopApp();
  90
  91   //wait for all the threads to finish
  92   t1.join();
  93   t2.join();
  94
  95   //match remtoe node information with that one of Application-1
  96 }
  97
  98
  99 CppUnit::Test* TestPeerHandshake::suite()
 100 {
 101   CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("TestPeerHandshake");
 102   CppUnit_addTest(pSuite, TestPeerHandshake, testHandshake);
 103
 104   return pSuite;
 105 }