108 lines
3.8 KiB
Python
108 lines
3.8 KiB
Python
import pytest
|
|
from selenium import webdriver
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.keys import Keys
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|
from selenium.common.exceptions import NoSuchElementException, TimeoutException
|
|
|
|
# 使用 Pytest 的 fixture 初始化 WebDriver 和 URL
|
|
@pytest.fixture(scope="module")
|
|
def driver():
|
|
# 初始化WebDriver
|
|
driver = webdriver.Edge(executable_path=r'D:\codeapp\python-3.8.5\Lib\site-packages\selenium\webdriver\edge\MicrosoftWebDriver.exe')
|
|
driver.implicitly_wait(10)
|
|
yield driver # yield 语句之后的代码在测试用例执行完毕后运行
|
|
driver.quit() # 测试用例执行完毕后关闭 WebDriver
|
|
|
|
@pytest.fixture(scope="module")
|
|
def URL():
|
|
return 'http://127.0.0.1:5000'
|
|
|
|
# 测试登录页面输入密码包含空格的情况
|
|
def test_login_password_with_space(driver, URL):
|
|
try:
|
|
driver.get(URL+"/login")
|
|
|
|
# 输入用户名
|
|
username_elem = driver.find_element_by_id('username')
|
|
username_elem.send_keys("test_user")
|
|
|
|
# 输入包含空格的密码
|
|
password_elem = driver.find_element_by_id('password')
|
|
password_elem.send_keys("password with space")
|
|
|
|
# 提交登录表单
|
|
elem = driver.find_element_by_class_name('btn') # 找到提交按钮
|
|
elem.click()
|
|
|
|
# 显式等待直到警告框出现
|
|
WebDriverWait(driver, 10).until(EC.alert_is_present())
|
|
|
|
# 检查是否弹出警告框
|
|
alert = driver.switch_to.alert
|
|
assert "输入不能包含空格!" in alert.text
|
|
except (NoSuchElementException, TimeoutException) as e:
|
|
pytest.fail("页面元素未找到或超时: {}".format(e))
|
|
|
|
|
|
# 测试注册页面输入密码包含空格的情况
|
|
|
|
def test_signup_password_with_space(driver, URL):
|
|
try:
|
|
driver.get(URL+"/signup")
|
|
|
|
# 输入用户名
|
|
username_elem = driver.find_element_by_id('username')
|
|
username_elem.send_keys("new_user")
|
|
|
|
# 输入包含空格的密码
|
|
password_elem = driver.find_element_by_id('password')
|
|
password_elem.send_keys("password with space")
|
|
|
|
# 再次输入密码
|
|
password2_elem = driver.find_element_by_id('password2')
|
|
password2_elem.send_keys("password with space")
|
|
|
|
# 提交注册表单
|
|
elem = driver.find_element_by_class_name('btn') # 找到提交按钮
|
|
elem.click()
|
|
|
|
# 显式等待直到警告框出现
|
|
WebDriverWait(driver, 10).until(EC.alert_is_present())
|
|
|
|
# 检查是否弹出警告框
|
|
alert = driver.switch_to.alert
|
|
assert "输入不能包含空格!" in alert.text
|
|
except (NoSuchElementException, TimeoutException) as e:
|
|
pytest.fail("页面元素未找到或超时: {}".format(e))
|
|
|
|
|
|
|
|
# 测试重设密码页面输入新密码包含空格的情况
|
|
|
|
def test_reset_password_with_space(driver, URL):
|
|
try:
|
|
driver.get(URL+"/reset")
|
|
|
|
# 输入用户名
|
|
username_elem = driver.find_element_by_id('username')
|
|
username_elem.send_keys("test_user")
|
|
|
|
# 输入包含空格的密码
|
|
password_elem = driver.find_element_by_id('password')
|
|
password_elem.send_keys("password with space")
|
|
|
|
# 提交重设密码表单
|
|
elem = driver.find_element_by_class_name('btn') # 找到提交按钮
|
|
elem.click()
|
|
|
|
# 显式等待直到警告框出现
|
|
WebDriverWait(driver, 10).until(EC.alert_is_present())
|
|
|
|
# 检查是否弹出警告框
|
|
alert = driver.switch_to.alert
|
|
assert "输入不能包含空格!" in alert.text
|
|
except (NoSuchElementException, TimeoutException) as e:
|
|
pytest.fail("页面元素未找到或超时: {}".format(e))
|