danicoin/tests/unit_tests/unit_tests_utils.h
2015-04-23 20:07:22 +04:00

38 lines
737 B
C++

// Copyright (c) 2011-2015 The Cryptonote developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
#pragma once
#include <atomic>
namespace unit_test
{
class call_counter
{
public:
call_counter()
: m_counter(0)
{
}
void inc() volatile
{
// memory_order_relaxed is enough for call counter
m_counter.fetch_add(1, std::memory_order_relaxed);
}
size_t get() volatile const
{
return m_counter.load(std::memory_order_relaxed);
}
void reset() volatile
{
m_counter.store(0, std::memory_order_relaxed);
}
private:
std::atomic<size_t> m_counter;
};
}