[util] Introduce a SetMockTime that takes chrono time

syscoin · Feb 16, 2021 · 47a7a16 · 47a7a16
1 parent df6a5fc
commit 47a7a16
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/util/time.cpp b/src/util/time.cpp
@@ -53,6 +53,11 @@ void SetMockTime(int64_t nMockTimeIn)
     nMockTime.store(nMockTimeIn, std::memory_order_relaxed);
 }
 
+void SetMockTime(std::chrono::seconds mock_time_in)
+{
+    nMockTime.store(mock_time_in.count(), std::memory_order_relaxed);
+}
+
 std::chrono::seconds GetMockTime()
 {
     return std::chrono::seconds(nMockTime.load(std::memory_order_relaxed));

diff --git a/src/util/time.h b/src/util/time.h
@@ -46,6 +46,9 @@ int64_t GetSystemTimeInSeconds(); // Like GetTime(), but not mockable
 /** For testing. Set e.g. with the setmocktime rpc, or -mocktime argument */
 void SetMockTime(int64_t nMockTimeIn);
 
+/** For testing. Set e.g. with the setmocktime rpc, or -mocktime argument */
+void SetMockTime(std::chrono::seconds mock_time_in);
+
 /** For testing */
 std::chrono::seconds GetMockTime();