Selenium+Pytest自动化测试框架

引言

Selenium+Pytest自动化测试框架是目前最流行的自动化测试工具之一，其强大的功能和易用性援助许多开发人员和测试人员。

当前最新消息显示，随着人们对软件质量的要求不断提高，自动化测试工作正在变得越来越重要。这也导致越来越多的公司和组织开始招聘具有自动化测试技能的人才，并愿意为其提供更高的薪酬和晋升机会。

在这种情况下，学习并掌握Selenium+Pytest自动化测试框架将成为一个非常有价值的能力，并且有望帮助你在职场上实现快速升职加薪的目标。

因此，如果你想迅速提高自己的技能并进一步发展职业生涯，请跟随我们的指引，学习Selenium+Pytest自动化测试框架，并使用它来实现高效的自动化测试工作！

前段时间有人问我登录携带登录的测试框架该怎么处理，今天就对框架做一点小升级吧，加入登录的测试功能。

selenium自动化+ pytest测试框架禅道实战

选用的测试网址为我电脑本地搭建的禅道

更改了以下的一些文件,框架为原文章框架主体

conftest.py更改

conftest.py

1. #!/usr/bin/env python3

2. # -*- coding:utf-8 -*-

3. import base64

4. import pytest

5. import allure

6. from py.xml import html

7. from selenium import webdriver

8. from page.webpage import WebPage

9. from common.readconfig import ini

10. from tools.send_mail import send_report

11. from tools.times import timestamp

12. from config.conf import cm

14. driver = None

17. @pytest.fixture(scope='session', autouse=True)

18. def drivers(request):

19. global driver

20. if driver is None:

21. driver = webdriver.Chrome()

22. web = WebPage(driver)

23. web.get_url(ini.url)

25. def fn():

26. driver.quit()

28. request.addfinalizer(fn)

29. return driver

32. @pytest.hookimpl(hookwrapper=True)

33. def pytest_runtest_makereport(item):

34. """

35. 当测试失败的时候，自动截图，展示到html报告中

36. :param item:

37. """

38. pytest_html = item.config.pluginmanager.getplugin('html')

39. outcome = yield

40. report = outcome.get_result()

41. extra = getattr(report, 'extra', [])

43. if report.when == 'call' or report.when == "setup":

44. xfail = hasattr(report, 'wasxfail')

45. if (report.skipped and xfail) or (report.failed and not xfail):

46. screen_img = _capture_screenshot()

47. if screen_img:

48. html = '<div><img src="data:image/png;base64,%s" alt="screenshot" style="width:1024px;height:768px;" ' \

49. 'onclick="window.open(this.src)" align="right"/></div>' % screen_img

50. extra.append(pytest_html.extras.html(html))

51. report.extra = extra

52. report.description = str(item.function.__doc__)

55. def pytest_html_results_table_header(cells):

56. cells.insert(1, html.th('用例名称'))

57. cells.insert(2, html.th('Test_nodeid'))

58. cells.pop(2)

61. def pytest_html_results_table_row(report, cells):

62. cells.insert(1, html.td(report.description))

63. cells.insert(2, html.td(report.nodeid))

64. cells.pop(2)

67. def pytest_html_results_table_html(report, data):

68. if report.passed:

69. del data[:]

70. data.append(html.div('通过的用例未捕获日志输出.', class_='empty log'))

73. def pytest_html_report_title(report):

74. report.title = "pytest示例项目测试报告"

77. def pytest_configure(config):

78. config._metadata.clear()

79. config._metadata['测试项目'] = "测试百度官网搜索"

80. config._metadata['测试地址'] = ini.url

83. def pytest_html_results_summary(prefix, summary, postfix):

84. # prefix.clear() # 清空summary中的内容

85. prefix.extend([html.p("所属部门: XX公司测试部")])

86. prefix.extend([html.p("测试执行人: 随风挥手")])

89. def pytest_terminal_summary(terminalreporter, exitstatus, config):

90. """收集测试结果"""

91. result = {

92. "total": terminalreporter._numcollected,

93. 'passed': len(terminalreporter.stats.get('passed', [])),

94. 'failed': len(terminalreporter.stats.get('failed', [])),

95. 'error': len(terminalreporter.stats.get('error', [])),

96. 'skipped': len(terminalreporter.stats.get('skipped', [])),

97. # terminalreporter._sessionstarttime 会话开始时间

98. 'total times': timestamp() - terminalreporter._sessionstarttime

99. }

100. print(result)

101. if result['failed'] or result['error']:

102. send_report()

105. def _capture_screenshot():

106. """截图保存为base64"""

107. now_time, screen_path = cm.screen_file

108. driver.save_screenshot(screen_path)

109. allure.attach.file(screen_path, "测试失败截图...{}".format(

110. now_time), allure.attachment_type.PNG)

111. with open(screen_path, 'rb') as f:

112. imagebase64 = base64.b64encode(f.read())

113. return imagebase64.decode()

 config.ini更改

1. [HOST]

2. HOST = http://127.0.0.1/zentao/user-login-L3plbnRhby9teS5odG1s.html

conf.py更改

1. #!/usr/bin/env python3

2. # -*- coding:utf-8 -*-

3. import os

4. from selenium.webdriver.common.by import By

5. from tools.times import datetime_strftime

8. class ConfigManager(object):

9. # 项目目录

10. BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

12. # 日志目录

13. LOG_PATH = os.path.join(BASE_DIR, 'logs')

15. # 报告目录

16. REPORT_PATH = os.path.join(BASE_DIR, 'report', 'report.html')

18. ELEMENT_PATH = os.path.join(BASE_DIR, 'page_element')

20. # 元素定位的类型

21. LOCATE_MODE = {

22. 'css': By.CSS_SELECTOR,

23. 'xpath': By.XPATH,

24. 'name': By.NAME,

25. 'id': By.ID,

26. 'class': By.CLASS_NAME

27. }

29. # 邮件信息

30. EMAIL_INFO = {

31. 'username': '1084502012@qq.com', # 切换成你自己的地址

32. 'password': 'QQ邮箱授权码',

33. 'smtp_host': 'smtp.qq.com',

34. 'smtp_port': 465

35. }

37. # 收件人

38. ADDRESSEE = [

39. '1084502012@qq.com',

40. ]

42. @property

43. def ini_file(self):

44. # 配置文件

45. _file = os.path.join(self.BASE_DIR, 'config', 'config.ini')

46. if not os.path.exists(_file):

47. raise FileNotFoundError("配置文件%s不存在！" % _file)

48. return _file

50. def element_file(self, name):

51. """页面元素文件"""

52. element_path = os.path.join(self.ELEMENT_PATH, '%s.yaml' % name)

53. if not os.path.exists(element_path):

54. raise FileNotFoundError("%s 文件不存在！" % element_path)

55. return element_path

57. @property

58. def log_path(self):

59. log_path = os.path.join(self.BASE_DIR, 'logs')

60. if not os.path.exists(log_path):

61. os.makedirs(log_path)

62. return os.path.join(log_path, "%s.log" % datetime_strftime())

64. @property

65. def screen_file(self):

66. now_time = datetime_strftime("%Y%m%d%H%M%S")

67. # 截图目录

68. screenshot_dir = os.path.join(self.BASE_DIR, 'screen_capture')

69. if not os.path.exists(screenshot_dir):

70. os.makedirs(screenshot_dir)

71. screen_path = os.path.join(screenshot_dir, "{}.png".format(now_time))

72. return now_time, screen_path

75. cm = ConfigManager()

76. if __name__ == '__main__':

77. print(cm.BASE_DIR)

page更改

webpage.py

添加了几个函数！

1. #!/usr/bin/env python3

2. # -*- coding:utf-8 -*-

3. """

4. selenium基类

5. 本文件存放了selenium基类的封装方法

6. """

7. from selenium.webdriver.support import expected_conditions as EC

8. from selenium.webdriver.support.ui import WebDriverWait

9. from selenium.common.exceptions import TimeoutException, NoSuchElementException

10. from config.conf import cm

11. from tools.times import sleep

12. from tools.logger import Logger

14. log = Logger(__name__).logger

17. class WebPage(object):

18. """selenium基类"""

20. def __init__(self, driver):

21. # self.driver = webdriver.Chrome()

22. self.driver = driver

23. self.timeout = 20

24. self.wait = WebDriverWait(self.driver, self.timeout)

26. def get_url(self, url):

27. """打开网址并验证"""

28. self.driver.maximize_window()

29. self.driver.set_page_load_timeout(60)

30. try:

31. self.driver.get(url)

32. self.driver.implicitly_wait(10)

33. log.info("打开网页：%s" % url)

34. except TimeoutException:

35. raise TimeoutException("打开%s超时请检查网络或网址服务器" % url)

37. @staticmethod

38. def element_locator(func, locator):

39. """元素定位器"""

40. name, value = locator

41. return func(cm.LOCATE_MODE[name], value)

43. def find_element(self, locator):

44. """寻找单个元素"""

45. return WebPage.element_locator(lambda *args: self.wait.until(

46. EC.presence_of_element_located(args)), locator)

48. def find_elements(self, locator):

49. """查找多个相同的元素"""

50. return WebPage.element_locator(lambda *args: self.wait.until(

51. EC.presence_of_all_elements_located(args)), locator)

53. def focus(self):

54. """聚焦元素"""

55. self.driver.execute_script("window.scrollTo(0,document.body.scrollHeight)")

57. def elements_num(self, locator):

58. """获取相同元素的个数"""

59. number = len(self.find_elements(locator))

60. log.info("相同元素：{}".format((locator, number)))

61. return number

63. def input_text(self, locator, txt):

64. """输入(输入前先清空)"""

65. sleep(0.5)

66. ele = self.find_element(locator)

67. ele.clear()

68. ele.send_keys(txt)

69. log.info("输入文本：{}".format(txt))

71. def is_click(self, locator):

72. """点击"""

73. ele = self.find_element(locator)

74. ele.click()

75. sleep()

76. log.info("点击元素：{}".format(locator))

78. def is_exists(self, locator):

79. """元素是否存在(DOM)"""

80. try:

81. WebPage.element_locator(lambda *args: EC.presence_of_element_located(args)(self.driver), locator)

82. return True

83. except NoSuchElementException:

84. return False

86. def alert_exists(self):

87. """判断弹框是否出现，并返回弹框的文字"""

88. alert = EC.alert_is_present()(self.driver)

89. if alert:

90. text = alert.text

91. log.info("Alert弹窗提示为：%s" % text)

92. alert.accept()

93. return text

94. else:

95. log.error("没有Alert弹窗提示!")

97. def element_text(self, locator):

98. """获取当前的text"""

99. _text = self.find_element(locator).text

100. log.info("获取文本：{}".format(_text))

101. return _text

103. def get_attribute(self, locator, name):

104. """获取元素属性"""

105. return self.find_element(locator).get_attribute(name)

107. @property

108. def get_source(self):

109. """获取页面源代码"""

110. return self.driver.page_source

112. def refresh(self):

113. """刷新页面F5"""

114. self.driver.refresh()

115. self.driver.implicitly_wait(30)

118. if __name__ == "__main__":

119. pass

page_element更改

login.yaml

1. 账号: "css==input[name=account]"

2. 密码: "css==input[name=password]"

3. 登录: "css==button#submit"

4. 我的地盘: "xpath==//nav[@id='navbar']//span[text()=' 我的地盘']"

5. 右上角名称: "css==.user-name"

6. 退出登录: "xpath==//a[text()='退出']"

product.yaml

1. 产品按钮: "xpath==//nav[@id='navbar']//a[text()='产品']"

2. 添加产品: "xpath==//div[@id='pageActions']//a[text()=' 添加产品']"

3. 产品名称: "css==#name"

4. 产品代号: "css==#code"

5. 保存产品: "css==#submit"

6. 产品列表: "xpath==//ul[@class='nav nav-stacked nav-secondary scrollbar-hover']//a[1]"

page_object更改

loginpage.py

1. #!/usr/bin/env python3

2. # -*- coding: utf-8 -*-

3. from page.webpage import WebPage

4. from common.readelement import Element

6. login = Element('login')

9. class LoginPage(WebPage):

10. """登录类"""

12. def username(self, name):

13. """用户名"""

14. self.input_text(login['账号'], name)

16. def password(self, pwd):

17. """密码"""

18. self.input_text(login['密码'], pwd)

20. def submit(self):

21. """登录"""

22. self.is_click(login['登录'])

24. def quit_login(self):

25. """退出登录"""

26. self.is_click(login['右上角名称'])

27. self.is_click(login['退出登录'])

29. def login_success(self):

30. """验证登录"""

31. return self.is_exists(login['我的地盘'])

productpage.py

1. #!/usr/bin/env python3

2. # -*- coding:utf-8 -*-

3. from page.webpage import WebPage, sleep

4. from common.readelement import Element

6. product = Element('product')

9. class ProductPage(WebPage):

10. """产品类"""

12. def click_product(self):

13. """点击产品"""

14. self.is_click(product['产品按钮'])

16. def add_product(self):

17. """添加产品"""

18. self.is_click(product['添加产品'])

20. def add_product_content(self, name, code):

21. """添加产品内容"""

22. self.input_text(product['产品名称'], name)

23. self.input_text(product['产品代号'], code)

25. def save_product(self):

26. """保存产品"""

27. self.focus()

28. self.is_click(product['保存产品'])

30. def product_list(self):

31. """产品列表"""

32. return [i.get_attribute('title') for i in self.find_elements(product['产品列表'])]

35. if __name__ == '__main__':

36. a = product['产品列表'][1] + "[1]"

37. print(a)

TestCase更改

test_login.py

1. #!/usr/bin/env python3

2. # -*- coding: utf-8 -*-

3. import pytest

4. from tools.times import sleep

5. from page_object.loginpage import LoginPage

8. class TestLogin:

9. """测试登录"""

11. @pytest.mark.parametrize("name,pwd", [('admin', 'Admin123456'), ('test', 'test123')])

12. def test_001(self, drivers, name, pwd):

13. login = LoginPage(drivers)

14. login.username(name)

15. login.password(pwd)

16. login.submit()

17. sleep(3)

18. res = login.alert_exists()

19. if res:

20. assert res == "登录失败，请检查您的用户名或密码是否填写正确。"

21. elif login.login_success():

22. login.quit_login()

test_product.py

1. #!/usr/bin/env python3

2. # -*- coding:utf-8 -*-

3. import pytest

4. import allure

5. from random import randint

6. from tools.times import sleep

7. from page_object.loginpage import LoginPage

8. from page_object.productpage import ProductPage

11. @allure.feature("测试产品模块")

12. class TestProduct:

14. @pytest.fixture(scope='class', autouse=True)

15. def is_login(self, request, drivers):

16. login = LoginPage(drivers)

17. login.username('admin')

18. login.password('Admin123456')

19. login.submit()

20. sleep(3)

22. def logout():

23. login.quit_login()

25. request.addfinalizer(logout)

27. @allure.story("测试添加产品")

28. def test_001(self, drivers):

29. """搜索"""

30. product = ProductPage(drivers)

31. product.click_product()

32. product.add_product()

33. name, code = randint(100, 999), randint(100, 999)

34. product.add_product_content(name, code)

35. product.save_product()

36. sleep(3)

37. product.click_product()

38. assert str(name) in product.product_list()

41. if __name__ == '__main__':

42. pytest.main(['TestCase/test_aproduct.py'])

测试结果

登录之后的测试用例:

测试登录的用例

总结：

感谢每一个认真阅读我文章的人！！！

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

 

          视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。