138 lines
5.5 KiB
Python
138 lines
5.5 KiB
Python
import unittest
|
|
import inspect
|
|
from selenium import webdriver
|
|
from selenium.webdriver.common.keys import Keys
|
|
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions
|
|
|
|
|
|
def wait_net_service(server, port, timeout=None):
|
|
""" Wait for network service to appear
|
|
@param timeout: in seconds, if None or 0 wait forever
|
|
@return: True of False, if timeout is None may return only True or
|
|
throw unhandled network exception
|
|
"""
|
|
import socket
|
|
import errno
|
|
|
|
s = socket.socket()
|
|
if timeout:
|
|
from time import time as now
|
|
# time module is needed to calc timeout shared between two exceptions
|
|
end = now() + timeout
|
|
|
|
while True:
|
|
try:
|
|
if timeout:
|
|
next_timeout = end - now()
|
|
if next_timeout < 0:
|
|
return False
|
|
else:
|
|
s.settimeout(next_timeout)
|
|
|
|
s.connect((server, port))
|
|
|
|
except socket.timeout as err:
|
|
# this exception occurs only if timeout is set
|
|
if timeout:
|
|
return False
|
|
|
|
except socket.error as err:
|
|
# catch timeout exception from underlying network library
|
|
# this one is different from socket.timeout
|
|
if type(err.args) != tuple or err[0] != errno.ETIMEDOUT:
|
|
raise
|
|
else:
|
|
s.close()
|
|
return True
|
|
|
|
|
|
class BaseTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.screen_shot_counter = 0
|
|
self.driver = webdriver.Remote(
|
|
command_executor='http://selenium:4444/wd/hub',
|
|
desired_capabilities=DesiredCapabilities.FIREFOX)
|
|
self.driver.get("http://rollerverbrauch:5000")
|
|
WebDriverWait(self.driver, 10).until(expected_conditions.presence_of_element_located((By.ID, "email")))
|
|
|
|
def create_screenshot(self):
|
|
self.driver.get_screenshot_as_file(
|
|
'/results/%s_%s_%s.png' % (self.__class__.__name__, inspect.stack()[1][3], str(self.screen_shot_counter)))
|
|
self.screen_shot_counter += 1
|
|
|
|
def tearDown(self):
|
|
self.driver.close()
|
|
|
|
|
|
class RegisterCheck(BaseTestCase):
|
|
def setUp(self):
|
|
self.screen_shot_counter = 0
|
|
self.driver = webdriver.Remote(
|
|
command_executor='http://selenium:4444/wd/hub',
|
|
desired_capabilities=DesiredCapabilities.FIREFOX)
|
|
self.driver.get("http://rollerverbrauch:5000")
|
|
WebDriverWait(self.driver, 10).until(expected_conditions.presence_of_element_located((By.ID, "email")))
|
|
|
|
def test_page_loads(self):
|
|
self.driver.get("http://rollerverbrauch:5000")
|
|
self.create_screenshot()
|
|
self.assertIn("refuel", self.driver.title, "Title must contain reload")
|
|
|
|
def test_can_register(self):
|
|
self.driver.get("http://rollerverbrauch:5000")
|
|
self.driver.find_element_by_partial_link_text('Register').click()
|
|
self.create_screenshot()
|
|
WebDriverWait(self.driver, 10).until(expected_conditions.presence_of_element_located((By.ID, "submit")))
|
|
self.create_screenshot()
|
|
self.driver.find_element_by_id('email').send_keys('test@test.com')
|
|
self.driver.find_element_by_id('password').send_keys('test123')
|
|
self.driver.find_element_by_id('password_confirm').send_keys('test123')
|
|
self.create_screenshot()
|
|
self.driver.find_element_by_id('submit').click()
|
|
WebDriverWait(self.driver, 10).until(expected_conditions.presence_of_element_located((By.ID, "i1")))
|
|
self.create_screenshot()
|
|
|
|
def test_register_must_repeat_pwd(self):
|
|
self.driver.get("http://rollerverbrauch:5000")
|
|
self.driver.find_element_by_partial_link_text('Register').click()
|
|
self.create_screenshot()
|
|
WebDriverWait(self.driver, 10).until(expected_conditions.presence_of_element_located((By.ID, "submit")))
|
|
self.create_screenshot()
|
|
self.driver.find_element_by_id('email').send_keys('test1@test.com')
|
|
self.driver.find_element_by_id('password').send_keys('test123')
|
|
self.create_screenshot()
|
|
self.driver.find_element_by_id('submit').click()
|
|
self.create_screenshot()
|
|
error = self.driver.find_elements_by_class_name('error')
|
|
self.assertIsNotNone(error[0], 'we expect an error')
|
|
self.assertIn('Passwords do not match', error[0].text, 'wrong error message')
|
|
|
|
def test_register_must_be_equal_pwd(self):
|
|
self.driver.get("http://rollerverbrauch:5000")
|
|
self.driver.find_element_by_partial_link_text('Register').click()
|
|
self.create_screenshot()
|
|
WebDriverWait(self.driver, 10).until(expected_conditions.presence_of_element_located((By.ID, "submit")))
|
|
self.create_screenshot()
|
|
self.driver.find_element_by_id('email').send_keys('test1@test.com')
|
|
self.driver.find_element_by_id('password').send_keys('test123')
|
|
self.driver.find_element_by_id('password_confirm').send_keys('test1234')
|
|
self.create_screenshot()
|
|
self.driver.find_element_by_id('submit').click()
|
|
self.create_screenshot()
|
|
error = self.driver.find_elements_by_class_name('error')
|
|
self.assertIsNotNone(error[0], 'we expect an error')
|
|
self.assertIn('Passwords do not match', error[0].text, 'wrong error message')
|
|
|
|
|
|
class LoginCheck(BaseTestCase):
|
|
|
|
def can_login(self):
|
|
pass
|
|
|
|
if __name__ == "__main__":
|
|
wait_net_service('selenium', 4444)
|
|
unittest.main()
|