轻松实现一个Python+Selenium的自动化测试框架

首先你得知道什么是Selenium？

Selenium是一个基于浏览器的自动化测试工具，它提供了一种跨平台、跨浏览器的端到端的web自动化解决方案。Selenium主要包括三部分：Selenium IDE、Selenium WebDriver 和Selenium Grid。

• Selenium IDE：Firefox的一个扩展，它可以进行录制回放，并把录制的操作以多种语言（例如java、python等）的形式导出成测试用例。
• Selenium WebDriver：提供Web自动化所需的API，主要用作浏览器控制、页面元素选择和调试。不同的浏览器需要不同的WebDriver。
• Selenium Grid：提供了在不同机器的不同浏览器上运行selenium测试的能力。

下面我会使用思维导图目录结构介绍基础测试框架，编写测试用例进行功能测试用例，希望对你的学习有所帮助。

设计思路

框架采用python3 + selenium3 + PO + yaml + ddt + unittest等技术编写成基础测试框架，能适应日常测试工作需要。

1、使用Page Object模式将页面定位和业务操作分开，分离测试对象（元素对象）和测试脚本（用例脚本），一个页面建一个对象类，提高用例的可维护性；

2、使用yaml管理页面控件元素数据和测试用例数据。例如元素ID等发生变化时，不需要去修改测试代码，只需要在对应的页面元素yaml文件中修改即可；

3、分模块管理，互不影响，随时组装，即拿即用。

测试框架分层设计

把常见的操作和查找封装成基础类，不管是什么产品，可直接拿来复用

• 业务层主要是封装对象页面类，一个页面建一个类，业务层页面继承基础层
• 用例层针对产品页面功能进行构造模拟执行测试
• 框架层提供基础组件，支撑整个流程执行及功能扩展，给用例层提供各页面的元素数据、用例测试数据，测试报告输出等

测试框架目录结构

如下思维导图目录结构介绍：

编写用例方法

如果对软件测试、接口测试、自动化测试、面试经验交流。感兴趣可以关注我个人公众号：程序员一凡，公众号内会有不定期的发放免费的资源链接，这些资料都是从各个技术网站搜集、整理出来的，如果你有好的学习资料可以私聊发我，我会注明出处之后分享给大家 

1. testinfo:

2. - id: test_login001

3. title: 登录测试

4. info: 打开抽屉首页

5. testcase:

6. - element_info: login-link-a

7. find_type: ID

8. operate_type: click

9. info: 打开登录对话框

10. - element_info: mobile

11. find_type: ID

12. operate_type: send_keys

13. info: 输入手机号

14. - element_info: mbpwd

15. find_type: ID

16. operate_type: send_keys

17. info: 输入密码

18. - element_info: //input[@class='keeplogin']

19. find_type: XPATH

20. operate_type: click

21. info: 单击取消自动登录单选框

22. - element_info: //span[text()='登录']

23. find_type: XPATH

24. operate_type: click

25. info: 单击登录按钮

26. - element_info: userProNick

27. find_type: ID

28. operate_type: perform

29. info: 鼠标悬停账户菜单

30. - element_info: //a[@class='logout']

31. find_type: XPATH

32. operate_type: click

33. info: 选择退出

34. check:

35. - element_info: //div[@class='box-mobilelogin']

36. /div[1]/span

37. find_type: XPATH

38. info: 检查输入手机号或密码，登录异常提示

39. - element_info: userProNick

40. find_type: ID

41. info: 成功登录

42. - element_info: reg-link-a

43. find_type: ID

44. info: 检查退出登录是否成功

45. login.yaml

 例如，我们要新增登录功能测试用例：

首先，只需在testyaml目录下新增一个页面对象yaml文件，参考login.yaml格式编写即可。这些文件是提供给封装页面对象类调用并执行定位识别操作。

1. -

2. id: test_login001.1

3. detail : 手机号和密码为空登录

4. screenshot : phone_pawd_empty

5. data:

6. phone: ""

7. password: ""

8. check :

9. - 手机号不能为空

10. -

11. id: test_login001.2

12. detail : 手机号为空登录

13. screenshot : phone_empty

14. data :

15. phone: ""

16. password : aa

17. check :

18. - 手机号不能为空

19. -

20. id: test_login001.3

21. detail : 密码为空登录

22. screenshot : pawd_empty

23. data :

24. phone : 13511112222

25. password: ""

26. check :

27. - 密码不能为空

28. -

29. id: test_login001.4

30. detail : 非法手机号登录

31. screenshot : phone_error

32. data :

33. phone : abc

34. password: aa

35. check :

36. - 手机号格式不对

37. -

38. id: test_login001.5

39. detail : 手机号或密码不匹配

40. screenshot : pawd_error

41. data :

42. phone : 13511112222

43. password: aa

44. check :

45. - 账号密码错误

46. -

47. id: test_login001.6

48. detail : 手机号和密码正确

49. screenshot : phone_pawd_success

50. data :

51. phone : 13865439800

52. password: ********

53. check :

54. - yingoja

55. login_data.yaml

57. login_data.yaml

其次，在testdata目录下新增一个login_data.yaml文件提供给登录接口传参的测试数据，编写格式参考login_data.yaml文件。

1. #!/usr/bin/env python

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

3. __author__ = 'YinJia'

4. import os,sys

5. sys.path.append(os.path.dirname(os.path.dirname

6. (os.path.dirname(__file__))))

7. from config import setting

8. from selenium.webdriver.support.select import Select

9. from selenium.webdriver.common.action_chains

10. import ActionChains

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

12. from public.page_obj.base import Page

13. from time import sleep

14. from public.models.GetYaml import getyaml

15. testData = getyaml(setting.TEST_Element_YAML

16. + '/' + 'login.yaml')

17. class login(Page):

18. """

19. 用户登录页面

20. """

21. url = '/'

22. dig_login_button_loc = (By.ID, testData.

23. get_elementinfo(0)) def dig_login(self):

24. """

25. 首页登录

26. :return:

27. """

28. self.find_element(*self.dig_login_button_loc)

29. .click() sleep(1)

30. # 定位器，通过元素属性定位元素对象

31. # 手机号输入框

32. login_phone_loc = (By.ID,testData.

33. get_elementinfo(1)) # 密码输入框

34. login_password_loc = (By.ID,testData.

35. get_elementinfo(2)) # 取消自动登录

36. keeplogin_button_loc = (By.XPATH,testData.

37. get_elementinfo(3)) # 单击登录

38. login_user_loc = (By.XPATH,testData.

39. get_elementinfo(4)) # 退出登录

40. login_exit_loc = (By.ID, testData.

41. get_elementinfo(5)) # 选择退出

42. login_exit_button_loc = (By.XPATH,testData.

43. get_elementinfo(6))def login_phone(self,phone):

44. """

45. 登录手机号

46. :param username:

47. :return:

48. """

49. self.find_element(*self.login_phone_loc).

50. send_keys(phone)def login_password(self,password):

51. """

52. 登录密码

53. :param password:

54. :return:

55. """

56. self.find_element(*self.login_password_loc).

57. send_keys(password) def keeplogin(self):

58. """

59. 取消单选自动登录

60. :return:

61. """

62. self.find_element(*self.keeplogin_button_loc).

63. click()def login_button(self):

64. """

65. 登录按钮

66. :return:

67. """

68. self.find_element(*self.login_user_loc).click()

69. def login_exit(self):

70. """

71. 退出系统

72. :return:

73. """

74. above = self.find_element(*self.login_exit_loc)

75. ActionChains(self.driver).move_to_element(above).

76. perform() sleep(2)

77. self.find_element(*self.login_exit_button_loc)

78. .click()def user_login(self,phone,password):

79. """

80. 登录入口

81. :param username: 用户名

82. :param password: 密码

83. :return:

84. """

85. self.open()

86. self.dig_login()

87. self.login_phone(phone)

88. self.login_password(password)

89. sleep(1)

90. self.keeplogin()

91. sleep(1)

92. self.login_button()

93. sleep(1)

94. phone_pawd_error_hint_loc = (By.XPATH,testData.

95. get_CheckElementinfo(0))

96. user_login_success_loc = (By.ID,testData.

97. get_CheckElementinfo(1))

98. exit_login_success_loc = (By.ID,testData.

99. get_CheckElementinfo(2))

100. # 手机号或密码错误提示

101. def phone_pawd_error_hint(self):

102. return self.find_element(*self.phone_pawd_error_

103. hint_loc).text# 登录成功用户名

104. def user_login_success_hint(self):

105. return self.find_element(*self.user_login_

106. success_loc).text # 退出登录

107. def exit_login_success_hint(self):

108. return self.find_element(*self.exit_login_

109. success_loc).textloginPage.py

然后，在page_obj目录下新增一个loginPage.py文件，是用来封装登录页面对象类，执行登录测试流程操作。

1. #!/usr/bin/env python

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

3. __author__ = 'YinJia'

4. import os,sys

5. sys.path.append(os.path.dirname(os.path.

6. dirname(__file__)))

7. import unittest,ddt,yaml

8. from config import setting

9. from public.models import myunit,screenshot

10. from public.page_obj.loginPage import login

11. from public.models.log import Log

12. try:

13. f =open(setting.TEST_DATA_YAML + '/' + 'login_data.yaml',encoding='utf-8')

14. testData = yaml.load(f)

15. except FileNotFoundError as file:

16. log = Log()

17. log.error("文件不存在：{0}".format(file))

18. @ddt.ddt

19. class Demo_UI(myunit.MyTest):

20. """抽屉新热榜登录测试"""

21. def user_login_verify(self,phone,password):

22. """

23. 用户登录

24. :param phone: 手机号

25. :param password: 密码

26. :return:

27. """

28. login(self.driver).user_login(phone,password)

29. def exit_login_check(self):

30. """

31. 退出登录

32. :return:

33. """

34. login(self.driver).login_exit()

35. @ddt.data(*testData)

36. def test_login(self,datayaml):

37. """

38. 登录测试

39. :param datayaml: 加载login_data登录测试数据

40. :return:

41. """

42. log = Log()

43. log.info("当前执行测试用例ID-> {0} ; 测试点-> {1}".format(datayaml['id'],datayaml['detail']))

44. # 调用登录方法

45. self.user_login_verify(datayaml['data']['phone'],

46. datayaml['data']['password'])

47. po = login(self.driver)

48. if datayaml['screenshot'] == 'phone_pawd_success':

49. log.info("检查点-> {0}".format

50. (po.user_login_success_hint()))

51. self.assertEqual(po.user_login_success_hint(), datayaml['check'][0], "成功登录，返回实际结果是->: {0}".format(po.user_login_success_hint()))

52. log.info("成功登录，返回实际结果是->: {0}".format(po.user_login_success_hint()))

53. screenshot.insert_img(self.driver, datayaml

54. ['screenshot'] + '.jpg')

55. log.info("-----> 开始执行退出流程操作")

56. self.exit_login_check()

57. po_exit = login(self.driver)

58. log.info("检查点-> 找到{0}元素,表示退出成功！".format(po_exit.exit_login_success_hint()))

59. self.assertEqual(po_exit.exit_login_success_hint(),

60. '注册',"退出登录，返回实际结果是->: {0}".format(po_exit.exit_login_success_hint()))

61. log.info("退出登录，返回实际结果是->: {0}".format(po_exit.exit_login_success_hint()))

62. else:

63. log.info("检查点-> {0}".format(po.phone

64. _pawd_error_hint()))

65. self.assertEqual(po.phone_pawd_error_hint(),

66. datayaml['check'][0] , "异常登录，返回实际结果是->: {0}".format(po.phone_pawd_error_hint()))

67. log.info("异常登录，返回实际结果是->: {0}".format(po.phone_pawd_error_hint()))

68. screenshot.insert_img(self.driver,datayaml

69. ['screenshot'] + '.jpg')

70. if __name__=='__main__':

71. unittest.main()

72. login_sta.py

最后，在testcase目录下创建测试用例文件login_sta.py，采用ddt数据驱动读取yaml测试数据文件

综上所述，编写用例方法只需要按以上四个步骤创建->编写即可。

执行如下主程序，可看输出的实际结果。

1. #!/usr/bin/env python

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

3. __author__ = 'YinJia' import os,sys

4. sys.path.append(os.path.dirname(__file__))

5. from config import setting

6. import unittest,time

7. from package.HTMLTestRunner import HTMLTestRunner

8. from public.models.newReport import new_report

9. from public.models.sendmail import send_mail

10. # 测试报告存放文件夹，如不存在，则自动创建

11. 一个report目录 if not os.path.exists(setting.TEST_REPORT):os.makedirs

12. (setting.TEST_REPORT + '/' + "screenshot")

13. def add_case(test_path=setting.TEST_DIR):

14. """加载所有的测试用例"""

15. discover = unittest.defaultTestLoader.discover

16. (test_path, pattern='*_sta.py')

17. return discover

18. def run_case(all_case,result_path=setting.TEST_REPORT):

19. """执行所有的测试用例"""

20. now = time.strftime("%Y-%m-%d %H_%M_%S")

21. filename = result_path + '/' + now + 'result.html'

22. fp = open(filename,'wb')

23. runner = HTMLTestRunner(stream=fp,title='

24. 抽屉新热榜UI自动化测试报告',

25. description='环境：windows 7 浏览器：chrome',

26. tester='Jason')

27. runner.run(all_case)

28. fp.close()

29. report = new_report(setting.TEST_REPORT)

30. #调用模块生成最新的报告

31. send_mail(report) #调用发送邮件模块

32. if __name__ =="__main__":

33. cases = add_case()

34. run_case(cases)

测试结果展示

HTML报告日志

HTML报告点击截图，弹出截图 

测试报告通过的日志

自动截图存放指定的目录

邮件测试报告

软件测试学习资料获取关注公众号：程序员雷叔