diff options
Diffstat (limited to 'test/BtBitfieldMessageTest.cc')
| -rw-r--r-- | test/BtBitfieldMessageTest.cc | 53 |
1 files changed, 30 insertions, 23 deletions
diff --git a/test/BtBitfieldMessageTest.cc b/test/BtBitfieldMessageTest.cc index 095fd9e8..a3059381 100644 --- a/test/BtBitfieldMessageTest.cc +++ b/test/BtBitfieldMessageTest.cc @@ -13,7 +13,7 @@ namespace aria2 { -class BtBitfieldMessageTest:public CppUnit::TestFixture { +class BtBitfieldMessageTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(BtBitfieldMessageTest); CPPUNIT_TEST(testCreate); @@ -22,11 +22,10 @@ class BtBitfieldMessageTest:public CppUnit::TestFixture { CPPUNIT_TEST(testDoReceivedAction_goodByeSeeder); CPPUNIT_TEST(testToString); CPPUNIT_TEST_SUITE_END(); -private: +private: public: - void setUp() { - } + void setUp() {} void testCreate(); void testCreateMessage(); @@ -35,11 +34,11 @@ public: void testToString(); }; - CPPUNIT_TEST_SUITE_REGISTRATION(BtBitfieldMessageTest); -void BtBitfieldMessageTest::testCreate() { - unsigned char msg[5+2]; +void BtBitfieldMessageTest::testCreate() +{ + unsigned char msg[5 + 2]; bittorrent::createPeerMessageString(msg, sizeof(msg), 3, 5); unsigned char bitfield[2]; memset(bitfield, 0xff, sizeof(bitfield)); @@ -54,47 +53,53 @@ void BtBitfieldMessageTest::testCreate() { bittorrent::createPeerMessageString(msg, sizeof(msg), 1, 5); BtBitfieldMessage::create(&msg[4], 1); CPPUNIT_FAIL("exception must be thrown."); - } catch(...) { + } + catch (...) { } // case: id is wrong try { - unsigned char msg[5+2]; + unsigned char msg[5 + 2]; bittorrent::createPeerMessageString(msg, sizeof(msg), 3, 6); BtBitfieldMessage::create(&msg[4], 3); CPPUNIT_FAIL("exception must be thrown."); - } catch(...) { + } + catch (...) { } } -void BtBitfieldMessageTest::testCreateMessage() { +void BtBitfieldMessageTest::testCreateMessage() +{ BtBitfieldMessage msg; unsigned char bitfield[2]; memset(bitfield, 0xff, sizeof(bitfield)); msg.setBitfield(bitfield, sizeof(bitfield)); - unsigned char data[5+2]; + unsigned char data[5 + 2]; bittorrent::createPeerMessageString(data, sizeof(data), 3, 5); memcpy(&data[5], bitfield, sizeof(bitfield)); unsigned char* rawmsg = msg.createMessage(); CPPUNIT_ASSERT(memcmp(rawmsg, data, 7) == 0); - delete [] rawmsg; + delete[] rawmsg; CPPUNIT_ASSERT_EQUAL((size_t)7, msg.getMessageLength()); } -void BtBitfieldMessageTest::testDoReceivedAction() { +void BtBitfieldMessageTest::testDoReceivedAction() +{ std::shared_ptr<Peer> peer(new Peer("host1", 6969)); peer->allocateSessionResource(16_k, 256_k); BtBitfieldMessage msg; msg.setPeer(peer); auto pieceStorage = make_unique<MockPieceStorage>(); msg.setPieceStorage(pieceStorage.get()); - unsigned char bitfield[] = { 0xff, 0xff }; + unsigned char bitfield[] = {0xff, 0xff}; msg.setBitfield(bitfield, sizeof(bitfield)); - CPPUNIT_ASSERT_EQUAL(std::string("0000"), util::toHex(peer->getBitfield(), - peer->getBitfieldLength())); + CPPUNIT_ASSERT_EQUAL( + std::string("0000"), + util::toHex(peer->getBitfield(), peer->getBitfieldLength())); msg.doReceivedAction(); - CPPUNIT_ASSERT_EQUAL(std::string("ffff"), util::toHex(peer->getBitfield(), - peer->getBitfieldLength())); + CPPUNIT_ASSERT_EQUAL( + std::string("ffff"), + util::toHex(peer->getBitfield(), peer->getBitfieldLength())); } void BtBitfieldMessageTest::testDoReceivedAction_goodByeSeeder() @@ -105,7 +110,7 @@ void BtBitfieldMessageTest::testDoReceivedAction_goodByeSeeder() msg.setPeer(peer); auto pieceStorage = make_unique<MockPieceStorage>(); msg.setPieceStorage(pieceStorage.get()); - unsigned char bitfield[] = { 0x00 }; + unsigned char bitfield[] = {0x00}; msg.setBitfield(bitfield, sizeof(bitfield)); // peer is not seeder and client have not completed download @@ -127,14 +132,16 @@ void BtBitfieldMessageTest::testDoReceivedAction_goodByeSeeder() try { msg.doReceivedAction(); CPPUNIT_FAIL("exception must be thrown."); - } catch(DlAbortEx& e) { + } + catch (DlAbortEx& e) { // success } } -void BtBitfieldMessageTest::testToString() { +void BtBitfieldMessageTest::testToString() +{ BtBitfieldMessage msg; - unsigned char bitfield[] = { 0xff, 0xff }; + unsigned char bitfield[] = {0xff, 0xff}; msg.setBitfield(bitfield, sizeof(bitfield)); CPPUNIT_ASSERT_EQUAL(std::string("bitfield ffff"), msg.toString()); |
