summaryrefslogtreecommitdiff
path: root/test/DefaultPeerListProcessorTest.cc
blob: 0d2e4ef7260abb44342a9fb255fa83b4ce620c44 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#include "DefaultPeerListProcessor.h"
#include "MetaFileUtil.h"
#include "Exception.h"
#include <cppunit/extensions/HelperMacros.h>

using namespace std;

class DefaultPeerListProcessorTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(DefaultPeerListProcessorTest);
  CPPUNIT_TEST(testExtractPeer);
  CPPUNIT_TEST(testExtract2Peers);
  CPPUNIT_TEST_SUITE_END();
private:

public:
  void setUp() {
  }

  void testExtractPeer();
  void testExtract2Peers();
};


CPPUNIT_TEST_SUITE_REGISTRATION( DefaultPeerListProcessorTest );

void DefaultPeerListProcessorTest::testExtractPeer() {
  DefaultPeerListProcessor proc(1024*1024, 10*1024*1024);
  string peersString = "d5:peersld2:ip11:192.168.0.17:peer id20:aria2-000000000000004:porti2006eeee";

  Dictionary* dic = (Dictionary*)MetaFileUtil::bdecoding(peersString.c_str(), peersString.size());
  
  CPPUNIT_ASSERT(proc.canHandle(dic->get("peers")));

  Peers peers = proc.extractPeer(dic->get("peers"));
  CPPUNIT_ASSERT_EQUAL((size_t)1, peers.size());
  PeerHandle peer = *peers.begin();
  CPPUNIT_ASSERT_EQUAL(string("192.168.0.1"), peer->ipaddr);
  CPPUNIT_ASSERT_EQUAL(2006, peer->port);
}

void DefaultPeerListProcessorTest::testExtract2Peers() {
  DefaultPeerListProcessor proc(1024*1024, 10*1024*1024);
  string peersString = "d5:peersld2:ip11:192.168.0.17:peer id20:aria2-000000000000004:porti2006eed2:ip11:192.168.0.27:peer id20:aria2-000000000000004:porti2007eeee";

  Dictionary* dic = (Dictionary*)MetaFileUtil::bdecoding(peersString.c_str(), peersString.size());

  Peers peers = proc.extractPeer(dic->get("peers"));
  CPPUNIT_ASSERT_EQUAL((size_t)2, peers.size());
  PeerHandle peer = *peers.begin();
  CPPUNIT_ASSERT_EQUAL(string("192.168.0.1"), peer->ipaddr);
  CPPUNIT_ASSERT_EQUAL(2006, peer->port);

  peer = *(peers.begin()+1);
  CPPUNIT_ASSERT_EQUAL(string("192.168.0.2"), peer->ipaddr);
  CPPUNIT_ASSERT_EQUAL(2007, peer->port);
}