summaryrefslogtreecommitdiff
path: root/test/DHTFindNodeReplyMessageTest.cc
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2008-12-14 05:46:45 +0000
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2008-12-14 05:46:45 +0000
commit9202fe23c8113cd02c77a2b1132372b1d0601190 (patch)
tree322fd603b573504fb4a2d472705887a33074b4da /test/DHTFindNodeReplyMessageTest.cc
parent057132cd5f03294433648c5f794b43c57fd62375 (diff)
2008-12-14 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Use BDE instead of Dictionary/List/Data. * src/DHTAbstractMessage.cc * src/DHTAbstractMessage.h * src/DHTAnnouncePeerMessage.cc * src/DHTAnnouncePeerMessage.h * src/DHTAnnouncePeerReplyMessage.cc * src/DHTAnnouncePeerReplyMessage.h * src/DHTFindNodeMessage.cc * src/DHTFindNodeMessage.h * src/DHTFindNodeReplyMessage.cc * src/DHTFindNodeReplyMessage.h * src/DHTGetPeersMessage.cc * src/DHTGetPeersMessage.h * src/DHTGetPeersReplyMessage.cc * src/DHTGetPeersReplyMessage.h * src/DHTMessageFactory.h * src/DHTMessageFactoryImpl.cc * src/DHTMessageFactoryImpl.h * src/DHTMessageReceiver.cc * src/DHTMessageTracker.cc * src/DHTMessageTracker.h * src/DHTPingMessage.cc * src/DHTPingMessage.h * src/DHTPingReplyMessage.cc * src/DHTPingReplyMessage.h * src/DHTQueryMessage.cc * src/DHTQueryMessage.h * src/DHTResponseMessage.cc * src/DHTResponseMessage.h * test/DHTAnnouncePeerMessageTest.cc * test/DHTAnnouncePeerReplyMessageTest.cc * test/DHTFindNodeMessageTest.cc * test/DHTFindNodeReplyMessageTest.cc * test/DHTGetPeersMessageTest.cc * test/DHTGetPeersReplyMessageTest.cc * test/DHTMessageFactoryImplTest.cc * test/DHTMessageTrackerTest.cc * test/DHTPingMessageTest.cc * test/DHTPingReplyMessageTest.cc * test/MockDHTMessageFactory.h
Diffstat (limited to 'test/DHTFindNodeReplyMessageTest.cc')
-rw-r--r--test/DHTFindNodeReplyMessageTest.cc27
1 files changed, 12 insertions, 15 deletions
diff --git a/test/DHTFindNodeReplyMessageTest.cc b/test/DHTFindNodeReplyMessageTest.cc
index cab09ad1..d4f3290e 100644
--- a/test/DHTFindNodeReplyMessageTest.cc
+++ b/test/DHTFindNodeReplyMessageTest.cc
@@ -1,14 +1,14 @@
#include "DHTFindNodeReplyMessage.h"
+
+#include <cppunit/extensions/HelperMacros.h>
+
#include "DHTNode.h"
#include "DHTUtil.h"
-#include "BencodeVisitor.h"
-#include "Dictionary.h"
-#include "Data.h"
#include "Exception.h"
#include "Util.h"
#include "DHTBucket.h"
#include "PeerMessageUtil.h"
-#include <cppunit/extensions/HelperMacros.h>
+#include "bencode.h"
namespace aria2 {
@@ -56,18 +56,15 @@ void DHTFindNodeReplyMessageTest::testGetBencodedMessage()
std::string msgbody = msg.getBencodedMessage();
- SharedHandle<Dictionary> cm(new Dictionary());
- cm->put("t", new Data(transactionID));
- cm->put("y", new Data("r"));
- Dictionary* r = new Dictionary();
- cm->put("r", r);
- r->put("id", new Data(localNode->getID(), DHT_ID_LENGTH));
- r->put("nodes", new Data(compactNodeInfo));
-
- BencodeVisitor v;
- cm->accept(&v);
+ bencode::BDE dict = bencode::BDE::dict();
+ dict["t"] = transactionID;
+ dict["y"] = bencode::BDE("r");
+ bencode::BDE rDict = bencode::BDE::dict();
+ rDict["id"] = bencode::BDE(localNode->getID(), DHT_ID_LENGTH);
+ rDict["nodes"] = compactNodeInfo;
+ dict["r"] = rDict;
- CPPUNIT_ASSERT_EQUAL(v.getBencodedData(), msgbody);
+ CPPUNIT_ASSERT_EQUAL(bencode::encode(dict), msgbody);
}
} // namespace aria2