/* * Copyright (c) 2010 Remko Tronçon * Licensed under the GNU General Public License v3. * See Documentation/Licenses/GPLv3.txt for more information. */ #include #include #include #include #include "Swiften/Component/ComponentConnector.h" #include "Swiften/Network/Connection.h" #include "Swiften/Network/ConnectionFactory.h" #include "Swiften/Network/HostAddressPort.h" #include "Swiften/Network/StaticDomainNameResolver.h" #include "Swiften/Network/DummyTimerFactory.h" #include "Swiften/EventLoop/MainEventLoop.h" #include "Swiften/EventLoop/DummyEventLoop.h" using namespace Swift; class ComponentConnectorTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(ComponentConnectorTest); CPPUNIT_TEST(testConnect); CPPUNIT_TEST(testConnect_FirstAddressHostFails); CPPUNIT_TEST(testConnect_NoHosts); CPPUNIT_TEST(testConnect_TimeoutDuringResolve); CPPUNIT_TEST(testConnect_TimeoutDuringConnect); CPPUNIT_TEST(testConnect_NoTimeout); CPPUNIT_TEST(testStop_Timeout); CPPUNIT_TEST_SUITE_END(); public: ComponentConnectorTest() : host1("1.1.1.1"), host2("2.2.2.2") { } void setUp() { eventLoop = new DummyEventLoop(); resolver = new StaticDomainNameResolver(); connectionFactory = new MockConnectionFactory(); timerFactory = new DummyTimerFactory(); } void tearDown() { delete timerFactory; delete connectionFactory; delete resolver; delete eventLoop; } void testConnect() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); resolver->addAddress("foo.com", host1); testling->start(); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(connections[0]); CPPUNIT_ASSERT(HostAddressPort(host1, 1234) == *(connections[0]->hostAddressPort)); } void testConnect_FirstAddressHostFails() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); resolver->addAddress("foo.com", host1); resolver->addAddress("foo.com", host2); connectionFactory->failingPorts.push_back(HostAddressPort(host1, 1234)); testling->start(); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(connections[0]); CPPUNIT_ASSERT(HostAddressPort(host2, 1234) == *(connections[0]->hostAddressPort)); } void testConnect_NoHosts() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); testling->start(); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(!connections[0]); } void testConnect_TimeoutDuringResolve() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); testling->setTimeoutMilliseconds(10); resolver->setIsResponsive(false); testling->start(); eventLoop->processEvents(); timerFactory->setTime(10); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(!connections[0]); } void testConnect_TimeoutDuringConnect() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); testling->setTimeoutMilliseconds(10); resolver->addAddress("foo.com", host1); connectionFactory->isResponsive = false; testling->start(); eventLoop->processEvents(); timerFactory->setTime(10); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(!connections[0]); } void testConnect_NoTimeout() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); testling->setTimeoutMilliseconds(10); resolver->addAddress("foo.com", host1); testling->start(); eventLoop->processEvents(); timerFactory->setTime(10); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(connections[0]); } void testStop_Timeout() { ComponentConnector::ref testling(createConnector("foo.com", 1234)); testling->setTimeoutMilliseconds(10); resolver->addAddress("foo.com", host1); testling->start(); testling->stop(); eventLoop->processEvents(); timerFactory->setTime(10); eventLoop->processEvents(); CPPUNIT_ASSERT_EQUAL(1, static_cast(connections.size())); CPPUNIT_ASSERT(!connections[0]); } private: ComponentConnector::ref createConnector(const String& hostname, int port) { ComponentConnector::ref connector = ComponentConnector::create(hostname, port, resolver, connectionFactory, timerFactory); connector->onConnectFinished.connect(boost::bind(&ComponentConnectorTest::handleConnectorFinished, this, _1)); return connector; } void handleConnectorFinished(boost::shared_ptr connection) { boost::shared_ptr c(boost::dynamic_pointer_cast(connection)); if (connection) { assert(c); } connections.push_back(c); } struct MockConnection : public Connection { public: MockConnection(const std::vector& failingPorts, bool isResponsive) : failingPorts(failingPorts), isResponsive(isResponsive) {} void listen() { assert(false); } void connect(const HostAddressPort& address) { hostAddressPort = address; if (isResponsive) { bool fail = std::find(failingPorts.begin(), failingPorts.end(), address) != failingPorts.end(); MainEventLoop::postEvent(boost::bind(boost::ref(onConnectFinished), fail)); } } void disconnect() { assert(false); } void write(const ByteArray&) { assert(false); } boost::optional hostAddressPort; std::vector failingPorts; bool isResponsive; }; struct MockConnectionFactory : public ConnectionFactory { MockConnectionFactory() : isResponsive(true) { } boost::shared_ptr createConnection() { return boost::shared_ptr(new MockConnection(failingPorts, isResponsive)); } bool isResponsive; std::vector failingPorts; }; private: HostAddress host1; HostAddress host2; DummyEventLoop* eventLoop; StaticDomainNameResolver* resolver; MockConnectionFactory* connectionFactory; DummyTimerFactory* timerFactory; std::vector< boost::shared_ptr > connections; }; CPPUNIT_TEST_SUITE_REGISTRATION(ComponentConnectorTest);