SetMockTime() for cleaner unit testing

2 files changed, 10 insertions, 0 deletions

diff --git a/src/util.cpp b/src/util.cpp
index 76a2700271..14ca87b62e 100644
--- a/src/util.cpp
+++ b/src/util.cpp
@@ -815,11 +815,20 @@ void ShrinkDebugFile()
 //  - Median of other nodes's clocks
 //  - The user (asking the user to fix the system clock if the first two disagree)
 //
+static int64 nMockTime = 0;  // For unit testing
+
 int64 GetTime()
 {
+    if (nMockTime) return nMockTime;
+
     return time(NULL);
 }
 
+void SetMockTime(int64 nMockTimeIn)
+{
+    nMockTime = nMockTimeIn;
+}
+
 static int64 nTimeOffset = 0;
 
 int64 GetAdjustedTime()
diff --git a/src/util.h b/src/util.h
index 3d7ef108b4..658d10e1eb 100644
--- a/src/util.h
+++ b/src/util.h
@@ -199,6 +199,7 @@ void ShrinkDebugFile();
 int GetRandInt(int nMax);
 uint64 GetRand(uint64 nMax);
 int64 GetTime();
+void SetMockTime(int64 nMockTimeIn);
 int64 GetAdjustedTime();
 void AddTimeData(unsigned int ip, int64 nTime);
 std::string FormatFullVersion();