/
test_login.py
38 lines (28 loc) · 1.19 KB
/
test_login.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# -*- coding: utf-8 -*-
import unittest
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.support.ui import WebDriverWait
from . import SeleniumTestCase, Annotator
class TestLogin(SeleniumTestCase):
def test_login(self):
driver = self.driver
driver.get(self.base_url + "/")
# Assert logged in with the right username
with Annotator(driver):
self.register()
self.logout()
self.login()
picker = (By.CLASS_NAME, 'user-picker')
ec = expected_conditions.visibility_of_element_located(picker)
WebDriverWait(self.driver, 10).until(ec)
picker = driver.find_element_by_class_name('user-picker')
dropdown = picker.find_element_by_class_name('dropdown-toggle')
# Some bugs were fixed in selenium 2.35 + FF23 combo
# Unfortunately, that means we need test both options
try:
self.assertEqual(dropdown.text, "test")
except AssertionError:
self.assertEqual(dropdown.text, "test/localhost")
if __name__ == "__main__":
unittest.main()