[fuzz] test new orphanage methods

glozow · Sep 13, 2023 · 2a8158e · 2a8158e
1 parent c9354f1
commit 2a8158e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/test/fuzz/txorphan.cpp b/src/test/fuzz/txorphan.cpp
@@ -101,9 +101,14 @@ FUZZ_TARGET(txorphan, .init = initialize_orphanage)
                     // AddTx should return false if tx is too big or already have it
                     // tx weight is unknown, we only check when tx is already in orphanage
                     {
+                        auto bytes_before = orphanage.BytesFromPeer(peer_id);
                         bool add_tx = orphanage.AddTx(tx, peer_id, {});
                         // have_tx == true -> add_tx == false
                         Assert(!have_tx || !add_tx);
+                        if (!have_tx && add_tx) {
+                            Assert(orphanage.BytesFromPeer(peer_id) == bytes_before + tx->GetTotalSize());
+                            Assert(orphanage.TotalOrphanBytes() == bytes_before + orphanage.TotalOrphanBytes());
+                        }
                     }
                     have_tx = orphanage.HaveTx(GenTxid::Txid(tx->GetHash())) || orphanage.HaveTx(GenTxid::Wtxid(tx->GetHash()));
                     {
@@ -127,6 +132,7 @@ FUZZ_TARGET(txorphan, .init = initialize_orphanage)
                 },
                 [&] {
                     orphanage.EraseForPeer(peer_id);
+                    Assert(orphanage.BytesFromPeer(peer_id) == 0);
                 },
                 [&] {
                     // test mocktime and expiry