LP 448319 : Moved event subscription to asynchronous mode

1 files changed, 10 insertions, 6 deletions

diff --git a/tests/enginetests.py b/tests/enginetests.py
index f1efe45..a035c2f 100644
--- a/tests/enginetests.py
+++ b/tests/enginetests.py
@@ -53,10 +53,13 @@ class MockProbeMgr(object):
     def uninstall(self, action_address):
         self.action = None
 
-    def subscribe(self, event, callback):
+    def subscribe(self, event, notif_cb, event_sub_cb, error_cb):
         self.event = event
-        self.cB = callback
-        self.event.install_handlers(callback)
+        self.cB = notif_cb
+        self.event.install_handlers(notif_cb)
+        # Trigger the installation callback
+        self.event_sub_cB = event_sub_cb
+        self._subscribe_error_cb = error_cb
         return str(event)
 
     def unsubscribe(self, address):
@@ -87,19 +90,20 @@ class TutorialRunnerTest(unittest.TestCase):
         tutorial.add_transition(state_name, (event, Tutorial.END))
 
         runner = TutorialRunner(tutorial, self.pM)
+
         runner.start()
+        self.pM.event_sub_cB('event1')
 
         assert runner._state == state_name, "Current state is: %s"%runner._state
         assert self.pM.action == None
         assert self.pM.event == event
-
+        
         event.do_callback()
+
         assert runner._state == Tutorial.END, "Current state is: %s"%runner._state
         assert self.pM.action == None, "Current action is %s"%str(self.pM.action)
         assert self.pM.event == None, "Current event is %s"%str(self.pM.event)
         
-        
-
     # Limit cases 
     def testEmptyTutorial(self):
         tutorial = Tutorial("TutorialRunner")