Update util.py

includes/util.py

@@ -5,6 +5,9 @@
 import unittest
 from contextlib import redirect_stdout
 from io import StringIO
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support import expected_conditions as EC
 
 
 def run_test(classname, data, modulename):
@@ -42,3 +45,22 @@ def parse_results(buffer):
         return 'FAIL'
     else:
         return 'Test failed to execute'
+
+ def login(workspace, username, password, driver):
+    ''' Function to log user in to workspace.
+    '''
+    # Select and submit workspace
+    driver.find_element_by_id('display_select_input').click()
+    driver.find_element_by_link_text(workspace).click()
+    driver.find_element_by_id('sentworkspace').click()
+
+    # Wait for login page to load
+    wait = WebDriverWait(driver, 30)
+    wait.until(EC.element_to_be_clickable((By.ID, 'form[BSUBMIT]')))
+
+    # Login
+    driver.find_element_by_id('form[USR_USERNAME]').send_keys(username)
+    driver.find_element_by_id('form[USR_PASSWORD_MASK]').send_keys(password)
+    driver.find_element_by_id('form[BSUBMIT]').click()
+
+    return driver