summaryrefslogtreecommitdiff
path: root/test/DHTPingMessageTest.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/DHTPingMessageTest.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/DHTPingMessageTest.cc')
-rw-r--r--test/DHTPingMessageTest.cc27
1 files changed, 12 insertions, 15 deletions
diff --git a/test/DHTPingMessageTest.cc b/test/DHTPingMessageTest.cc
index d931c3b7..aa1c621c 100644
--- a/test/DHTPingMessageTest.cc
+++ b/test/DHTPingMessageTest.cc
@@ -1,15 +1,15 @@
#include "DHTPingMessage.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 "MockDHTMessageFactory.h"
#include "MockDHTMessageDispatcher.h"
#include "MockDHTMessage.h"
-#include <cppunit/extensions/HelperMacros.h>
+#include "bencode.h"
namespace aria2 {
@@ -57,18 +57,15 @@ void DHTPingMessageTest::testGetBencodedMessage()
std::string msgbody = msg.getBencodedMessage();
- SharedHandle<Dictionary> cm(new Dictionary());
- cm->put("t", new Data(transactionID));
- cm->put("y", new Data("q"));
- cm->put("q", new Data("ping"));
- Dictionary* a = new Dictionary();
- cm->put("a", a);
- a->put("id", new Data(localNode->getID(), DHT_ID_LENGTH));
-
- BencodeVisitor v;
- cm->accept(&v);
+ bencode::BDE dict = bencode::BDE::dict();
+ dict["t"] = transactionID;
+ dict["y"] = bencode::BDE("q");
+ dict["q"] = bencode::BDE("ping");
+ bencode::BDE aDict = bencode::BDE::dict();
+ aDict["id"] = bencode::BDE(localNode->getID(), DHT_ID_LENGTH);
+ dict["a"] = aDict;
- CPPUNIT_ASSERT_EQUAL(v.getBencodedData(), msgbody);
+ CPPUNIT_ASSERT_EQUAL(bencode::encode(dict), msgbody);
}
void DHTPingMessageTest::testDoReceivedAction()