A Discrete-Event Network Simulator
API
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
watchdog-test-suite.cc
Go to the documentation of this file.
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 namespace ns3 {
24 
25 class WatchdogTestCase : public TestCase
26 {
27 public:
28  WatchdogTestCase ();
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
50 WatchdogTestCase::DoRun (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:
74  WatchdogTestSuite()
75  : TestSuite ("watchdog", UNIT)
76  {
77  AddTestCase (new WatchdogTestCase ());
78  }
79 } g_watchdogTestSuite;
80 
81 } // namespace ns3
82