SetMockTime() for cleaner unit testing

This commit is contained in:
Gavin Andresen 2011-09-15 08:55:15 -04:00
parent 700f942692
commit 54d02f158d
2 changed files with 10 additions and 0 deletions

View File

@ -815,11 +815,20 @@ void ShrinkDebugFile()
// - Median of other nodes's clocks // - Median of other nodes's clocks
// - The user (asking the user to fix the system clock if the first two disagree) // - The user (asking the user to fix the system clock if the first two disagree)
// //
static int64 nMockTime = 0; // For unit testing
int64 GetTime() int64 GetTime()
{ {
if (nMockTime) return nMockTime;
return time(NULL); return time(NULL);
} }
void SetMockTime(int64 nMockTimeIn)
{
nMockTime = nMockTimeIn;
}
static int64 nTimeOffset = 0; static int64 nTimeOffset = 0;
int64 GetAdjustedTime() int64 GetAdjustedTime()

View File

@ -199,6 +199,7 @@ void ShrinkDebugFile();
int GetRandInt(int nMax); int GetRandInt(int nMax);
uint64 GetRand(uint64 nMax); uint64 GetRand(uint64 nMax);
int64 GetTime(); int64 GetTime();
void SetMockTime(int64 nMockTimeIn);
int64 GetAdjustedTime(); int64 GetAdjustedTime();
void AddTimeData(unsigned int ip, int64 nTime); void AddTimeData(unsigned int ip, int64 nTime);
std::string FormatFullVersion(); std::string FormatFullVersion();