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

Posted 2020-09-20

预期框架整理目标：

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
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="找找看"]‘]
    }
}

config_01.py

1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 # @Time    : 2017-05-15 13:20
4 
5 # 常量部分（固定不变使用频繁的参数维护在此处）
6 LOGIN_URL = ‘https://www.cnblogs.com/‘

constant_1.py

 1 #!/usr/bin/env python
 2 # -*- coding: utf-8 -*-
 3 # @Time    : 2017-05-15 13:20
 4 
 5 # 封装部分维护在此
 6 
 7 from config.config_01 import locat_config
 8 from log.log import Logger
 9 from selenium.webdriver.support.wait import WebDriverWait
10 
11 class UIHandle():
12     logger = Logger()
13 
14     # 构造方法，用来接收selenium的driver对象
15     @classmethod
16     def __init__(cls, driver):
17         cls.driver = driver
18 
19     # 输入地址
20     @classmethod
21     def get(cls, url):
22         cls.logger.loginfo(url)
23         cls.driver.get(url)
24 
25     # 关闭浏览器驱动
26     @classmethod
27     def quit(cls):
28         cls.driver.quit()
29 
30     # element对象（还可加入try，截图等。。。）
31     @classmethod
32     def element(cls, page, element):
33         # 加入日志
34         cls.logger.loginfo(page)
35         # 加入隐性等待
36         WebDriverWait(cls.driver, 10)
37         # 此处便可以传入config_o1中的dict定位参数
38         el = cls.driver.find_element(*locat_config[page][element])
39         # 加入日志
40         cls.logger.loginfo(page+‘OK‘)
41         return el
42     # element对象(还未完成。。。)
43     def elements(cls, page, element):
44         # 加入日志
45         cls.logger.loginfo(page)
46         # 加入隐性等待
47         WebDriverWait(cls.driver, 10)
48         els = cls.driver.find_elements(*locat_config[page][element])
49         # 注意返回的是list
50         return els
51 
52     # send_keys方法
53     @classmethod
54     def Input(cls, page, element, msg):
55         el = cls.element(page, element)
56         el.send_keys(msg)
57 
58     # click方法
59     @classmethod
60     def Click(cls, page, element):
61         el = cls.element(page, element)
62         el.click()

encapsulation.py

 1 #!/usr/bin/env python
 2 # -*- coding: utf-8 -*-
 3 # @Time    : 2017-05-15 13:22
 4 
 5 # 业务功能脚本（用例脚本可调用此处的功能脚本）
 6 
 7 from encapsulation.encapsulation import UIHandle
 8 from constant.constant_1 import LOGIN_URL
 9 from config.config_01 import browser_config
10 from time import sleep
11 
12 # 打开博客园首页，进行找找看搜索功能
13 def search(msg):
14     # 打开浏览器
15     driver = browser_config[‘chrome‘]()
16     # 传入driver对象
17     uihandle = UIHandle(driver)
18     #输入url地址
19     uihandle.get(LOGIN_URL)
20     # 调用二次封装后的方法，此处可见操作了哪个页面，哪个元素，msg是要插入的值，插入值得操作在另外一个用例文件中传入
21     uihandle.Input(‘博客园首页‘, ‘找找看输入框‘, msg)
22     uihandle.Click(‘博客园首页‘, ‘找找看按钮‘)
23     uihandle.quit()

function_01.py

 1 #!/usr/bin/env python
 2 # -*- coding: utf-8 -*-
 3 # @Time    : 2017-05-17 11:19
 4 import logging
 5 import logging.handlers
 6 
 7 # 日志类
 8 class Logger():
 9     LOG_FILE = ‘tst.log‘
10 
11     handler = logging.handlers.RotatingFileHandler(LOG_FILE, maxBytes = 1024*1024, backupCount = 5) # 实例化handler
12     fmt = ‘%(asctime)s - %(filename)s:%(lineno)s - %(name)s - %(message)s‘
13 
14     formatter = logging.Formatter(fmt)   # 实例化formatter
15     handler.setFormatter(formatter)      # 为handler添加formatter
16 
17     logger = logging.getLogger(‘tst‘)    # 获取名为tst的logger
18     logger.addHandler(handler)           # 为logger添加handler
19     logger.setLevel(logging.DEBUG)
20     def loginfo(self, message):
21         self.logger.info(message)
22 
23     def logdebug(self, message):
24         self.logger.debug(message)

log.py

 1 #!/usr/bin/env python
 2 #!/usr/bin/env python
 3 # -*- coding: utf-8 -*-
 4 # @Time    : 2017-05-15 15:30
 5 import unittest
 6 from function.function_01 import *
 7 # 用例
 8 class Case_02(unittest.TestCase):
 9     u‘‘‘哇塞好玩‘‘‘
10     def setUp(self):
11         pass
12 
13     def test_zzk(self):
14         u‘‘‘输入哇塞好玩后点击找找看‘‘‘
15         search("哇塞好玩")
16         print(‘打印方法名：test_zzk‘)
17 
18     def tearDown(self):
19         pass
20 
21 if __name__ == "__main__":
22     unittest.main()

start_case_01.py

 1 #!/usr/bin/env python
 2 # -*- coding: utf-8 -*-
 3 # @Time    : 2017-05-10 16:34
 4 import unittest
 5 import HTMLTestRunner
 6 import time,os,datetime
 7 import smtplib
 8 from email.mime.text import MIMEText
 9 from email.mime.multipart import MIMEMultipart
10 from email.mime.image import MIMEImage
11 
12 
13 
14 # 取test_case文件夹下所有用例文件
15 def creatsuitel(lists):
16     testunit = unittest.TestSuite()
17     # discover 方法定义
18     discover = unittest.defaultTestLoader.discover(lists, pattern=‘start_*.py‘, top_level_dir=None)
19     #discover 方法筛选出来的用例，循环添加到测试套件中
20     for test_suite in discover:
21         for test_case in test_suite:
22             testunit.addTests(test_case)
23             print(testunit)
24     return testunit
25 list_1 = ‘test_case\\\\test_case_1‘
26 alltestnames = creatsuitel(list_1)
27 
28 #取前面时间加入到测试报告文件名中
29 now = time.strftime("%Y-%m-%M-%H_%M_%S", time.localtime(time.time()))
30 filename = "report\\\\"+now+‘result.html‘ #定义个报告存放路径，支持相对路径。
31 fp = open(filename, ‘wb‘)
32 runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title=‘Report_title‘, description=‘Report_description‘)
33 
34 if __name__ == "__main__":
35     # 执行测试用例集并生成报告
36     runner = unittest.TextTestRunner()

all_case.py

 

架子比较简陋，可以跑通，感谢@大师兄[email protected]我去热饭的指导与帮助。框架后续还会进一步优化，欢迎建议，指点，本人QQ：915069792

 

为什么选择？

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

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