falk-werner_webfuse/test/webfuse/test_util/timeout_watcher.cc

#include "webfuse/test_util/timeout_watcher.hpp"
#include <stdexcept>
#include <thread>

using std::chrono::milliseconds;
using std::chrono::duration_cast;
using std::chrono::steady_clock;

namespace
{
    milliseconds now()
    {
        return duration_cast<milliseconds>(steady_clock::now().time_since_epoch());
    }
}

namespace webfuse_test
{

TimeoutWatcher::TimeoutWatcher(milliseconds timeout)
: startedAt(now())
, timeout_(timeout)
{

}

TimeoutWatcher::~TimeoutWatcher()
{

}

bool TimeoutWatcher::isTimeout()
{
    return (now() - startedAt) > timeout_;
}

void TimeoutWatcher::check()
{
    if (isTimeout())
    {
        throw std::runtime_error("timeout");
    }
}

bool TimeoutWatcher::waitUntil(std::function<bool()> predicate)
{
    bool result = predicate();
    while ((!result) && (!isTimeout()))
    {
        std::this_thread::yield();
        result = predicate();
    }

    return result;
}


}