commit 3566f2f3ddc62224f4df4568ae3a9deb732e2859
author Zhenkai Zhu <zhenkai@cs.ucla.edu> Fri Jan 25 23:47:59 2013 -0800
committer Zhenkai Zhu <zhenkai@cs.ucla.edu> Fri Jan 25 23:47:59 2013 -0800
tree b8012119b9b3307092aaf6ff8f4d7fbf6a3023b5
parent 95160100432b58d59cd01a528a20b34676068572

fix bug in rescheduleTask;
use PeriodicTask instead (so it won't delete itself)

test/test-event-scheduler.cc

diff --git a/test/test-event-scheduler.cc b/test/test-event-scheduler.cc
index 4db9a0e..0d70abf 100644
--- a/test/test-event-scheduler.cc
+++ b/test/test-event-scheduler.cc
@@ -26,6 +26,7 @@
 #include "random-interval-generator.h"
 
 #include <boost/test/unit_test.hpp>
+#include <boost/make_shared.hpp>
 #include <map>
 #include <unistd.h>
 
@@ -136,6 +137,26 @@
   scheduler->shutdown();
 }
 
+void reschedule();
+SchedulerPtr schd0(new Scheduler());
+int resCount;
+TaskPtr task0(new PeriodicTask(boost::bind(reschedule), "testtest", schd0, boost::make_shared<SimpleIntervalGenerator>(0.5)));
+void reschedule()
+{
+  schd0->rescheduleTask(task0);
+  resCount++;
+}
+
+BOOST_AUTO_TEST_CASE(RescheduleTest)
+{
+  resCount = 0;
+  schd0->start();
+  schd0->addTask(task0);
+  usleep(5100000);
+  BOOST_CHECK_EQUAL(resCount, 10);
+  schd0->shutdown();
+}
+
 BOOST_AUTO_TEST_CASE(GeneratorTest)
 {
   double interval = 10;