├── .gitattributes
├── .gitignore
├── LICENSE
├── MANIFEST.in
├── README.md
├── demo
    ├── __init__.py
    ├── debug_case.py
    ├── report
    │   ├── 2019-03-13-18-46-13-style-1.html
    │   └── 2019-03-13-18-46-13-style-2.html
    ├── run.py
    ├── testcase
    │   ├── __init__.py
    │   ├── battle
    │   │   ├── __init__.py
    │   │   └── test_tattle.py
    │   ├── chat
    │   │   ├── __init__.py
    │   │   └── test_chat.py
    │   └── legion
    │   │   ├── __init__.py
    │   │   └── test_legion.py
    └── use_report.py
├── img
    ├── print_info.png
    ├── style1.png
    └── style2.png
├── requirements.txt
├── setup.py
└── utx
    ├── __init__.py
    ├── core.py
    ├── log.py
    ├── report
        ├── __init__.py
        ├── style_1.py
        └── style_2.py
    ├── runner.py
    ├── setting.py
    └── tag.py


/.gitattributes:
--------------------------------------------------------------------------------
1 | *.html linguist-language=python


--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
 1 | # Byte-compiled / optimized / DLL files
 2 | __pycache__/
 3 | *.py[cod]
 4 | *$py.class
 5 | 
 6 | # C extensions
 7 | *.so
 8 | 
 9 | # Distribution / packaging
10 | .Python
11 | env/
12 | build/
13 | develop-eggs/
14 | dist/
15 | downloads/
16 | eggs/
17 | .eggs/
18 | lib/
19 | lib64/
20 | parts/
21 | sdist/
22 | var/
23 | *.egg-info/
24 | .installed.cfg
25 | *.egg
26 | 
27 | # PyInstaller
28 | #  Usually these files are written by a python script from a template
29 | #  before PyInstaller builds the exe, so as to inject date/other infos into it.
30 | *.manifest
31 | *.spec
32 | 
33 | # Installer logs
34 | pip-log.txt
35 | pip-delete-this-directory.txt
36 | 
37 | # Unit test / coverage reports
38 | htmlcov/
39 | .tox/
40 | .coverage
41 | .coverage.*
42 | .cache
43 | nosetests.xml
44 | coverage.xml
45 | *,cover
46 | .hypothesis/
47 | 
48 | # Translations
49 | *.mo
50 | *.pot
51 | 
52 | # Django stuff:
53 | *.log
54 | local_settings.py
55 | 
56 | # Sphinx documentation
57 | docs/_build/
58 | 
59 | # PyBuilder
60 | target/
61 | 
62 | #Ipython Notebook
63 | .ipynb_checkpoints
64 | 
65 | # pyenv
66 | .python-version
67 | 
68 | # pycharm
69 | .idea/
70 | 
71 | # 本地测试
72 | local_test/
73 | wxglade/
74 | kivy-2014/
75 | designer/
76 | youtube_dl_gui/
77 | 
78 | 
79 | # 本地配置
80 | local_config.py
81 | 
82 | *.bak
83 | 


--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
 1 | MIT License
 2 | 
 3 | Copyright (c) [2017] [jianbing.g]
 4 | 
 5 | Permission is hereby granted, free of charge, to any person obtaining a copy
 6 | of this software and associated documentation files (the "Software"), to deal
 7 | in the Software without restriction, including without limitation the rights
 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 | 
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 | 
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.


--------------------------------------------------------------------------------
/MANIFEST.in:
--------------------------------------------------------------------------------
1 | include README.md
2 | include LICENSE
3 | include requirements.txt


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
  1 | # utx
  2 | 
  3 | > 支持Python3.6及以上版本
  4 | 
  5 | utx扩展了Python unittest框架的功能，起因是需要控制测试用例的执行顺序，而unittest的默认执行顺序是按照用例函数的名称进行排序，所以想要做一个可以无缝接入unittest的扩展功能库。
  6 | 
  7 | ## 当前版本
  8 | 
  9 | ```python
 10 | V0.0.8
 11 | ```
 12 | 
 13 | ## 安装
 14 | 
 15 | ```python
 16 | python setup.py install
 17 | ```
 18 | 
 19 | ## 更新
 20 | 
 21 | ```python
 22 | pip uninstall utx   # 需要先卸载旧的utx
 23 | python setup.py install
 24 | ```
 25 | 
 26 | 
 27 | 
 28 | ## 功能列表
 29 | 
 30 | 
 31 | - 用例排序，只需要导入utx库，用例的执行顺序就会和编写顺序一致
 32 | 
 33 | - 用例自定义标签，在 tag.py 里边添加标签，可以对测试用例指定多个标签。全部用例默认带有`ALL`标签。
 34 | 
 35 | ```python
 36 | @unique
 37 | class Tag(Enum):
 38 |     ALL = NewTag("完整")  # 完整测试标记，可以重命名，不要删除
 39 |     SMOKE = NewTag("冒烟")  # 冒烟测试标记
 40 | 
 41 |     # 以下开始为扩展标签，自行调整
 42 |     V1_0_0 = NewTag("V1.0.0版本")
 43 |     V2_0_0 = NewTag("V2.0.0版本")
 44 | ```
 45 | 
 46 | ```python
 47 | class TestLegion(unittest.TestCase):
 48 | 
 49 |     @tag(Tag.SMOKE)
 50 |     def test_create_legion(self):
 51 |         pass
 52 | 
 53 |     @tag(Tag.V1_0_0, Tag.ALL)
 54 |     def test_quit_legion(self):
 55 |         """测试退出军团
 56 | 
 57 |         :return:
 58 |         """
 59 |         print("测试退出军团")
 60 |         assert 1 == 2
 61 | ```
 62 | 
 63 | - 运行指定标签的测试用例
 64 | 
 65 | ```python
 66 | from utx import *
 67 | 
 68 | if __name__ == '__main__':
 69 |     setting.run_case = {Tag.ALL}  # 运行全部测试用例
 70 |     # setting.run_case = {Tag.SMOKE}  # 只运行SMOKE标记的测试用例
 71 |     # setting.run_case = {Tag.SMOKE, Tag.V1_0_0}   # 只运行SMOKE和V1_0_0标记的测试用例
 72 | 
 73 |     runner = TestRunner()
 74 |     runner.add_case_dir(r"testcase")
 75 |     runner.run_test(report_title='接口自动化测试报告')
 76 | ```
 77 | 
 78 | - 数据驱动
 79 | ```python
 80 | class TestLegion(unittest.TestCase):
 81 | 
 82 |     @data(["gold", 100], ["diamond", 500])
 83 |     def test_bless(self, bless_type, cost):
 84 |         """测试公会祈福
 85 | 
 86 |         :param bless_type: 祈福类型
 87 |         :param cost: 消耗数量
 88 |         :return:
 89 |         """
 90 |         print(bless_type)
 91 |         print(cost)
 92 |         
 93 |     @data(10001, 10002, 10003)
 94 |     def test_receive_bless_box(self, box_id):
 95 |         """ 测试领取祈福宝箱
 96 | 
 97 |         :return:
 98 |         """
 99 |         print(box_id)
100 | 
101 | # 默认会解包测试数据来一一对应函数参数，可以使用unpack=False，不进行解包
102 | 
103 | class TestBattle(unittest.TestCase):
104 |     @data({"gold": 1000, "diamond": 100}, {"gold": 2000, "diamond": 200}, unpack=False)
105 |     def test_get_battle_reward(self, reward):
106 |         """ 测试领取战斗奖励
107 | 
108 |         :return:
109 |         """
110 |         print(reward)
111 |         print("获得的钻石数量是：{}".format(reward['diamond']))
112 |  ```
113 | 
114 | - 检测用例是否编写了用例描述
115 | ```python
116 | 2017-11-13 12:00:19,334 WARNING legion.test_legion.test_bless没有用例描述
117 | ```
118 | 
119 | - 采集测试用例的print打印的信息到测试报告里边
120 | ```python
121 | @data({"gold": 1000, "diamond": 100}, {"gold": 2000, "diamond": 200}, unpack=False)
122 | def test_get_battle_reward(self, reward):
123 |     """ 领取战斗奖励
124 | 
125 |     :return:
126 |     """
127 |     print(reward)
128 |     print("获得的钻石数量是：{}".format(reward['diamond']))
129 | ```
130 | 
131 | ![](https://github.com/jianbing/utx/raw/master/img/print_info.png)
132 | 
133 | 
134 | - 执行测试时，显示测试进度
135 | ```
136 | 2019-03-13 18:46:13,810 INFO 开始测试，用例数量总共15个，跳过5个，实际运行10个
137 | 2019-03-13 18:46:13,910 INFO start to test battle.test_tattle.test_start_battle (1/10)
138 | 2019-03-13 18:46:14,010 INFO start to test battle.test_tattle.test_skill_buff (2/10)
139 | 2019-03-13 18:46:14,111 INFO start to test battle.test_tattle.test_normal_attack (3/10)
140 | 2019-03-13 18:46:14,211 INFO start to test battle.test_tattle.test_get_battle_reward (4/10)
141 | 2019-03-13 18:46:14,211 DEBUG 测试领取战斗奖励，获得的钻石数量是：100
142 | 2019-03-13 18:46:14,311 INFO start to test battle.test_tattle.test_get_battle_reward (5/10)
143 | ```
144 | 
145 | - setting类提供多个设置选项进行配置
146 | ```python
147 | class setting:
148 | 
149 |     # 只运行的用例类型
150 |     run_case = {Tag.SMOKE}
151 | 
152 |     # 开启用例排序
153 |     sort_case = True
154 | 
155 |     # 每个用例的执行间隔，单位是秒
156 |     execute_interval = 0.1
157 | 
158 |     # 开启检测用例描述
159 |     check_case_doc = True
160 | 
161 |     # 显示完整用例名字（函数名字+参数信息）
162 |     full_case_name = False
163 | 
164 |     # 测试报告显示的用例名字最大程度
165 |     max_case_name_len = 80
166 | 
167 |     # 执行用例的时候，显示报错信息
168 |     show_error_traceback = True
169 | 
170 |     # 测试报告样式1
171 |     create_report_by_style_1 = True
172 | 
173 |     # 测试报告样式2
174 |     create_report_by_style_2 = True
175 |     
176 |     # 在控制台显示print打印的内容
177 |     show_print_in_console = False
178 | ```
179 | 
180 | 
181 | - 集成两种测试报告样式，感谢两位作者的测试报告模版  
182 |     
183 | [测试报告1](https://github.com/findyou/HTMLTestRunnerCN)
184 | 
185 | ![](https://github.com/jianbing/utx/raw/master/img/style1.png)
186 | 
187 | [测试报告2](https://github.com/zhangfei19841004/ztest)
188 | 
189 | ![](https://github.com/jianbing/utx/raw/master/img/style2.png)
190 | 
191 | - 无缝接入unittest项目，导入utx包即可开始使用扩展功能，无需修改之前的代码
192 | 
193 | 
194 | ## 例子
195 | 
196 | demo目录下，有几个例子：
197 | 
198 | - ```run.py```  一个完整使用utx功能的demo小项目
199 | 
200 | - ```use_report.py``` 单独使用测试报告组件，不需要utx的其他扩展功能
201 | 
202 | - ```debug_case.py``` 使用utx后，单独调试某个用例的例子


--------------------------------------------------------------------------------
/demo/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | 
4 | 
5 | 


--------------------------------------------------------------------------------
/demo/debug_case.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | 
 4 | import utx
 5 | from demo.testcase.battle.test_tattle import TestBattle
 6 | 
 7 | utx.run_case(TestBattle, "test_get_battle_reward")
 8 | 
 9 | 
10 | 


--------------------------------------------------------------------------------
/demo/report/2019-03-13-18-46-13-style-1.html:
--------------------------------------------------------------------------------
  1 | 
  2 | <?xml version="1.0" encoding="UTF-8"?>
  3 | <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
  4 | <html xmlns="http://www.w3.org/1999/xhtml">
  5 | <head>
  6 |     <title>接口自动化测试报告</title>
  7 |     <meta http-equiv="Content-Type" content="text/html; charset=UTF-8"/>
  8 |     <link href="http://libs.baidu.com/bootstrap/3.0.3/css/bootstrap.min.css" rel="stylesheet">
  9 |     <script src="http://libs.baidu.com/jquery/2.0.0/jquery.min.js"></script>
 10 |     <script src="http://libs.baidu.com/bootstrap/3.0.3/js/bootstrap.min.js"></script>
 11 | 
 12 |     <style type="text/css" media="screen">
 13 |         pre {
 14 |             white-space: pre-wrap;
 15 |             white-space: -moz-pre-wrap;
 16 |             white-space: -pre-wrap;
 17 |             white-space: -o-pre-wrap;
 18 |             word-wrap: break-word;
 19 |             text-align: left;
 20 |         }
 21 |     
 22 |         body {
 23 |             font-family: Microsoft YaHei, Tahoma, arial, helvetica, sans-serif;
 24 |             padding: 20px;
 25 |             font-size: 80%;
 26 |         }
 27 | 
 28 |         table {
 29 |             font-size: 100%;
 30 |         }
 31 | 
 32 |         /* -- heading ---------------------------------------------------------------------- */
 33 |         .heading {
 34 |             margin-top: 0ex;
 35 |             margin-bottom: 1ex;
 36 |         }
 37 | 
 38 |         .heading .description {
 39 |             margin-top: 4ex;
 40 |             margin-bottom: 6ex;
 41 |         }
 42 | 
 43 |         /* -- report ------------------------------------------------------------------------ */
 44 |         #total_row {
 45 |             font-weight: bold;
 46 |         }
 47 | 
 48 |         .passCase {
 49 |             color: #5cb85c;
 50 |         }
 51 | 
 52 |         .failCase {
 53 |             color: #f0ad4e;
 54 |             font-weight: bold;
 55 |         }
 56 | 
 57 |         .errorCase {        
 58 |             color: #d9534f;
 59 |             font-weight: bold;
 60 |         }
 61 | 
 62 |         .hiddenRow {
 63 |             display: none;
 64 |         }
 65 | 
 66 |         .testcase {
 67 |             margin-left: 2em;
 68 |         }
 69 |     </style>
 70 | 
 71 | </head>
 72 | <script language="javascript" type="text/javascript">
 73 |     output_list = Array();
 74 | 
 75 | 
 76 |     function showCase(level) {
 77 |         trs = document.getElementsByTagName("tr");
 78 |         for (var i = 0; i < trs.length; i++) {
 79 |             tr = trs[i];
 80 |             id = tr.id;
 81 |             if (id.substr(0, 2) == 'pt') {
 82 |                 if (level == 1 || level == 4) {
 83 |                     tr.className = '';
 84 |                 }
 85 |                 else {
 86 |                     tr.className = 'hiddenRow';                    
 87 |                 }
 88 |             }
 89 |             if (id.substr(0, 2) == 'ft') {
 90 |                 if (level == 2 || level == 4) {
 91 |                     tr.className = '';
 92 |                 }
 93 |                 else {
 94 |                     tr.className = 'hiddenRow';
 95 |                 }
 96 |             }
 97 |             
 98 |             if (id.substr(0, 2) == 'et') {
 99 |                 if (level == 3 || level == 4) {
100 |                     tr.className = '';
101 |                 }
102 |                 else {
103 |                     tr.className = 'hiddenRow';
104 |                 }
105 |             }
106 |         }
107 | 
108 |         detail_class = document.getElementsByClassName('detail');
109 |         //console.log(detail_class.length)
110 |         if (level == 3) {
111 |             for (var i = 0; i < detail_class.length; i++) {
112 |                 detail_class[i].innerHTML = "收起"
113 |             }
114 |         }
115 |         else {
116 |             for (var i = 0; i < detail_class.length; i++) {
117 |                 detail_class[i].innerHTML = "详细"
118 |             }
119 |         }
120 |     }
121 | 
122 |     function showClassDetail(cid, count) {
123 |         var id_list = Array(count);
124 |         var toHide = 1;
125 |         for (var i = 0; i < count; i++) {
126 |             tid0 = 't' + cid.substr(1) + '_' + (i + 1);
127 |             console.log(tid0);
128 |             tid = 'f' + tid0;
129 |             tr = document.getElementById(tid);
130 |             if (!tr) {
131 |                 tid = 'p' + tid0;
132 |                 tr = document.getElementById(tid);
133 |             }
134 |             if (!tr) {
135 |                 tid = 'e' + tid0;
136 |                 tr = document.getElementById(tid);
137 |             }
138 |             id_list[i] = tid;
139 |             if (tr.className) {
140 |                 toHide = 0;
141 |             }
142 |         }
143 |         for (var i = 0; i < count; i++) {
144 |             tid = id_list[i];
145 |             if (toHide) {
146 |                 document.getElementById(tid).className = 'hiddenRow';
147 |                 document.getElementById(cid).innerText = "详细"
148 |             }
149 |             else {
150 |                 document.getElementById(tid).className = '';
151 |                 document.getElementById(cid).innerText = "收起"
152 |             }
153 |         }
154 |     }
155 | 
156 |     function html_escape(s) {
157 |         s = s.replace(/&/g, '&amp;');
158 |         s = s.replace(/</g, '&lt;');
159 |         s = s.replace(/>/g, '&gt;');
160 |         return s;
161 |     }
162 | </script>
163 | 
164 | 
165 | <body>
166 | 
167 | <div class='heading'>
168 |     <h1 style="font-family: Microsoft YaHei">接口自动化测试报告</h1>
169 |     <p class='attribute'><strong>开始时间 : </strong> 2019-03-13 18:46:13</p>
170 |     <p class='attribute'><strong>合计耗时 : </strong> 0:00:01.003057</p>
171 |     <p class='attribute'><strong>测试结果 : </strong> 共 15，通过 8，失败 1，错误 1，跳过 5，通过率= 53.33%</p>
172 | 
173 |     <p class='description'></p>
174 | </div>
175 | 
176 | 
177 | <p id='show_detail_line'>
178 |     <a class="btn btn-primary" href='javascript:showCase(0)'>概要{ 53.33% }</a>
179 |     <a class="btn btn-success" href='javascript:showCase(1)'>通过{ 8 }</a>
180 |     <a class="btn btn-warning" href='javascript:showCase(2)'>失败{ 1 }</a>
181 |     <a class="btn btn-danger" href='javascript:showCase(3)'>错误{ 1 }</a>
182 |     <a class="btn btn-info" href='javascript:showCase(4)'>所有{ 15 }</a>
183 | </p>
184 | <table id='result_table' class="table table-condensed table-bordered table-hover">
185 |     <colgroup>
186 |         <col align='left'/>
187 |         <col align='right'/>
188 |         <col align='right'/>
189 |         <col align='right'/>
190 |         <col align='right'/>
191 |         <col align='right'/>
192 |         <col align='right'/>
193 |     </colgroup>
194 |     <tr id='header_row' class="text-center success" style="font-weight: bold;font-size: 14px;">
195 |         <td>用例集/测试用例</td>
196 |         <td>总计</td>
197 |         <td>通过</td>
198 |         <td>失败</td>
199 |         <td>错误</td>
200 |         <td>跳过</td>
201 |         <td>详细</td>
202 |     </tr>
203 | 
204 |  
205 |     
206 | <tr class="failClass warning">
207 |         <td>battle.test_tattle.TestBattle</td>
208 |         <td class="text-center">5</td>
209 |         <td class="text-center">5</td>
210 |         <td class="text-center">0</td>
211 |         <td class="text-center">0</td>
212 |         <td class="text-center">0</td>
213 |         <td class="text-center"><a href="javascript:showClassDetail('c1',5)" class="detail" id="c1">详细</a></td>
214 |     </tr>
215 | 
216 | <tr id="pt1_1" class="">
217 |         <td class="passCase">
218 |             <div class="testcase">test_00001_start_battle（测试开始战斗）</div>
219 |         </td>
220 |         <td colspan="6" align="center">
221 |        
222 |         <button id='btn_pt1_1' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt1_1'>通过</button>
223 |         <div id='div_pt1_1' class="collapse">
224 |                 <pre>测试开始战斗
225 | </pre>
226 |             </div>
227 |         </td>
228 |     </tr>
229 | 
230 | <tr id="pt1_2" class="">
231 |         <td class="passCase">
232 |             <div class="testcase">test_00002_skill_buff（测试技能buff）</div>
233 |         </td>
234 |         <td colspan="6" align="center">
235 |        
236 |         <button id='btn_pt1_2' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt1_2'>通过</button>
237 |         <div id='div_pt1_2' class="collapse">
238 |                 <pre>测试技能buff
239 | </pre>
240 |             </div>
241 |         </td>
242 |     </tr>
243 | 
244 | <tr id="pt1_3" class="">
245 |         <td class="passCase">
246 |             <div class="testcase">test_00003_normal_attack（测试普通攻击）</div>
247 |         </td>
248 |         <td colspan="6" align="center">
249 |        
250 |         <button id='btn_pt1_3' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt1_3'>通过</button>
251 |         <div id='div_pt1_3' class="collapse">
252 |                 <pre>测试普通攻击
253 | </pre>
254 |             </div>
255 |         </td>
256 |     </tr>
257 | 
258 | <tr id="pt1_4" class="">
259 |         <td class="passCase">
260 |             <div class="testcase">test_00004_get_battle_reward_00001_1000_100（测试领取战斗奖励）</div>
261 |         </td>
262 |         <td colspan="6" align="center">
263 |        
264 |         <button id='btn_pt1_4' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt1_4'>通过</button>
265 |         <div id='div_pt1_4' class="collapse">
266 |                 <pre>{'gold': 1000, 'diamond': 100}
267 | 测试领取战斗奖励，获得的钻石数量是：100
268 | </pre>
269 |             </div>
270 |         </td>
271 |     </tr>
272 | 
273 | <tr id="pt1_5" class="">
274 |         <td class="passCase">
275 |             <div class="testcase">test_00005_get_battle_reward_00002_2000_200（测试领取战斗奖励）</div>
276 |         </td>
277 |         <td colspan="6" align="center">
278 |        
279 |         <button id='btn_pt1_5' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt1_5'>通过</button>
280 |         <div id='div_pt1_5' class="collapse">
281 |                 <pre>{'gold': 2000, 'diamond': 200}
282 | 测试领取战斗奖励，获得的钻石数量是：200
283 | </pre>
284 |             </div>
285 |         </td>
286 |     </tr>
287 | 
288 | <tr class="failClass warning">
289 |         <td>chat.test_chat.TestChat</td>
290 |         <td class="text-center">3</td>
291 |         <td class="text-center">0</td>
292 |         <td class="text-center">0</td>
293 |         <td class="text-center">1</td>
294 |         <td class="text-center">2</td>
295 |         <td class="text-center"><a href="javascript:showClassDetail('c2',3)" class="detail" id="c2">详细</a></td>
296 |     </tr>
297 | 
298 | <tr id="et2_1" class="">
299 |     <td class="errorCase"><div class="testcase">test_00006_chat_in_world_channel（测试世界聊天）</div></td>
300 |     <td colspan="6" align="center">
301 |     <!--默认收起错误信息
302 |     <button id='btn_et1_5' type="button"  class="btn btn-danger btn-xs collapsed" data-toggle="collapse" data-target='#div_et1_5'>错误</button>
303 |     <div id='div_et1_5' class="collapse">  -->
304 | 
305 |     <!-- 默认展开错误信息 -->
306 |     <button id="btn_et2_1" type="button" class="btn btn-danger btn-xs" data-toggle="collapse" data-target="#div_et2_1">错误</button>
307 |     <div id="div_et2_1" class="collapse in">
308 |         <pre>测试世界聊天
309 | Traceback (most recent call last):
310 |   File "D:\github\utx\utx\core.py", line 101, in wrap
311 |     result = func(*args, **kwargs)
312 |   File "D:\github\utx\demo\testcase\chat\test_chat.py", line 15, in test_chat_in_world_channel
313 |     raise Exception("运行报错了")
314 | Exception: 运行报错了
315 | </pre>
316 |     </div>
317 |     </td>
318 | </tr>
319 | 
320 | <tr id="pt2_2" class="">
321 |         <td class="passCase">
322 |             <div class="testcase">test_00007_chat_in_personal_channel（测试私聊）</div>
323 |         </td>
324 |         <td colspan="6" align="center">
325 |         <span class="label label-info success">跳过</span>
326 |         </td>
327 |     </tr>
328 | 
329 | <tr id="pt2_3" class="">
330 |         <td class="passCase">
331 |             <div class="testcase">test_00008_chat_in_union_channel（测试公会聊天）</div>
332 |         </td>
333 |         <td colspan="6" align="center">
334 |         <span class="label label-info success">跳过</span>
335 |         </td>
336 |     </tr>
337 | 
338 | <tr class="failClass warning">
339 |         <td>legion.test_legion.TestLegion</td>
340 |         <td class="text-center">7</td>
341 |         <td class="text-center">3</td>
342 |         <td class="text-center">1</td>
343 |         <td class="text-center">0</td>
344 |         <td class="text-center">3</td>
345 |         <td class="text-center"><a href="javascript:showClassDetail('c3',7)" class="detail" id="c3">详细</a></td>
346 |     </tr>
347 | 
348 | <tr id="pt3_1" class="">
349 |         <td class="passCase">
350 |             <div class="testcase">test_00009_create_legion（测试创建军团）</div>
351 |         </td>
352 |         <td colspan="6" align="center">
353 |        
354 |         <button id='btn_pt3_1' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt3_1'>通过</button>
355 |         <div id='div_pt3_1' class="collapse">
356 |                 <pre>运行setUp方法
357 | 运行tearDown方法
358 | </pre>
359 |             </div>
360 |         </td>
361 |     </tr>
362 | 
363 | <tr id="pt3_2" class="">
364 |         <td class="passCase">
365 |             <div class="testcase">test_00010_bless_00001_gold_100（测试公会祈福）</div>
366 |         </td>
367 |         <td colspan="6" align="center">
368 |        
369 |         <button id='btn_pt3_2' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt3_2'>通过</button>
370 |         <div id='div_pt3_2' class="collapse">
371 |                 <pre>运行setUp方法
372 | gold
373 | 100
374 | 运行tearDown方法
375 | </pre>
376 |             </div>
377 |         </td>
378 |     </tr>
379 | 
380 | <tr id="pt3_3" class="">
381 |         <td class="passCase">
382 |             <div class="testcase">test_00011_bless_00002_diamond_500（测试公会祈福）</div>
383 |         </td>
384 |         <td colspan="6" align="center">
385 |        
386 |         <button id='btn_pt3_3' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt3_3'>通过</button>
387 |         <div id='div_pt3_3' class="collapse">
388 |                 <pre>运行setUp方法
389 | diamond
390 | 500
391 | 运行tearDown方法
392 | </pre>
393 |             </div>
394 |         </td>
395 |     </tr>
396 | 
397 | <tr id="pt3_4" class="">
398 |         <td class="passCase">
399 |             <div class="testcase">test_00012_receive_bless_box_00001_10001（测试领取祈福宝箱）</div>
400 |         </td>
401 |         <td colspan="6" align="center">
402 |         <span class="label label-info success">跳过</span>
403 |         </td>
404 |     </tr>
405 | 
406 | <tr id="pt3_5" class="">
407 |         <td class="passCase">
408 |             <div class="testcase">test_00013_receive_bless_box_00002_10002（测试领取祈福宝箱）</div>
409 |         </td>
410 |         <td colspan="6" align="center">
411 |         <span class="label label-info success">跳过</span>
412 |         </td>
413 |     </tr>
414 | 
415 | <tr id="pt3_6" class="">
416 |         <td class="passCase">
417 |             <div class="testcase">test_00014_receive_bless_box_00003_10003（测试领取祈福宝箱）</div>
418 |         </td>
419 |         <td colspan="6" align="center">
420 |         <span class="label label-info success">跳过</span>
421 |         </td>
422 |     </tr>
423 | 
424 | <tr id="ft3_7" class="">
425 |         <td class="failCase">
426 |             <div class="testcase">test_00015_quit_legion（测试退出军团）</div>
427 |         </td>
428 |         <td colspan="6" align="center">
429 |             <!--默认收起错误信息
430 |             <button id='btn_ft1_2' type="button"  class="btn btn-danger btn-xs collapsed" data-toggle="collapse" data-target='#div_ft1_2'>失败</button>
431 |             <div id='div_ft1_2' class="collapse">  -->
432 | 
433 |             <!-- 默认展开错误信息 -->
434 |             <button id="btn_ft3_7" type="button" class="btn btn-warning btn-xs" data-toggle="collapse" data-target="#div_ft3_7">失败</button>
435 |             <div id="div_ft3_7" class="collapse in">
436 |                 <pre>运行setUp方法
437 | 测试退出军团
438 | 运行tearDown方法
439 | Traceback (most recent call last):
440 |   File "D:\github\utx\utx\core.py", line 101, in wrap
441 |     result = func(*args, **kwargs)
442 |   File "D:\github\utx\demo\testcase\legion\test_legion.py", line 50, in test_quit_legion
443 |     assert 1 == 2
444 | AssertionError
445 | </pre>
446 |             </div>
447 |         </td>
448 |     </tr>
449 | 
450 | 
451 |     <tr id='total_row' class="text-center active">
452 |         <td>总计</td>
453 |         <td>15</td>
454 |         <td>8</td>
455 |         <td>1</td>
456 |         <td>1</td>
457 |         <td>5</td>
458 |         <td>通过率：53.33%</td>
459 |     </tr>
460 | </table>
461 | 
462 | <div id='ending'>&nbsp;</div>
463 | <div style=" position:fixed;right:50px; bottom:30px; width:20px; height:20px;cursor:pointer">
464 |     <a href="#"><span class="glyphicon glyphicon-eject" style="font-size:30px;" aria-hidden="true">
465 |     </span></a></div>
466 | 
467 | 
468 | </body>
469 | </html>
470 | 


--------------------------------------------------------------------------------
/demo/run.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | 
 4 | from utx import *
 5 | import logging
 6 | 
 7 | if __name__ == '__main__':
 8 |     setting.run_case = {Tag.ALL}  # 运行全部测试用例
 9 |     # setting.run_case = {Tag.SMOKE}  # 只运行SMOKE标记的测试用例
10 |     # setting.run_case = {Tag.SMOKE, Tag.V1_0_0}   # 只运行SMOKE和V1_0_0标记的测试用例
11 | 
12 |     setting.check_case_doc = False  # 关闭检测是否编写了测试用例描述
13 |     setting.full_case_name = True
14 |     setting.max_case_name_len = 80  # 测试报告内，显示用例名字的最大程度
15 |     setting.show_error_traceback = True  # 执行用例的时候，显示报错信息
16 |     setting.sort_case = True  # 是否按照编写顺序，对用例进行排序
17 |     setting.create_report_by_style_1 = True  # 测试报告样式1
18 |     setting.create_report_by_style_2 = True  # 测试报告样式2
19 |     setting.show_print_in_console = True
20 | 
21 |     log.set_level(logging.DEBUG)  # 设置utx的log级别
22 |     # log.set_level_to_debug()     # 设置log级别的另外一种方法
23 | 
24 |     runner = TestRunner()
25 |     runner.add_case_dir(r"testcase")
26 |     runner.run_test(report_title='接口自动化测试报告')
27 | 


--------------------------------------------------------------------------------
/demo/testcase/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | 
4 | 
5 | if __name__ == '__main__':
6 |     pass
7 | 
8 | 


--------------------------------------------------------------------------------
/demo/testcase/battle/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | """
4 | Created by jianbing on 2017-10-27
5 | """


--------------------------------------------------------------------------------
/demo/testcase/battle/test_tattle.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | 
 4 | import unittest
 5 | from utx import *
 6 | 
 7 | 
 8 | class TestBattle(unittest.TestCase):
 9 |     def test_start_battle(self):
10 |         """测试开始战斗
11 | 
12 |         :return:
13 |         """
14 |         print("测试开始战斗")
15 | 
16 |     def test_skill_buff(self):
17 |         """测试技能buff
18 | 
19 |         :return:
20 |         """
21 |         print("测试技能buff")
22 | 
23 |     @tag(Tag.V1_0_0)
24 |     def test_normal_attack(self):
25 |         """测试普通攻击
26 | 
27 |         :return:
28 |         """
29 |         print("测试普通攻击")
30 | 
31 |     @data({"gold": 1000, "diamond": 100}, {"gold": 2000, "diamond": 200}, unpack=False)
32 |     def test_get_battle_reward(self, reward):
33 |         """ 测试领取战斗奖励
34 | 
35 |         :return:
36 |         """
37 |         print(reward)
38 |         print("测试领取战斗奖励，获得的钻石数量是：{}".format(reward['diamond']))
39 |         log.debug("测试领取战斗奖励，获得的钻石数量是：{}".format(reward['diamond']))
40 | 


--------------------------------------------------------------------------------
/demo/testcase/chat/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | """
4 | Created by jianbing on 2017-10-27
5 | """


--------------------------------------------------------------------------------
/demo/testcase/chat/test_chat.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | 
 4 | import unittest
 5 | import utx
 6 | 
 7 | 
 8 | class TestChat(unittest.TestCase):
 9 |     def test_chat_in_world_channel(self):
10 |         """测试世界聊天
11 | 
12 |         :return:
13 |         """
14 |         print("测试世界聊天")
15 |         raise Exception("运行报错了")
16 | 
17 |     @unittest.skip("跳过此用例")
18 |     def test_chat_in_personal_channel(self):
19 |         """测试私聊
20 | 
21 |         :return:
22 |         """
23 |         print("测试私聊")
24 | 
25 |     @utx.skip("跳过此用例")
26 |     def test_chat_in_union_channel(self):
27 |         """测试公会聊天
28 | 
29 |         :return:
30 |         """
31 |         print("测试公会聊天")
32 | 


--------------------------------------------------------------------------------
/demo/testcase/legion/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | """
4 | Created by jianbing on 2017-10-27
5 | """


--------------------------------------------------------------------------------
/demo/testcase/legion/test_legion.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | 
 4 | import unittest
 5 | from utx import *
 6 | 
 7 | 
 8 | class TestLegion(unittest.TestCase):
 9 |     def setUp(self):
10 |         print('运行setUp方法')
11 | 
12 |     def tearDown(self):
13 |         print('运行tearDown方法')
14 | 
15 |     @tag(Tag.SMOKE)
16 |     def test_create_legion(self):
17 |         """测试创建军团
18 | 
19 |         :return:
20 |         """
21 | 
22 |     @tag(Tag.ALL)
23 |     @data(["gold", 100], ["diamond", 500])
24 |     def test_bless(self, bless_type, cost):
25 |         """测试公会祈福
26 | 
27 |         :param bless_type: 祈福类型
28 |         :param cost: 消耗数量
29 |         :return:
30 |         """
31 |         print(bless_type)
32 |         print(cost)
33 | 
34 |     @skip("跳过的原因")
35 |     @data(10001, 10002, 10003)
36 |     def test_receive_bless_box(self, box_id):
37 |         """ 测试领取祈福宝箱
38 | 
39 |         :return:
40 |         """
41 |         print(box_id)
42 | 
43 |     @tag(Tag.V1_0_0, Tag.ALL)
44 |     def test_quit_legion(self):
45 |         """测试退出军团
46 | 
47 |         :return:
48 |         """
49 |         print("测试退出军团")
50 |         assert 1 == 2
51 | 


--------------------------------------------------------------------------------
/demo/use_report.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | """
 4 | 单独使用测试报告组件，不需要utx的其他扩展功能
 5 | """
 6 | import utx
 7 | 
 8 | if __name__ == '__main__':
 9 | 
10 |     utx.stop_patch()
11 | 
12 |     runner = utx.TestRunner()
13 |     runner.add_case_dir(r"testcase\chat")
14 |     runner.run_test(report_title='接口自动化测试报告')


--------------------------------------------------------------------------------
/img/print_info.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/jianbing/utx/43498227162707bc5d3460d8435540033082dfcf/img/print_info.png


--------------------------------------------------------------------------------
/img/style1.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/jianbing/utx/43498227162707bc5d3460d8435540033082dfcf/img/style1.png


--------------------------------------------------------------------------------
/img/style2.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/jianbing/utx/43498227162707bc5d3460d8435540033082dfcf/img/style2.png


--------------------------------------------------------------------------------
/requirements.txt:
--------------------------------------------------------------------------------
1 | colorama
2 | 


--------------------------------------------------------------------------------
/setup.py:
--------------------------------------------------------------------------------
 1 | #!/usr/bin/env python
 2 | # -*- coding: utf-8 -*-
 3 | 
 4 | import io
 5 | import os
 6 | import sys
 7 | from shutil import rmtree
 8 | from setuptools import setup, Command, find_packages
 9 | 
10 | NAME = 'utx'
11 | DESCRIPTION = '对Python unittest的功能进行了扩展'
12 | URL = 'https://github.com/jianbing/utx'
13 | EMAIL = '326333381@qq.com'
14 | AUTHOR = 'jianbing'
15 | VERSION = '0.0.7'
16 | REQUIRED = [
17 |     'colorama'
18 | ]
19 | 
20 | here = os.path.abspath(os.path.dirname(__file__))
21 | 
22 | with io.open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
23 |     long_description = '\n' + f.read()
24 | 
25 | 
26 | class UploadCommand(Command):
27 |     """Support setup.py upload."""
28 | 
29 |     description = 'Build and publish the package.'
30 |     user_options = []
31 | 
32 |     @staticmethod
33 |     def status(s):
34 |         """Prints things in bold."""
35 |         print('\033[1m{0}\033[0m'.format(s))
36 | 
37 |     def initialize_options(self):
38 |         pass
39 | 
40 |     def finalize_options(self):
41 |         pass
42 | 
43 |     def run(self):
44 |         try:
45 |             self.status('Removing previous builds…')
46 |             rmtree(os.path.join(here, 'dist'))
47 |         except OSError:
48 |             pass
49 | 
50 |         self.status('Building Source and Wheel (universal) distribution…')
51 |         os.system('{0} setup.py sdist bdist_wheel --universal'.format(sys.executable))
52 | 
53 |         self.status('Uploading the package to PyPi via Twine…')
54 |         os.system('twine upload dist/*')
55 | 
56 |         sys.exit()
57 | 
58 | 
59 | setup(
60 |     name=NAME,
61 |     version=VERSION,
62 |     description=DESCRIPTION,
63 |     long_description=long_description,
64 |     author=AUTHOR,
65 |     author_email=EMAIL,
66 |     url=URL,
67 |     python_requires='>=3.6.0',
68 |     packages=find_packages(),
69 |     install_requires=REQUIRED,
70 |     include_package_data=True,
71 |     license='MIT',
72 |     classifiers=[
73 |         'License :: OSI Approved :: MIT License',
74 |         'Programming Language :: Python',
75 |         'Programming Language :: Python :: 3.6',
76 |         'Programming Language :: Python :: 3.7',
77 |         'Programming Language :: Python :: 3.8',
78 |         'Programming Language :: Python :: Implementation :: CPython',
79 |     ],
80 |     cmdclass={
81 |         'upload': UploadCommand,
82 |     },
83 | )
84 | 


--------------------------------------------------------------------------------
/utx/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | from utx.tag import Tag
4 | from utx import log
5 | from utx.setting import setting
6 | from utx.core import *
7 | from utx.runner import TestRunner
8 | 


--------------------------------------------------------------------------------
/utx/core.py:
--------------------------------------------------------------------------------
  1 | #! /usr/bin/env python
  2 | # -*- coding: UTF-8 -*-
  3 | import re
  4 | import functools
  5 | import time
  6 | import unittest
  7 | from .setting import setting
  8 | from . import log
  9 | from .tag import Tag
 10 | 
 11 | CASE_TAG_FLAG = "__case_tag__"
 12 | CASE_DATA_FLAG = "__case_data__"
 13 | CASE_DATA_UNPACK_FLAG = "__case_data_unpack__"
 14 | CASE_ID_FLAG = "__case_id__"
 15 | CASE_INFO_FLAG = "__case_info__"
 16 | CASE_RUN_INDEX_FlAG = "__case_run_index_flag__"
 17 | CASE_SKIP_FLAG = "__unittest_skip__"
 18 | CASE_SKIP_REASON_FLAG = "__unittest_skip_why__"
 19 | 
 20 | __all__ = ["skip", "skip_if", "data", "tag", "stop_patch", "run_case"]
 21 | 
 22 | 
 23 | def skip(reason):
 24 |     def wrap(func):
 25 |         return unittest.skip(reason)(func)
 26 | 
 27 |     return wrap
 28 | 
 29 | 
 30 | def skip_if(condition, reason):
 31 |     def wrap(func):
 32 |         return unittest.skipIf(condition, reason)(func)
 33 | 
 34 |     return wrap
 35 | 
 36 | 
 37 | def data(*values, unpack=True):
 38 |     """注入测试数据，可以做为测试用例的数据驱动
 39 |     1. 单一参数的测试用例
 40 |     @data(10001, 10002, 10003)
 41 |     def test_receive_bless_box(self, box_id):
 42 |         print(box_id)
 43 | 
 44 |     2. 多个参数的测试用例
 45 |     @data(["gold", 100], ["diamond", 500])
 46 |     def test_bless(self, bless_type, award):
 47 |         print(bless_type)
 48 |         print(award)
 49 | 
 50 |     3. 是否对测试数据进行解包
 51 |     @data({"gold": 1000, "diamond": 100}, {"gold": 2000, "diamond": 200}, unpack=False)
 52 |     def test_get_battle_reward(self, reward):
 53 |         print(reward)
 54 |         print("获得的钻石数量是：{}".format(reward['diamond']))
 55 | 
 56 |     :param values:测试数据
 57 |     :param unpack: 是否解包
 58 |     :return:
 59 |     """
 60 | 
 61 |     def wrap(func):
 62 |         if hasattr(func, CASE_DATA_FLAG):
 63 |             log.error("{}的测试数据只能初始化一次".format(func.__name__))
 64 |         else:
 65 |             setattr(func, CASE_DATA_FLAG, values)
 66 |             setattr(func, CASE_DATA_UNPACK_FLAG, unpack)
 67 |         return func
 68 | 
 69 |     return wrap
 70 | 
 71 | 
 72 | def tag(*tag_type):
 73 |     """指定测试用例的标签，可以作为测试用例分组使用，用例默认会有Tag.ALL标签，支持同时设定多个标签，如：
 74 |     @tag(Tag.V1_0_0, Tag.SMOKE)
 75 |     def test_func(self):
 76 |         pass
 77 | 
 78 |     :param tag_type:标签类型，在tag.py里边自定义
 79 |     :return:
 80 |     """
 81 | 
 82 |     def wrap(func):
 83 |         if not hasattr(func, CASE_TAG_FLAG):
 84 |             tags = {Tag.ALL}
 85 |             tags.update(tag_type)
 86 |             setattr(func, CASE_TAG_FLAG, tags)
 87 |         else:
 88 |             getattr(func, CASE_TAG_FLAG).update(tag_type)
 89 |         return func
 90 | 
 91 |     return wrap
 92 | 
 93 | 
 94 | def _handler(func):
 95 |     @functools.wraps(func)
 96 |     def wrap(*args, **kwargs):
 97 |         time.sleep(setting.execute_interval)
 98 |         msg = "start to test {} ({}/{})".format(getattr(func, CASE_INFO_FLAG),
 99 |                                                 getattr(func, CASE_RUN_INDEX_FlAG),
100 |                                                 Tool.actual_case_num)
101 |         log.info(msg)
102 |         result = func(*args, **kwargs)
103 |         return result
104 | 
105 |     return wrap
106 | 
107 | 
108 | class Tool:
109 |     actual_case_num = 0
110 |     total_case_num = 0
111 | 
112 |     @classmethod
113 |     def create_case_id(cls):
114 |         cls.total_case_num += 1
115 |         return cls.total_case_num
116 | 
117 |     @classmethod
118 |     def create_actual_run_index(cls):
119 |         cls.actual_case_num += 1
120 |         return cls.actual_case_num
121 | 
122 |     @staticmethod
123 |     def modify_func_name(func):
124 |         """修改函数名字，实现排序 eg test_fight ---> test_00001_fight
125 | 
126 |         :param func:
127 |         :return:
128 |         """
129 |         case_id = Tool.create_case_id()
130 |         setattr(func, CASE_ID_FLAG, case_id)
131 |         if setting.sort_case:
132 |             func_name = func.__name__.replace("test_", "test_{:05d}_".format(case_id))
133 |         else:
134 |             func_name = func.__name__
135 |         return func_name
136 | 
137 |     @staticmethod
138 |     def general_case_name_with_test_data(func_name, index, test_data):
139 |         if setting.full_case_name:
140 |             params_str = "_".join([str(_) for _ in test_data]).replace(".", "")
141 |             func_name += "_{:05d}_{}".format(index, params_str)
142 |         else:
143 |             func_name += "_{:05d}".format(index)
144 |         if len(func_name) > setting.max_case_name_len:
145 |             func_name = func_name[:setting.max_case_name_len] + "……"
146 |         return func_name
147 | 
148 |     @staticmethod
149 |     def create_case_with_case_data(func):
150 |         result = dict()
151 |         for index, test_data in enumerate(getattr(func, CASE_DATA_FLAG), 1):
152 |             if not hasattr(func, CASE_SKIP_FLAG):
153 |                 setattr(func, CASE_RUN_INDEX_FlAG, Tool.create_actual_run_index())
154 | 
155 |             func_name = Tool.modify_func_name(func)
156 |             if isinstance(test_data, list):
157 |                 func_name = Tool.general_case_name_with_test_data(func_name, index, test_data)
158 |                 if getattr(func, CASE_DATA_UNPACK_FLAG, None):
159 |                     result[func_name] = _handler(_feed_data(*test_data)(func))
160 |                 else:
161 |                     result[func_name] = _handler(_feed_data(test_data)(func))
162 | 
163 |             elif isinstance(test_data, dict):
164 |                 func_name = Tool.general_case_name_with_test_data(func_name, index, test_data.values())
165 |                 if getattr(func, CASE_DATA_UNPACK_FLAG, None):
166 |                     result[func_name] = _handler(_feed_data(**test_data)(func))
167 |                 else:
168 |                     result[func_name] = _handler(_feed_data(test_data)(func))
169 | 
170 |             elif isinstance(test_data, (int, str, bool, float)):
171 |                 func_name = Tool.general_case_name_with_test_data(func_name, index, [test_data])
172 |                 result[func_name] = _handler(_feed_data(test_data)(func))
173 | 
174 |             else:
175 |                 raise Exception("无法解析{}".format(test_data))
176 | 
177 |         return result
178 | 
179 |     @staticmethod
180 |     def create_case_without_case_data(func):
181 |         if not hasattr(func, CASE_SKIP_FLAG):
182 |             setattr(func, CASE_RUN_INDEX_FlAG, Tool.create_actual_run_index())
183 | 
184 |         result = dict()
185 |         func_name = Tool.modify_func_name(func)
186 |         if len(func_name) > setting.max_case_name_len:
187 |             func_name = func_name[:setting.max_case_name_len] + "……"
188 |         result[func_name] = _handler(func)
189 |         return result
190 | 
191 |     @staticmethod
192 |     def filter_test_case(funcs_dict):
193 |         funcs = dict()
194 |         cases = dict()
195 |         for i in funcs_dict:
196 |             if i.startswith("test_"):
197 |                 cases[i] = funcs_dict[i]
198 |             else:
199 |                 funcs[i] = funcs_dict[i]
200 | 
201 |         return funcs, cases
202 | 
203 | 
204 | def _feed_data(*args, **kwargs):
205 |     def wrap(func):
206 |         @functools.wraps(func)
207 |         def _wrap(self):
208 |             return func(self, *args, **kwargs)
209 | 
210 |         return _wrap
211 | 
212 |     return wrap
213 | 
214 | 
215 | class Meta(type):
216 |     def __new__(cls, clsname, bases, attrs):
217 |         funcs, cases = Tool.filter_test_case(attrs)
218 |         for test_case in cases.values():
219 |             if not hasattr(test_case, CASE_TAG_FLAG):
220 |                 setattr(test_case, CASE_TAG_FLAG, {Tag.ALL})  # 没有指定tag的用例，默认带有tag：ALL
221 | 
222 |             # 注入用例信息
223 |             case_info = "{}.{}".format(test_case.__module__, test_case.__name__)
224 |             setattr(test_case, CASE_INFO_FLAG, case_info)
225 | 
226 |             # 检查用例描述
227 |             if setting.check_case_doc and not test_case.__doc__:
228 |                 log.warn("{}没有用例描述".format(case_info))
229 | 
230 |             # 过滤不执行的用例
231 |             if not getattr(test_case, CASE_TAG_FLAG) & set(setting.run_case):
232 |                 continue
233 | 
234 |             # 注入测试数据
235 |             if hasattr(test_case, CASE_DATA_FLAG):
236 |                 funcs.update(Tool.create_case_with_case_data(test_case))
237 |             else:
238 |                 funcs.update(Tool.create_case_without_case_data(test_case))
239 | 
240 |         return super(Meta, cls).__new__(cls, clsname, bases, funcs)
241 | 
242 | 
243 | class _TestCase(unittest.TestCase, metaclass=Meta):
244 |     def shortDescription(self):
245 |         """覆盖父类的方法，获取函数的注释
246 | 
247 |         :return:
248 |         """
249 |         doc = self._testMethodDoc
250 |         doc = doc and doc.split()[0].strip() or None
251 |         return doc
252 | 
253 | 
254 | TestCaseBackup = unittest.TestCase
255 | unittest.TestCase = _TestCase
256 | 
257 | 
258 | def stop_patch():
259 |     unittest.TestCase = TestCaseBackup
260 | 
261 | 
262 | def run_case(case_class, case_name: str):
263 |     setting.execute_interval = 0.3
264 |     r = re.compile(case_name.replace("test_", "test(_\d+)?_"))
265 |     suite = unittest.TestSuite()
266 |     for i in unittest.TestLoader().loadTestsFromTestCase(case_class):
267 |         if r.match(getattr(i, "_testMethodName")):
268 |             suite.addTest(i)
269 |     unittest.TextTestRunner(verbosity=0).run(suite)
270 | 


--------------------------------------------------------------------------------
/utx/log.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | """
 4 | Created by jianbing on 2017-10-30
 5 | """
 6 | import sys
 7 | import logging.handlers
 8 | from colorama import Fore, Style
 9 | 
10 | _logger = logging.getLogger('utx')
11 | _logger.setLevel(logging.DEBUG)
12 | _handler = logging.StreamHandler(sys.stdout)
13 | _handler.setFormatter(logging.Formatter('%(asctime)s %(message)s'))
14 | _logger.addHandler(_handler)
15 | 
16 | 
17 | def debug(msg):
18 |     _logger.debug("DEBUG " + str(msg))
19 | 
20 | 
21 | def info(msg):
22 |     _logger.info(Fore.GREEN + "INFO " + str(msg) + Style.RESET_ALL)
23 | 
24 | 
25 | def error(msg):
26 |     _logger.error(Fore.RED + "ERROR " + str(msg) + Style.RESET_ALL)
27 | 
28 | 
29 | def warn(msg):
30 |     _logger.warning(Fore.YELLOW + "WARNING " + str(msg) + Style.RESET_ALL)
31 | 
32 | 
33 | def _print(msg):
34 |     _logger.debug(Fore.BLUE + "PRINT " + str(msg) + Style.RESET_ALL)
35 | 
36 | 
37 | def set_level(level):
38 |     """ 设置log级别
39 | 
40 |     :param level: logging.DEBUG, logging.INFO, logging.WARN, logging.ERROR
41 |     :return:
42 |     """
43 |     _logger.setLevel(level)
44 | 
45 | 
46 | def set_level_to_debug():
47 |     _logger.setLevel(logging.DEBUG)
48 | 
49 | 
50 | def set_level_to_info():
51 |     _logger.setLevel(logging.INFO)
52 | 
53 | 
54 | def set_level_to_warn():
55 |     _logger.setLevel(logging.WARN)
56 | 
57 | 
58 | def set_level_to_error():
59 |     _logger.setLevel(logging.ERROR)
60 | 


--------------------------------------------------------------------------------
/utx/report/__init__.py:
--------------------------------------------------------------------------------
1 | #! /usr/bin/env python
2 | # -*- coding: UTF-8 -*-
3 | """
4 | Created by Administrator on 2019-03-12
5 | """


--------------------------------------------------------------------------------
/utx/report/style_1.py:
--------------------------------------------------------------------------------
  1 | #! /usr/bin/env python
  2 | # -*- coding: UTF-8 -*-
  3 | 
  4 | html_head = r"""
  5 | <?xml version="1.0" encoding="UTF-8"?>
  6 | <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
  7 | <html xmlns="http://www.w3.org/1999/xhtml">
  8 | <head>
  9 |     <title>{report_name}</title>
 10 |     <meta http-equiv="Content-Type" content="text/html; charset=UTF-8"/>
 11 |     <link href="http://libs.baidu.com/bootstrap/3.0.3/css/bootstrap.min.css" rel="stylesheet">
 12 |     <script src="http://libs.baidu.com/jquery/2.0.0/jquery.min.js"></script>
 13 |     <script src="http://libs.baidu.com/bootstrap/3.0.3/js/bootstrap.min.js"></script>
 14 | 
 15 |     <style type="text/css" media="screen">
 16 |         pre {{
 17 |             white-space: pre-wrap;
 18 |             white-space: -moz-pre-wrap;
 19 |             white-space: -pre-wrap;
 20 |             white-space: -o-pre-wrap;
 21 |             word-wrap: break-word;
 22 |             text-align: left;
 23 |         }}
 24 |     
 25 |         body {{
 26 |             font-family: Microsoft YaHei, Tahoma, arial, helvetica, sans-serif;
 27 |             padding: 20px;
 28 |             font-size: 80%;
 29 |         }}
 30 | 
 31 |         table {{
 32 |             font-size: 100%;
 33 |         }}
 34 | 
 35 |         /* -- heading ---------------------------------------------------------------------- */
 36 |         .heading {{
 37 |             margin-top: 0ex;
 38 |             margin-bottom: 1ex;
 39 |         }}
 40 | 
 41 |         .heading .description {{
 42 |             margin-top: 4ex;
 43 |             margin-bottom: 6ex;
 44 |         }}
 45 | 
 46 |         /* -- report ------------------------------------------------------------------------ */
 47 |         #total_row {{
 48 |             font-weight: bold;
 49 |         }}
 50 | 
 51 |         .passCase {{
 52 |             color: #5cb85c;
 53 |         }}
 54 | 
 55 |         .failCase {{
 56 |             color: #f0ad4e;
 57 |             font-weight: bold;
 58 |         }}
 59 | 
 60 |         .errorCase {{        
 61 |             color: #d9534f;
 62 |             font-weight: bold;
 63 |         }}
 64 | 
 65 |         .hiddenRow {{
 66 |             display: none;
 67 |         }}
 68 | 
 69 |         .testcase {{
 70 |             margin-left: 2em;
 71 |         }}
 72 |     </style>
 73 | 
 74 | </head>
 75 | <script language="javascript" type="text/javascript">
 76 |     output_list = Array();
 77 | 
 78 | 
 79 |     function showCase(level) {{
 80 |         trs = document.getElementsByTagName("tr");
 81 |         for (var i = 0; i < trs.length; i++) {{
 82 |             tr = trs[i];
 83 |             id = tr.id;
 84 |             if (id.substr(0, 2) == 'pt') {{
 85 |                 if (level == 1 || level == 4) {{
 86 |                     tr.className = '';
 87 |                 }}
 88 |                 else {{
 89 |                     tr.className = 'hiddenRow';                    
 90 |                 }}
 91 |             }}
 92 |             if (id.substr(0, 2) == 'ft') {{
 93 |                 if (level == 2 || level == 4) {{
 94 |                     tr.className = '';
 95 |                 }}
 96 |                 else {{
 97 |                     tr.className = 'hiddenRow';
 98 |                 }}
 99 |             }}
100 |             
101 |             if (id.substr(0, 2) == 'et') {{
102 |                 if (level == 3 || level == 4) {{
103 |                     tr.className = '';
104 |                 }}
105 |                 else {{
106 |                     tr.className = 'hiddenRow';
107 |                 }}
108 |             }}
109 |         }}
110 | 
111 |         detail_class = document.getElementsByClassName('detail');
112 |         //console.log(detail_class.length)
113 |         if (level == 3) {{
114 |             for (var i = 0; i < detail_class.length; i++) {{
115 |                 detail_class[i].innerHTML = "收起"
116 |             }}
117 |         }}
118 |         else {{
119 |             for (var i = 0; i < detail_class.length; i++) {{
120 |                 detail_class[i].innerHTML = "详细"
121 |             }}
122 |         }}
123 |     }}
124 | 
125 |     function showClassDetail(cid, count) {{
126 |         var id_list = Array(count);
127 |         var toHide = 1;
128 |         for (var i = 0; i < count; i++) {{
129 |             tid0 = 't' + cid.substr(1) + '_' + (i + 1);
130 |             console.log(tid0);
131 |             tid = 'f' + tid0;
132 |             tr = document.getElementById(tid);
133 |             if (!tr) {{
134 |                 tid = 'p' + tid0;
135 |                 tr = document.getElementById(tid);
136 |             }}
137 |             if (!tr) {{
138 |                 tid = 'e' + tid0;
139 |                 tr = document.getElementById(tid);
140 |             }}
141 |             id_list[i] = tid;
142 |             if (tr.className) {{
143 |                 toHide = 0;
144 |             }}
145 |         }}
146 |         for (var i = 0; i < count; i++) {{
147 |             tid = id_list[i];
148 |             if (toHide) {{
149 |                 document.getElementById(tid).className = 'hiddenRow';
150 |                 document.getElementById(cid).innerText = "详细"
151 |             }}
152 |             else {{
153 |                 document.getElementById(tid).className = '';
154 |                 document.getElementById(cid).innerText = "收起"
155 |             }}
156 |         }}
157 |     }}
158 | 
159 |     function html_escape(s) {{
160 |         s = s.replace(/&/g, '&amp;');
161 |         s = s.replace(/</g, '&lt;');
162 |         s = s.replace(/>/g, '&gt;');
163 |         return s;
164 |     }}
165 | </script>
166 | """
167 | 
168 | html_body = """
169 | 
170 | <body>
171 | 
172 | <div class='heading'>
173 |     <h1 style="font-family: Microsoft YaHei">{report_name}</h1>
174 |     <p class='attribute'><strong>开始时间 : </strong> {begin_time}</p>
175 |     <p class='attribute'><strong>合计耗时 : </strong> {total_time}</p>
176 |     <p class='attribute'><strong>测试结果 : </strong> 共 {test_all}，通过 {test_pass}，失败 {test_fail}，错误 {test_error}，跳过 {test_skip}，通过率= {test_pass_per}</p>
177 | 
178 |     <p class='description'></p>
179 | </div>
180 | 
181 | 
182 | <p id='show_detail_line'>
183 |     <a class="btn btn-primary" href='javascript:showCase(0)'>概要{{ {test_pass_per} }}</a>
184 |     <a class="btn btn-success" href='javascript:showCase(1)'>通过{{ {test_pass} }}</a>
185 |     <a class="btn btn-warning" href='javascript:showCase(2)'>失败{{ {test_fail} }}</a>
186 |     <a class="btn btn-danger" href='javascript:showCase(3)'>错误{{ {test_error} }}</a>
187 |     <a class="btn btn-info" href='javascript:showCase(4)'>所有{{ {test_all} }}</a>
188 | </p>
189 | <table id='result_table' class="table table-condensed table-bordered table-hover">
190 |     <colgroup>
191 |         <col align='left'/>
192 |         <col align='right'/>
193 |         <col align='right'/>
194 |         <col align='right'/>
195 |         <col align='right'/>
196 |         <col align='right'/>
197 |         <col align='right'/>
198 |     </colgroup>
199 |     <tr id='header_row' class="text-center success" style="font-weight: bold;font-size: 14px;">
200 |         <td>用例集/测试用例</td>
201 |         <td>总计</td>
202 |         <td>通过</td>
203 |         <td>失败</td>
204 |         <td>错误</td>
205 |         <td>跳过</td>
206 |         <td>详细</td>
207 |     </tr>
208 | 
209 |  
210 |     {case_info}
211 | 
212 |     <tr id='total_row' class="text-center active">
213 |         <td>总计</td>
214 |         <td>{test_all}</td>
215 |         <td>{test_pass}</td>
216 |         <td>{test_fail}</td>
217 |         <td>{test_error}</td>
218 |         <td>{test_skip}</td>
219 |         <td>通过率：{test_pass_per}</td>
220 |     </tr>
221 | </table>
222 | 
223 | <div id='ending'>&nbsp;</div>
224 | <div style=" position:fixed;right:50px; bottom:30px; width:20px; height:20px;cursor:pointer">
225 |     <a href="#"><span class="glyphicon glyphicon-eject" style="font-size:30px;" aria-hidden="true">
226 |     </span></a></div>
227 | 
228 | 
229 | </body>
230 | </html>
231 | """
232 | 
233 | html_case_group = """
234 | <tr class="failClass warning">
235 |         <td>{group_name}</td>
236 |         <td class="text-center">{test_all}</td>
237 |         <td class="text-center">{test_pass}</td>
238 |         <td class="text-center">{test_fail}</td>
239 |         <td class="text-center">{test_error}</td>
240 |         <td class="text-center">{test_skip}</td>
241 |         <td class="text-center"><a href="javascript:showClassDetail('c{group_index}',{test_all})" class="detail" id="c{group_index}">详细</a></td>
242 |     </tr>
243 | """
244 | 
245 | html_case_skip = """
246 | <tr id="pt{case_id}" class="">
247 |         <td class="passCase">
248 |             <div class="testcase">{method_name}</div>
249 |         </td>
250 |         <td colspan="6" align="center">
251 |         <span class="label label-info success">跳过</span>
252 |         </td>
253 |     </tr>
254 | """
255 | 
256 | html_case_pass_no_log = """
257 | <tr id="pt{case_id}" class="">
258 |         <td class="passCase">
259 |             <div class="testcase">{method_name}</div>
260 |         </td>
261 |         <td colspan="6" align="center">
262 |         <span class="label label-success success">通过</span>
263 |         </td>
264 |     </tr>
265 | """
266 | 
267 | html_case_pass_with_log = """
268 | <tr id="pt{case_id}" class="">
269 |         <td class="passCase">
270 |             <div class="testcase">{method_name}</div>
271 |         </td>
272 |         <td colspan="6" align="center">
273 |        
274 |         <button id='btn_pt{case_id}' type="button"  class="btn btn-success btn-xs collapsed" data-toggle="collapse" data-target='#div_pt{case_id}'>通过</button>
275 |         <div id='div_pt{case_id}' class="collapse">
276 |                 <pre>{case_log}</pre>
277 |             </div>
278 |         </td>
279 |     </tr>
280 | """
281 | 
282 | html_case_fail = """
283 | <tr id="ft{case_id}" class="">
284 |         <td class="failCase">
285 |             <div class="testcase">{method_name}</div>
286 |         </td>
287 |         <td colspan="6" align="center">
288 |             <!--默认收起错误信息
289 |             <button id='btn_ft1_2' type="button"  class="btn btn-danger btn-xs collapsed" data-toggle="collapse" data-target='#div_ft1_2'>失败</button>
290 |             <div id='div_ft1_2' class="collapse">  -->
291 | 
292 |             <!-- 默认展开错误信息 -->
293 |             <button id="btn_ft{case_id}" type="button" class="btn btn-warning btn-xs" data-toggle="collapse" data-target="#div_ft{case_id}">失败</button>
294 |             <div id="div_ft{case_id}" class="collapse in">
295 |                 <pre>{case_log}</pre>
296 |             </div>
297 |         </td>
298 |     </tr>
299 | """
300 | 
301 | html_case_error = """
302 | <tr id="et{case_id}" class="">
303 |     <td class="errorCase"><div class="testcase">{method_name}</div></td>
304 |     <td colspan="6" align="center">
305 |     <!--默认收起错误信息
306 |     <button id='btn_et1_5' type="button"  class="btn btn-danger btn-xs collapsed" data-toggle="collapse" data-target='#div_et1_5'>错误</button>
307 |     <div id='div_et1_5' class="collapse">  -->
308 | 
309 |     <!-- 默认展开错误信息 -->
310 |     <button id="btn_et{case_id}" type="button" class="btn btn-danger btn-xs" data-toggle="collapse" data-target="#div_et{case_id}">错误</button>
311 |     <div id="div_et{case_id}" class="collapse in">
312 |         <pre>{case_log}</pre>
313 |     </div>
314 |     </td>
315 | </tr>
316 | """
317 | 
318 | 
319 | def build_report(file_path, data):
320 |     html = html_head.format(report_name=data['reportName'])
321 | 
322 |     case_info = ""
323 |     from collections import defaultdict
324 |     filter_case = defaultdict(list)
325 |     for i in data['testResult']:
326 |         filter_case[i["className"]].append(i)
327 | 
328 |     for group_index, group in enumerate(filter_case.keys(), 1):
329 |         pass_num = len([i for i in filter_case[group] if i['status'] == '成功'])
330 |         fail_num = len([i for i in filter_case[group] if i['status'] == '失败'])
331 |         error_num = len([i for i in filter_case[group] if i['status'] == '错误'])
332 |         skip_num = len([i for i in filter_case[group] if i['status'] == '跳过'])
333 |         case_info += html_case_group.format(group_name=group,
334 |                                             group_index=group_index,
335 |                                             test_all=pass_num + fail_num + error_num + skip_num,
336 |                                             test_pass=pass_num,
337 |                                             test_fail=fail_num,
338 |                                             test_error=error_num,
339 |                                             test_skip=skip_num
340 |                                             )
341 |         cases = filter_case[group]
342 |         for case_index, case in enumerate(cases, 1):
343 |             if case['status'] == '成功':
344 |                 if case['log']:
345 |                     case_info += html_case_pass_with_log.format(case_id="{}_{}".format(group_index, case_index),
346 |                                                                 method_name="{}（{}）".format(case['methodName'],
347 |                                                                                             case['description']),
348 |                                                                 case_log=case['log'])
349 |                 else:
350 |                     case_info += html_case_pass_no_log.format(case_id="{}_{}".format(group_index, case_index),
351 |                                                               method_name="{}（{}）".format(case['methodName'],
352 |                                                                                           case['description']),
353 |                                                               )
354 |             elif case['status'] == '失败':
355 |                 case_info += html_case_fail.format(case_id="{}_{}".format(group_index, case_index),
356 |                                                    method_name="{}（{}）".format(case['methodName'],
357 |                                                                                case['description']),
358 |                                                    case_log=case['log'])
359 |             elif case['status'] == '错误':
360 |                 case_info += html_case_error.format(case_id="{}_{}".format(group_index, case_index),
361 |                                                     method_name="{}（{}）".format(case['methodName'],
362 |                                                                                 case['description']),
363 |                                                     case_log=case['log'])
364 |             elif case['status'] == '跳过':
365 |                 case_info += html_case_skip.format(case_id="{}_{}".format(group_index, case_index),
366 |                                                    method_name="{}（{}）".format(case['methodName'],
367 |                                                                                case['description']),
368 |                                                    )
369 | 
370 |     html += html_body.format(begin_time=data['beginTime'],
371 |                              total_time=data['totalTime'],
372 |                              test_all=data['testAll'],
373 |                              test_pass=data['testPass'],
374 |                              test_fail=data['testFail'],
375 |                              test_skip=data['testSkip'],
376 |                              test_error=data['testError'],
377 |                              test_pass_per="{:.2%}".format(data['testPass'] / data['testAll']),
378 |                              report_name=data['reportName'],
379 |                              case_info=case_info,
380 |                              )
381 | 
382 |     with open(file_path, 'w', encoding='utf-8') as f:
383 |         f.write(html)
384 | 


--------------------------------------------------------------------------------
/utx/runner.py:
--------------------------------------------------------------------------------
  1 | #! /usr/bin/env python
  2 | # -*- coding: UTF-8 -*-
  3 | import time
  4 | import os
  5 | import io
  6 | import datetime
  7 | import sys
  8 | import unittest
  9 | from utx.core import Tool
 10 | from . import log, setting
 11 | 
 12 | result_data = dict()
 13 | result_data['testResult'] = []
 14 | current_class_name = ""
 15 | STATUS = {
 16 |     0: 'Pass',
 17 |     1: 'Fail',
 18 |     2: 'Error',
 19 |     3: 'Skip',
 20 | }
 21 | 
 22 | 
 23 | class _TestResult(unittest.TestResult):
 24 |     def __init__(self, verbosity=1):
 25 |         super().__init__(verbosity)
 26 |         self.outputBuffer = io.StringIO()
 27 |         self.raw_stdout = None
 28 |         self.raw_stderr = None
 29 |         self.success_count = 0
 30 |         self.failure_count = 0
 31 |         self.skip_count = 0
 32 |         self.error_count = 0
 33 |         self.verbosity = verbosity
 34 |         self.result = []
 35 |         self._case_start_time = 0
 36 |         self._case_run_time = 0
 37 | 
 38 |     def startTest(self, test):
 39 |         self._case_start_time = time.time()
 40 |         super().startTest(test)
 41 |         self.raw_stdout = sys.stdout
 42 |         self.raw_stderr = sys.stderr
 43 |         sys.stdout = self.outputBuffer
 44 |         sys.stderr = self.outputBuffer
 45 | 
 46 |     def complete_output(self):
 47 |         self._case_run_time = time.time() - self._case_start_time
 48 |         if self.raw_stdout:
 49 |             sys.stdout = self.raw_stdout
 50 |             sys.stderr = self.raw_stderr
 51 | 
 52 |         result = self.outputBuffer.getvalue()
 53 |         self.outputBuffer.seek(0)
 54 |         self.outputBuffer.truncate()
 55 |         if result and setting.show_print_in_console:
 56 |             log._print(result.strip())
 57 |         return result
 58 | 
 59 |     def stopTest(self, test):
 60 |         self.complete_output()
 61 | 
 62 |     def addSuccess(self, test):
 63 |         self.success_count += 1
 64 |         super().addSuccess(test)
 65 |         output = self.complete_output()
 66 |         self.result.append((0, test, output, '', self._case_run_time))
 67 | 
 68 |     def addError(self, test, err):
 69 |         self.error_count += 1
 70 |         super().addError(test, err)
 71 |         _, _exc_str = self.errors[-1]
 72 |         output = self.complete_output()
 73 |         self.result.append((2, test, output, _exc_str, self._case_run_time))
 74 |         log.error('TestCase Error')
 75 |         if setting.show_error_traceback:
 76 |             log.error(_exc_str)
 77 | 
 78 |     def addSkip(self, test, reason):
 79 |         self.skip_count += 1
 80 |         super().addSkip(test, reason)
 81 |         self.result.append((3, test, "", "", 0.0))
 82 | 
 83 |     def addFailure(self, test, err):
 84 |         self.failure_count += 1
 85 |         super().addFailure(test, err)
 86 |         _, _exc_str = self.failures[-1]
 87 |         output = self.complete_output()
 88 |         self.result.append((1, test, output, _exc_str, self._case_run_time))
 89 |         log.error('TestCase Failed')
 90 |         if setting.show_error_traceback:
 91 |             log.error(_exc_str)
 92 | 
 93 | 
 94 | class _TestRunner:
 95 |     def __init__(self, report_title, report_dir, verbosity=1, description=""):
 96 |         self.report_dir = report_dir
 97 |         self.verbosity = verbosity
 98 |         self.title = report_title
 99 |         self.description = description
100 |         self.start_time = datetime.datetime.now()
101 |         self.stop_time = None
102 | 
103 |     def run(self, test):
104 |         msg = "开始测试，用例数量总共{}个，跳过{}个，实际运行{}个"
105 |         log.info(msg.format(Tool.total_case_num,
106 |                             Tool.total_case_num - Tool.actual_case_num,
107 |                             Tool.actual_case_num))
108 |         result = _TestResult(self.verbosity)
109 |         test(result)
110 |         self.stop_time = datetime.datetime.now()
111 |         self.analyze_test_result(result)
112 |         log.info('Time Elapsed: {}'.format(self.stop_time - self.start_time))
113 | 
114 |         from utx.report import style_1, style_2
115 |         if setting.create_report_by_style_1:
116 |             file_path = os.path.join(self.report_dir,
117 |                                      r"{}-style-1.html".format(self.start_time.strftime("%Y-%m-%d-%H-%M-%S")))
118 |             style_1.build_report(file_path, result_data)
119 | 
120 |         if setting.create_report_by_style_2:
121 |             file_path = os.path.join(self.report_dir,
122 |                                      r"{}-style-2.html".format(self.start_time.strftime("%Y-%m-%d-%H-%M-%S")))
123 |             style_2.build_report(file_path, result_data)
124 | 
125 |     @staticmethod
126 |     def sort_result(case_results):
127 |         rmap = {}
128 |         classes = []
129 |         for n, t, o, e, run_time in case_results:
130 |             cls = t.__class__
131 |             if cls not in rmap:
132 |                 rmap[cls] = []
133 |                 classes.append(cls)
134 |             rmap[cls].append((n, t, o, e, run_time))
135 |         r = [(cls, rmap[cls]) for cls in classes]
136 |         return r
137 | 
138 |     def analyze_test_result(self, result):
139 |         result_data["reportName"] = self.title
140 |         result_data["beginTime"] = str(self.start_time)[:19]
141 |         result_data["totalTime"] = str(self.stop_time - self.start_time)
142 | 
143 |         sorted_result = self.sort_result(result.result)
144 |         for cid, (cls, cls_results) in enumerate(sorted_result):
145 |             pass_num = fail_num = error_num = skip_num = 0
146 |             for case_state, *_ in cls_results:
147 |                 if case_state == 0:
148 |                     pass_num += 1
149 |                 elif case_state == 1:
150 |                     fail_num += 1
151 |                 elif case_state == 2:
152 |                     error_num += 1
153 |                 else:
154 |                     skip_num += 1
155 | 
156 |             name = "{}.{}".format(cls.__module__, cls.__name__)
157 |             global current_class_name
158 |             current_class_name = name
159 | 
160 |             for tid, (state_id, t, o, e, run_time) in enumerate(cls_results):
161 | 
162 |                 name = t.id().split('.')[-1]
163 |                 doc = t.shortDescription() or ""
164 |                 case_data = dict()
165 |                 case_data['className'] = current_class_name
166 |                 case_data['methodName'] = name
167 |                 case_data['spendTime'] = "{:.2}S".format(run_time)
168 |                 case_data['description'] = doc
169 |                 case_data['log'] = o + e
170 |                 if STATUS[state_id] == "Pass":
171 |                     case_data['status'] = "成功"
172 |                 if STATUS[state_id] == "Fail":
173 |                     case_data['status'] = "失败"
174 |                 if STATUS[state_id] == "Error":
175 |                     case_data['status'] = "错误"
176 |                 if STATUS[state_id] == "Skip":
177 |                     case_data['status'] = "跳过"
178 |                 result_data['testResult'].append(case_data)
179 | 
180 |         result_data["testPass"] = result.success_count
181 |         result_data["testAll"] = result.success_count + result.failure_count + result.error_count + result.skip_count
182 |         result_data["testFail"] = result.failure_count
183 |         result_data["testSkip"] = result.skip_count
184 |         result_data["testError"] = result.error_count
185 | 
186 | 
187 | class TestRunner:
188 | 
189 |     def __init__(self):
190 |         self.case_dirs = []
191 | 
192 |     def add_case_dir(self, dir_path):
193 |         """添加测试用例文件夹，多次调用可以添加多个文件夹，会按照文件夹的添加顺序执行用例
194 | 
195 |             runner = TestRunner()
196 |             runner.add_case_dir(r"testcase\chat")
197 |             runner.add_case_dir(r"testcase\battle")
198 |             runner.run_test(report_title='接口自动化测试报告')
199 | 
200 |         :param dir_path:
201 |         :return:
202 |         """
203 |         if not os.path.exists(dir_path):
204 |             raise Exception("测试用例文件夹不存在：{}".format(dir_path))
205 |         if dir_path in self.case_dirs:
206 |             log.warn("测试用例文件夹已经存在了：{}".format(dir_path))
207 |         else:
208 |             self.case_dirs.append(dir_path)
209 | 
210 |     def run_test(self, report_title='接口自动化测试报告'):
211 | 
212 |         if not self.case_dirs:
213 |             raise Exception("请先调用add_case_dir方法，添加测试用例文件夹")
214 | 
215 |         if not os.path.exists("report"):
216 |             os.mkdir("report")
217 | 
218 |         report_dir = os.path.abspath("report")
219 |         suite = unittest.TestSuite()
220 |         for case_path in self.case_dirs:
221 |             suite.addTests(unittest.TestLoader().discover(case_path))
222 |         _TestRunner(report_dir=report_dir, report_title=report_title).run(suite)
223 | 
224 |         print("测试完成，请查看报告")
225 |         os.system("start report")
226 | 


--------------------------------------------------------------------------------
/utx/setting.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | """
 4 | Created by jianbing on 2017-11-04
 5 | """
 6 | from utx.tag import Tag
 7 | 
 8 | 
 9 | class setting:
10 |     # 只运行的用例类型
11 |     run_case = {Tag.ALL}
12 | 
13 |     # 开启用例排序
14 |     sort_case = True
15 | 
16 |     # 每个用例的执行间隔，单位是秒
17 |     execute_interval = 0.1
18 | 
19 |     # 开启检测用例描述
20 |     check_case_doc = True
21 | 
22 |     # 显示完整用例名字（函数名字+参数信息）
23 |     full_case_name = False
24 | 
25 |     # 测试报告显示的用例名字最大程度
26 |     max_case_name_len = 80
27 | 
28 |     # 执行用例的时候，显示报错信息
29 |     show_error_traceback = True
30 | 
31 |     # 测试报告样式1
32 |     create_report_by_style_1 = True
33 | 
34 |     # 测试报告样式2
35 |     create_report_by_style_2 = True
36 | 
37 |     # 在控制台显示print打印的内容
38 |     show_print_in_console = False
39 | 


--------------------------------------------------------------------------------
/utx/tag.py:
--------------------------------------------------------------------------------
 1 | #! /usr/bin/env python
 2 | # -*- coding: UTF-8 -*-
 3 | """
 4 | 测试用例标签类
 5 | """
 6 | 
 7 | from enum import Enum, unique
 8 | 
 9 | 
10 | class NewTag:
11 |     def __init__(self, desc=""):
12 |         self.desc = desc
13 | 
14 | 
15 | @unique
16 | class Tag(Enum):
17 |     SMOKE = NewTag("冒烟")  # 冒烟测试标记，可以重命名，不要删除
18 |     ALL = NewTag("完整")  # 完整测试标记，可以重命名，不要删除
19 | 
20 |     # 以下开始为扩展标签，自行调整
21 |     V1_0_0 = NewTag("V1.0.0版本")
22 |     V2_0_0 = NewTag("V2.0.0版本")
23 | 
24 | 


--------------------------------------------------------------------------------