上一篇：
（四）selenium自动化测试之上传本地文件_要开朗的spookypop的博客-CSDN博客_selenium上传本地文件
先看下测试代码运行效果：
在做自动化测试时，测试框架设计很重要，测试代码写法也很多种。最简单的莫过于流水式的写法，比如一个简单的登录功能
import time
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service as ChromeService
from selenium.webdriver.support.relative_locator import locate_with
from webdriver_manager.chrome import ChromeDriverManager
service = ChromeService(executable_path=ChromeDriverManager().install())
driver = webdriver.Chrome(service=service)
driver.get('http://www.softwarebox.club/pages/OnlineTools/AutoLearn')
driver.implicitly_wait(3)
driver.maximize_window()
driver.find_element(By.ID, 'username').send_keys('小明')
driver.find_element(By.ID, 'password').send_keys('123456Xm')
# 通过XPath定位登录按钮，并点击
driver.find_element(By.XPATH, '//*[@id="contentmain"]/section/div[2]/form/button[1]').click()
time.sleep(3)
assert driver.current_url == 'http://www.softwarebox.club/', '用例不通过'
print('用例通过')
上面的写法没有问题，只是测试方法与定位器耦合过于严重，维护起来很头疼，尤其是当UI发生变化的时候。
可以使用selenium官网推荐的最佳实践PO(page object)模式，就是将测试代码与页面定位代码分离。当页面元素发生变化时只需要改定位代码，测试用例本身是不需要更改的。
还是一样的登录功能，将上面的代码再重新设计，将页面的元素定位和该页面提供的方法放在独立的一个类中（Pages->LoginPage.py），登录的用例代码单独一个类（Testcase->LoginCase.py）
登录页面包含元素/方法：
1、元素定位：用户名、密码、登录按钮位置
2、页面的地址信息
3、登录方法
4、登录失败弹窗获取方法
LoginPage.py
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions
# 登录页面
class LoginPage(object):
def __init__(self, driver):
# 浏览器驱动
self.driver = driver
# 页面地址
self.driver.get('http://www.softwarebox.club/pages/OnlineTools/AutoLearn')
try:
# 用户名输入框
self.usernameBy = self.driver.find_element(By.ID, 'username')
# 密码输入框
self.passwordBy = self.driver.find_element(By.ID, 'password')
# 登录按钮
self.submitBy = self.driver.find_element(By.XPATH, '//*[@id="contentmain"]/section/div[2]/form/button[1]')
except Exception as e:
message = '元素定位失败'
print(e)
print(message)
# 登录功能
def login_valid(self, username, password):
# 输入用户名
self.usernameBy.send_keys(username)
# 输入密码
self.passwordBy.send_keys(password)
# 点击登录按钮
self.submitBy.click()
# 获取页面弹窗
def alert_message(self):
wait = WebDriverWait(self.driver, 10)
alert = wait.until(expected_conditions.alert_is_present())
# 返回弹窗和弹窗信息
return alert, alert.text登录的3个测试用例示例：
LoginCase.py
import time
from Pages.LoginPage import LoginPage
from PublicFunc.assertion import assertion_equal
from selenium import webdriver
from selenium.webdriver.chrome.service import Service as ChromeService
from webdriver_manager.chrome import ChromeDriverManager
class LoginCase:
def __init__(self):
service = ChromeService(executable_path=ChromeDriverManager().install())
self.driver = webdriver.Chrome(service=service)
def test_login_success(self):
case_description = '输入正确的用户名密码，登录成功，跳转到首页'
login = LoginPage(self.driver)
login.login_valid('小明', '123456Xm')
time.sleep(3)
assertion_equal(self.driver.current_url, 'http://www.softwarebox.club/', case_description)
def test_login_fail_username(self):
case_description = '输入的用户名不存在，系统提示用户不存在'
login = LoginPage(self.driver)
login.login_valid('小朱', '123')
time.sleep(3)
alert, message = login.alert_message()
assertion_equal(message, '用户不存在', case_description)
# 关闭弹窗
alert.accept()
def test_login_fail_password(self):
case_description = '输入的密码不正确，登录失败'
login = LoginPage(self.driver)
login.login_valid('小明', '123')
time.sleep(3)
alert, message = login.alert_message()
assertion_equal(message, '账号或密码不正确', case_description)
alert.accept()
def __str__(self):
print('**登录模块**')
# 执行用例
def run_test(self):
self.__str__()
self.test_login_success()
self.test_login_fail_username()
self.test_login_fail_password()
self.driver.quit()上面代码有个自己写的断言方法assertion_equal()，判断实际结果与预期是否一致，返回用例执行结果。
assertion.py
def assertion_equal(actual, expect, case_description='用例', success_msg='成功', fail_msg='失败'):
if actual == expect:
print(case_description + '--' + success_msg)
else:
print(case_description + '--' + fail_msg)执行用例，main.py
from TestCase.LoginCase import LoginCase
if __name__ == '__main__':
login_case = LoginCase()
login_case.run_test()运行结果：
完整代码地址：
https://github.com/spookypop/UiTeast.git