You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

test.cpp 10KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339
  1. #include "test.h"
  2. Test::Test(QObject *parent) : QObject(parent)
  3. {
  4. #ifdef PT_DEBUG
  5. qDebug()<<"Debug mode";
  6. #else
  7. qDebug()<<"Release mode";
  8. #warning PTSocket is not in debug mode
  9. #endif
  10. m_serv = new PTServer(this);
  11. m_timeoutTimer.setSingleShot(true);
  12. connect(&m_timeoutTimer, SIGNAL(timeout()), this, SLOT(timedout()));
  13. m_testsStates<<(QList<TestState>());
  14. m_testsStates<<(QList<TestState>()//Basic
  15. <<TestState(1, PTSocket::HostLookUp)
  16. <<TestState(1, PTSocket::Connecting)
  17. <<TestState(1, PTSocket::Connected)
  18. <<TestState(1, PTSocket::Handshaked)
  19. <<TestState(1, PTSocket::Disconnecting)
  20. <<TestState(1, PTSocket::Disconnected)
  21. <<TestState(2, PTSocket::Disconnecting)
  22. <<TestState(2, PTSocket::Disconnected));
  23. m_testsStates<<(QList<TestState>()//Ping
  24. <<TestState(1, PTSocket::HostLookUp)
  25. <<TestState(1, PTSocket::Connecting)
  26. <<TestState(1, PTSocket::Connected)
  27. <<TestState(1, PTSocket::Handshaked)
  28. <<TestState(2, PTSocket::Error)
  29. <<TestState(1, PTSocket::Disconnecting)
  30. <<TestState(1, PTSocket::Disconnected));
  31. m_testsStates<<(QList<TestState>()//Handshake
  32. <<TestState(1, PTSocket::HostLookUp)
  33. <<TestState(1, PTSocket::Connecting)
  34. <<TestState(1, PTSocket::Connected)
  35. <<TestState(1, PTSocket::Handshaked)
  36. <<TestState(1, PTSocket::Disconnecting)
  37. <<TestState(1, PTSocket::Disconnected));
  38. m_testsStates<<(QList<TestState>()//Handshake Fake 1
  39. <<TestState(1, PTSocket::HostLookUp)
  40. <<TestState(1, PTSocket::Connecting)
  41. <<TestState(1, PTSocket::Connected)
  42. <<TestState(1, PTSocket::Handshaked)
  43. <<TestState(1, PTSocket::Disconnecting)
  44. <<TestState(1, PTSocket::Disconnected));
  45. m_testsStates<<(QList<TestState>()//Handshake Fake 2
  46. <<TestState(1, PTSocket::HostLookUp)
  47. <<TestState(1, PTSocket::Connecting)
  48. <<TestState(1, PTSocket::Connected)
  49. <<TestState(1, PTSocket::Handshaked)
  50. <<TestState(1, PTSocket::Disconnecting)
  51. <<TestState(1, PTSocket::Disconnected));
  52. m_testsStates<<(QList<TestState>()//Error Invalid Host
  53. <<TestState(1, PTSocket::HostLookUp)
  54. <<TestState(1, PTSocket::Error));
  55. m_testsStates<<(QList<TestState>()//Error Invalid Port
  56. <<TestState(1, PTSocket::HostLookUp)
  57. <<TestState(1, PTSocket::Connecting)
  58. <<TestState(1, PTSocket::Error));
  59. m_currentState = 0;
  60. m_currentTest = -1;
  61. }
  62. void Test::test0IntConvert()
  63. {
  64. preTest();
  65. QList<int> tests;
  66. tests << 0 << 1 << 255 << 256 << 65535 << 65536 << -1 << -65535 << -65536 << 123456789 << -123456789;
  67. for(int i = 0; i < tests.size(); ++i)
  68. {
  69. if(PTSocket::byteArrayToInt(PTSocket::intToByteArray(tests[i])) != tests[i])
  70. {
  71. qDebug()<<tests[i]<<PTSocket::intToByteArray(tests[i])<<PTSocket::byteArrayToInt(PTSocket::intToByteArray(tests[i]));
  72. Q_ASSERT(false);
  73. }
  74. }
  75. timeout(0);
  76. }
  77. void Test::testListen(quint16 p)
  78. {
  79. Q_ASSERT(m_serv->listen(QHostAddress::Any, p));
  80. }
  81. void Test::test1Basic()
  82. {
  83. preTest();
  84. connect(m_serv, SIGNAL(newClient(PTSocket*)), this, SLOT(m_newClient1(PTSocket*)));
  85. PTSocket* sock = newSocket();
  86. connect(sock, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged1(PTSocket::State)));
  87. m_socketsIds.insert(sock, 1);
  88. sock->connectToHost("127.0.0.1", m_serv->serverPort());
  89. }
  90. void Test::test2Ping()
  91. {
  92. preTest();
  93. connect(m_serv, SIGNAL(newClient(PTSocket*)), this, SLOT(m_newClient2(PTSocket*)));
  94. PTSocket* sock = newSocket();
  95. sock->dbgSetWillPing(false);
  96. connect(sock, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged2(PTSocket::State)));
  97. m_socketsIds.insert(sock, 1);
  98. sock->connectToHost("127.0.0.1", m_serv->serverPort());
  99. }
  100. void Test::test3Handshake()
  101. {
  102. preTest();
  103. connect(m_serv, SIGNAL(newClient(PTSocket*)), this, SLOT(m_newClient3(PTSocket*)));
  104. PTSocket* sock = newSocket();
  105. sock->dbgSetWillHandshake(false);
  106. connect(sock, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged3(PTSocket::State)));
  107. m_socketsIds.insert(sock, 1);
  108. sock->connectToHost("127.0.0.1", m_serv->serverPort());
  109. }
  110. void Test::test4HandshakeFake1()
  111. {
  112. preTest();
  113. connect(m_serv, SIGNAL(newClient(PTSocket*)), this, SLOT(m_newClient4(PTSocket*)));
  114. PTSocket* sock = newSocket();
  115. sock->dbgSetWillHandshakeFake1(true);
  116. connect(sock, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged4(PTSocket::State)));
  117. m_socketsIds.insert(sock, 1);
  118. sock->connectToHost("127.0.0.1", m_serv->serverPort());
  119. }
  120. void Test::test5HandshakeFake2()
  121. {
  122. preTest();
  123. connect(m_serv, SIGNAL(newClient(PTSocket*)), this, SLOT(m_newClient5(PTSocket*)));
  124. PTSocket* sock = newSocket();
  125. sock->dbgSetWillHandshakeFake2(true);
  126. connect(sock, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged5(PTSocket::State)));
  127. m_socketsIds.insert(sock, 1);
  128. sock->connectToHost("127.0.0.1", m_serv->serverPort());
  129. }
  130. void Test::test6Error1()
  131. {
  132. preTest();
  133. PTSocket* sock = newSocket();
  134. m_socketsIds.insert(sock, 1);
  135. sock->connectToHost("", m_serv->serverPort());
  136. }
  137. void Test::test7Error2()
  138. {
  139. preTest();
  140. PTSocket* sock = newSocket();
  141. m_socketsIds.insert(sock, 1);
  142. sock->connectToHost("127.0.0.1", 15963);
  143. }
  144. void Test::timeout(int i)
  145. {
  146. m_timeoutTimer.start(i);
  147. }
  148. void Test::stopTimeout()
  149. {
  150. m_timeoutTimer.stop();
  151. }
  152. void Test::timedout()
  153. {
  154. if(m_currentState == m_testsStates.at(m_currentTest).size())
  155. {
  156. qDebug()<<"\n\nTest"<<m_currentTest<<"OK\n\n";
  157. switch(m_currentTest)
  158. {
  159. case 0:
  160. test1Basic();
  161. break;
  162. case 1:
  163. test2Ping();
  164. break;
  165. case 2:
  166. test3Handshake();
  167. break;
  168. case 3:
  169. test4HandshakeFake1();
  170. break;
  171. case 4:
  172. test5HandshakeFake2();
  173. break;
  174. case 5:
  175. test6Error1();
  176. break;
  177. case 6:
  178. test7Error2();
  179. break;
  180. default:
  181. qApp->quit();
  182. }
  183. }
  184. else
  185. {
  186. qDebug()<<"Timedout\n\n\n";
  187. qDebug()<<m_currentState+1<<m_testsStates.at(m_currentTest).at(m_currentState)<<m_testsStates.at(m_currentTest);
  188. Q_ASSERT(false);
  189. }
  190. }
  191. void Test::m_clientStateChangedTest(PTSocket::State s)
  192. {
  193. PTSocket* sock = qobject_cast<PTSocket*>(sender());
  194. Q_ASSERT(sock != 0);
  195. qDebug()<<"State"<<(sock->isServerSide()? "server": "client")<<sock<<s;
  196. if(s == PTSocket::Error)
  197. qDebug()<<"\t"<<sock->getError();
  198. if(m_currentState >= m_testsStates.at(m_currentTest).size())
  199. {
  200. qDebug()<<"Out of TestState\n\n\n"<<m_socketsIds[sock]<<s;
  201. Q_ASSERT(false);
  202. }
  203. TestState t = m_testsStates.at(m_currentTest).at(m_currentState);
  204. if(!m_socketsIds.keys().contains(sock) && !m_socketsIds.values().contains(t.first))
  205. m_socketsIds.insert(sock, t.first);
  206. TestState r(m_socketsIds[sock], s);
  207. if(r != t)
  208. {
  209. qDebug()<<"Test not matching\n\n\n"<<"Expected"<<t<<"but got"<<r;
  210. Q_ASSERT(false);
  211. }
  212. ++m_currentState;
  213. if(m_currentState == m_testsStates.at(m_currentTest).size())
  214. timeout(1500);
  215. else
  216. timeout();
  217. }
  218. void Test::m_newClientTest(PTSocket *s)
  219. {
  220. qDebug()<<"New client:"<<s->peerAddress().toString()<<s->peerPort();
  221. timeout();
  222. connect(s, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChangedTest(PTSocket::State)));
  223. }
  224. void Test::preTest()
  225. {
  226. ASSERT_LISTEN;
  227. ++m_currentTest;
  228. m_currentState = 0;
  229. m_socketsIds.clear();
  230. m_serv->disconnect(this);
  231. connect(m_serv, SIGNAL(newConnection()), this, SLOT(timeout()));
  232. connect(m_serv, SIGNAL(newClient(PTSocket*)), this, SLOT(m_newClientTest(PTSocket*)));
  233. }
  234. void Test::m_newClient1(PTSocket* s)
  235. {
  236. connect(s, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged1(PTSocket::State)));
  237. }
  238. void Test::m_clientStateChanged1(PTSocket::State s)
  239. {
  240. PTSocket* sock = qobject_cast<PTSocket*>(sender());
  241. if(s == PTSocket::Handshaked)
  242. {
  243. qDebug()<<"";
  244. QTimer::singleShot(1500, sock, SLOT(disconnectFromServer()));
  245. }
  246. }
  247. void Test::m_newClient2(PTSocket* s)
  248. {
  249. connect(s, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged2(PTSocket::State)));
  250. }
  251. void Test::m_clientStateChanged2(PTSocket::State s)
  252. {
  253. if(s == PTSocket::Handshaked)
  254. qDebug()<<"";
  255. }
  256. void Test::m_newClient3(PTSocket* s)
  257. {
  258. connect(s, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged3(PTSocket::State)));
  259. }
  260. void Test::m_clientStateChanged3(PTSocket::State s)
  261. {
  262. if(s == PTSocket::Handshaked)
  263. qDebug()<<"";
  264. }
  265. void Test::m_newClient4(PTSocket* s)
  266. {
  267. connect(s, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged4(PTSocket::State)));
  268. }
  269. void Test::m_clientStateChanged4(PTSocket::State s)
  270. {
  271. if(s == PTSocket::Handshaked)
  272. qDebug()<<"";
  273. }
  274. void Test::m_newClient5(PTSocket* s)
  275. {
  276. connect(s, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChanged5(PTSocket::State)));
  277. }
  278. void Test::m_clientStateChanged5(PTSocket::State s)
  279. {
  280. if(s == PTSocket::Handshaked)
  281. qDebug()<<"";
  282. }
  283. PTSocket *Test::newSocket()
  284. {
  285. PTSocket* sock = new PTSocket(this);
  286. connect(sock, SIGNAL(stateChanged(PTSocket::State)), this, SLOT(m_clientStateChangedTest(PTSocket::State)));
  287. return sock;
  288. }
  289. bool operator ==(TestState t1, TestState t2)
  290. {
  291. return t1.first == t2.first && t1.second == t2.second;
  292. }
  293. bool operator !=(TestState t1, TestState t2)
  294. {
  295. return !(t1 == t2);
  296. }
  297. QDebug operator <<(QDebug dbg, TestState t)
  298. {
  299. return dbg<<"TestState("<<t.first<<","<<t.second<<")";
  300. }