diff --git a/functional_tests/tests.py b/functional_tests/tests.py index d008385..3485829 100644 --- a/functional_tests/tests.py +++ b/functional_tests/tests.py @@ -1,17 +1,20 @@ -from django.test import LiveServerTestCase +from django.contrib.staticfiles.testing import StaticLiveServerTestCase from selenium import webdriver from selenium.common.exceptions import WebDriverException from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys +import os import time import unittest MAX_WAIT = 5 -class NewVisitorTest(LiveServerTestCase): +class NewVisitorTest(StaticLiveServerTestCase): # Helper methods def setUp(self): self.browser = webdriver.Firefox() + if test_server := os.environ.get('TEST_SERVER'): + self.live_server_url = 'http://' + test_server def tearDown(self): self.browser.quit()