Adding screen to payment selenium tests

payment/tests.py

@@ -20,10 +20,14 @@ import time
 from selenium import webdriver
 from selenium.webdriver.support.ui import WebDriverWait
 import logging
-
+import os
 from decimal import Decimal as D
 import datetime
 
+def setup_selenium():
+    # Set the display window for our xvfb
+    os.environ['DISPLAY'] = ':99'
+
 def set_test_logging():
     
     # Setup debug logging to our console so we can watch
@@ -95,6 +99,7 @@ class PledgeTest(TestCase):
         # This is an empty array where we will store any verification errors
         # we find in our tests
 
+        setup_selenium()
         self.selenium = webdriver.Firefox()
         set_test_logging()
 
@@ -191,6 +196,7 @@ class AuthorizeTest(TestCase):
         # This is an empty array where we will store any verification errors
         # we find in our tests
 
+        setup_selenium()
         self.selenium = webdriver.Firefox()
         set_test_logging()
     
@@ -269,4 +275,4 @@ def suite():
     suites = unittest.TestSuite([unittest.TestLoader().loadTestsFromTestCase(testcase) for testcase in testcases])
     return suites    
         
-       
+