tests/test_CheckNotifications.py

   1 #!/usr/bin/env python
   2 '''
   3    Copyright 2013 Emily O'Leary
   4
   5    Licensed under the Apache License, Version 2.0 (the "License");
   6    you may not use this file except in compliance with the License.
   7    You may obtain a copy of the License at
   8
   9        http://www.apache.org/licenses/LICENSE-2.0
  10
  11    Unless required by applicable law or agreed to in writing, software
  12    distributed under the License is distributed on an "AS IS" BASIS,
  13    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14    See the License for the specific language governing permissions and
  15    limitations under the License.
  16 '''
  17 from util import login
  18 from selenium import webdriver
  19 from selenium.webdriver.common.by import By
  20 from selenium.webdriver.support.ui import Select
  21 from selenium.common.exceptions import NoSuchElementException
  22 import unittest, time, re
  23
  24 class CheckNotifications(unittest.TestCase):
  25     def setUp(self):
  26         self.driver = webdriver.Firefox()
  27         self.driver.implicitly_wait(30)
  28         self.base_url = "http://127.0.0.1:6543"
  29         self.verificationErrors = []
  30         self.accept_next_alert = True
  31
  32     def test_check_notifications(self):
  33         driver = self.driver
  34         driver.get(self.base_url + "/")
  35         tLogin = login.Login()
  36         data = {"username" : "socialgoblin", "password" : "WhycantIholdalltheserhymes"}
  37         tLogin.log_in(self, data)
  38         self.assertTrue(self.is_element_present(By.LINK_TEXT, "1"))
  39         driver.find_element_by_link_text("1").click()
  40         driver.find_element_by_css_selector("div.button_action.header_dropdown_down").click()
  41         try: self.assertEqual("This is a test comment.", driver.find_element_by_css_selector("div.comment_content > p").text)
  42         except AssertionError as e: self.verificationErrors.append(str(e))
  43
  44     def is_element_present(self, how, what):
  45         try: self.driver.find_element(by=how, value=what)
  46         except NoSuchElementException, e: return False
  47         return True
  48
  49     def is_alert_present(self):
  50         try: self.driver.switch_to_alert()
  51         except NoAlertPresentException, e: return False
  52         return True
  53
  54     def close_alert_and_get_its_text(self):
  55         try:
  56             alert = self.driver.switch_to_alert()
  57             alert_text = alert.text
  58             if self.accept_next_alert:
  59                 alert.accept()
  60             else:
  61                 alert.dismiss()
  62             return alert_text
  63         finally: self.accept_next_alert = True
  64
  65     def tearDown(self):
  66         self.driver.quit()
  67         self.assertEqual([], self.verificationErrors)
  68
  69 if __name__ == "__main__":
  70     unittest.main()