diff options
Diffstat (limited to 'test/BtRejectMessageTest.cc')
| -rw-r--r-- | test/BtRejectMessageTest.cc | 67 |
1 files changed, 37 insertions, 30 deletions
diff --git a/test/BtRejectMessageTest.cc b/test/BtRejectMessageTest.cc index 53004554..3e3b7f8e 100644 --- a/test/BtRejectMessageTest.cc +++ b/test/BtRejectMessageTest.cc @@ -11,7 +11,7 @@ namespace aria2 { -class BtRejectMessageTest:public CppUnit::TestFixture { +class BtRejectMessageTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(BtRejectMessageTest); CPPUNIT_TEST(testCreate); @@ -21,8 +21,8 @@ class BtRejectMessageTest:public CppUnit::TestFixture { CPPUNIT_TEST(testDoReceivedActionFastExtensionDisabled); CPPUNIT_TEST(testToString); CPPUNIT_TEST_SUITE_END(); -private: +private: public: void testCreate(); void testCreateMessage(); @@ -35,40 +35,40 @@ public: public: std::unique_ptr<RequestSlot> slot; - void setRequestSlot(std::unique_ptr<RequestSlot> s) - { - slot = std::move(s); - } + void setRequestSlot(std::unique_ptr<RequestSlot> s) { slot = std::move(s); } - virtual const RequestSlot* getOutstandingRequest - (size_t index, int32_t begin, int32_t length) CXX11_OVERRIDE + virtual const RequestSlot* + getOutstandingRequest(size_t index, int32_t begin, + int32_t length) CXX11_OVERRIDE { - if(slot && - slot->getIndex() == index && slot->getBegin() == begin && - slot->getLength() == length) { + if (slot && slot->getIndex() == index && slot->getBegin() == begin && + slot->getLength() == length) { return slot.get(); - } else { + } + else { return nullptr; } } virtual void removeOutstandingRequest(const RequestSlot* s) CXX11_OVERRIDE { - if(slot->getIndex() == s->getIndex() && - slot->getBegin() == s->getBegin() && - slot->getLength() == s->getLength()) { + if (slot->getIndex() == s->getIndex() && + slot->getBegin() == s->getBegin() && + slot->getLength() == s->getLength()) { slot.reset(); } } }; - typedef std::shared_ptr<MockBtMessageDispatcher2> MockBtMessageDispatcher2Handle; + typedef std::shared_ptr<MockBtMessageDispatcher2> + MockBtMessageDispatcher2Handle; std::shared_ptr<Peer> peer; std::shared_ptr<MockBtMessageDispatcher2> dispatcher; std::shared_ptr<BtRejectMessage> msg; - void setUp() { + void setUp() + { peer.reset(new Peer("host", 6969)); peer->allocateSessionResource(1_k, 1_m); @@ -83,10 +83,10 @@ public: } }; - CPPUNIT_TEST_SUITE_REGISTRATION(BtRejectMessageTest); -void BtRejectMessageTest::testCreate() { +void BtRejectMessageTest::testCreate() +{ unsigned char msg[17]; bittorrent::createPeerMessageString(msg, sizeof(msg), 13, 16); bittorrent::setIntParam(&msg[5], 12345); @@ -104,7 +104,8 @@ void BtRejectMessageTest::testCreate() { bittorrent::createPeerMessageString(msg, sizeof(msg), 14, 16); BtRejectMessage::create(&msg[4], 14); CPPUNIT_FAIL("exception must be thrown."); - } catch(...) { + } + catch (...) { } // case: id is wrong try { @@ -112,11 +113,13 @@ void BtRejectMessageTest::testCreate() { bittorrent::createPeerMessageString(msg, sizeof(msg), 13, 17); BtRejectMessage::create(&msg[4], 13); CPPUNIT_FAIL("exception must be thrown."); - } catch(...) { + } + catch (...) { } } -void BtRejectMessageTest::testCreateMessage() { +void BtRejectMessageTest::testCreateMessage() +{ BtRejectMessage msg; msg.setIndex(12345); msg.setBegin(256); @@ -128,10 +131,11 @@ void BtRejectMessageTest::testCreateMessage() { bittorrent::setIntParam(&data[13], 1_k); unsigned char* rawmsg = msg.createMessage(); CPPUNIT_ASSERT(memcmp(rawmsg, data, 17) == 0); - delete [] rawmsg; + delete[] rawmsg; } -void BtRejectMessageTest::testDoReceivedAction() { +void BtRejectMessageTest::testDoReceivedAction() +{ peer->setFastExtensionEnabled(true); dispatcher->setRequestSlot(make_unique<RequestSlot>(1, 16, 32, 2)); @@ -142,7 +146,8 @@ void BtRejectMessageTest::testDoReceivedAction() { CPPUNIT_ASSERT(!dispatcher->getOutstandingRequest(1, 16, 32)); } -void BtRejectMessageTest::testDoReceivedActionNoMatch() { +void BtRejectMessageTest::testDoReceivedActionNoMatch() +{ peer->setFastExtensionEnabled(true); dispatcher->setRequestSlot(make_unique<RequestSlot>(2, 16, 32, 2)); @@ -151,10 +156,10 @@ void BtRejectMessageTest::testDoReceivedActionNoMatch() { msg->doReceivedAction(); CPPUNIT_ASSERT(dispatcher->getOutstandingRequest(2, 16, 32)); - } -void BtRejectMessageTest::testDoReceivedActionFastExtensionDisabled() { +void BtRejectMessageTest::testDoReceivedActionFastExtensionDisabled() +{ RequestSlot slot(1, 16, 32, 2); dispatcher->setRequestSlot(make_unique<RequestSlot>(1, 16, 32, 2)); @@ -162,11 +167,13 @@ void BtRejectMessageTest::testDoReceivedActionFastExtensionDisabled() { try { msg->doReceivedAction(); CPPUNIT_FAIL("exception must be thrown."); - } catch(...) {} - + } + catch (...) { + } } -void BtRejectMessageTest::testToString() { +void BtRejectMessageTest::testToString() +{ CPPUNIT_ASSERT_EQUAL(std::string("reject index=1, begin=16, length=32"), msg->toString()); } |
