diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2016-03-26 17:12:52 +0900 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2016-03-26 17:20:27 +0900 |
| commit | d64089632e82a2548752c7a6baaa68365a795a74 (patch) | |
| tree | 57fc7b85a8c1290975d58fa4761b04a7695bf7e3 /test/DHTFindNodeMessageTest.cc | |
| parent | c81b1321122e8509b55c94c0cbca0d8d6d319f0c (diff) | |
| parent | a7cd943a06b1a5ff4515b57060b84037f8d01f9f (diff) | |
Merge branch 'master' into random-webseeding
Diffstat (limited to 'test/DHTFindNodeMessageTest.cc')
| -rw-r--r-- | test/DHTFindNodeMessageTest.cc | 23 |
1 files changed, 11 insertions, 12 deletions
diff --git a/test/DHTFindNodeMessageTest.cc b/test/DHTFindNodeMessageTest.cc index 2cd13a10..d5101c35 100644 --- a/test/DHTFindNodeMessageTest.cc +++ b/test/DHTFindNodeMessageTest.cc @@ -13,12 +13,13 @@ namespace aria2 { -class DHTFindNodeMessageTest:public CppUnit::TestFixture { +class DHTFindNodeMessageTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(DHTFindNodeMessageTest); CPPUNIT_TEST(testGetBencodedMessage); CPPUNIT_TEST(testDoReceivedAction); CPPUNIT_TEST_SUITE_END(); + public: std::shared_ptr<DHTNode> localNode_; std::shared_ptr<DHTNode> remoteNode_; @@ -34,23 +35,21 @@ public: void testGetBencodedMessage(); void testDoReceivedAction(); - class MockDHTMessageFactory2:public MockDHTMessageFactory { + class MockDHTMessageFactory2 : public MockDHTMessageFactory { public: - virtual std::unique_ptr<DHTFindNodeReplyMessage> - createFindNodeReplyMessage - (const std::shared_ptr<DHTNode>& remoteNode, - std::vector<std::shared_ptr<DHTNode>> closestKNodes, - const std::string& transactionID) CXX11_OVERRIDE + virtual std::unique_ptr<DHTFindNodeReplyMessage> createFindNodeReplyMessage( + const std::shared_ptr<DHTNode>& remoteNode, + std::vector<std::shared_ptr<DHTNode>> closestKNodes, + const std::string& transactionID) CXX11_OVERRIDE { - auto m = make_unique<DHTFindNodeReplyMessage> - (AF_INET, localNode_, remoteNode, transactionID); + auto m = make_unique<DHTFindNodeReplyMessage>(AF_INET, localNode_, + remoteNode, transactionID); m->setClosestKNodes(std::move(closestKNodes)); return m; } }; }; - CPPUNIT_TEST_SUITE_REGISTRATION(DHTFindNodeMessageTest); void DHTFindNodeMessageTest::testGetBencodedMessage() @@ -102,8 +101,8 @@ void DHTFindNodeMessageTest::testDoReceivedAction() msg.doReceivedAction(); CPPUNIT_ASSERT_EQUAL((size_t)1, dispatcher.messageQueue_.size()); - auto m = dynamic_cast<DHTFindNodeReplyMessage*> - (dispatcher.messageQueue_[0].message_.get()); + auto m = dynamic_cast<DHTFindNodeReplyMessage*>( + dispatcher.messageQueue_[0].message_.get()); CPPUNIT_ASSERT(*localNode_ == *m->getLocalNode()); CPPUNIT_ASSERT(*remoteNode_ == *m->getRemoteNode()); CPPUNIT_ASSERT_EQUAL(std::string("find_node"), m->getMessageType()); |
