highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2026-01-20 23:29:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

fuzz: create FeeEstimatorTestingSetup to set fee_estimator

Browse Source
This commit is contained in:
brunoerg
2025-08-15 10:13:47 -03:00
parent ff10a37e99
commit adf67eb21b
1 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/wallet/test/fuzz/fees.cpp
View File

@@ -14,7 +14,23 @@
namespace wallet {
namespace {
TestingSetup* g_setup;
struct FeeEstimatorTestingSetup : public TestingSetup {
FeeEstimatorTestingSetup(const ChainType chain_type, TestOpts opts) : TestingSetup{chain_type, opts}
{
}
~FeeEstimatorTestingSetup() {
m_node.fee_estimator.reset();
}
void SetFeeEstimator(std::unique_ptr<CBlockPolicyEstimator> fee_estimator)
{
m_node.fee_estimator = std::move(fee_estimator);
}
};
FeeEstimatorTestingSetup* g_setup;
class FuzzedBlockPolicyEstimator : public CBlockPolicyEstimator
{
@@ -37,7 +53,7 @@ public:
void initialize_setup()
{
static const auto testing_setup = MakeNoLogFileContext<TestingSetup>();
static const auto testing_setup = MakeNoLogFileContext<FeeEstimatorTestingSetup>();
g_setup = testing_setup.get();
}
@@ -57,7 +73,7 @@ FUZZ_TARGET(wallet_fees, .init = initialize_setup)
};
node.mempool = std::make_unique<CTxMemPool>(mempool_opts, error);
std::unique_ptr<CBlockPolicyEstimator> fee_estimator = std::make_unique<FuzzedBlockPolicyEstimator>(fuzzed_data_provider);
node.fee_estimator = std::move(fee_estimator);
g_setup->SetFeeEstimator(std::move(fee_estimator));
std::unique_ptr<CWallet> wallet_ptr{std::make_unique<CWallet>(node.chain.get(), "", CreateMockableWalletDatabase())};
CWallet& wallet{*wallet_ptr};
{