Differential D10092 Diff 29921 src/avalanche/test/peermanager_tests.cpp

Changeset View

Standalone View

src/avalanche/test/peermanager_tests.cpp

Show First 20 Lines • Show All 452 Lines • ▼ Show 20 Lines	for (int i = 0; i < 10; i++) {
BOOST_CHECK(!TestPeerManager::nodeBelongToPeer(pm, i, peerid));		BOOST_CHECK(!TestPeerManager::nodeBelongToPeer(pm, i, peerid));
}		}
BOOST_CHECK(pm.verify());		BOOST_CHECK(pm.verify());
}		}

BOOST_AUTO_TEST_CASE(node_binding_reorg) {		BOOST_AUTO_TEST_CASE(node_binding_reorg) {
avalanche::PeerManager pm;		avalanche::PeerManager pm;

ProofBuilder pb(0, 0, CPubKey());		ProofBuilder pb(0, 0, CKey::MakeCompressedKey());
auto key = CKey::MakeCompressedKey();		auto key = CKey::MakeCompressedKey();
const CScript script = GetScriptForDestination(PKHash(key.GetPubKey()));		const CScript script = GetScriptForDestination(PKHash(key.GetPubKey()));
COutPoint utxo(TxId(GetRandHash()), 0);		COutPoint utxo(TxId(GetRandHash()), 0);
Amount amount = 1 * COIN;		Amount amount = 1 * COIN;
const int height = 1234;		const int height = 1234;
BOOST_CHECK(pb.addUTXO(utxo, amount, height, false, key));		BOOST_CHECK(pb.addUTXO(utxo, amount, height, false, key));
auto proof = std::make_shared<Proof>(pb.build());		auto proof = std::make_shared<Proof>(pb.build());
const ProofId &proofid = proof->getId();		const ProofId &proofid = proof->getId();
▲ Show 20 Lines • Show All 70 Lines • ▼ Show 20 Lines	const int height = 1234;
coins.AddCoin(COutPoint(txid1, i),		coins.AddCoin(COutPoint(txid1, i),
Coin(CTxOut(v, script), height, false), false);		Coin(CTxOut(v, script), height, false), false);
coins.AddCoin(COutPoint(txid2, i),		coins.AddCoin(COutPoint(txid2, i),
Coin(CTxOut(v, script), height, false), false);		Coin(CTxOut(v, script), height, false), false);
}		}
}		}

avalanche::PeerManager pm;		avalanche::PeerManager pm;
		CKey masterKey = CKey::MakeCompressedKey();
const auto getPeerId = [&](const std::vector<COutPoint> &outpoints) {		const auto getPeerId = [&](const std::vector<COutPoint> &outpoints) {
ProofBuilder pb(0, 0, CPubKey());		ProofBuilder pb(0, 0, masterKey);
for (const auto &o : outpoints) {		for (const auto &o : outpoints) {
BOOST_CHECK(pb.addUTXO(o, v, height, false, key));		BOOST_CHECK(pb.addUTXO(o, v, height, false, key));
}		}

return pm.getPeerId(std::make_shared<Proof>(pb.build()));		return pm.getPeerId(std::make_shared<Proof>(pb.build()));
};		};

// Add one peer.		// Add one peer.
Show All 15 Lines	BOOST_AUTO_TEST_CASE(proof_conflict) {
BOOST_CHECK_EQUAL(getPeerId({}), NO_PEER);		BOOST_CHECK_EQUAL(getPeerId({}), NO_PEER);

// Multiple inputs.		// Multiple inputs.
const PeerId peer4 = getPeerId({COutPoint(txid1, 2), COutPoint(txid2, 2)});		const PeerId peer4 = getPeerId({COutPoint(txid1, 2), COutPoint(txid2, 2)});
BOOST_CHECK(peer4 != NO_PEER && peer4 != peer1);		BOOST_CHECK(peer4 != NO_PEER && peer4 != peer1);

// Duplicated input.		// Duplicated input.
{		{
ProofBuilder pb(0, 0, CPubKey());		ProofBuilder pb(0, 0, CKey::MakeCompressedKey());
COutPoint o(txid1, 3);		COutPoint o(txid1, 3);
BOOST_CHECK(pb.addUTXO(o, v, height, false, key));		BOOST_CHECK(pb.addUTXO(o, v, height, false, key));
PeerId peerid = pm.getPeerId(std::make_shared<Proof>(		PeerId peerid = pm.getPeerId(std::make_shared<Proof>(
TestProofBuilder::buildDuplicatedStakes(pb)));		TestProofBuilder::buildDuplicatedStakes(pb)));
BOOST_CHECK_EQUAL(peerid, NO_PEER);		BOOST_CHECK_EQUAL(peerid, NO_PEER);
}		}

// Multiple inputs, collision on first input.		// Multiple inputs, collision on first input.
Show All 19 Lines	BOOST_AUTO_TEST_CASE(orphan_proofs) {
COutPoint outpoint2 = COutPoint(TxId(GetRandHash()), 0);		COutPoint outpoint2 = COutPoint(TxId(GetRandHash()), 0);
COutPoint outpoint3 = COutPoint(TxId(GetRandHash()), 0);		COutPoint outpoint3 = COutPoint(TxId(GetRandHash()), 0);

const Amount v = 5 * COIN;		const Amount v = 5 * COIN;
const int height = 1234;		const int height = 1234;
const int wrongHeight = 12345;		const int wrongHeight = 12345;

const auto makeProof = [&](const COutPoint &outpoint, const int h) {		const auto makeProof = [&](const COutPoint &outpoint, const int h) {
ProofBuilder pb(0, 0, CPubKey());		ProofBuilder pb(0, 0, CKey::MakeCompressedKey());
BOOST_CHECK(pb.addUTXO(outpoint, v, h, false, key));		BOOST_CHECK(pb.addUTXO(outpoint, v, h, false, key));
return std::make_shared<Proof>(pb.build());		return std::make_shared<Proof>(pb.build());
};		};

auto proof1 = makeProof(outpoint1, height);		auto proof1 = makeProof(outpoint1, height);
auto proof2 = makeProof(outpoint2, height);		auto proof2 = makeProof(outpoint2, height);
auto proof3 = makeProof(outpoint3, wrongHeight);		auto proof3 = makeProof(outpoint3, wrongHeight);

▲ Show 20 Lines • Show All 172 Lines • Show Last 20 Lines