diff options
Diffstat (limited to 'test/DefaultBtMessageFactoryTest.cc')
| -rw-r--r-- | test/DefaultBtMessageFactoryTest.cc | 24 |
1 files changed, 13 insertions, 11 deletions
diff --git a/test/DefaultBtMessageFactoryTest.cc b/test/DefaultBtMessageFactoryTest.cc index ad870c1f..d9d871f9 100644 --- a/test/DefaultBtMessageFactoryTest.cc +++ b/test/DefaultBtMessageFactoryTest.cc @@ -17,17 +17,19 @@ namespace aria2 { -class DefaultBtMessageFactoryTest:public CppUnit::TestFixture { +class DefaultBtMessageFactoryTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(DefaultBtMessageFactoryTest); CPPUNIT_TEST(testCreateBtMessage_BtExtendedMessage); CPPUNIT_TEST(testCreatePortMessage); CPPUNIT_TEST_SUITE_END(); + private: std::unique_ptr<DownloadContext> dctx_; std::shared_ptr<Peer> peer_; std::shared_ptr<MockExtensionMessageFactory> exmsgFactory_; std::unique_ptr<DefaultBtMessageFactory> factory_; + public: void setUp() { @@ -49,28 +51,27 @@ public: void testCreatePortMessage(); }; - CPPUNIT_TEST_SUITE_REGISTRATION(DefaultBtMessageFactoryTest); void DefaultBtMessageFactoryTest::testCreateBtMessage_BtExtendedMessage() { // payload:{4:name3:foo}->11bytes std::string payload = "4:name3:foo"; - char msg[17];// 6+11bytes - bittorrent::createPeerMessageString((unsigned char*)msg, sizeof(msg), - 13, 20); + char msg[17]; // 6+11bytes + bittorrent::createPeerMessageString((unsigned char*)msg, sizeof(msg), 13, 20); msg[5] = 1; // Set dummy extended message ID 1 - memcpy(msg+6, payload.c_str(), payload.size()); + memcpy(msg + 6, payload.c_str(), payload.size()); auto m = - factory_->createBtMessage((const unsigned char*)msg+4, sizeof(msg)-4); + factory_->createBtMessage((const unsigned char*)msg + 4, sizeof(msg) - 4); CPPUNIT_ASSERT(BtExtendedMessage::ID == m->getId()); try { // disable extended messaging peer_->setExtendedMessagingEnabled(false); - factory_->createBtMessage((const unsigned char*)msg+4, sizeof(msg)-4); + factory_->createBtMessage((const unsigned char*)msg + 4, sizeof(msg) - 4); CPPUNIT_FAIL("exception must be thrown."); - } catch(Exception& e) { + } + catch (Exception& e) { std::cerr << e.stackTrace() << std::endl; } } @@ -82,11 +83,12 @@ void DefaultBtMessageFactoryTest::testCreatePortMessage() bittorrent::createPeerMessageString(data, sizeof(data), 3, 9); bittorrent::setShortIntParam(&data[5], 6881); try { - auto r = factory_->createBtMessage(&data[4], sizeof(data)-4); + auto r = factory_->createBtMessage(&data[4], sizeof(data) - 4); CPPUNIT_ASSERT(BtPortMessage::ID == r->getId()); auto m = static_cast<const BtPortMessage*>(r.get()); CPPUNIT_ASSERT_EQUAL((uint16_t)6881, m->getPort()); - } catch(Exception& e) { + } + catch (Exception& e) { CPPUNIT_FAIL(e.stackTrace()); } } |
