add test for not exists in __init__

cgoldberg · Jul 7, 2016 · 4941894 · 4941894
1 parent b4e8aac
commit 4941894
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test_xvfb.py b/test_xvfb.py
@@ -19,6 +19,12 @@ def reset_display(self):
     def setUp(self):
         self.reset_display()
 
+    def test_xvfb_binary_not_exists(self):
+        with patch('xvfbwrapper.Xvfb.xvfb_exists') as xvfb_exists:
+            xvfb_exists.return_value = False
+            with self.assertRaises(EnvironmentError):
+                Xvfb()
+
     def test_start(self):
         xvfb = Xvfb()
         self.addCleanup(xvfb.stop)