A Discrete-Event Network Simulator
API
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
watchdog-test-suite.cc
1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3  * Copyright (c) 2007 INRIA
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation;
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17  *
18  * Author: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
19  */
20 #include "ns3/watchdog.h"
21 #include "ns3/test.h"
22 
23 using namespace ns3;
24 
25 class WatchdogTestCase : public TestCase
26 {
27 public:
29  virtual void DoRun (void);
30  void Expire (Time expected);
31  bool m_expired;
32  Time m_expiredTime;
33  Time m_expiredArgument;
34 };
35 
36 WatchdogTestCase::WatchdogTestCase()
37  : TestCase ("Check that we can keepalive a watchdog")
38 {
39 }
40 
41 void
42 WatchdogTestCase::Expire (Time expected)
43 {
44  m_expired = true;
45  m_expiredTime = Simulator::Now ();
46  m_expiredArgument = expected;
47 }
48 
49 void
51 {
52  m_expired = false;
53  m_expiredArgument = Seconds (0);
54  m_expiredTime = Seconds (0);
55  Watchdog watchdog;
56  watchdog.SetFunction (&WatchdogTestCase::Expire, this);
57  watchdog.SetArguments (MicroSeconds (40));
58  watchdog.Ping (MicroSeconds (10));
59  Simulator::Schedule (MicroSeconds (5), &Watchdog::Ping, &watchdog, MicroSeconds (20));
60  Simulator::Schedule (MicroSeconds (20), &Watchdog::Ping, &watchdog, MicroSeconds (2));
61  Simulator::Schedule (MicroSeconds (23), &Watchdog::Ping, &watchdog, MicroSeconds (17));
62  Simulator::Run ();
63  Simulator::Destroy ();
64  NS_TEST_ASSERT_MSG_EQ (m_expired, true, "The timer did not expire ??");
65  NS_TEST_ASSERT_MSG_EQ (m_expiredTime, MicroSeconds (40), "The timer did not expire at the expected time ?");
66  NS_TEST_ASSERT_MSG_EQ (m_expiredArgument, MicroSeconds (40), "We did not get the right argument");
67 }
68 
69 
70 
71 static class WatchdogTestSuite : public TestSuite
72 {
73 public:
75  : TestSuite ("watchdog", UNIT)
76  {
77  AddTestCase (new WatchdogTestCase (), TestCase::QUICK);
78  }
79 } g_watchdogTestSuite;
keep track of time unit.
Definition: nstime.h:149
void Ping(Time delay)
Definition: watchdog.cc:42
A suite of tests to run.
Definition: test.h:962
virtual void DoRun(void)
Implementation to actually run this test case.
encapsulates test code
Definition: test.h:834
TestSuite(std::string name, Type type=UNIT)
Constuct a new test suite.
Definition: test.cc:354
void SetFunction(FN fn)
Definition: watchdog.h:144
a very simple watchdog
Definition: watchdog.h:37
void AddTestCase(TestCase *testCase) NS_DEPRECATED
Add an individual test case to this test suite.
Definition: test.cc:172
Time Seconds(double seconds)
create ns3::Time instances in units of seconds.
Definition: nstime.h:586
void SetArguments(T1 a1)
Definition: watchdog.h:159
Time MicroSeconds(uint64_t us)
create ns3::Time instances in units of microseconds.
Definition: nstime.h:615