Ported networking test.

Engine/source/platform/test/netTest.cpp

@@ -0,0 +1,182 @@
+//-----------------------------------------------------------------------------
+// Copyright (c) 2014 GarageGames, LLC
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to
+// deal in the Software without restriction, including without limitation the
+// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
+// sell copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+// IN THE SOFTWARE.
+//-----------------------------------------------------------------------------
+
+#ifdef TORQUE_TESTS_ENABLED
+#include "testing/unitTesting.h"
+#include "platform/platformNet.h"
+#include "core/util/journal/process.h"
+
+TEST(Net, TCPRequest)
+{
+   struct handle
+   {
+      NetSocket mSocket;
+      S32 mDataReceived;
+
+      void notify(NetSocket sock, U32 state) 
+      {
+         // Only consider our own socket.
+         if(mSocket != sock)
+            return;
+
+         // Ok - what's the state? We do some dumb responses to given states
+         // in order to fulfill the request.
+         if(state == Net::Connected)
+         {
+            U8 reqBuffer[] = {
+               "GET / HTTP/1.0\nUser-Agent: Torque/1.0\n\n"
+            };
+
+            Net::Error e = Net::sendtoSocket(mSocket, reqBuffer, sizeof(reqBuffer));
+
+            EXPECT_EQ(e, Net::NoError)
+               << "Got an error sending our HTTP request!";
+         }
+         else if(state == Net::Disconnected)
+         {
+            Process::requestShutdown();
+            mSocket = NULL;
+         }
+      }
+
+      void receive(NetSocket sock, RawData incomingData)
+      {
+         // Only consider our own socket.
+         if(mSocket != sock)
+            return;
+
+         char buff[4096];
+         dMemcpy(buff, incomingData.data, incomingData.size);
+         buff[incomingData.size] = 0;
+
+         mDataReceived += incomingData.size;
+      }
+   } handler;
+
+   handler.mSocket = InvalidSocket;
+   handler.mDataReceived = 0;
+
+   // Hook into the signals.
+   Net::smConnectionNotify. notify(&handler, &handle::notify);
+   Net::smConnectionReceive.notify(&handler, &handle::receive);
+
+   // Open a TCP connection to garagegames.com
+   handler.mSocket = Net::openConnectTo("72.246.107.193:80");
+   const U32 limit = Platform::getRealMilliseconds() + (5*1000);
+   while(Process::processEvents() && (Platform::getRealMilliseconds() < limit) ) {}
+
+   // Unhook from the signals.
+   Net::smConnectionNotify. remove(&handler, &handle::notify);
+   Net::smConnectionReceive.remove(&handler, &handle::receive);
+
+   EXPECT_GT(handler.mDataReceived, 0)
+      << "Didn't get any data back!";
+}
+
+TEST(Net, JournalTCPRequest)
+{
+   struct handle
+   {
+      NetSocket mSocket;
+      S32 mDataReceived;
+
+      void notify(NetSocket sock, U32 state)
+      {
+         // Only consider our own socket.
+         if(mSocket != sock)
+            return;
+
+         // Ok - what's the state? We do some dumb responses to given states
+         // in order to fulfill the request.
+         if(state == Net::Connected)
+         {
+            U8 reqBuffer[] = {
+               "GET / HTTP/1.0\nUser-Agent: Torque/1.0\n\n"
+            };
+
+            Net::Error e = Net::sendtoSocket(mSocket, reqBuffer, sizeof(reqBuffer));
+
+            EXPECT_EQ(e, Net::NoError)
+               << "Got an error sending our HTTP request!";
+         }
+         else if(state == Net::Disconnected)
+         {
+            Process::requestShutdown();
+            mSocket = NULL;
+         }
+      }
+
+      void receive(NetSocket sock, RawData incomingData)
+      {
+         // Only consider our own socket.
+         if(mSocket != sock)
+            return;
+
+         char buff[4096];
+         dMemcpy(buff, incomingData.data, incomingData.size);
+         buff[incomingData.size] = 0;
+
+         mDataReceived += incomingData.size;
+      }
+
+      void makeRequest()
+      {
+         mSocket = InvalidSocket;
+         mDataReceived = 0;
+
+         // Initialize networking - done by initLibraries currently
+         //test(Net::init(), "Failed to initialize networking!");
+
+         // Hook into the signals.
+         Net::smConnectionNotify. notify(this, &handle::notify);
+         Net::smConnectionReceive.notify(this, &handle::receive);
+
+         // Open a TCP connection to garagegames.com
+         mSocket = Net::openConnectTo("72.246.107.193:80");
+
+         // Let the callbacks enable things to process.
+         while(Process::processEvents()) {}
+
+         // Unhook from the signals.
+         Net::smConnectionNotify. remove(this, &handle::notify);
+         Net::smConnectionReceive.remove(this, &handle::receive);
+
+         EXPECT_GT(mDataReceived, 0)
+            << "Didn't get any data back!";
+      }
+   } handler;
+
+   Journal::Record("journalTCP.jrn");
+   ASSERT_TRUE(Journal::IsRecording());
+   handler.makeRequest();
+   S32 bytesRead = handler.mDataReceived;
+   Journal::Stop();
+
+   Journal::Play("journalTCP.jrn");
+   handler.makeRequest();
+   Journal::Stop();
+
+   EXPECT_EQ(bytesRead, handler.mDataReceived)
+      << "Didn't get same data back from journal playback.";
+}
+
+#endif

Engine/source/platform/test/testNet.cpp

@@ -1,200 +0,0 @@
-//-----------------------------------------------------------------------------
-// Copyright (c) 2012 GarageGames, LLC
-//
-// Permission is hereby granted, free of charge, to any person obtaining a copy
-// of this software and associated documentation files (the "Software"), to
-// deal in the Software without restriction, including without limitation the
-// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
-// sell copies of the Software, and to permit persons to whom the Software is
-// furnished to do so, subject to the following conditions:
-//
-// The above copyright notice and this permission notice shall be included in
-// all copies or substantial portions of the Software.
-//
-// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
-// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
-// IN THE SOFTWARE.
-//-----------------------------------------------------------------------------
-
-#include "platform/platformNet.h"
-#include "unit/test.h"
-#include "core/util/journal/process.h"
-
-using namespace UnitTesting;
-
-CreateUnitTest( TestTCPRequest, "Platform/Net/TCPRequest")
-{
-   NetSocket mSocket;
-   S32 mDataRecved;
-
-   void handleNotify(NetSocket sock, U32 state) 
-   {
-      // Only consider our own socket.
-      if(mSocket != sock)
-         return;
-
-      // Ok - what's the state? We do some dumb responses to given states
-      // in order to fulfill the request.
-      if(state == Net::Connected)
-      {
-         U8 reqBuffer[] = {
-            "GET / HTTP/1.0\nUser-Agent: Torque/1.0\n\n"
-         };
-
-         Net::Error e = Net::sendtoSocket(mSocket, reqBuffer, sizeof(reqBuffer));
-
-         test(e == Net::NoError, "Got an error sending our HTTP request!");
-      }
-      else if(state == Net::Disconnected)
-      {
-         Process::requestShutdown();
-         mSocket = NULL;
-      }
-   }
-
-   void handleReceive(NetSocket sock, RawData incomingData)
-   {
-      // Only consider our own socket.
-      if(mSocket != sock)
-         return;
-
-      char buff[4096];
-      dMemcpy(buff, incomingData.data, incomingData.size);
-      buff[incomingData.size] = 0;
-
-      UnitPrint("Got a message...\n");
-      UnitPrint(buff);
-      UnitPrint("------\n");
-
-      mDataRecved += incomingData.size;
-   }
-
-   void run()
-   {
-      mSocket = InvalidSocket;
-      mDataRecved = 0;
-
-      // Initialize networking - done by initLibraries currently
-      //test(Net::init(), "Failed to initialize networking!");
-
-      // Hook into the signals.
-      Net::smConnectionNotify. notify(this, &TestTCPRequest::handleNotify);
-      Net::smConnectionReceive.notify(this, &TestTCPRequest::handleReceive);
-
-      // Open a TCP connection to garagegames.com
-      mSocket = Net::openConnectTo("ip:72.246.107.193:80");
-      U32 limit = Platform::getRealMilliseconds() + (5*1000);
-      while(Process::processEvents() && (Platform::getRealMilliseconds() < limit) )
-         ;
-
-      // Unhook from the signals.
-      Net::smConnectionNotify. remove(this, &TestTCPRequest::handleNotify);
-      Net::smConnectionReceive.remove(this, &TestTCPRequest::handleReceive);
-
-      test(mDataRecved > 0, "Didn't get any data back!");
-   }
-};
-
-CreateUnitTest( TestTCPRequestJournal, "Platform/Net/JournalTCPRequest")
-{
-   NetSocket mSocket;
-   S32 mDataRecved;
-
-   void handleNotify(NetSocket sock, U32 state) 
-   {
-      // Only consider our own socket.
-      if(mSocket != sock)
-         return;
-
-      // Ok - what's the state? We do some dumb responses to given states
-      // in order to fulfill the request.
-      if(state == Net::Connected)
-      {
-         U8 reqBuffer[] = {
-            "GET / HTTP/1.0\nUser-Agent: Torque/1.0\n\n"
-         };
-
-         Net::Error e = Net::sendtoSocket(mSocket, reqBuffer, sizeof(reqBuffer));
-
-         test(e == Net::NoError, "Got an error sending our HTTP request!");
-      }
-      else if(state == Net::Disconnected)
-      {
-         Process::requestShutdown();
-         mSocket = NULL;
-      }
-   }
-
-   void handleReceive(NetSocket sock, RawData incomingData)
-   {
-      // Only consider our own socket.
-      if(mSocket != sock)
-         return;
-
-      char buff[4096];
-      dMemcpy(buff, incomingData.data, incomingData.size);
-      buff[incomingData.size] = 0;
-
-      UnitPrint("Got a message...\n");
-      UnitPrint(buff);
-      UnitPrint("------\n");
-
-      mDataRecved += incomingData.size;
-   }
-
-   void makeRequest()
-   {
-      mSocket = InvalidSocket;
-      mDataRecved = 0;
-
-      // Initialize networking - done by initLibraries currently
-      //test(Net::init(), "Failed to initialize networking!");
-
-      // Hook into the signals.
-      Net::smConnectionNotify. notify(this, &TestTCPRequestJournal::handleNotify);
-      Net::smConnectionReceive.notify(this, &TestTCPRequestJournal::handleReceive);
-
-      // Open a TCP connection to garagegames.com
-      mSocket = Net::openConnectTo("ip:72.246.107.193:80");
-
-      // Let the callbacks enable things to process.
-      while(Process::processEvents())
-         ;
-
-      // Unhook from the signals.
-      Net::smConnectionNotify. remove(this, &TestTCPRequestJournal::handleNotify);
-      Net::smConnectionReceive.remove(this, &TestTCPRequestJournal::handleReceive);
-
-      test(mDataRecved > 0, "Didn't get any data back!");
-   }
-
-   void run()
-   {
-      Journal::Record("journalTCP.jrn");
-
-      if( !Journal::IsRecording() )
-      {
-         test(0, "Failed.");
-         return;
-      }
-
-      makeRequest();
-
-      S32 bytesRead = mDataRecved;
-
-      Journal::Stop();
-
-      Journal::Play("journalTCP.jrn");
-
-      makeRequest();
-
-      Journal::Stop();
-
-      test(bytesRead == mDataRecved, "Didn't get same data back from journal playback.");
-
-   }
-};