1 files changed, 159 insertions, 135 deletions
diff --git a/Swiften/QA/NetworkTest/BoostConnectionTest.cpp b/Swiften/QA/NetworkTest/BoostConnectionTest.cpp
index 53faff7..e0890bf 100644..100755
--- a/Swiften/QA/NetworkTest/BoostConnectionTest.cpp
+++ b/Swiften/QA/NetworkTest/BoostConnectionTest.cpp
@@ -1,155 +1,179 @@
 /*
- * Copyright (c) 2010-2013 Remko Tronçon
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
+#include <memory>
+#include <string>
+
+#include <boost/date_time/posix_time/posix_time_types.hpp>
+
 #include <cppunit/extensions/HelperMacros.h>
 #include <cppunit/extensions/TestFactoryRegistry.h>
-#include <boost/shared_ptr.hpp>
-#include <boost/smart_ptr/make_shared.hpp>
 
-#include <string>
 #include <Swiften/Base/Algorithm.h>
 #include <Swiften/Base/sleep.h>
+#include <Swiften/EventLoop/DummyEventLoop.h>
 #include <Swiften/Network/BoostConnection.h>
+#include <Swiften/Network/BoostIOServiceThread.h>
 #include <Swiften/Network/HostAddress.h>
 #include <Swiften/Network/HostAddressPort.h>
-#include <Swiften/Network/BoostIOServiceThread.h>
-#include <Swiften/EventLoop/DummyEventLoop.h>
-
-static const unsigned char* address = reinterpret_cast<const unsigned char*>("\x4A\x32\x3e\x31");
 
 using namespace Swift;
 
 class BoostConnectionTest : public CppUnit::TestFixture {
-		CPPUNIT_TEST_SUITE(BoostConnectionTest);
-		CPPUNIT_TEST(testDestructor);
-		CPPUNIT_TEST(testDestructor_PendingEvents);
-		CPPUNIT_TEST(testWrite);
-		CPPUNIT_TEST(testWriteMultipleSimultaniouslyQueuesWrites);
+        CPPUNIT_TEST_SUITE(BoostConnectionTest);
+        CPPUNIT_TEST(testDestructor);
+        CPPUNIT_TEST(testDestructor_PendingEvents);
+        CPPUNIT_TEST(testWrite);
+        CPPUNIT_TEST(testWriteMultipleSimultaniouslyQueuesWrites);
 #ifdef TEST_IPV6
-		CPPUNIT_TEST(testWrite_IPv6);
+        CPPUNIT_TEST(testWrite_IPv6);
 #endif
-		CPPUNIT_TEST_SUITE_END();
-
-	public:
-		void setUp() {
-			boostIOServiceThread_ = new BoostIOServiceThread();
-			boostIOService = boost::make_shared<boost::asio::io_service>();
-			eventLoop_ = new DummyEventLoop();
-			disconnected = false;
-			connectFinished = false;
-		}
-
-		void tearDown() {
-			delete eventLoop_;
-			delete boostIOServiceThread_;
-		}
-
-		void testDestructor() {
-			{
-				BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
-				testling->connect(HostAddressPort(HostAddress(address, 4), 5222));
-			}
-		}
-
-		void testDestructor_PendingEvents() {
-			{
-				BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
-				testling->connect(HostAddressPort(HostAddress(address, 4), 5222));
-				while (!eventLoop_->hasEvents()) {
-					Swift::sleep(10);
-				}
-			}
-			eventLoop_->processEvents();
-		}
-
-		void testWrite() {
-			BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
-			testling->onConnectFinished.connect(boost::bind(&BoostConnectionTest::doWrite, this, testling.get()));
-			testling->onDataRead.connect(boost::bind(&BoostConnectionTest::handleDataRead, this, _1));
-			testling->onDisconnected.connect(boost::bind(&BoostConnectionTest::handleDisconnected, this));
-			testling->connect(HostAddressPort(HostAddress("74.50.62.49"), 5222));
-			while (receivedData.empty()) {
-				Swift::sleep(10);
-				eventLoop_->processEvents();
-			}
-			testling->disconnect();
-		}
-
-		void testWrite_IPv6() {
-			BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
-			testling->onConnectFinished.connect(boost::bind(&BoostConnectionTest::doWrite, this, testling.get()));
-			testling->onDataRead.connect(boost::bind(&BoostConnectionTest::handleDataRead, this, _1));
-			testling->onDisconnected.connect(boost::bind(&BoostConnectionTest::handleDisconnected, this));
-			testling->connect(HostAddressPort(HostAddress("2001:470:1f0e:852::2"), 80));
-			while (receivedData.empty()) {
-				Swift::sleep(10);
-				eventLoop_->processEvents();
-			}
-			testling->disconnect();
-		}
-
-
-		void testWriteMultipleSimultaniouslyQueuesWrites() {
-			BoostConnection::ref testling(BoostConnection::create(boostIOService, eventLoop_));
-			testling->onConnectFinished.connect(boost::bind(&BoostConnectionTest::handleConnectFinished, this));
-			testling->onDataRead.connect(boost::bind(&BoostConnectionTest::handleDataRead, this, _1));
-			testling->onDisconnected.connect(boost::bind(&BoostConnectionTest::handleDisconnected, this));
-			testling->connect(HostAddressPort(HostAddress("74.50.62.49"), 5222));
-			while (!connectFinished) {
-				boostIOService->run_one();
-				eventLoop_->processEvents();
-			}
-
-			testling->write(createSafeByteArray("<stream:strea"));
-			testling->write(createSafeByteArray("m"));
-			testling->write(createSafeByteArray(">"));
-
-			 // Check that we only did one write event, the others are queued
-			/*int runHandlers = */boostIOService->poll();
-			// Disabling this test, because poll runns all handlers that are added during poll() as well, so
-			// this test doesn't really work any more. We'll have to trust that things are queued.
-			//CPPUNIT_ASSERT_EQUAL(1, runHandlers);
-			// Process the other events
-			while (receivedData.empty()) {
-				boostIOService->run_one();
-				eventLoop_->processEvents();
-			}
-
-			// Disconnect & clean up
-			testling->disconnect();
-			while (!disconnected) {
-				boostIOService->run_one();
-				eventLoop_->processEvents();
-			}
-		}
-
-		void doWrite(BoostConnection* connection) {
-			connection->write(createSafeByteArray("<stream:stream>"));
-			connection->write(createSafeByteArray("\r\n\r\n")); // Temporarily, while we don't have an xmpp server running on ipv6
-		}
-
-		void handleDataRead(boost::shared_ptr<SafeByteArray> data) {
-			append(receivedData, *data);
-		}
-
-		void handleDisconnected() {
-			disconnected = true;
-		}
-
-		void handleConnectFinished() {
-			connectFinished = true;
-		}
-
-	private:
-		BoostIOServiceThread* boostIOServiceThread_;
-		boost::shared_ptr<boost::asio::io_service> boostIOService;
-		DummyEventLoop* eventLoop_;
-		ByteArray receivedData;
-		bool disconnected;
-		bool connectFinished;
+        CPPUNIT_TEST_SUITE_END();
+
+    public:
+        void setUp() {
+            eventLoop_ = new DummyEventLoop();
+            boostIOServiceThread_ = new BoostIOServiceThread();
+            boostIOService_ = std::make_shared<boost::asio::io_service>();
+            disconnected_ = false;
+            connectFinished_ = false;
+        }
+
+        void tearDown() {
+            delete boostIOServiceThread_;
+            while (eventLoop_->hasEvents()) {
+                eventLoop_->processEvents();
+            }
+            delete eventLoop_;
+        }
+
+        void testDestructor() {
+            {
+                BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
+                auto hostAddress = HostAddress::fromString(getenv("SWIFT_NETWORK_TEST_IPV4"));
+                CPPUNIT_ASSERT_EQUAL(true, hostAddress.is_initialized());
+                testling->connect(HostAddressPort(hostAddress.get(), 5222));
+            }
+        }
+
+        void testDestructor_PendingEvents() {
+            {
+                BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
+                auto hostAddress = HostAddress::fromString(getenv("SWIFT_NETWORK_TEST_IPV4"));
+                CPPUNIT_ASSERT_EQUAL(true, hostAddress.is_initialized());
+                testling->connect(HostAddressPort(hostAddress.get(), 5222));
+                while (!eventLoop_->hasEvents()) {
+                    Swift::sleep(10);
+                }
+            }
+            eventLoop_->processEvents();
+        }
+
+        void testWrite() {
+            using namespace boost::posix_time;
+
+            BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
+            testling->onConnectFinished.connect(boost::bind(&BoostConnectionTest::doWrite, this, testling.get()));
+            testling->onDataRead.connect(boost::bind(&BoostConnectionTest::handleDataRead, this, _1));
+            testling->onDisconnected.connect(boost::bind(&BoostConnectionTest::handleDisconnected, this));
+            auto hostAddress = HostAddress::fromString(getenv("SWIFT_NETWORK_TEST_IPV4"));
+            CPPUNIT_ASSERT_EQUAL(true, hostAddress.is_initialized());
+            testling->connect(HostAddressPort(hostAddress.get(), 5222));
+
+            boost::posix_time::ptime start = second_clock::local_time();
+            while (receivedData_.empty() && ((second_clock::local_time() - start) < seconds(60)))  {
+                Swift::sleep(10);
+                eventLoop_->processEvents();
+            }
+            CPPUNIT_ASSERT_EQUAL(false, receivedData_.empty());
+            testling->disconnect();
+        }
+
+        void testWrite_IPv6() {
+            using namespace boost::posix_time;
+
+            BoostConnection::ref testling(BoostConnection::create(boostIOServiceThread_->getIOService(), eventLoop_));
+            testling->onConnectFinished.connect(boost::bind(&BoostConnectionTest::doWrite, this, testling.get()));
+            testling->onDataRead.connect(boost::bind(&BoostConnectionTest::handleDataRead, this, _1));
+            testling->onDisconnected.connect(boost::bind(&BoostConnectionTest::handleDisconnected, this));
+            auto hostAddress = HostAddress::fromString(getenv("SWIFT_NETWORK_TEST_IPV6"));
+            CPPUNIT_ASSERT_EQUAL(true, hostAddress.is_initialized());
+            testling->connect(HostAddressPort(hostAddress.get(), 5222));
+
+            boost::posix_time::ptime start = second_clock::local_time();
+            while (receivedData_.empty() && ((second_clock::local_time() - start) < seconds(60)))  {
+                Swift::sleep(10);
+                eventLoop_->processEvents();
+            }
+            CPPUNIT_ASSERT_EQUAL(false, receivedData_.empty());
+            testling->disconnect();
+        }
+
+
+        void testWriteMultipleSimultaniouslyQueuesWrites() {
+            BoostConnection::ref testling(BoostConnection::create(boostIOService_, eventLoop_));
+            testling->onConnectFinished.connect(boost::bind(&BoostConnectionTest::handleConnectFinished, this));
+            testling->onDataRead.connect(boost::bind(&BoostConnectionTest::handleDataRead, this, _1));
+            testling->onDisconnected.connect(boost::bind(&BoostConnectionTest::handleDisconnected, this));
+
+            auto hostAddress = HostAddress::fromString(getenv("SWIFT_NETWORK_TEST_IPV4"));
+            CPPUNIT_ASSERT_EQUAL(true, hostAddress.is_initialized());
+            testling->connect(HostAddressPort(hostAddress.get(), 5222));
+            while (!connectFinished_) {
+                boostIOService_->run_one();
+                eventLoop_->processEvents();
+            }
+
+            testling->write(createSafeByteArray("<stream:strea"));
+            testling->write(createSafeByteArray("m"));
+            testling->write(createSafeByteArray(">"));
+
+             // Check that we only did one write event, the others are queued
+            /*int runHandlers = */boostIOService_->poll();
+            // Disabling this test, because poll runns all handlers that are added during poll() as well, so
+            // this test doesn't really work any more. We'll have to trust that things are queued.
+            //CPPUNIT_ASSERT_EQUAL(1, runHandlers);
+            // Process the other events
+            while (receivedData_.empty()) {
+                boostIOService_->run_one();
+                eventLoop_->processEvents();
+            }
+
+            // Disconnect & clean up
+            testling->disconnect();
+            while (!disconnected_) {
+                boostIOService_->run_one();
+                eventLoop_->processEvents();
+            }
+        }
+
+        void doWrite(BoostConnection* connection) {
+            connection->write(createSafeByteArray("<stream:stream>"));
+            connection->write(createSafeByteArray("\r\n\r\n")); // Temporarily, while we don't have an xmpp server running on ipv6
+        }
+
+        void handleDataRead(std::shared_ptr<SafeByteArray> data) {
+            append(receivedData_, *data);
+        }
+
+        void handleDisconnected() {
+            disconnected_ = true;
+        }
+
+        void handleConnectFinished() {
+            connectFinished_ = true;
+        }
+
+    private:
+        BoostIOServiceThread* boostIOServiceThread_;
+        std::shared_ptr<boost::asio::io_service> boostIOService_;
+        DummyEventLoop* eventLoop_;
+        ByteArray receivedData_;
+        bool disconnected_;
+        bool connectFinished_;
 };
 
 CPPUNIT_TEST_SUITE_REGISTRATION(BoostConnectionTest);