Rewritten ExtensionMessageRegistry

2012-09-26 22:02:48 +09:00
parent f0000a8754
commit c13dc166de
20 changed files with 321 additions and 143 deletions
--- a/test/UTMetadataRequestExtensionMessageTest.cc
+++ b/test/UTMetadataRequestExtensionMessageTest.cc
@@ -16,6 +16,7 @@
 #include "PieceStorage.h"
 #include "extension_message_test_helper.h"
 #include "DlAbortEx.h"
+#include "ExtensionMessageRegistry.h"

 namespace aria2 {

@@ -44,7 +45,7 @@ public:
    dctx_->setAttribute(CTX_ATTR_BT, attrs);
    peer_.reset(new Peer("host", 6880));
    peer_->allocateSessionResource(0, 0);
-    peer_->setExtension("ut_metadata", 1);
+    peer_->setExtension(ExtensionMessageRegistry::UT_METADATA, 1);
  }

  template<typename T>