check_various_pages.py

import unittest
from base_test_class import BaseTestCase
from selenium.webdriver.common.by import By
import sys


class VariousPagesTest(BaseTestCase):
    def test_user_status(self):
        driver = self.driver
        driver.get(self.base_url + "user")

    def test_calendar_status(self):
        driver = self.driver
        driver.get(self.base_url + "calendar")
        # click apply to see if this helps webdriver to catch the javascript errors we're seeing
        driver.find_element(By.CSS_SELECTOR, "input.btn.btn-primary").click()


def suite():
    suite = unittest.TestSuite()
    suite.addTest(BaseTestCase('test_login'))
    suite.addTest(VariousPagesTest('test_user_status'))
    suite.addTest(VariousPagesTest('test_calendar_status'))
    return suite


if __name__ == "__main__":
    runner = unittest.TextTestRunner(descriptions=True, failfast=True, verbosity=2)
    ret = not runner.run(suite()).wasSuccessful()
    BaseTestCase.tearDownDriver()
    sys.exit(ret)