ignore absence of gtk when testing

Bauble · Jan 3, 2015 · 3cc9cb8 · 3cc9cb8
1 parent aeb515a
commit 3cc9cb8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/bauble/fake_gtk.py b/bauble/fake_gtk.py
@@ -2,10 +2,13 @@ class fake:
     MESSAGE_ERROR = ''
     MESSAGE_WARNING = ''
     MESSAGE_INFO = ''
-    BUTTONS_OK = None
+    DIALOG_MODAL = DIALOG_DESTROY_WITH_PARENT = BUTTONS_OK = None
     VBox = object
-    Action = Expander = ScrolledWindow = object
+    Menu = Action = Expander = ScrolledWindow = object
     EventBox = object
 
+    def MessageDialog(*args):
+        pass
+
 gtk = fake()
 gobject = fake()
diff --git a/bauble/task.py b/bauble/task.py
@@ -7,8 +7,12 @@
 import Queue
 
 import fibra
-import gobject
-import gtk
+
+try:
+    import gtk
+    import gobject
+except:
+    from bauble.fake_gtk import gtk, gobject
 
 import bauble
 import bauble.utils as utils