python+selenium+unittest自动化测试框架

前言

关于自动化测试的介绍，网上已有很多资料，这里不再赘述，UI自动化测试是自动化测试的一种，也是测试金字塔最上面的一层，selenium是应用于web的自动化测试工具，支持多平台、多浏览器、多语言来实现自动化，优点如下：

①开源、免费且对web界面有良好的支持；；

②多浏览器支持：chrome、Firefox、IE、Edge等；

③多平台支持：Linux、Windows、MAC；

④多语言支持：java、python、Ruby、C#、JavaScript、C++；

⑤简单（API简单）、灵活（开发语言驱动），支持分布式测试用例执行；

环境准备

• 浏览器（Chrome）
• Python
• Selenium
• chromedriver（请下载于Chrome对应版本的驱动）
• IDE（Pycharm）

1、python、pycharm安装

2、selenium安装

pip install selenium ​

关于selenium元素定位是开展web自动化测试的基础和关键，总共有八种定位方式，包括id，class，xpath等

3、chromedriver驱动安装

驱动下载：http://chromedriver.storage.googleapis.com/index.html

然后把chromedriver.exe拷贝到到chrome的安装目录下…\Google\Chrome\Application\ ，同时把chrome的安装目录加入到path环境变量。

4、快速验证测试

自动化测试框架

• config目录中存放的是测试配置相关的文件，文件类型ini，包括测试的网址、浏览器驱动等信息
• framework目录中存放的是页面基础类base_page: 封装一些常用的页面操作方法；日志类 Logger: 封装日志输出及控制台输出方法
• logs用来存放输出的日志文件
• pageobjects目录存放的是页面对象，一个页面封装为一个类，一个元素封装为一个方法
• screenshots目录存放的是测试过程中的相关截图
• test_report用来存放输出的测试报告
• testsuites目录存放测试用例，包括test_base和单个测试用例

1、config.ini配置信息如下：

2、framework目录下base_page的封装：

2. import time

4. from selenium.webdriver.common.action_chains import ActionChains

6. from selenium.webdriver.support.wait import WebDriverWait

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

10. import os.path

12. from framework.logger import Logger

15. # 创建一个日志实例

17. logger = Logger(logger="BasePage").getlog()

20. class BasePage(object):

22. """

24. 定义一个页面基类，让所有页面都继承这个类，封装一些常用的页面操作方法

26. """

28. def __init__(self, driver):

30. self.driver = driver

33. #get an url link

35. def open(self,url):

37. self.driver.get(url)

40. #quit browser and end testing

42. def quit_browser(self):

44. self.driver.quit()

47. # 浏览器前进操作

49. def forward(self):

51. self.driver.forward()

53. logger.info("Click forward on current page.")

56. # 浏览器后退操作

58. def back(self):

60. self.driver.back()

62. logger.info("Click back on current page.")

65. # 显示等待

67. def wait(self,loc,seconds):

69. try:

71. wait_=WebDriverWait(self.driver,seconds)

73. wait_.until(lambda driver:driver.find_element(*loc))

75. logger.info("wait for %d seconds." % seconds)

77. except NameError as e:

79. logger.error("Failed to load the element with %s" % e)

82. # 保存图片

84. def get_windows_img(self):

86. """

88. 把file_path保存到我们项目根目录的一个文件夹.\Screenshots下

90. """

92. file_path = os.path.dirname(os.path.abspath('.')) + 'Selenium/screenshots/'

94. rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))

96. screen_name = file_path + rq + '.png'

98. try:

100. self.driver.get_screenshot_as_file(screen_name)

102. logger.info("Had take screenshot and save to folder : /screenshots")

104. except NameError as e:

106. logger.error("Failed to take screenshot! %s" % e)

108. self.get_windows_img()

111. # 定位元素方法

113. def find_element(self,loc):

115. """

117. :return: element

119. """

121. return self.driver.find_element(*loc)

124. # 输入

126. def send_keys(self, selector, text):

129. el = self.find_element(selector)

131. el.clear()

133. try:

135. el.send_keys(text)

137. logger.info("Had type \' %s \' in inputBox" % text)

139. except NameError as e:

141. logger.error("Failed to select in input box with %s" % e)

143. self.get_windows_img()

146. # 清除文本框

148. def clear(self, selector):

151. el = self.find_element(selector)

153. try:

155. el.clear()

157. logger.info("Clear text in input box before typing.")

159. except NameError as e:

161. logger.error("Failed to clear in input box with %s" % e)

163. self.get_windows_img()

166. # 点击元素

168. def click(self, selector):

171. el = self.find_element(selector)

173. try:

175. el.click()

177. logger.info("The element \'%s\' was clicked." % el.text)

179. except NameError as e:

181. logger.error("Failed to click the element with %s" % e)

185. #鼠标事件（左键点击）

187. def move_element(self,loc,sloc):

189. mouse=self.find_element(loc)

191. try:

193. ActionChains(self.driver).move_to_element(mouse).perform()

195. self.click(sloc)

197. pass

199. except Exception as e:

201. logger.error("Failed to click move_element with %s" % e)

203. self.get_windows_img()

206. # 强制等待

208. @staticmethod

210. def sleep(seconds):

212. time.sleep(seconds)

214. logger.info("Sleep for %d seconds" % seconds)

 logger日志输入方法：

2. import logging

4. import os.path

6. import time

9. class Logger(object):

11. def __init__(self, logger):

13. '''''

15. 指定保存日志的文件路径，日志级别，以及调用文件

17. 将日志存入到指定的文件中

19. '''

21. # 创建一个logger

23. self.logger = logging.getLogger(logger)

25. self.logger.setLevel(logging.DEBUG)

28. # 创建一个handler，用于写入日志文件

30. rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))

32. log_path = os.path.dirname(os.path.abspath('.')) + 'Selenium/logs/'

34. log_name = log_path + rq + '.log'

36. fh = logging.FileHandler(log_name)

38. fh.setLevel(logging.INFO)

41. # 再创建一个handler，用于输出到控制台

43. ch = logging.StreamHandler()

45. ch.setLevel(logging.INFO)

48. # 定义handler的输出格式

50. formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

52. fh.setFormatter(formatter)

54. ch.setFormatter(formatter)

57. # 给logger添加handler

59. self.logger.addHandler(fh)

61. self.logger.addHandler(ch)

64. def getlog(self):

66. return self.logger

3、关于页面对象的写法参考：

页面对象继承于base_page基类

2. from framework.base_page import BasePage

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

6. from framework.logger import Logger

9. logger = Logger(logger="cloud_page").getlog()

11. class Cloud(BasePage):

13. #定位器

15. input_box = (By.ID,'kw')

17. search_submit = (By.XPATH,'//*[@id="su"]')

20. def value_input(self, text):

22. self.wait(self.input_box,5)

24. self.send_keys(self.input_box, text)

27. def submit_btn(self):

29. self.click(self.search_submit)

31. logger.info("show results!")

33. self.sleep(2)

4、testsuite部分，每个用例都要执行的用例前的准备setup和清理teardown写在test_base文件里，单个测试用例文件继承于它

2. from selenium import webdriver

4. import unittest

7. class TestBase(unittest.TestCase):

9. def setUp(self):

11. self.driver=webdriver.Chrome() #驱动浏览器

13. self.driver.implicitly_wait(10) #设置隐式等待

15. self.driver.maximize_window() #最大化浏览器

18. def tearDown(self):

20. self.driver.quit()

23. if __name__=='__main__':

25. unittest.main()

测试用例参考，注意测试用例方法命名一定要以test开头

2. import time

4. from testsuites.test_base import TestBase

6. from pageobjects.baidu_page import Cloud

9. class BaiduSearch(TestBase):

12. def test_baidu_search(self):

14. """

16. 这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。

18. :return:

20. """

22. input = Cloud(self.driver)

24. input.open('https://www.baidu.com/')

26. input.value_input('selenium') # 调用页面对象中的方法

28. input.submit_btn() # 调用页面对象类中的点击搜索按钮方法

30. time.sleep(2)

32. input.get_windows_img() # 调用基类截图方法

34. try:

36. assert 'selenium' in 'selenium'

38. print('Test Pass.')

40. except Exception as e:

42. print('Test Fail.', format(e))

5、最后就是核心代码，testrunner.py 用来批量执行测试用例，引用HTMLTestRunner模块生成测试报告。

2. import unittest

4. from HTMLTestRunner import HTMLTestRunner

6. import time

8. import os

11. # 定义输出的文件位置和名字

13. DIR = os.path.dirname(os.path.abspath(__file__))

15. now = time.strftime("%Y%m%d%H%M", time.localtime(time.time()))

18. filename =now+"report.html"

20. #discover方法执行测试套件

22. testsuite = unittest.defaultTestLoader.discover(

24. start_dir='./testsuites',

26. pattern='*case.py',

28. top_level_dir=None

30. )

33. with open(DIR+'/test_report/'+filename,'wb') as f:

35. runner = HTMLTestRunner(

37. stream=f,

39. verbosity=2,

41. title='gateway UI report',

43. description='执行情况',

45. tester='tester'

47. )

49. runner.run(testsuite)

6、测试报告展示

基于selenium实现的web自动化框架不仅轻量级而且灵活，可以快速的开发自动化测试用例。本篇中的框架设计比较简单，希望对大家以后的web自动化框架的设计和实现有所帮助，共同交流，共同进步！