blob: f172cff52c5105680e32dbd430ce9f056d4d452f [file] [log] [blame]
Ashlesh Gawande08784d42017-09-06 23:40:21 -05001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
2/*
3 * Copyright (c) 2012-2017 University of California, Los Angeles
Qiuhan Dingd71a08a2014-12-30 18:56:15 -08004 *
Ashlesh Gawande08784d42017-09-06 23:40:21 -05005 * This file is part of ChronoSync, synchronization library for distributed realtime
6 * applications for NDN.
Qiuhan Dingd71a08a2014-12-30 18:56:15 -08007 *
Ashlesh Gawande08784d42017-09-06 23:40:21 -05008 * ChronoSync is free software: you can redistribute it and/or modify it under the terms
9 * of the GNU General Public License as published by the Free Software Foundation, either
10 * version 3 of the License, or (at your option) any later version.
Qiuhan Dingd71a08a2014-12-30 18:56:15 -080011 *
Ashlesh Gawande08784d42017-09-06 23:40:21 -050012 * ChronoSync is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
13 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
14 * PURPOSE. See the GNU General Public License for more details.
Qiuhan Dingd71a08a2014-12-30 18:56:15 -080015 *
Ashlesh Gawande08784d42017-09-06 23:40:21 -050016 * You should have received a copy of the GNU General Public License along with
17 * ChronoSync, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
Qiuhan Dingd71a08a2014-12-30 18:56:15 -080018 */
19
20#ifndef NDN_TESTS_UNIT_TESTS_UNIT_TEST_TIME_FIXTURE_HPP
21#define NDN_TESTS_UNIT_TESTS_UNIT_TEST_TIME_FIXTURE_HPP
22
23#include <ndn-cxx/util/time-unit-test-clock.hpp>
Ashlesh Gawande08784d42017-09-06 23:40:21 -050024
Qiuhan Dingd71a08a2014-12-30 18:56:15 -080025#include <boost/asio.hpp>
26
27namespace ndn {
28namespace tests {
29
30class UnitTestTimeFixture
31{
32public:
33 UnitTestTimeFixture()
34 : steadyClock(make_shared<time::UnitTestSteadyClock>())
35 , systemClock(make_shared<time::UnitTestSystemClock>())
36 {
37 time::setCustomClocks(steadyClock, systemClock);
38 }
39
40 ~UnitTestTimeFixture()
41 {
42 time::setCustomClocks(nullptr, nullptr);
43 }
44
45 void
46 advanceClocks(const time::nanoseconds& tick, size_t nTicks = 1)
47 {
48 for (size_t i = 0; i < nTicks; ++i) {
49 steadyClock->advance(tick);
50 systemClock->advance(tick);
51
52 if (io.stopped())
53 io.reset();
54 io.poll();
55 }
56 }
57
58public:
59 shared_ptr<time::UnitTestSteadyClock> steadyClock;
60 shared_ptr<time::UnitTestSystemClock> systemClock;
61 boost::asio::io_service io;
62};
63
64} // namespace tests
65} // namespace ndn
66
67#endif // NDN_TESTS_UNIT_TESTS_UNIT_TEST_TIME_FIXTURE_HPP