summaryrefslogtreecommitdiff
path: root/test/DHTFindNodeMessageTest.cc
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2016-03-26 17:12:52 +0900
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2016-03-26 17:20:27 +0900
commitd64089632e82a2548752c7a6baaa68365a795a74 (patch)
tree57fc7b85a8c1290975d58fa4761b04a7695bf7e3 /test/DHTFindNodeMessageTest.cc
parentc81b1321122e8509b55c94c0cbca0d8d6d319f0c (diff)
parenta7cd943a06b1a5ff4515b57060b84037f8d01f9f (diff)
Merge branch 'master' into random-webseeding
Diffstat (limited to 'test/DHTFindNodeMessageTest.cc')
-rw-r--r--test/DHTFindNodeMessageTest.cc23
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());