Add logged_in decorator

Author: Brandon Duffany

.gitignore

@@ -1,2 +1,3 @@
 .tox
 *.pyc
+*.swp

codebender_testing/config.py

@@ -7,3 +7,8 @@
     "firefox": webdriver.Firefox,
     "chrome": webdriver.Chrome
 }
+
+TEST_CREDENTIALS = {
+    "username": "tester",
+    "password": "testerPASS"
+}

codebender_testing/utils.py

@@ -2,10 +2,13 @@
 import sys
 
 from selenium import webdriver
+from selenium.common.exceptions import NoSuchElementException
+from selenium.webdriver.common.keys import Keys
 import pytest
 
-from codebender_testing.config import WEBDRIVERS
 from codebender_testing.config import BASE_URL
+from codebender_testing.config import TEST_CREDENTIALS
+from codebender_testing.config import WEBDRIVERS
 
 
 class SeleniumTestCase(object):
@@ -20,14 +23,40 @@ def setup(self, request):
         webdriver_cls = WEBDRIVERS[request.param]
         self.driver = webdriver_cls()
 
-    def open(self, url):
+    def open(self, url=None):
         """Open the resource specified by `url`.
         If an absolute URL is specified (like 'http://codebender.cc') we
         use that URL. Otherwise the resource is relative to `BASE_URL` in
         `codebender_testing.config`.
         """
+        if url is None:
+            url = ''
         if re.match(".+?://^", url):
+            # url specifies an absolute path.
             return self.driver.get(url)
         else:
             return self.driver.get("%s/%s" % (BASE_URL, url))
 
+
+def logged_in(func):
+    """Decorator to ensure the user is logged in before performing a test.
+    If not logged in, a login will be performed."""
+    def inner(*args, **kwargs):
+        test = args[0]
+        try:
+            test.open()
+            login_button = test.driver.find_element_by_id('login_btn')
+            login_button.send_keys(Keys.ENTER)
+            # Enter credentials and log in
+            user_field = test.driver.find_element_by_id('username')
+            user_field.send_keys(TEST_CREDENTIALS['username'])
+            pass_field = test.driver.find_element_by_id('password')
+            pass_field.send_keys(TEST_CREDENTIALS['password'])
+            do_login = test.driver.find_element_by_id('_submit')
+            do_login.send_keys(Keys.ENTER)
+        except NoSuchElementException:
+            # 'Log In' is not displayed, so we're already logged in.
+            pass
+        return func(*args, **kwargs)
+    return inner
+