2008-02-11 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>

	Added test 'testDoReceivedAction' for DHTQueryMessage 
subclasses.
	* test/DHTPingMessageTest.cc
	* test/DHTGetPeersMessageTest.cc
	* test/DHTFindNodeMessageTest.cc
	* test/DHTAnnouncePeerMessageTest.cc
	* test/MockDHTMessage.h
	* test/MockDHTMessageFactory.h
	* test/MockDHTMessageDispatcher.h: New class.

1 files changed, 53 insertions, 0 deletions

diff --git a/test/DHTFindNodeMessageTest.cc b/test/DHTFindNodeMessageTest.cc
index 3193c627..86dbeaa7 100644
--- a/test/DHTFindNodeMessageTest.cc
+++ b/test/DHTFindNodeMessageTest.cc
@@ -6,6 +6,10 @@
 #include "Data.h"
 #include "Exception.h"
 #include "Util.h"
+#include "MockDHTMessageFactory.h"
+#include "MockDHTMessage.h"
+#include "MockDHTMessageDispatcher.h"
+#include "DHTRoutingTable.h"
 #include <cppunit/extensions/HelperMacros.h>
 
 namespace aria2 {
@@ -14,6 +18,7 @@ class DHTFindNodeMessageTest:public CppUnit::TestFixture {
 
   CPPUNIT_TEST_SUITE(DHTFindNodeMessageTest);
   CPPUNIT_TEST(testGetBencodedMessage);
+  CPPUNIT_TEST(testDoReceivedAction);
   CPPUNIT_TEST_SUITE_END();
 public:
   void setUp() {}
@@ -21,6 +26,21 @@ public:
   void tearDown() {}
 
   void testGetBencodedMessage();
+  void testDoReceivedAction();
+
+  class MockDHTMessageFactory2:public MockDHTMessageFactory {
+  public:
+    virtual SharedHandle<DHTMessage>
+    createFindNodeReplyMessage(const SharedHandle<DHTNode>& remoteNode,
+			       const std::deque<SharedHandle<DHTNode> >& closestKNodes,
+			       const std::string& transactionID)
+    {
+      SharedHandle<MockDHTMessage> m =
+	new MockDHTMessage(_localNode, remoteNode, "find_node", transactionID);
+      m->_nodes = closestKNodes;
+      return m;
+    }
+  };
 };
 
 
@@ -56,4 +76,37 @@ void DHTFindNodeMessageTest::testGetBencodedMessage()
   CPPUNIT_ASSERT_EQUAL(v.getBencodedData(), msgbody);
 }
 
+void DHTFindNodeMessageTest::testDoReceivedAction()
+{
+  SharedHandle<DHTNode> localNode = new DHTNode();
+  SharedHandle<DHTNode> remoteNode = new DHTNode();
+
+  char tid[DHT_TRANSACTION_ID_LENGTH];
+  DHTUtil::generateRandomData(tid, DHT_TRANSACTION_ID_LENGTH);
+  std::string transactionID(&tid[0], &tid[DHT_TRANSACTION_ID_LENGTH]);
+
+  SharedHandle<DHTNode> targetNode = new DHTNode();
+
+  MockDHTMessageDispatcher dispatcher;
+  MockDHTMessageFactory2 factory;
+  factory.setLocalNode(localNode);
+  DHTRoutingTable routingTable(localNode);
+  routingTable.addNode(targetNode);
+
+  DHTFindNodeMessage msg(localNode, remoteNode, targetNode->getID(), transactionID);
+  msg.setMessageDispatcher(&dispatcher);
+  msg.setMessageFactory(&factory);
+  msg.setRoutingTable(&routingTable);
+
+  msg.doReceivedAction();
+
+  CPPUNIT_ASSERT_EQUAL((size_t)1, dispatcher._messageQueue.size());
+  SharedHandle<MockDHTMessage> m = dispatcher._messageQueue[0]._message;
+  CPPUNIT_ASSERT(localNode == m->getLocalNode());
+  CPPUNIT_ASSERT(remoteNode == m->getRemoteNode());
+  CPPUNIT_ASSERT_EQUAL(std::string("find_node"), m->getMessageType());
+  CPPUNIT_ASSERT_EQUAL(msg.getTransactionID(), m->getTransactionID());
+  CPPUNIT_ASSERT_EQUAL((size_t)1, m->_nodes.size());
+}
+
 } // namespace aria2