Make headless tests configurable via setting

jonasundderwolf · Apr 3, 2015 · 0a22b61 · 0a22b61
1 parent fd1e3bf
commit 0a22b61
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/example/settings.py b/example/settings.py
@@ -86,3 +86,4 @@
 IMAGE_CROPPING_THUMB_SIZE = (300, 300)
 IMAGE_CROPPING_JQUERY_URL = 'js/jquery.min.js'
 THUMBNAIL_DEBUG = True
+HEADLESS = False
diff --git a/example/tests/test_browser.py b/example/tests/test_browser.py
@@ -17,15 +17,17 @@ class BrowserTestBase(object):
 
     @classmethod
     def setUpClass(cls):
-        cls.display = Display(visible=0, size=(1024, 768))
-        cls.display.start()
+        if settings.HEADLESS:
+            cls.display = Display(visible=0, size=(1024, 768))
+            cls.display.start()
         cls.selenium = WebDriver()
         super(BrowserTestBase, cls).setUpClass()
 
     @classmethod
     def tearDownClass(cls):
         cls.selenium.quit()
-        cls.display.stop()
+        if settings.HEADLESS:
+            cls.display.stop()
         super(BrowserTestBase, cls).tearDownClass()
 
     def setUp(self):