Page MenuHomePhabricator
Differential D9768 Diff 29154 src/avalanche/test/processor_tests.cpp

Changeset View

Standalone View

View Options

src/avalanche/test/processor_tests.cpp

Show First 20 Lines • Show All 127 Lines • ▼ Show 20 Linesstruct AvalancheTestingSetup : public TestChain100Setup {
bool addNode(NodeId nodeid, const ProofId &proofid) { bool addNode(NodeId nodeid, const ProofId &proofid) {
return m_processor->withPeerManager([&](avalanche::PeerManager &pm) { return m_processor->withPeerManager([&](avalanche::PeerManager &pm) {
return pm.addNode(nodeid, proofid); return pm.addNode(nodeid, proofid);
}); });
} }
bool addNode(NodeId nodeid) { bool addNode(NodeId nodeid) {
auto proof = GetProof(); auto proof = GetProof();
BOOST_CHECK(m_processor->addProof(proof)); return m_processor->withPeerManager([&](avalanche::PeerManager &pm) {
return addNode(nodeid, proof->getId()); return pm.registerProof(proof) &&
pm.addNode(nodeid, proof->getId());
});
} }
std::array<CNode *, 8> ConnectNodes() { std::array<CNode *, 8> ConnectNodes() {
auto proof = GetProof(); auto proof = GetProof();
BOOST_CHECK(m_processor->addProof(proof)); BOOST_CHECK(
m_processor->withPeerManager([&](avalanche::PeerManager &pm) {
return pm.registerProof(proof);
}));
const ProofId &proofid = proof->getId(); const ProofId &proofid = proof->getId();
std::array<CNode *, 8> nodes; std::array<CNode *, 8> nodes;
for (CNode *&n : nodes) { for (CNode *&n : nodes) {
n = ConnectNode(NODE_AVALANCHE); n = ConnectNode(NODE_AVALANCHE);
BOOST_CHECK(addNode(n->GetId(), proofid)); BOOST_CHECK(addNode(n->GetId(), proofid));
} }
▲ Show 20 Lines • Show All 588 Lines • ▼ Show 20 Lines for (int i = 0; i < 10; i++) {
BOOST_CHECK( BOOST_CHECK(
!m_processor->registerVotes(avanodeid, next(resp), updates)); !m_processor->registerVotes(avanodeid, next(resp), updates));
} }
} }
BOOST_AUTO_TEST_CASE(poll_inflight_count) { BOOST_AUTO_TEST_CASE(poll_inflight_count) {
// Create enough nodes so that we run into the inflight request limit. // Create enough nodes so that we run into the inflight request limit.
auto proof = GetProof(); auto proof = GetProof();
BOOST_CHECK(m_processor->addProof(proof)); BOOST_CHECK(m_processor->withPeerManager(
[&](avalanche::PeerManager &pm) { return pm.registerProof(proof); }));
std::array<CNode *, AVALANCHE_MAX_INFLIGHT_POLL + 1> nodes; std::array<CNode *, AVALANCHE_MAX_INFLIGHT_POLL + 1> nodes;
for (auto &n : nodes) { for (auto &n : nodes) {
n = ConnectNode(NODE_AVALANCHE); n = ConnectNode(NODE_AVALANCHE);
BOOST_CHECK(addNode(n->GetId(), proof->getId())); BOOST_CHECK(addNode(n->GetId(), proof->getId()));
} }
// Add a block to poll // Add a block to poll
▲ Show 20 Lines • Show All 218 Lines • Show Last 20 Lines