time.h

Go to the documentation of this file.

// Copyright (c) The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_TEST_UTIL_TIME_H
#define BITCOIN_TEST_UTIL_TIME_H
 
#include <util/check.h>
#include <util/time.h>

template <class T>
class LimitOne
{
public:
    LimitOne() { Assert(g_T_available) = false; }
    ~LimitOne() { g_T_available = true; }
    LimitOne(const LimitOne&) = delete;
    LimitOne& operator=(const LimitOne&) = delete;
 
private:
    static inline bool g_T_available{true};
};
 

class FakeSteadyClock : public LimitOne<FakeSteadyClock>
{
    MockableSteadyClock::mock_time_point::duration t{MockableSteadyClock::INITIAL_MOCK_TIME};
 
public:
    explicit FakeSteadyClock() { (*this) += 0s; }

    ~FakeSteadyClock() { MockableSteadyClock::ClearMockTime(); }
 
    FakeSteadyClock(const FakeSteadyClock&) = delete;
    FakeSteadyClock& operator=(const FakeSteadyClock&) = delete;

    void operator+=(std::chrono::milliseconds d)
    {
        Assert(d >= 0s); // Steady time can only increase monotonically.
        t += d;
        MockableSteadyClock::SetMockTime(t);
    }
};

class FakeNodeClock : public LimitOne<FakeNodeClock>
{
    NodeSeconds m_t{std::chrono::seconds::max()};
 
public:
    explicit FakeNodeClock(NodeSeconds init_time) { set(init_time); }
    explicit FakeNodeClock(std::chrono::seconds init_time) { set(init_time); }
    explicit FakeNodeClock() { set(Now<NodeSeconds>()); }

    ~FakeNodeClock() { set(0s); }
 
    FakeNodeClock(const FakeNodeClock&) = delete;
    FakeNodeClock& operator=(const FakeNodeClock&) = delete;

    void set(NodeSeconds t) { SetMockTime(m_t = t); }
    void set(std::chrono::seconds t) { set(NodeSeconds{t}); }

    void operator+=(std::chrono::seconds d) { set(m_t += d); }
    void operator-=(std::chrono::seconds d) { set(m_t -= d); }
};
 
#endif // BITCOIN_TEST_UTIL_TIME_H