Branch data     Line data    Source code

       1                 :            : // Copyright (c) 2021-2022 The Bitcoin Core developers
       2                 :            : // Distributed under the MIT software license, see the accompanying
       3                 :            : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4                 :            : 
       5                 :            : #include <node/txreconciliation.h>
       6                 :            : 
       7                 :            : #include <test/util/setup_common.h>
       8                 :            : 
       9                 :            : #include <boost/test/unit_test.hpp>
      10                 :            : 
      11                 :          0 : BOOST_FIXTURE_TEST_SUITE(txreconciliation_tests, BasicTestingSetup)
      12                 :            : 
      13                 :          0 : BOOST_AUTO_TEST_CASE(RegisterPeerTest)
      14                 :            : {
      15                 :          0 :     TxReconciliationTracker tracker(TXRECONCILIATION_VERSION);
      16                 :          0 :     const uint64_t salt = 0;
      17                 :          0 : 
      18                 :          0 :     // Prepare a peer for reconciliation.
      19                 :          0 :     tracker.PreRegisterPeer(0);
      20                 :            : 
      21                 :            :     // Invalid version.
      22                 :          0 :     BOOST_CHECK_EQUAL(tracker.RegisterPeer(/*peer_id=*/0, /*is_peer_inbound=*/true,
      23                 :            :                                            /*peer_recon_version=*/0, salt),
      24                 :            :                       ReconciliationRegisterResult::PROTOCOL_VIOLATION);
      25                 :            : 
      26                 :            :     // Valid registration (inbound and outbound peers).
      27                 :          0 :     BOOST_REQUIRE(!tracker.IsPeerRegistered(0));
      28                 :          0 :     BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(0, true, 1, salt), ReconciliationRegisterResult::SUCCESS);
      29                 :          0 :     BOOST_CHECK(tracker.IsPeerRegistered(0));
      30                 :          0 :     BOOST_REQUIRE(!tracker.IsPeerRegistered(1));
      31                 :          0 :     tracker.PreRegisterPeer(1);
      32                 :          0 :     BOOST_REQUIRE(tracker.RegisterPeer(1, false, 1, salt) == ReconciliationRegisterResult::SUCCESS);
      33                 :          0 :     BOOST_CHECK(tracker.IsPeerRegistered(1));
      34                 :            : 
      35                 :            :     // Reconciliation version is higher than ours, should be able to register.
      36                 :          0 :     BOOST_REQUIRE(!tracker.IsPeerRegistered(2));
      37                 :          0 :     tracker.PreRegisterPeer(2);
      38                 :          0 :     BOOST_REQUIRE(tracker.RegisterPeer(2, true, 2, salt) == ReconciliationRegisterResult::SUCCESS);
      39                 :          0 :     BOOST_CHECK(tracker.IsPeerRegistered(2));
      40                 :            : 
      41                 :            :     // Try registering for the second time.
      42                 :          0 :     BOOST_REQUIRE(tracker.RegisterPeer(1, false, 1, salt) == ReconciliationRegisterResult::ALREADY_REGISTERED);
      43                 :            : 
      44                 :            :     // Do not register if there were no pre-registration for the peer.
      45                 :          0 :     BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(100, true, 1, salt), ReconciliationRegisterResult::NOT_FOUND);
      46                 :          0 :     BOOST_CHECK(!tracker.IsPeerRegistered(100));
      47                 :          0 : }
      48                 :            : 
      49                 :          0 : BOOST_AUTO_TEST_CASE(ForgetPeerTest)
      50                 :            : {
      51                 :          0 :     TxReconciliationTracker tracker(TXRECONCILIATION_VERSION);
      52                 :          0 :     NodeId peer_id0 = 0;
      53                 :            : 
      54                 :            :     // Removing peer after pre-registring works and does not let to register the peer.
      55                 :          0 :     tracker.PreRegisterPeer(peer_id0);
      56                 :          0 :     tracker.ForgetPeer(peer_id0);
      57                 :          0 :     BOOST_CHECK_EQUAL(tracker.RegisterPeer(peer_id0, true, 1, 1), ReconciliationRegisterResult::NOT_FOUND);
      58                 :            : 
      59                 :            :     // Removing peer after it is registered works.
      60                 :          0 :     tracker.PreRegisterPeer(peer_id0);
      61                 :          0 :     BOOST_REQUIRE(!tracker.IsPeerRegistered(peer_id0));
      62                 :          0 :     BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(peer_id0, true, 1, 1), ReconciliationRegisterResult::SUCCESS);
      63                 :          0 :     BOOST_CHECK(tracker.IsPeerRegistered(peer_id0));
      64                 :          0 :     tracker.ForgetPeer(peer_id0);
      65                 :          0 :     BOOST_CHECK(!tracker.IsPeerRegistered(peer_id0));
      66                 :          0 : }
      67                 :            : 
      68                 :          0 : BOOST_AUTO_TEST_CASE(IsPeerRegisteredTest)
      69                 :            : {
      70                 :          0 :     TxReconciliationTracker tracker(TXRECONCILIATION_VERSION);
      71                 :          0 :     NodeId peer_id0 = 0;
      72                 :            : 
      73                 :          0 :     BOOST_REQUIRE(!tracker.IsPeerRegistered(peer_id0));
      74                 :          0 :     tracker.PreRegisterPeer(peer_id0);
      75                 :          0 :     BOOST_REQUIRE(!tracker.IsPeerRegistered(peer_id0));
      76                 :            : 
      77                 :          0 :     BOOST_REQUIRE_EQUAL(tracker.RegisterPeer(peer_id0, true, 1, 1), ReconciliationRegisterResult::SUCCESS);
      78                 :          0 :     BOOST_CHECK(tracker.IsPeerRegistered(peer_id0));
      79                 :            : 
      80                 :          0 :     tracker.ForgetPeer(peer_id0);
      81                 :          0 :     BOOST_CHECK(!tracker.IsPeerRegistered(peer_id0));
      82                 :          0 : }
      83                 :          0 : 
      84                 :          0 : BOOST_AUTO_TEST_SUITE_END()