前言
有人問我登錄攜帶登錄的測試框架該怎么處理,今天就對框架做一點小升級吧,加入登錄的測試功能。
selenium自動化+ pytest測試框架禪道實戰
原文章
選用的測試網址為我電腦本地搭建的禪道
更改了以下的一些文件,框架為原文章框架主體
conftest.py更改
conftest.py
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import base64
import pytest
import allure
from py.xml import html
from selenium import webdriver
from page.webpage import WebPage
from common.readconfig import ini
from tools.send_mail import send_report
from tools.times import timestamp
from config.conf import cm
driver = None
@pytest.fixture(scope='session', autouse=True)
def drivers(request):
global driver
if driver is None:
driver = webdriver.Chrome()
web = WebPage(driver)
web.get_url(ini.url)
def fn():
driver.quit()
request.addfinalizer(fn)
return driver
@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item):
"""
當測試失敗的時候,自動截圖,展示到html報告中
:param item:
"""
pytest_html = item.config.pluginmanager.getplugin('html')
outcome = yield
report = outcome.get_result()
extra = getattr(report, 'extra', [])
if report.when == 'call' or report.when == "setup":
xfail = hasattr(report, 'wasxfail')
if (report.skipped and xfail) or (report.failed and not xfail):
screen_img = _capture_screenshot()
if screen_img:
html = '<div><img src="data:image/png;base64,%s" alt="screenshot" style="width:1024px;height:768px;" ' \
'onclick="window.open(this.src)" align="right"/></div>' % screen_img
extra.append(pytest_html.extras.html(html))
report.extra = extra
report.description = str(item.function.__doc__)
def pytest_html_results_table_header(cells):
cells.insert(1, html.th('用例名稱'))
cells.insert(2, html.th('Test_nodeid'))
cells.pop(2)
def pytest_html_results_table_row(report, cells):
cells.insert(1, html.td(report.description))
cells.insert(2, html.td(report.nodeid))
cells.pop(2)
def pytest_html_results_table_html(report, data):
if report.passed:
del data[:]
data.append(html.div('通過的用例未捕獲日志輸出.', class_='empty log'))
def pytest_html_report_title(report):
report.title = "pytest示例項目測試報告"
def pytest_configure(config):
config._metadata.clear()
config._metadata['測試項目'] = "測試百度官網搜索"
config._metadata['測試地址'] = ini.url
def pytest_html_results_summary(prefix, summary, postfix):
# prefix.clear() # 清空summary中的內容
prefix.extend([html.p("所屬部門: XX公司測試部")])
prefix.extend([html.p("測試執行人: 隨風揮手")])
def pytest_terminal_summary(terminalreporter, exitstatus, config):
"""收集測試結果"""
result = {
"total": terminalreporter._numcollected,
'passed': len(terminalreporter.stats.get('passed', [])),
'failed': len(terminalreporter.stats.get('failed', [])),
'error': len(terminalreporter.stats.get('error', [])),
'skipped': len(terminalreporter.stats.get('skipped', [])),
# terminalreporter._sessionstarttime 會話開始時間
'total times': timestamp() - terminalreporter._sessionstarttime
}
print(result)
if result['failed'] or result['error']:
send_report()
def _capture_screenshot():
"""截圖保存為base64"""
now_time, screen_path = cm.screen_file
driver.save_screenshot(screen_path)
allure.attach.file(screen_path, "測試失敗截圖...{}".format(
now_time), allure.attachment_type.PNG)
with open(screen_path, 'rb') as f:
imagebase64 = base64.b64encode(f.read())
return imagebase64.decode()
config.ini更改
[HOST]
HOST = http://127.0.0.1/zentao/user-login-L3plbnRhby9teS5odG1s.html
conf.py更改
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import os
from selenium.webdriver.common.by import By
from tools.times import datetime_strftime
class ConfigManager(object):
# 項目目錄
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# 日志目錄
LOG_PATH = os.path.join(BASE_DIR, 'logs')
# 報告目錄
REPORT_PATH = os.path.join(BASE_DIR, 'report', 'report.html')
ELEMENT_PATH = os.path.join(BASE_DIR, 'page_element')
# 元素定位的類型
LOCATE_MODE = {
'css': By.CSS_SELECTOR,
'xpath': By.XPATH,
'name': By.NAME,
'id': By.ID,
'class': By.CLASS_NAME
}
# 郵件信息
EMAIL_INFO = {
'username': '1084502012@qq.com', # 切換成你自己的地址
'password': 'QQ郵箱授權碼',
'smtp_host': 'smtp.qq.com',
'smtp_port': 465
}
# 收件人
ADDRESSEE = [
'1084502012@qq.com',
]
@property
def ini_file(self):
# 配置文件
_file = os.path.join(self.BASE_DIR, 'config', 'config.ini')
if not os.path.exists(_file):
raise FileNotFoundError("配置文件%s不存在!" % _file)
return _file
def element_file(self, name):
"""頁面元素文件"""
element_path = os.path.join(self.ELEMENT_PATH, '%s.yaml' % name)
if not os.path.exists(element_path):
raise FileNotFoundError("%s 文件不存在!" % element_path)
return element_path
@property
def log_path(self):
log_path = os.path.join(self.BASE_DIR, 'logs')
if not os.path.exists(log_path):
os.makedirs(log_path)
return os.path.join(log_path, "%s.log" % datetime_strftime())
@property
def screen_file(self):
now_time = datetime_strftime("%Y%m%d%H%M%S")
# 截圖目錄
screenshot_dir = os.path.join(self.BASE_DIR, 'screen_capture')
if not os.path.exists(screenshot_dir):
os.makedirs(screenshot_dir)
screen_path = os.path.join(screenshot_dir, "{}.png".format(now_time))
return now_time, screen_path
cm = ConfigManager()
if __name__ == '__main__':
print(cm.BASE_DIR)
page更改
webpage.py
添加了幾個函數!
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
"""
selenium基類
本文件存放了selenium基類的封裝方法
"""
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from selenium.common.exceptions import TimeoutException, NoSuchElementException
from config.conf import cm
from tools.times import sleep
from tools.logger import Logger
log = Logger(__name__).logger
class WebPage(object):
"""selenium基類"""
def __init__(self, driver):
# self.driver = webdriver.Chrome()
self.driver = driver
self.timeout = 20
self.wait = WebDriverWait(self.driver, self.timeout)
def get_url(self, url):
"""打開網址並驗證"""
self.driver.maximize_window()
self.driver.set_page_load_timeout(60)
try:
self.driver.get(url)
self.driver.implicitly_wait(10)
log.info("打開網頁:%s" % url)
except TimeoutException:
raise TimeoutException("打開%s超時請檢查網絡或網址服務器" % url)
@staticmethod
def element_locator(func, locator):
"""元素定位器"""
name, value = locator
return func(cm.LOCATE_MODE[name], value)
def find_element(self, locator):
"""尋找單個元素"""
return WebPage.element_locator(lambda *args: self.wait.until(
EC.presence_of_element_located(args)), locator)
def find_elements(self, locator):
"""查找多個相同的元素"""
return WebPage.element_locator(lambda *args: self.wait.until(
EC.presence_of_all_elements_located(args)), locator)
def focus(self):
"""聚焦元素"""
self.driver.execute_script("window.scrollTo(0,document.body.scrollHeight)")
def elements_num(self, locator):
"""獲取相同元素的個數"""
number = len(self.find_elements(locator))
log.info("相同元素:{}".format((locator, number)))
return number
def input_text(self, locator, txt):
"""輸入(輸入前先清空)"""
sleep(0.5)
ele = self.find_element(locator)
ele.clear()
ele.send_keys(txt)
log.info("輸入文本:{}".format(txt))
def is_click(self, locator):
"""點擊"""
ele = self.find_element(locator)
ele.click()
sleep()
log.info("點擊元素:{}".format(locator))
def is_exists(self, locator):
"""元素是否存在(DOM)"""
try:
WebPage.element_locator(lambda *args: EC.presence_of_element_located(args)(self.driver), locator)
return True
except NoSuchElementException:
return False
def alert_exists(self):
"""判斷彈框是否出現,並返回彈框的文字"""
alert = EC.alert_is_present()(self.driver)
if alert:
text = alert.text
log.info("Alert彈窗提示為:%s" % text)
alert.accept()
return text
else:
log.error("沒有Alert彈窗提示!")
def element_text(self, locator):
"""獲取當前的text"""
_text = self.find_element(locator).text
log.info("獲取文本:{}".format(_text))
return _text
def get_attribute(self, locator, name):
"""獲取元素屬性"""
return self.find_element(locator).get_attribute(name)
@property
def get_source(self):
"""獲取頁面源代碼"""
return self.driver.page_source
def refresh(self):
"""刷新頁面F5"""
self.driver.refresh()
self.driver.implicitly_wait(30)
if __name__ == "__main__":
pass
page_element更改
login.yaml
賬號: "css==input[name=account]"
密碼: "css==input[name=password]"
登錄: "css==button#submit"
我的地盤: "xpath==//nav[@id='navbar']//span[text()=' 我的地盤']"
右上角名稱: "css==.user-name"
退出登錄: "xpath==//a[text()='退出']"
product.yaml
產品按鈕: "xpath==//nav[@id='navbar']//a[text()='產品']"
添加產品: "xpath==//div[@id='pageActions']//a[text()=' 添加產品']"
產品名稱: "css==#name"
產品代號: "css==#code"
保存產品: "css==#submit"
產品列表: "xpath==//ul[@class='nav nav-stacked nav-secondary scrollbar-hover']//a[1]"
page_object更改
loginpage.py
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from page.webpage import WebPage
from common.readelement import Element
login = Element('login')
class LoginPage(WebPage):
"""登錄類"""
def username(self, name):
"""用戶名"""
self.input_text(login['賬號'], name)
def password(self, pwd):
"""密碼"""
self.input_text(login['密碼'], pwd)
def submit(self):
"""登錄"""
self.is_click(login['登錄'])
def quit_login(self):
"""退出登錄"""
self.is_click(login['右上角名稱'])
self.is_click(login['退出登錄'])
def login_success(self):
"""驗證登錄"""
return self.is_exists(login['我的地盤'])
productpage.py
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
from page.webpage import WebPage, sleep
from common.readelement import Element
product = Element('product')
class ProductPage(WebPage):
"""產品類"""
def click_product(self):
"""點擊產品"""
self.is_click(product['產品按鈕'])
def add_product(self):
"""添加產品"""
self.is_click(product['添加產品'])
def add_product_content(self, name, code):
"""添加產品內容"""
self.input_text(product['產品名稱'], name)
self.input_text(product['產品代號'], code)
def save_product(self):
"""保存產品"""
self.focus()
self.is_click(product['保存產品'])
def product_list(self):
"""產品列表"""
return [i.get_attribute('title') for i in self.find_elements(product['產品列表'])]
if __name__ == '__main__':
a = product['產品列表'][1] + "[1]"
print(a)
TestCase更改
test_login.py
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import pytest
from tools.times import sleep
from page_object.loginpage import LoginPage
class TestLogin:
"""測試登錄"""
@pytest.mark.parametrize("name,pwd", [('admin', 'Admin123456'), ('test', 'test123')])
def test_001(self, drivers, name, pwd):
login = LoginPage(drivers)
login.username(name)
login.password(pwd)
login.submit()
sleep(3)
res = login.alert_exists()
if res:
assert res == "登錄失敗,請檢查您的用戶名或密碼是否填寫正確。"
elif login.login_success():
login.quit_login()
test_product.py
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import pytest
import allure
from random import randint
from tools.times import sleep
from page_object.loginpage import LoginPage
from page_object.productpage import ProductPage
@allure.feature("測試產品模塊")
class TestProduct:
@pytest.fixture(scope='class', autouse=True)
def is_login(self, request, drivers):
login = LoginPage(drivers)
login.username('admin')
login.password('Admin123456')
login.submit()
sleep(3)
def logout():
login.quit_login()
request.addfinalizer(logout)
@allure.story("測試添加產品")
def test_001(self, drivers):
"""搜索"""
product = ProductPage(drivers)
product.click_product()
product.add_product()
name, code = randint(100, 999), randint(100, 999)
product.add_product_content(name, code)
product.save_product()
sleep(3)
product.click_product()
assert str(name) in product.product_list()
if __name__ == '__main__':
pytest.main(['TestCase/test_aproduct.py'])
測試結果
登錄之后的測試用例:

測試登錄的用例


開源地址
本次示例也開源在碼雲