1 files changed, 6 insertions, 0 deletions
diff --git a/tests/probetests.py b/tests/probetests.py
index f073135..99fbf4d 100644
--- a/tests/probetests.py
+++ b/tests/probetests.py
@@ -29,6 +29,7 @@ from sugar.tutorius.TProbe import TProbe, ProbeProxy, ProbeManager
 from sugar.tutorius import addon
 from sugar.tutorius.actions import Action
 from sugar.tutorius.properties import TIntProperty, TStringProperty
+from sugar.tutorius.gtkutils import find_widget
 
 
 #Create a substitute addon create function
@@ -72,6 +73,11 @@ class MockActivity(object):
     def get_id(self):
         return "unique_id_1"
 
+
+class MockProbe(object):
+    def find_widget(self, base, path, ignore_errors=True):
+        return find_widget(base,path,ignore_errors)
+
 class MockProbeProxy(object):
     _MockProxyCache = {}
     def __new__(cls, activityName, unique_id):