blob: 9c349fbbc4711c0666bcff817e64d7d080dc6c60 [file] [log] [blame]
Alexander Afanasyevd3a55b22014-11-18 19:23:28 -08001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Spyridon Mastorakis429634f2015-02-19 17:35:33 -08003 * Copyright (c) 2013-2015 Regents of the University of California.
Alexander Afanasyevd3a55b22014-11-18 19:23:28 -08004 *
5 * This file is part of ndn-cxx library (NDN C++ library with eXperimental eXtensions).
6 *
7 * ndn-cxx library is free software: you can redistribute it and/or modify it under the
8 * terms of the GNU Lesser General Public License as published by the Free Software
9 * Foundation, either version 3 of the License, or (at your option) any later version.
10 *
11 * ndn-cxx library is distributed in the hope that it will be useful, but WITHOUT ANY
12 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
13 * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
14 *
15 * You should have received copies of the GNU General Public License and GNU Lesser
16 * General Public License along with ndn-cxx, e.g., in COPYING.md file. If not, see
17 * <http://www.gnu.org/licenses/>.
18 *
19 * See AUTHORS.md for complete list of ndn-cxx authors and contributors.
20 */
21
22#ifndef NDN_TESTS_UNIT_TESTS_UNIT_TEST_TIME_FIXTURE_HPP
23#define NDN_TESTS_UNIT_TESTS_UNIT_TEST_TIME_FIXTURE_HPP
24
25#include "util/time-unit-test-clock.hpp"
26#include <boost/asio.hpp>
27
28namespace ndn {
29namespace tests {
30
31class UnitTestTimeFixture
32{
33public:
34 UnitTestTimeFixture()
35 : steadyClock(make_shared<time::UnitTestSteadyClock>())
36 , systemClock(make_shared<time::UnitTestSystemClock>())
37 {
38 time::setCustomClocks(steadyClock, systemClock);
39 }
40
41 ~UnitTestTimeFixture()
42 {
43 time::setCustomClocks(nullptr, nullptr);
44 }
45
46 void
47 advanceClocks(const time::nanoseconds& tick, size_t nTicks = 1)
48 {
49 for (size_t i = 0; i < nTicks; ++i) {
50 steadyClock->advance(tick);
51 systemClock->advance(tick);
52
53 if (io.stopped())
54 io.reset();
55 io.poll();
56 }
57 }
58
59public:
60 shared_ptr<time::UnitTestSteadyClock> steadyClock;
61 shared_ptr<time::UnitTestSystemClock> systemClock;
62 boost::asio::io_service io;
63};
64
65} // namespace tests
66} // namespace ndn
67
68#endif // NDN_TESTS_UNIT_TESTS_UNIT_TEST_TIME_FIXTURE_HPP