diff options
Diffstat (limited to 'test/DHTRoutingTableTest.cc')
| -rw-r--r-- | test/DHTRoutingTableTest.cc | 25 |
1 files changed, 13 insertions, 12 deletions
diff --git a/test/DHTRoutingTableTest.cc b/test/DHTRoutingTableTest.cc index 61bc66ca..5a20657b 100644 --- a/test/DHTRoutingTableTest.cc +++ b/test/DHTRoutingTableTest.cc @@ -13,13 +13,14 @@ namespace aria2 { -class DHTRoutingTableTest:public CppUnit::TestFixture { +class DHTRoutingTableTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(DHTRoutingTableTest); CPPUNIT_TEST(testAddNode); CPPUNIT_TEST(testAddNode_localNode); CPPUNIT_TEST(testGetClosestKNodes); CPPUNIT_TEST_SUITE_END(); + public: void setUp() {} @@ -30,7 +31,6 @@ public: void testGetClosestKNodes(); }; - CPPUNIT_TEST_SUITE_REGISTRATION(DHTRoutingTableTest); void DHTRoutingTableTest::testAddNode() @@ -42,8 +42,8 @@ void DHTRoutingTableTest::testAddNode() auto taskQueue = make_unique<MockDHTTaskQueue>(); table.setTaskQueue(taskQueue.get()); uint32_t count = 0; - for(int i = 0; i < 100; ++i) { - if(table.addNode(std::make_shared<DHTNode>())) { + for (int i = 0; i < 100; ++i) { + if (table.addNode(std::make_shared<DHTNode>())) { ++count; } } @@ -64,11 +64,12 @@ void DHTRoutingTableTest::testAddNode_localNode() } namespace { -void createID(unsigned char* id, unsigned char firstChar, unsigned char lastChar) +void createID(unsigned char* id, unsigned char firstChar, + unsigned char lastChar) { memset(id, 0, DHT_ID_LENGTH); id[0] = firstChar; - id[DHT_ID_LENGTH-1] = lastChar; + id[DHT_ID_LENGTH - 1] = lastChar; } } // namespace @@ -83,17 +84,17 @@ void DHTRoutingTableTest::testGetClosestKNodes() std::shared_ptr<DHTNode> nodes1[8]; std::shared_ptr<DHTNode> nodes2[8]; std::shared_ptr<DHTNode> nodes3[8]; - for(size_t i = 0; i < DHTBucket::K; ++i) { + for (size_t i = 0; i < DHTBucket::K; ++i) { createID(id, 0xf0, i); nodes1[i] = std::make_shared<DHTNode>(id); CPPUNIT_ASSERT(table.addNode(nodes1[i])); } - for(size_t i = 0; i < DHTBucket::K; ++i) { + for (size_t i = 0; i < DHTBucket::K; ++i) { createID(id, 0x80, i); nodes2[i] = std::make_shared<DHTNode>(id); CPPUNIT_ASSERT(table.addNode(nodes2[i])); } - for(size_t i = 0; i < DHTBucket::K; ++i) { + for (size_t i = 0; i < DHTBucket::K; ++i) { createID(id, 0x70, i); nodes3[i] = std::make_shared<DHTNode>(id); CPPUNIT_ASSERT(table.addNode(nodes3[i])); @@ -103,9 +104,9 @@ void DHTRoutingTableTest::testGetClosestKNodes() std::vector<std::shared_ptr<DHTNode>> nodes; table.getClosestKNodes(nodes, id); CPPUNIT_ASSERT_EQUAL((size_t)8, nodes.size()); - for(size_t i = 0; i < nodes.size(); ++i) { - CPPUNIT_ASSERT(memcmp(nodes2[0]->getID(), nodes[0]->getID(), - DHT_ID_LENGTH) == 0); + for (size_t i = 0; i < nodes.size(); ++i) { + CPPUNIT_ASSERT( + memcmp(nodes2[0]->getID(), nodes[0]->getID(), DHT_ID_LENGTH) == 0); } } } |
