From 230a2f4cc3fab9f66b6c24ba809ddbea77755cb7 Mon Sep 17 00:00:00 2001
From: Ryan Ofsky <ryan@ofsky.org>
Date: Wed, 8 Jun 2022 23:05:16 -0300
Subject: wallet test: Add unit test for wallet scan save_progress option

---
 src/wallet/wallet.h | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/wallet/wallet.h')

diff --git a/src/wallet/wallet.h b/src/wallet/wallet.h
index 206a4417a7..5be2ad166a 100644
--- a/src/wallet/wallet.h
+++ b/src/wallet/wallet.h
@@ -896,8 +896,11 @@ void MaybeResendWalletTxs(WalletContext& context);
 class WalletRescanReserver
 {
 private:
+    using Clock = std::chrono::steady_clock;
+    using NowFn = std::function<Clock::time_point()>;
     CWallet& m_wallet;
     bool m_could_reserve;
+    NowFn m_now;
 public:
     explicit WalletRescanReserver(CWallet& w) : m_wallet(w), m_could_reserve(false) {}
 
@@ -918,6 +921,10 @@ public:
         return (m_could_reserve && m_wallet.fScanningWallet);
     }
 
+    Clock::time_point now() const { return m_now ? m_now() : Clock::now(); };
+
+    void setNow(NowFn now) { m_now = std::move(now); }
+
     ~WalletRescanReserver()
     {
         if (m_could_reserve) {
-- 
cgit v1.2.3