extracttests.py: Added tool for extracting test information
This commit is contained in:
Daniel Marjamäki
parent
e0b50719f4
commit
9c626b941a
|
|
@ -0,0 +1,176 @@
|
||||||
|
#!/usr/bin/python
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import re
|
||||||
|
|
||||||
|
class Extract:
|
||||||
|
nodes = []
|
||||||
|
|
||||||
|
def parseFile(self,filename):
|
||||||
|
name = '[0-9a-zA-Z_]+'
|
||||||
|
str = '\\"(.+)\\"'
|
||||||
|
|
||||||
|
testclass = None
|
||||||
|
functionName = None
|
||||||
|
|
||||||
|
fin = open(filename, 'r')
|
||||||
|
for line in fin:
|
||||||
|
# testclass starts
|
||||||
|
res = re.match('class ('+name+')', line)
|
||||||
|
if res != None:
|
||||||
|
testclass = res.group(1)
|
||||||
|
|
||||||
|
# end of testclass
|
||||||
|
if re.match('};', line) != None:
|
||||||
|
testclass = None
|
||||||
|
|
||||||
|
# function start
|
||||||
|
res = re.match('\\s+void ('+name+')\\(\\)', line)
|
||||||
|
if res != None:
|
||||||
|
functionName = res.group(1)
|
||||||
|
|
||||||
|
elif re.match('\\s+}', line) != None:
|
||||||
|
functionName = None
|
||||||
|
|
||||||
|
if functionName == None:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# check
|
||||||
|
res = re.match('\s+check\('+str, line)
|
||||||
|
if res != None:
|
||||||
|
code = res.group(1)
|
||||||
|
|
||||||
|
# code..
|
||||||
|
res = re.match('\\s+'+str, line)
|
||||||
|
if res != None:
|
||||||
|
code = code + res.group(1)
|
||||||
|
|
||||||
|
# assert
|
||||||
|
res = re.match('\\s+ASSERT_EQUALS\\(\\"(.*)\\"', line)
|
||||||
|
if res != None and len(code) > 10:
|
||||||
|
node = { 'testclass':testclass, 'functionName':functionName, 'code':code, 'expected':res.group(1) }
|
||||||
|
self.nodes.append(node)
|
||||||
|
code = ''
|
||||||
|
|
||||||
|
def strtoxml(s):
|
||||||
|
return s.replace('&','&').replace('"', '"').replace('<','<').replace('>','>')
|
||||||
|
|
||||||
|
# parse command line
|
||||||
|
xml = False
|
||||||
|
filename = None
|
||||||
|
htmldir = None
|
||||||
|
i = 1
|
||||||
|
while i < len(sys.argv):
|
||||||
|
if sys.argv[i] == '--xml':
|
||||||
|
xml = True
|
||||||
|
elif sys.argv[i].startswith('--html='):
|
||||||
|
arg = sys.argv[i]
|
||||||
|
htmldir = arg[7:]
|
||||||
|
else:
|
||||||
|
filename = sys.argv[i]
|
||||||
|
i = i + 1
|
||||||
|
|
||||||
|
# extract test cases
|
||||||
|
if filename != None:
|
||||||
|
# parse test file
|
||||||
|
e = Extract()
|
||||||
|
e.parseFile(filename)
|
||||||
|
|
||||||
|
# generate output
|
||||||
|
if xml:
|
||||||
|
print '<?xml version="1.0"?>'
|
||||||
|
print '<tree>'
|
||||||
|
count = 0
|
||||||
|
for node in e.nodes:
|
||||||
|
s = ' <node'
|
||||||
|
s += ' function="' + node['functionName'] + '"'
|
||||||
|
s += ' code="' + strtoxml(node['code']) + '"'
|
||||||
|
s += ' expected="' + strtoxml(node['expected']) + '"'
|
||||||
|
s += '/>'
|
||||||
|
print s
|
||||||
|
print '</tree>'
|
||||||
|
elif htmldir != None:
|
||||||
|
if not htmldir.endswith('/'):
|
||||||
|
htmldir += '/'
|
||||||
|
findex = open(htmldir + 'index.htm', 'w')
|
||||||
|
findex.write('<html>\n')
|
||||||
|
findex.write('<head>\n')
|
||||||
|
findex.write(' <style type="text/css">\n')
|
||||||
|
findex.write(' table { font-size: 0.8em }\n')
|
||||||
|
findex.write(' th { background-color: #A3C159; text-transform: uppercase }\n')
|
||||||
|
findex.write(' td { background-color: #F0FFE0; vertical-align: text-top }\n')
|
||||||
|
findex.write(' A:link { text-decoration: none }\n')
|
||||||
|
findex.write(' A:visited { text-decoration: none }\n')
|
||||||
|
findex.write(' A:active { text-decoration: none }\n')
|
||||||
|
findex.write(' A:hover { text-decoration: underline; color: blue }\n')
|
||||||
|
findex.write(' </style>\n')
|
||||||
|
findex.write('</head>\n')
|
||||||
|
findex.write('<body>\n')
|
||||||
|
findex.write('<h1>' + filename + '</h1>\n')
|
||||||
|
|
||||||
|
functionNames = []
|
||||||
|
for node in e.nodes:
|
||||||
|
functionname = node['functionName']
|
||||||
|
while functionname[-1].isdigit():
|
||||||
|
functionname = functionname[:-1]
|
||||||
|
if functionname[-1] == '_':
|
||||||
|
functionname = functionname[:-1]
|
||||||
|
if not functionname in functionNames:
|
||||||
|
functionNames.append(functionname)
|
||||||
|
functionNames.sort()
|
||||||
|
|
||||||
|
findex.write('<table border="0" cellspacing="0">\n')
|
||||||
|
findex.write(' <tr><th>Name</th><th>Number</th></tr>\n')
|
||||||
|
for functionname in functionNames:
|
||||||
|
findex.write(' <tr><td><a href="'+functionname+'.htm">'+functionname+'</a></td>')
|
||||||
|
num = 0
|
||||||
|
for node in e.nodes:
|
||||||
|
name = node['functionName']
|
||||||
|
while name[-1].isdigit():
|
||||||
|
name = name[:-1]
|
||||||
|
if name[-1] == '_':
|
||||||
|
name = name[:-1]
|
||||||
|
if name == functionname:
|
||||||
|
num = num + 1
|
||||||
|
findex.write('<td><div align="right">' + str(num) + '</div></td></tr>\n')
|
||||||
|
|
||||||
|
findex.write('</table>\n')
|
||||||
|
|
||||||
|
findex.write('</body></html>')
|
||||||
|
findex.close()
|
||||||
|
|
||||||
|
# create files for each functionName
|
||||||
|
for functionName in functionNames:
|
||||||
|
fout = open(htmldir + functionName + '.htm', 'w')
|
||||||
|
fout.write('<html>\n')
|
||||||
|
fout.write('<head>\n')
|
||||||
|
fout.write(' <style type="text/css">\n')
|
||||||
|
fout.write(' body { font-size: 0.8em }\n')
|
||||||
|
fout.write(' th { background-color: #A3C159; text-transform: uppercase }\n')
|
||||||
|
fout.write(' td { background-color: white; vertical-align: text-top }\n')
|
||||||
|
fout.write(' pre { background-color: #EEEEEE }\n')
|
||||||
|
fout.write(' </style>\n')
|
||||||
|
fout.write('</head>\n')
|
||||||
|
fout.write('<body>\n')
|
||||||
|
fout.write('<h1>' + node['testclass'] + '::' + functionName + '</h1>')
|
||||||
|
fout.write('<table border="0" cellspacing="0">\n')
|
||||||
|
fout.write(' <tr><th>Nr</th><th>Code</th><th>Expected</th></tr>\n')
|
||||||
|
num = 0
|
||||||
|
for node in e.nodes:
|
||||||
|
name = node['functionName']
|
||||||
|
while name[-1].isdigit():
|
||||||
|
name = name[:-1]
|
||||||
|
if name[-1] == '_':
|
||||||
|
name = name[:-1]
|
||||||
|
if name == functionName:
|
||||||
|
num = num + 1
|
||||||
|
fout.write(' <tr><td>' + str(num) + '</td>')
|
||||||
|
fout.write('<td><pre>' + strtoxml(node['code']).replace('\\n', '\n') + '</pre></td>')
|
||||||
|
fout.write('<td>' + strtoxml(node['expected']).replace('\\n', '<br>') + '</td>')
|
||||||
|
fout.write('</tr>\n')
|
||||||
|
fout.write('</table></body></html>\n')
|
||||||
|
fout.close()
|
||||||
|
else:
|
||||||
|
for node in e.nodes:
|
||||||
|
print node['functionName']
|
||||||
|
|
||||||
Loading…
Reference in New Issue