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