Add selenium base for testing

requirements-dev.txt

@@ -3,3 +3,5 @@ log_colorizer
 wsreload
 cutter
 pytest
+selenium
+pyvirtualdisplay

test/selenium/conftest.py

@@ -0,0 +1,105 @@
+from selenium import webdriver
+from pytest import fixture
+from time import sleep
+from multiprocessing import Process, active_children, Lock
+from signal import signal, SIGUSR1, SIGKILL
+import atexit
+import os
+import re
+import sys
+try:
+    from wdb.ext import add_w_builtin
+    add_w_builtin()
+except ImportError:
+    pass
+
+display = None
+fs_path = {}
+
+
+def pytest_report_teststatus(report):
+    try:
+        browser = re.search('(\[.*\])', report.nodeid).groups()[0]
+    except:
+        return
+
+    if browser not in fs_path:
+        fs_path[browser] = [report.fspath]
+        sys.stdout.write('\t' + browser + '\t')
+
+    elif report.fspath not in fs_path[browser]:
+        fs_path[browser].append(report.fspath)
+        sys.stdout.write('\t' + browser + '\t')
+
+if not os.environ.get('SELENIUM_BROWSERS_VISIBLE'):
+    from pyvirtualdisplay import Display
+    display = Display(visible=0, size=(1440, 900))
+    display.start()
+
+
+class App(Process):
+    def __init__(self):
+        # self.parent_pid = os.getpid()
+        super(App, self).__init__()
+        self.lock = Lock()
+        self.start()
+
+    def run(self):
+        from http.server import HTTPServer
+        old_serve_forever = HTTPServer.serve_forever
+        print('Acquire lock')
+        l = self.lock
+        l.acquire()
+
+        # Inform parent that we are ready
+        def new_serve_forever(self):
+            print('Release lock')
+            l.release()
+            # from signal import SIGUSR1
+            # os.kill(ppid, SIGUSR1)
+            old_serve_forever(self)
+
+        HTTPServer.serve_forever = new_serve_forever
+
+        from app import app
+        app.run(
+            debug=True,
+            threaded=True,
+            use_reloader=False,
+            port=29013)
+
+    def wait_for_lock(self):
+        ok = self.lock.acquire(20)
+        if not ok:
+            raise RuntimeError(
+                'Impossible to get app lock.'
+                ' App may not have been started successfuly')
+        self.lock.release()
+
+
+@fixture(scope='session')
+def app(request):
+    app = App()
+
+    def end_app():
+        app.terminate()
+
+    request.addfinalizer(end_app)
+    return app
+
+@fixture(scope='session', params=('firefox', 'chrome'))
+def s(request, app): # s = Selenium Browser
+    browser = getattr(webdriver, request.param.capitalize())()
+
+    def close_browser():
+        browser.close()
+
+    request.addfinalizer(close_browser)
+    app.wait_for_lock()
+    return browser
+
+
+@atexit.register
+def quit():
+    if display:
+        display.stop()

test/selenium/test_404.py

@@ -0,0 +1,4 @@
+def test_not_found(s):
+    s.get('http://localhost:29013/some_random_page')
+    h1 = s.find_element_by_css_selector('h1')
+    assert h1.text == 'Not Found'

test/selenium/test_index.py

@@ -0,0 +1,11 @@
+
+def test_index(s):
+    s.get('http://localhost:29013/')
+    assert 'Apparatus' in s.title
+    assert s.find_element_by_css_selector('h1')
+
+
+def test_index_2(s):
+    s.get('http://localhost:29013/')
+    assert 'Apparatus' in s.title
+    assert s.find_element_by_css_selector('ol')