[tools] add PoissonNextSend method that returns mockable time

1 files changed, 6 insertions, 4 deletions

diff --git a/src/net.h b/src/net.h
index 44655abf80..ab69e8b82b 100644
--- a/src/net.h
+++ b/src/net.h
@@ -943,11 +943,13 @@ public:
     void MaybeSetAddrName(const std::string& addrNameIn);
 };
 
-
-
-
-
 /** Return a timestamp in the future (in microseconds) for exponentially distributed events. */
 int64_t PoissonNextSend(int64_t now, int average_interval_seconds);
 
+/** Wrapper to return mockable type */
+inline std::chrono::microseconds PoissonNextSend(std::chrono::microseconds now, std::chrono::seconds average_interval)
+{
+    return std::chrono::microseconds{PoissonNextSend(now.count(), average_interval.count())};
+}
+
 #endif // BITCOIN_NET_H