Add MCN failure experiment
refs: #3272
Change-Id: I663680898bf4a0b9da21e0c408933cef6200564b
diff --git a/ndn/experiments/mcn_failure_experiment.py b/ndn/experiments/mcn_failure_experiment.py
new file mode 100644
index 0000000..cc3fa09
--- /dev/null
+++ b/ndn/experiments/mcn_failure_experiment.py
@@ -0,0 +1,67 @@
+# -*- Mode:python; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
+#
+# Copyright (C) 2015 The University of Memphis,
+# Arizona Board of Regents,
+# Regents of the University of California.
+#
+# This file is part of Mini-NDN.
+# See AUTHORS.md for a complete list of Mini-NDN authors and contributors.
+#
+# Mini-NDN is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Mini-NDN is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Mini-NDN, e.g., in COPYING.md file.
+# If not, see <http://www.gnu.org/licenses/>.
+
+from ndn.experiments.experiment import Experiment
+from ndn.nlsr import Nlsr
+
+import time
+
+class FailureExperiment(Experiment):
+
+ def __init__(self, args):
+ args["nPings"] = 300
+ Experiment.__init__(self, args)
+
+ self.PING_COLLECTION_TIME_BEFORE_FAILURE = 60
+ self.PING_COLLECTION_TIME_AFTER_RECOVERY = 120
+
+ def getMostConnectedNode(self):
+ mcn = max(self.net.hosts, key=lambda host: len(host.intfNames()))
+ print "The most connected node is: %s" % mcn.name
+ return mcn
+
+ def run(self):
+ mostConnectedNode = self.getMostConnectedNode()
+
+ self.startPctPings()
+
+ # After the pings are scheduled, collect pings for 1 minute
+ time.sleep(self.PING_COLLECTION_TIME_BEFORE_FAILURE)
+
+ # Bring down MCN
+ self.failNode(mostConnectedNode)
+
+ # MCN is down for 2 minutes
+ time.sleep(120)
+
+ # Bring MCN back up
+ self.recoverNode(mostConnectedNode)
+
+ # Restart pings
+ for nodeToPing in self.pingedDict[mostConnectedNode]:
+ self.ping(mostConnectedNode, nodeToPing, self.PING_COLLECTION_TIME_AFTER_RECOVERY)
+
+ # Collect pings for more seconds after MCN is up
+ time.sleep(self.PING_COLLECTION_TIME_AFTER_RECOVERY)
+
+Experiment.register("failure-mcn", FailureExperiment)