diff options
Diffstat (limited to 'tests/enginetests.py')
-rw-r--r-- | tests/enginetests.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tests/enginetests.py b/tests/enginetests.py index f24aedb..1723954 100644 --- a/tests/enginetests.py +++ b/tests/enginetests.py @@ -72,17 +72,17 @@ class MockProbeMgrMultiAddons(object): def uninstall(self, action_address): del self.action_dict[action_address] - def subscribe(self, event_name, event, notif_cb, subscribe_cb, error_cb): + def subscribe(self, event, notif_cb, subscribe_cb, error_cb): event_address = str(uuid1()) - self.event_dict[event_name] = event_address - self.event_cb_dict[event_name] = notif_cb + self.event_dict[event_address] = event + self.event_cb_dict[event_address] = notif_cb self._event_subscribed_cb_list.append(subscribe_cb) self._subscribe_error_cb_list.append(error_cb) def unsubscribe(self, address): - for (event_name, other_event) in self.event_dict.values(): - if event == othet_event: - del self.event_dict[event_name] + for (event_address, other_event) in self.event_dict.values(): + if event == other_event: + del self.event_dict[address] break class MockProbeMgr(object): @@ -110,7 +110,7 @@ class MockProbeMgr(object): def uninstall(self, action_address): self.action = None - def subscribe(self, event_name, event, notif_cb, event_sub_cb, error_cb): + def subscribe(self, event, notif_cb, event_sub_cb, error_cb): self.event = event self.cB = notif_cb self.event.install_handlers(notif_cb) |