← Back to branch summary

~ubuntu-branches/ubuntu/wily/aria2/wily-proposed

« back to all changes in this revision

Viewing changes to test/DefaultBtMessageFactoryTest.cc

  • Committer: Package Import Robot
  • Author(s): Kartik Mistry, Kartik Mistry, Patrick Ruckstuhl
  • Date: 2013-09-22 18:52:14 UTC
  • mfrom: (2.5.19 sid)
  • Revision ID: package-import@ubuntu.com-20130922185214-upeu2ljgeqi7e7oo
Tags: 1.18.0-1
http://bugs.debian.org/697659
http://bugs.debian.org/720977
[ Kartik Mistry ]
* New upstream release.
* debian/control:
  + (really) Set priority to optional from extra (Closes: #697659).
  + wrap-and-sort some fields.

[ Patrick Ruckstuhl ]
* debian/rules:
  + Allow parallel building (Closes: #720977)
* debian/tests, debian/control:
  + autopkgtest infrastructure

Show diffs side-by-side

added added

removed removed

Lines of Context:
test/DefaultBtMessageFactoryTest.cc
24
24
  CPPUNIT_TEST(testCreatePortMessage);
25
25
  CPPUNIT_TEST_SUITE_END();
26
26
private:
27
 
  SharedHandle<DownloadContext> dctx_;
28
 
  SharedHandle<Peer> peer_;
29
 
  SharedHandle<MockExtensionMessageFactory> exmsgFactory_;
30
 
  SharedHandle<DefaultBtMessageFactory> factory_;
 
27
  std::unique_ptr<DownloadContext> dctx_;
 
28
  std::shared_ptr<Peer> peer_;
 
29
  std::shared_ptr<MockExtensionMessageFactory> exmsgFactory_;
 
30
  std::unique_ptr<DefaultBtMessageFactory> factory_;
31
31
public:
32
32
  void setUp()
33
33
  {
34
 
    dctx_.reset(new DownloadContext());
 
34
    dctx_ = make_unique<DownloadContext>();
35
35
 
36
 
    peer_.reset(new Peer("192.168.0.1", 6969));
 
36
    peer_ = std::make_shared<Peer>("192.168.0.1", 6969);
37
37
    peer_->allocateSessionResource(1024, 1024*1024);
38
38
    peer_->setExtendedMessagingEnabled(true);
39
39
 
40
 
    exmsgFactory_.reset(new MockExtensionMessageFactory());
 
40
    exmsgFactory_ = std::make_shared<MockExtensionMessageFactory>();
41
41
 
42
 
    factory_.reset(new DefaultBtMessageFactory());
43
 
    factory_->setDownloadContext(dctx_);
 
42
    factory_ = make_unique<DefaultBtMessageFactory>();
 
43
    factory_->setDownloadContext(dctx_.get());
44
44
    factory_->setPeer(peer_);
45
 
    factory_->setExtensionMessageFactory(exmsgFactory_);
 
45
    factory_->setExtensionMessageFactory(exmsgFactory_.get());
46
46
  }
47
47
 
48
48
  void testCreateBtMessage_BtExtendedMessage();
62
62
  msg[5] = 1; // Set dummy extended message ID 1
63
63
  memcpy(msg+6, payload.c_str(), payload.size());
64
64
 
65
 
  SharedHandle<BtExtendedMessage> m
66
 
    (dynamic_pointer_cast<BtExtendedMessage>
67
 
     (factory_->createBtMessage((const unsigned char*)msg+4, sizeof(msg))));
68
 
 
 
65
  auto m =
 
66
    factory_->createBtMessage((const unsigned char*)msg+4, sizeof(msg));
 
67
  CPPUNIT_ASSERT(BtExtendedMessage::ID == m->getId());
69
68
  try {
70
69
    // disable extended messaging
71
70
    peer_->setExtendedMessagingEnabled(false);
83
82
    bittorrent::createPeerMessageString(data, sizeof(data), 3, 9);
84
83
    bittorrent::setShortIntParam(&data[5], 6881);
85
84
    try {
86
 
      SharedHandle<BtPortMessage> m
87
 
        (dynamic_pointer_cast<BtPortMessage>
88
 
         (factory_->createBtMessage(&data[4], sizeof(data)-4)));
89
 
      CPPUNIT_ASSERT(m);
 
85
      auto r = factory_->createBtMessage(&data[4], sizeof(data)-4);
 
86
      CPPUNIT_ASSERT(BtPortMessage::ID == r->getId());
 
87
      auto m = static_cast<const BtPortMessage*>(r.get());
90
88
      CPPUNIT_ASSERT_EQUAL((uint16_t)6881, m->getPort());
91
89
    } catch(Exception& e) {
92
90
      CPPUNIT_FAIL(e.stackTrace());
93
91
    }
94
92
  }
95
93
  {
96
 
    SharedHandle<BtPortMessage> m
97
 
      (dynamic_pointer_cast<BtPortMessage>(factory_->createPortMessage(6881)));
 
94
    auto m = factory_->createPortMessage(6881);
98
95
    CPPUNIT_ASSERT_EQUAL((uint16_t)6881, m->getPort());
99
96
  }
100
97
}