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>


class SteadyClockContext
{
    MockableSteadyClock::mock_time_point::duration t{MockableSteadyClock::INITIAL_MOCK_TIME};

public:
    explicit SteadyClockContext() { (*this) += 0s; }

    ~SteadyClockContext() { MockableSteadyClock::ClearMockTime(); }

    SteadyClockContext(const SteadyClockContext&) = delete;
    SteadyClockContext& operator=(const SteadyClockContext&) = delete;

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

#endif // BITCOIN_TEST_UTIL_TIME_H