Page MenuHomePhabricator
Differential D6824 Diff 22017 src/avalanche/test/peermanager_tests.cpp

Changeset View

Standalone View

View Options

src/avalanche/test/peermanager_tests.cpp

Show First 20 Lines • Show All 344 Lines • ▼ Show 20 LinesBOOST_AUTO_TEST_CASE(compact_slots) {
} }
BOOST_CHECK_EQUAL(pm.compact(), 300); BOOST_CHECK_EQUAL(pm.compact(), 300);
BOOST_CHECK(pm.verify()); BOOST_CHECK(pm.verify());
BOOST_CHECK_EQUAL(pm.getSlotCount(), 0); BOOST_CHECK_EQUAL(pm.getSlotCount(), 0);
BOOST_CHECK_EQUAL(pm.getFragmentation(), 0); BOOST_CHECK_EQUAL(pm.getFragmentation(), 0);
} }
BOOST_AUTO_TEST_CASE(add_and_get_node) { BOOST_AUTO_TEST_CASE(node_crud) {
PeerManager pm; PeerManager pm;
// Create one peer. // Create one peer.
PeerId peerid = pm.addPeer(100); PeerId peerid = pm.addPeer(100);
BOOST_CHECK_EQUAL(pm.getSuitableNodeToQuery(), NO_NODE); BOOST_CHECK_EQUAL(pm.getSuitableNodeToQuery(), NO_NODE);
// Add 4 nodes. // Add 4 nodes.
for (int i = 0; i < 4; i++) { for (int i = 0; i < 4; i++) {
BOOST_CHECK(pm.addNodeToPeer(peerid, i, CPubKey())); BOOST_CHECK(pm.addNodeToPeer(peerid, i, CPubKey()));
} }
for (int i = 0; i < 100; i++) { for (int i = 0; i < 100; i++) {
NodeId n = pm.getSuitableNodeToQuery(); NodeId n = pm.getSuitableNodeToQuery();
BOOST_CHECK((n >= 0 && n < 4) || n == NO_PEER); BOOST_CHECK(n >= 0 && n < 4);
}
// Remove a node, check that it doesn't show up.
BOOST_CHECK(pm.removeNode(2));
for (int i = 0; i < 100; i++) {
NodeId n = pm.getSuitableNodeToQuery();
BOOST_CHECK(n == 0 || n == 1 || n == 3);
} }
} }
BOOST_AUTO_TEST_SUITE_END() BOOST_AUTO_TEST_SUITE_END()