-
Notifications
You must be signed in to change notification settings - Fork 29
/
Copy pathFreemind.py
241 lines (213 loc) · 11.8 KB
/
Freemind.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
# -*- coding: utf-8 -*-
import logging.config
import os
from xml.etree import ElementTree as ET
from lxml import etree as lxmlET
import sys
def CDATA(text=None):
element = ET.Element('![CDATA[')
element.text = text
return element
def get_freemindfile(path):
''' for windows'''
L = []
for root, dirs, files in os.walk(path):
for file in files:
if os.path.splitext(file)[1] == '.mm':
# L.append(os.path.join(root, file))
L.append(path + '/' + file)
if len(L) == 0:
logging.error(path + '\nThere is no FreeMind file,Please check out!')
return L
class FreeMind(object):
def __init__(self, logger, ):
self.logger = logger
self.log_prefix = 'FreeMind:'
def Generate_TCs_from_TDS(self, file, node_list='', reuse_tc='0', link_rule='BY_TDS_ID_THEN_SMART_NAME',
suite_structure='TDS'):
"""
The generated xml file need to be imported into testlink manually.
"""
self.logger.info(self.log_prefix + "Starting to export %s to xml file." % file)
tc_tds_dict = {}
tc_pfs_dict = {}
fm_tree = lxmlET.parse(file)
tds_root = fm_tree.getroot()
node_list = node_list.split('|')
node_list = [item.strip() for item in node_list]
tc_root = lxmlET.Element('testsuite', {'name': ''})
lxmlET.SubElement(tc_root, 'node_order').text = lxmlET.CDATA('')
lxmlET.SubElement(tc_root, 'details').text = lxmlET.CDATA('')
self._gen_tc_xml_from_tds(tc_root, tds_root, tc_tds_dict, tc_pfs_dict, node_list, reuse_tc, link_rule,
suite_structure)
xml_file_name = file[:-2] + 'xml'
f = open(xml_file_name, 'wb')
f.write(lxmlET.tostring(tc_root, xml_declaration=True, encoding='utf-8', pretty_print=True))
f.close()
self.logger.info(self.log_prefix + \
"Successfully generated the test cases xml file (%s).\n\n\n\n" % xml_file_name)
def _gen_tc_xml_from_tds(self, ts_node, root_node, tc_tds_dict, tc_pfs_dict, node_list,
reuse_tc, link_rule, suite_structure):
existing_tc_list = []
if node_list == ['']:
self.logger.info(self.log_prefix + \
"Generating test cases xml file for all TDS nodes.")
self._gen_tc_xml_from_tds_node(ts_node, root_node, tc_tds_dict, tc_pfs_dict, existing_tc_list,
reuse_tc, link_rule, suite_structure)
return
for tds_item in root_node.iter('node'):
if tds_item.attrib['ID'] in node_list:
self.logger.info(self.log_prefix + \
"Generating test cases xml file for TDS node (%s)." % \
(tds_item.attrib['ID']))
child_ts_node = lxmlET.SubElement(ts_node, 'testsuite', {'name': tds_item.attrib['TEXT'].strip()})
lxmlET.SubElement(child_ts_node, 'node_order').text = lxmlET.CDATA('')
lxmlET.SubElement(child_ts_node, 'details').text = lxmlET.CDATA('')
self._gen_tc_xml_from_tds_node(child_ts_node, tds_item, tc_tds_dict, tc_pfs_dict, existing_tc_list,
reuse_tc, link_rule, suite_structure)
def _gen_tc_xml_from_tds_node(self, ts_node, root_node, tc_tds_dict, tc_pfs_dict, existing_tc_list,
reuse_tc, link_rule, suite_structure):
ts_node_order = -1
tc_node_order = -1
for tds_item in root_node.findall('node'):
if tds_item.attrib.has_key('LINK'):
continue
is_testsuite = False
has_steps = False
for item_icon in tds_item.findall('icon'):
if item_icon.attrib['BUILTIN'] == 'flag':
ts_node_order += 1
child_ts_node = lxmlET.SubElement(ts_node, 'testsuite', {'name': tds_item.attrib['TEXT'].strip()})
self.logger.info(
self.log_prefix + "Creating Test Suite [%s]" % tds_item.attrib['TEXT'].strip())
lxmlET.SubElement(child_ts_node, 'node_order').text = lxmlET.CDATA(str(ts_node_order))
lxmlET.SubElement(child_ts_node, 'details').text = lxmlET.CDATA('')
# ts_node = child_ts_node
is_testsuite = True
break
'''如果有星星的，后面的是步骤'''
if item_icon in tds_item.findall('icon'):
if item_icon.attrib['BUILTIN'] == 'bookmark':
tc_node_order += 1
step_ts_node = lxmlET.SubElement(ts_node, 'testcase',
{'name': str(tds_item.attrib['TEXT']).split('&')[0]})
has_steps = True
break
if is_testsuite:
self._gen_tc_xml_from_tds_node(child_ts_node, tds_item, tc_tds_dict, tc_pfs_dict, existing_tc_list,
reuse_tc, link_rule, suite_structure)
continue
'''存在步骤的，添加步骤内容'''
if has_steps:
tds_item_str = str(tds_item.attrib['TEXT']).split('&')
self.logger.info(
self.log_prefix + "Generating Test case -<has steps> %s" % tds_item_str[0])
lxmlET.SubElement(step_ts_node, 'node_order').text = lxmlET.CDATA(str(tc_node_order))
lxmlET.SubElement(step_ts_node, 'externalid').text = lxmlET.CDATA('')
lxmlET.SubElement(step_ts_node, 'version').text = lxmlET.CDATA('1')
lxmlET.SubElement(step_ts_node, 'summary').text = lxmlET.CDATA(tds_item_str[0])
if len(tds_item_str) > 1:
lxmlET.SubElement(step_ts_node, 'preconditions').text = lxmlET.CDATA(tds_item_str[1])
lxmlET.SubElement(step_ts_node, 'execution_type').text = lxmlET.CDATA('1')
node_reg_lvl = 2
for node_icon in tds_item.findall('icon'):
if node_icon.attrib['BUILTIN'].strip().count('full-') == 1:
node_reg_lvl = 4 - int(node_icon.attrib['BUILTIN'].strip()[-1])
if node_reg_lvl < 1:
node_reg_lvl = 2
lxmlET.SubElement(step_ts_node, 'importance').text = lxmlET.CDATA(str(node_reg_lvl))
# print(type(step_ts_node))
'''xml写入steps'''
steps = lxmlET.SubElement(step_ts_node, 'steps')
'''循环写入step'''
sn = 1
for item in tds_item.findall('node'):
step = lxmlET.SubElement(steps, 'step')
lxmlET.SubElement(step, 'step_number').text = lxmlET.CDATA(str(sn))
'''步骤和期望结果用 & 符号区分'''
stritem = str(item.attrib['TEXT']).split('&')
# print(stritem)
lxmlET.SubElement(step, 'actions').text = lxmlET.CDATA(stritem[0])
self.logger.info(self.log_prefix + "add Step actions %s" % stritem[0])
if len(stritem) > 1:
lxmlET.SubElement(step, 'expectedresults').text = lxmlET.CDATA(stritem[1])
self.logger.info(self.log_prefix + "add Expected Results ->%s" % stritem[1])
lxmlET.SubElement(step, 'execution_type').text = lxmlET.CDATA('1')
sn += 1
continue
if not self._last_tds_node(tds_item):
self._gen_tc_xml_from_tds_node(ts_node, tds_item, tc_tds_dict, tc_pfs_dict, existing_tc_list,
reuse_tc, link_rule, suite_structure)
continue
# This must be the last TDS node
tc_list = []
if not tc_list:
# There is no linked test case nodes (which mainly used for reusing test cases between projects)
if tds_item.attrib['ID'].strip() in existing_tc_list:
continue
existing_tc_list.append(tds_item.attrib['ID'].strip())
tc_node_order += 1
self._add_dummy_testcase(ts_node, tds_item, tc_tds_dict, tc_pfs_dict, tc_node_order)
# If this node already have test cases associated, update its traceability if necessary.
# Get the test case from original xml file and copy it into the new xml file
for tc_id in tc_list:
if tc_id in existing_tc_list:
continue
existing_tc_list.append(tc_id)
tc_node_order += 1
# tc_node = self._get_tc_node_from_xml_by_id(self.based_tc_url, tc_id)
if tc_node is None:
return
# res = self._update_tc_node(tc_node, tc_node_order, tds_item, tc_tds_dict, tc_pfs_dict, tc_id)
ts_node.append(tc_node)
def _add_dummy_testcase(self, ts_node, tds_item, tc_tds_dict, tc_pfs_dict, tc_node_order):
if not tds_item.attrib.has_key('TEXT'):
self.logger.error(self.log_prefix + \
"Please check node (%s) since it may use a long name. Please convert it to plain text via FreeMind Menu Format=>Use Plaine Text." % \
(tds_item.attrib['ID'].strip()))
exit(-1)
tds_item_str = str(tds_item.attrib['TEXT']).split('&')
self.logger.info(self.log_prefix + "Generating test case -%s" % tds_item_str[0])
node_reg_lvl = 2
for node_icon in tds_item.findall('icon'):
if node_icon.attrib['BUILTIN'].strip().count('full-') == 1:
node_reg_lvl = 4 - int(node_icon.attrib['BUILTIN'].strip()[-1])
if node_reg_lvl < 1:
node_reg_lvl = 2
testcase = lxmlET.SubElement(ts_node, 'testcase', {'name': tds_item_str[0]})
lxmlET.SubElement(testcase, 'node_order').text = lxmlET.CDATA(str(tc_node_order))
lxmlET.SubElement(testcase, 'externalid').text = lxmlET.CDATA('')
lxmlET.SubElement(testcase, 'version').text = lxmlET.CDATA('1')
lxmlET.SubElement(testcase, 'summary').text = lxmlET.CDATA(tds_item_str[0])
if len(tds_item_str) >1:
lxmlET.SubElement(testcase, 'preconditions').text = lxmlET.CDATA(tds_item_str[1])
lxmlET.SubElement(testcase, 'execution_type').text = lxmlET.CDATA('1')
lxmlET.SubElement(testcase, 'importance').text = lxmlET.CDATA(str(node_reg_lvl))
def _last_tds_node(self, node):
if node.attrib.has_key('LINK'):
return False
for child in node.findall('node'):
if not (child.attrib.has_key('LINK')):
return False
return True
def start_main():
# path = os.path.dirname('./') # for windows
# path = os.getcwd() # for windows
path = os.path.dirname(sys.executable) # for mac
logging.basicConfig(handlers=[logging.FileHandler(path + '/logger.log', 'w', 'utf-8')],
format='%(asctime)s:%(levelname)s %(message)s',
datefmt='%Y-%m-%d %H:%M:%S',
level=logging.DEBUG,
)
logger = logging.getLogger(__name__)
console = logging.StreamHandler() # 定义console handler
console.setLevel(logging.INFO) # 定义该handler级别
formatter = logging.Formatter('%(asctime)s %(filename)s : %(levelname)s %(message)s') # 定义该handler格式
console.setFormatter(formatter)
# Create an instance
logging.getLogger().addHandler(console)
file_list = get_freemindfile(path)
for file in file_list:
FreeMind(logger).Generate_TCs_from_TDS(file)
if __name__ == '__main__':
start_main()