Added #!/usr/bin/env python to tests.
[mediagoblin-automation:mediagoblin-selenium.git] / tests / test_Logout.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 LogOut(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_log_out(self):
33         driver = self.driver
34         driver.get(self.base_url + "/")
35         tLogin = login.Login()
36         data = {"username" : "testlogout", "password" : "OkayTestingPassworD42"}
37         tLogin.log_in(self, data)
38         self.assertTrue(self.is_element_present(By.CSS_SELECTOR, "div.button_action.header_dropdown_down"))
39         driver.find_element_by_css_selector("div.button_action.header_dropdown_down").click()
40         self.assertTrue(self.is_element_present(By.LINK_TEXT, "Log out"))
41         driver.find_element_by_link_text("Log out").click()
42         try: self.assertTrue(self.is_element_present(By.LINK_TEXT, "Create an account at this site"))
43         except AssertionError as e: self.verificationErrors.append(str(e))
44         try: self.assertTrue(self.is_element_present(By.LINK_TEXT, "Log in"))
45         except AssertionError as e: self.verificationErrors.append(str(e))
46     
47     def is_element_present(self, how, what):
48         try: self.driver.find_element(by=how, value=what)
49         except NoSuchElementException, e: return False
50         return True
51     
52     def is_alert_present(self):
53         try: self.driver.switch_to_alert()
54         except NoAlertPresentException, e: return False
55         return True
56     
57     def close_alert_and_get_its_text(self):
58         try:
59             alert = self.driver.switch_to_alert()
60             alert_text = alert.text
61             if self.accept_next_alert:
62                 alert.accept()
63             else:
64                 alert.dismiss()
65             return alert_text
66         finally: self.accept_next_alert = True
67     
68     def tearDown(self):
69         self.driver.quit()
70         self.assertEqual([], self.verificationErrors)
71
72 if __name__ == "__main__":
73     unittest.main()