tests/test_CheckNotifications.py

   1 from util import login
   2 from selenium import webdriver
   3 from selenium.webdriver.common.by import By
   4 from selenium.webdriver.support.ui import Select
   5 from selenium.common.exceptions import NoSuchElementException
   6 import unittest, time, re
   7
   8 class CheckNotifications(unittest.TestCase):
   9     def setUp(self):
  10         self.driver = webdriver.Firefox()
  11         self.driver.implicitly_wait(30)
  12         self.base_url = "http://127.0.0.1:6543"
  13         self.verificationErrors = []
  14         self.accept_next_alert = True
  15
  16     def test_check_notifications(self):
  17         driver = self.driver
  18         driver.get(self.base_url + "/")
  19         tLogin = login.Login()
  20         #data = {"username" : "", "password" : ""}
  21         tLogin.log_in(self, data)
  22         self.assertTrue(self.is_element_present(By.LINK_TEXT, "1"))
  23         driver.find_element_by_link_text("1").click()
  24         driver.find_element_by_css_selector("div.button_action.header_dropdown_down").click()
  25         try: self.assertEqual("This is a test comment.", driver.find_element_by_css_selector("div.comment_content > p").text)
  26         except AssertionError as e: self.verificationErrors.append(str(e))
  27
  28     def is_element_present(self, how, what):
  29         try: self.driver.find_element(by=how, value=what)
  30         except NoSuchElementException, e: return False
  31         return True
  32
  33     def is_alert_present(self):
  34         try: self.driver.switch_to_alert()
  35         except NoAlertPresentException, e: return False
  36         return True
  37
  38     def close_alert_and_get_its_text(self):
  39         try:
  40             alert = self.driver.switch_to_alert()
  41             alert_text = alert.text
  42             if self.accept_next_alert:
  43                 alert.accept()
  44             else:
  45                 alert.dismiss()
  46             return alert_text
  47         finally: self.accept_next_alert = True
  48
  49     def tearDown(self):
  50         self.driver.quit()
  51         self.assertEqual([], self.verificationErrors)
  52
  53 if __name__ == "__main__":
  54     unittest.main()