Changeset View
Changeset View
Standalone View
Standalone View
src/avalanche/test/peermanager_tests.cpp
| Show First 20 Lines • Show All 335 Lines • ▼ Show 20 Lines | BOOST_AUTO_TEST_CASE(remove_peer) { | ||||
| // No peers. | // No peers. | ||||
| avalanche::PeerManager pm(chainman); | avalanche::PeerManager pm(chainman); | ||||
| BOOST_CHECK_EQUAL(pm.selectPeer(), NO_PEER); | BOOST_CHECK_EQUAL(pm.selectPeer(), NO_PEER); | ||||
| CChainState &active_chainstate = chainman.ActiveChainstate(); | CChainState &active_chainstate = chainman.ActiveChainstate(); | ||||
| // Add 4 peers. | // Add 4 peers. | ||||
| std::array<PeerId, 8> peerids; | std::array<PeerId, 8> peerids; | ||||
| for (int i = 0; i < 4; i++) { | for (int i = 0; i < 4; i++) { | ||||
| auto p = buildRandomProof(active_chainstate, 100); | auto p = buildRandomProof(active_chainstate, MIN_VALID_PROOF_SCORE); | ||||
| peerids[i] = TestPeerManager::registerAndGetPeerId(pm, p); | peerids[i] = TestPeerManager::registerAndGetPeerId(pm, p); | ||||
| BOOST_CHECK(pm.addNode(InsecureRand32(), p->getId())); | BOOST_CHECK(pm.addNode(InsecureRand32(), p->getId())); | ||||
| } | } | ||||
| BOOST_CHECK_EQUAL(pm.getSlotCount(), 400); | BOOST_CHECK_EQUAL(pm.getSlotCount(), 400); | ||||
| BOOST_CHECK_EQUAL(pm.getFragmentation(), 0); | BOOST_CHECK_EQUAL(pm.getFragmentation(), 0); | ||||
| for (int i = 0; i < 100; i++) { | for (int i = 0; i < 100; i++) { | ||||
| Show All 15 Lines | BOOST_AUTO_TEST_CASE(remove_peer) { | ||||
| for (int i = 0; i < 100; i++) { | for (int i = 0; i < 100; i++) { | ||||
| PeerId p = pm.selectPeer(); | PeerId p = pm.selectPeer(); | ||||
| BOOST_CHECK(p == peerids[0] || p == peerids[1] || p == peerids[3]); | BOOST_CHECK(p == peerids[0] || p == peerids[1] || p == peerids[3]); | ||||
| } | } | ||||
| // Add 4 more peers. | // Add 4 more peers. | ||||
| for (int i = 0; i < 4; i++) { | for (int i = 0; i < 4; i++) { | ||||
| auto p = buildRandomProof(active_chainstate, 100); | auto p = buildRandomProof(active_chainstate, MIN_VALID_PROOF_SCORE); | ||||
| peerids[i + 4] = TestPeerManager::registerAndGetPeerId(pm, p); | peerids[i + 4] = TestPeerManager::registerAndGetPeerId(pm, p); | ||||
| BOOST_CHECK(pm.addNode(InsecureRand32(), p->getId())); | BOOST_CHECK(pm.addNode(InsecureRand32(), p->getId())); | ||||
| } | } | ||||
| BOOST_CHECK_EQUAL(pm.getSlotCount(), 700); | BOOST_CHECK_EQUAL(pm.getSlotCount(), 700); | ||||
| BOOST_CHECK_EQUAL(pm.getFragmentation(), 0); | BOOST_CHECK_EQUAL(pm.getFragmentation(), 0); | ||||
| BOOST_CHECK(pm.removePeer(peerids[0])); | BOOST_CHECK(pm.removePeer(peerids[0])); | ||||
| Show All 26 Lines | |||||
| BOOST_AUTO_TEST_CASE(compact_slots) { | BOOST_AUTO_TEST_CASE(compact_slots) { | ||||
| ChainstateManager &chainman = *Assert(m_node.chainman); | ChainstateManager &chainman = *Assert(m_node.chainman); | ||||
| avalanche::PeerManager pm(chainman); | avalanche::PeerManager pm(chainman); | ||||
| // Add 4 peers. | // Add 4 peers. | ||||
| std::array<PeerId, 4> peerids; | std::array<PeerId, 4> peerids; | ||||
| for (int i = 0; i < 4; i++) { | for (int i = 0; i < 4; i++) { | ||||
| auto p = buildRandomProof(chainman.ActiveChainstate(), 100); | auto p = buildRandomProof(chainman.ActiveChainstate(), | ||||
| MIN_VALID_PROOF_SCORE); | |||||
| peerids[i] = TestPeerManager::registerAndGetPeerId(pm, p); | peerids[i] = TestPeerManager::registerAndGetPeerId(pm, p); | ||||
| BOOST_CHECK(pm.addNode(InsecureRand32(), p->getId())); | BOOST_CHECK(pm.addNode(InsecureRand32(), p->getId())); | ||||
| } | } | ||||
| // Remove all peers. | // Remove all peers. | ||||
| for (auto p : peerids) { | for (auto p : peerids) { | ||||
| pm.removePeer(p); | pm.removePeer(p); | ||||
| } | } | ||||
| ▲ Show 20 Lines • Show All 1,628 Lines • Show Last 20 Lines | |||||