From 91757a5b113841e2c1a465c184e111aeac8fd1d6 Mon Sep 17 00:00:00 2001
From: Vincent Vinet <vince.vinet@gmail.com>
Date: Thu, 29 Oct 2009 14:37:57 +0000
Subject: add the probe tests

---
(limited to 'tutorius/TProbe.py')

diff --git a/tutorius/TProbe.py b/tutorius/TProbe.py
index 867ef1c..afeba6d 100644
--- a/tutorius/TProbe.py
+++ b/tutorius/TProbe.py
@@ -195,7 +195,11 @@ class TProbe(dbus.service.Object):
     # The actual method we will call on the probe to send events
     def notify(self, event):
         LOGGER.debug("TProbe :: notify event %s", str(event))
-        self.eventOccured(pickle.dumps(event))
+        #Check that this event is even allowed
+        if event in self._subscribedEvents.values():
+            self.eventOccured(pickle.dumps(event))
+        else:
+            raise RuntimeWarning("Attempted to raise an unregistered event")
 
     # Return a unique name for this action 
     def _generate_action_reference(self, action):
@@ -400,8 +404,8 @@ class ProbeProxy:
         Detach the ProbeProxy from it's TProbe. All installed actions and 
         subscribed events should be removed.
         """
-        for action in self._actions.keys():
-            self.uninstall(action, block)
+        for action_addr in self._actions.keys():
+            self.uninstall(action_addr, block)
 
         for address in self._subscribedEvents.keys():
             self.unsubscribe(address, block)
@@ -414,7 +418,12 @@ class ProbeManager(object):
     For now, it only handles one at a time, though.
     Actually it doesn't do much at all. But it keeps your encapsulation happy
     """
-    def __init__(self):
+    def __init__(self, proxy_class=ProbeProxy):
+        """Constructor
+        @param proxy_class Class to use for creating Proxies to activities.
+            The class should support the same interface as ProbeProxy.
+        """
+        self._ProxyClass = proxy_class
         self._probes = {}
         self._current_activity = None
 
@@ -431,7 +440,7 @@ class ProbeManager(object):
         if activity_id in self._probes:
             raise RuntimeWarning("Activity already attached")
 
-        self._probes[activity_id] = ProbeProxy(activity_id)
+        self._probes[activity_id] = self._ProxyClass(activity_id)
         #TODO what do we do with this? Raise something?
         if self._probes[activity_id].isAlive():
             print "Alive!"
@@ -442,6 +451,8 @@ class ProbeManager(object):
         if activity_id in self._probes:
             probe = self._probes.pop(activity_id)
             probe.detach()
+            if self._current_activity == activity_id:
+                self._current_activity = None
 
     def install(self, action, block=False):
         """
--
cgit v0.9.1