mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-02-04 10:07:27 -05:00
306ccd4927
-BEGIN VERIFY SCRIPT- ./contrib/devtools/copyright_header.py update ./ -END VERIFY SCRIPT- Commits of previous years: - 2021:f47dda2c58- 2020:fa0074e2d8- 2019:aaaaad6ac9
84 lines
3.3 KiB
C++
84 lines
3.3 KiB
C++
// Copyright (c) 2021-2022 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include <node/txreconciliation.h>
|
|
|
|
#include <test/util/setup_common.h>
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
BOOST_FIXTURE_TEST_SUITE(txreconciliation_tests, BasicTestingSetup)
|
|
|
|
BOOST_AUTO_TEST_CASE(RegisterPeerTest)
|
|
{
|
|
TxReconciliationTracker tracker(TXRECONCILIATION_VERSION);
|
|
const uint64_t salt = 0;
|
|
|
|
// Prepare a peer for reconciliation.
|
|
tracker.PreRegisterPeer(0);
|
|
|
|
// Invalid version.
|
|
BOOST_CHECK_EQUAL(tracker.RegisterPeer(/*peer_id=*/0, /*is_peer_inbound=*/true,
|
|
/*peer_recon_version=*/0, salt),
|
|
ReconciliationRegisterResult::PROTOCOL_VIOLATION);
|
|
|
|
// Valid registration (inbound and outbound peers).
|
|
BOOST_REQUIRE(!tracker.IsPeerRegistered(0));
|
|
BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(0, true, 1, salt), ReconciliationRegisterResult::SUCCESS);
|
|
BOOST_CHECK(tracker.IsPeerRegistered(0));
|
|
BOOST_REQUIRE(!tracker.IsPeerRegistered(1));
|
|
tracker.PreRegisterPeer(1);
|
|
BOOST_REQUIRE(tracker.RegisterPeer(1, false, 1, salt) == ReconciliationRegisterResult::SUCCESS);
|
|
BOOST_CHECK(tracker.IsPeerRegistered(1));
|
|
|
|
// Reconciliation version is higher than ours, should be able to register.
|
|
BOOST_REQUIRE(!tracker.IsPeerRegistered(2));
|
|
tracker.PreRegisterPeer(2);
|
|
BOOST_REQUIRE(tracker.RegisterPeer(2, true, 2, salt) == ReconciliationRegisterResult::SUCCESS);
|
|
BOOST_CHECK(tracker.IsPeerRegistered(2));
|
|
|
|
// Try registering for the second time.
|
|
BOOST_REQUIRE(tracker.RegisterPeer(1, false, 1, salt) == ReconciliationRegisterResult::ALREADY_REGISTERED);
|
|
|
|
// Do not register if there were no pre-registration for the peer.
|
|
BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(100, true, 1, salt), ReconciliationRegisterResult::NOT_FOUND);
|
|
BOOST_CHECK(!tracker.IsPeerRegistered(100));
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(ForgetPeerTest)
|
|
{
|
|
TxReconciliationTracker tracker(TXRECONCILIATION_VERSION);
|
|
NodeId peer_id0 = 0;
|
|
|
|
// Removing peer after pre-registring works and does not let to register the peer.
|
|
tracker.PreRegisterPeer(peer_id0);
|
|
tracker.ForgetPeer(peer_id0);
|
|
BOOST_CHECK_EQUAL(tracker.RegisterPeer(peer_id0, true, 1, 1), ReconciliationRegisterResult::NOT_FOUND);
|
|
|
|
// Removing peer after it is registered works.
|
|
tracker.PreRegisterPeer(peer_id0);
|
|
BOOST_REQUIRE(!tracker.IsPeerRegistered(peer_id0));
|
|
BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(peer_id0, true, 1, 1), ReconciliationRegisterResult::SUCCESS);
|
|
BOOST_CHECK(tracker.IsPeerRegistered(peer_id0));
|
|
tracker.ForgetPeer(peer_id0);
|
|
BOOST_CHECK(!tracker.IsPeerRegistered(peer_id0));
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(IsPeerRegisteredTest)
|
|
{
|
|
TxReconciliationTracker tracker(TXRECONCILIATION_VERSION);
|
|
NodeId peer_id0 = 0;
|
|
|
|
BOOST_REQUIRE(!tracker.IsPeerRegistered(peer_id0));
|
|
tracker.PreRegisterPeer(peer_id0);
|
|
BOOST_REQUIRE(!tracker.IsPeerRegistered(peer_id0));
|
|
|
|
BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(peer_id0, true, 1, 1), ReconciliationRegisterResult::SUCCESS);
|
|
BOOST_CHECK(tracker.IsPeerRegistered(peer_id0));
|
|
|
|
tracker.ForgetPeer(peer_id0);
|
|
BOOST_CHECK(!tracker.IsPeerRegistered(peer_id0));
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|