summaryrefslogtreecommitdiff
path: root/test/BtRejectMessageTest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'test/BtRejectMessageTest.cc')
-rw-r--r--test/BtRejectMessageTest.cc67
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());
}