mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-03-15 04:11:02 +01:00
c61d3f02f5
The wallet tests and benchmarks both had helper functions for loading and unloading the wallet for the test that were almost identical. These functions are consolidated and reused.
74 lines
2.3 KiB
C++
74 lines
2.3 KiB
C++
// Copyright (c) 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 <bench/bench.h>
|
|
#include <interfaces/chain.h>
|
|
#include <node/context.h>
|
|
#include <test/util/mining.h>
|
|
#include <test/util/setup_common.h>
|
|
#include <wallet/test/util.h>
|
|
#include <util/translation.h>
|
|
#include <validationinterface.h>
|
|
#include <wallet/context.h>
|
|
#include <wallet/receive.h>
|
|
#include <wallet/wallet.h>
|
|
|
|
#include <optional>
|
|
|
|
namespace wallet{
|
|
static void AddTx(CWallet& wallet)
|
|
{
|
|
CMutableTransaction mtx;
|
|
mtx.vout.push_back({COIN, GetScriptForDestination(*Assert(wallet.GetNewDestination(OutputType::BECH32, "")))});
|
|
mtx.vin.push_back(CTxIn());
|
|
|
|
wallet.AddToWallet(MakeTransactionRef(mtx), TxStateInactive{});
|
|
}
|
|
|
|
static void WalletLoading(benchmark::Bench& bench, bool legacy_wallet)
|
|
{
|
|
const auto test_setup = MakeNoLogFileContext<TestingSetup>();
|
|
|
|
WalletContext context;
|
|
context.args = &test_setup->m_args;
|
|
context.chain = test_setup->m_node.chain.get();
|
|
|
|
// Setup the wallet
|
|
// Loading the wallet will also create it
|
|
uint64_t create_flags = 0;
|
|
if (!legacy_wallet) {
|
|
create_flags = WALLET_FLAG_DESCRIPTORS;
|
|
}
|
|
auto database = CreateMockableWalletDatabase();
|
|
auto wallet = TestLoadWallet(std::move(database), context, create_flags);
|
|
|
|
// Generate a bunch of transactions and addresses to put into the wallet
|
|
for (int i = 0; i < 1000; ++i) {
|
|
AddTx(*wallet);
|
|
}
|
|
|
|
database = DuplicateMockDatabase(wallet->GetDatabase());
|
|
|
|
// reload the wallet for the actual benchmark
|
|
TestUnloadWallet(std::move(wallet));
|
|
|
|
bench.epochs(5).run([&] {
|
|
wallet = TestLoadWallet(std::move(database), context, create_flags);
|
|
|
|
// Cleanup
|
|
database = DuplicateMockDatabase(wallet->GetDatabase());
|
|
TestUnloadWallet(std::move(wallet));
|
|
});
|
|
}
|
|
|
|
#ifdef USE_BDB
|
|
static void WalletLoadingLegacy(benchmark::Bench& bench) { WalletLoading(bench, /*legacy_wallet=*/true); }
|
|
BENCHMARK(WalletLoadingLegacy, benchmark::PriorityLevel::HIGH);
|
|
#endif
|
|
|
|
#ifdef USE_SQLITE
|
|
static void WalletLoadingDescriptors(benchmark::Bench& bench) { WalletLoading(bench, /*legacy_wallet=*/false); }
|
|
BENCHMARK(WalletLoadingDescriptors, benchmark::PriorityLevel::HIGH);
|
|
#endif
|
|
} // namespace wallet
|