python+unittest框架整理（一点点学习前辈们的封装思路，一点点成长。。。）

预期框架整理目标：

1.单个用例维护在单个.py文件中可单个执行，也可批量生成组件批量执行

2.对定位参数，定位方法，业务功能脚本，用例脚本，用例批量执行脚本，常用常量进行分层独立，各自维护在单独的.py文件中

3.加入日志，htlm报表,发送邮件功能

框架结构

结构说明：

config:配置部分，浏览器种类和定位信息维护在此处

constant:常量部分，固定不变的数据维护在此处

data:存放用于参数化的文本表格等文件

encapsulation:定位等selenium功能二次封装在此处

error_picture:存放错误截图

function:业务功能脚本维护在此处

log:存放log类

report:存放测试报告文件

test_case:存放用例文件

all_case.py:用来执行所有用例

debug_case.py:本人调试用的，可以忽略

tst.log：生成的日志

逐个介绍各个包下面的.py文件，并附上源码（说明见注释哈哈~）:

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-11 13:42
 # config/config_01.py
 from selenium import webdriver
 import time
 from selenium.webdriver.common.action_chains import *

 # config配置部分

 # 浏览器种类维护在此处
 browser_config = {
     'ie': webdriver.Ie,
     'chrome': webdriver.Chrome
 }

 # 定位信息维护在此处，维护结构由外到内为：页面名称--页面下元素名称--元素的定位方式+参数
 locat_config = {
     '博客园首页': {
         '找找看输入框': ['id', 'zzk_q'],
         '找找看按钮': ['xpath', '//input[@value="找找看"]']
     }
 }

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-15 13:20
 # constant/constant_1.py

 # 常量部分（固定不变使用频繁的参数维护在此处）
 LOGIN_URL = 'https://www.cnblogs.com/'

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-15 13:20
 # encapsulation/encapsulation.py

 # 封装部分维护在此

 from config.config_01 import locat_config
 from log.log import Logger
 from selenium.webdriver.support.wait import WebDriverWait

 from selenium.webdriver.support import expected_conditions as EC

 class UIHandle():
     logger = Logger()

     # 构造方法，用来接收selenium的driver对象
     @classmethod
     def __init__(cls, driver):
         cls.driver = driver

     # 输入地址
     @classmethod
     def get(cls, url):
         cls.logger.loginfo(url)
         cls.driver.get(url)

     # 关闭浏览器驱动
     @classmethod
     def quit(cls):
         cls.driver.quit()

     # element对象（还可加入try，截图等。。。）
     @classmethod
     def element(cls, page, element):
         # 加入日志
         cls.logger.loginfo(page)
         # 加入隐性等待
         # 此处便可以传入config_o1中的dict定位参数
         el = WebDriverWait(cls.driver, 10).until(EC.presence_of_element_located(locat_config[page][element]))
         # 加入日志
         cls.logger.loginfo(page+'OK')
         return el
     # element对象(还未完成。。。)
     def elements(cls, page, element):
         # 加入日志
         cls.logger.loginfo(page)
         # 加入隐性等待
         WebDriverWait(cls.driver, 10)
         els = cls.driver.find_elements(*locat_config[page][element])
         # 注意返回的是list
         return els

     # send_keys方法
     @classmethod
     def Input(cls, page, element, msg):
         el = cls.element(page, element)
         el.send_keys(msg)

     # click方法
     @classmethod
     def Click(cls, page, element):
         el = cls.element(page, element)
         el.click()

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-15 13:22
 # function/function_01.py
 # 业务功能脚本（用例脚本可调用此处的功能脚本）

 from encapsulation.encapsulation import UIHandle
 from constant.constant_1 import LOGIN_URL
 from config.config_01 import browser_config
 from time import sleep

 # 打开博客园首页，进行找找看搜索功能
 def search(msg):
     # 打开浏览器
     driver = browser_config['chrome']()
     # 传入driver对象
     uihandle = UIHandle(driver)
     #输入url地址
     uihandle.get(LOGIN_URL)
     # 调用二次封装后的方法，此处可见操作了哪个页面，哪个元素，msg是要插入的值，插入值得操作在另外一个用例文件中传入
     uihandle.Input('博客园首页', '找找看输入框', msg)
     uihandle.Click('博客园首页', '找找看按钮')
     uihandle.quit()

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-17 11:19
 # log/log.py

 import logging
 import logging.handlers

 # 日志类
 class Logger():
     LOG_FILE = 'tst.log'

     handler = logging.handlers.RotatingFileHandler(LOG_FILE, maxBytes = 1024*1024, backupCount = 5) # 实例化handler
     fmt = '%(asctime)s - %(filename)s:%(lineno)s - %(name)s - %(message)s'

     formatter = logging.Formatter(fmt)   # 实例化formatter
     handler.setFormatter(formatter)      # 为handler添加formatter

     logger = logging.getLogger('tst')    # 获取名为tst的logger
     logger.addHandler(handler)           # 为logger添加handler
     logger.setLevel(logging.DEBUG)
     def loginfo(self, message):
         self.logger.info(message)

     def logdebug(self, message):
         self.logger.debug(message)

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-15 15:30
 # test_case/test_case_1/start_case_01.py

 import unittest
 from function.function_01 import *
 # 用例
 class Case_02(unittest.TestCase):
     u'''哇塞好玩'''
     def setUp(self):
         pass

     def test_zzk(self):
         u'''输入哇塞好玩后点击找找看'''
         search("哇塞好玩")
         print('打印方法名：test_zzk')

     def tearDown(self):
         pass

 if __name__ == "__main__":
     unittest.main()

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 # @Time    : 2017-05-10 16:34
 # all_case.py

 import unittest
 import HTMLTestRunner
 import time,os,datetime
 import smtplib
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
 from email.mime.image import MIMEImage

 # 取test_case文件夹下所有用例文件
 def creatsuitel(lists):
     testunit = unittest.TestSuite()
     # discover 方法定义
     discover = unittest.defaultTestLoader.discover(lists, pattern='start_*.py', top_level_dir=None)
     #discover 方法筛选出来的用例，循环添加到测试套件中
     for test_suite in discover:
         for test_case in test_suite:
             testunit.addTests(test_case)
             print(testunit)
     return testunit
 list_1 = 'test_case\\test_case_1'
 alltestnames = creatsuitel(list_1)

 #取前面时间加入到测试报告文件名中
 now = time.strftime("%Y-%m-%M-%H_%M_%S", time.localtime(time.time()))
 filename = "report\\"+now+'result.html' #定义个报告存放路径，支持相对路径。
 fp = open(filename, 'wb')
 runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title='Report_title', description='Report_description')

 if __name__ == "__main__":
     # 执行测试用例集并生成报告
     runner = unittest.TextTestRunner()

感谢@大师兄+@我去热饭的指导与帮助。

 

为什么选择？

有的人喜欢创造世界，他们做了程序员

有的人喜欢拯救世界，他们做了测试员