def testWriteHTMLForTracesToFile(self):
output_filename = os.path.join(self._tmpdir,
'WriteHTMLForTracesToFile.html')
with codecs.open(output_filename, 'w',
encoding='utf-8') as output_file:
trace2html.WriteHTMLForTracesToFile([
self.BIG_TRACE_PATH, self.SIMPLE_TRACE_PATH,
self.NON_JSON_TRACE_PATH
], output_file)
def testGzippedTraceIsntDoubleGzipped(self):
# |input_filename| will contain plain JSON data at one point, then gzipped
# JSON data at another point for reasons that will be explained below.
input_filename = os.path.join(self._tmpdir,
'GzippedTraceIsntDoubleGzipped')
output_filename = os.path.join(self._tmpdir,
'GzippedTraceIsntDoubleGzipped.html')
# trace2html-ify SIMPLE_TRACE, but from a controlled input filename so
# that when ViewerDataScript gzips it, it uses the same filename for both
# the unzipped SIMPLE_TRACE here and the gzipped SIMPLE_TRACE below.
open(input_filename, 'w').write(open(self.SIMPLE_TRACE_PATH).read())
with codecs.open(output_filename, 'w',
encoding='utf-8') as output_file:
trace2html.WriteHTMLForTracesToFile([input_filename], output_file)
# Hash the contents of the output file that was generated from an unzipped
# json input file.
unzipped_hash = hash(open(output_filename).read())
os.unlink(output_filename)
# Gzip SIMPLE_TRACE into |input_filename|.
# trace2html should automatically gunzip it and start building the html from
# the same input as if the input weren't gzipped.
with gzip.GzipFile(input_filename, mode='w') as input_gzipfile:
input_gzipfile.write(open(self.SIMPLE_TRACE_PATH).read())
# trace2html-ify the zipped version of SIMPLE_TRACE from the same input
# filename as the unzipped version so that the gzipping process is stable.
with codecs.open(output_filename, 'w',
encoding='utf-8') as output_file:
trace2html.WriteHTMLForTracesToFile([input_filename], output_file)
# Hash the contents of the output file that was generated from the zipped
# json input file.
zipped_hash = hash(open(output_filename).read())
# Compare the hashes, not the file contents directly so that, if they are
# different, python shouldn't try to print megabytes of html.
self.assertEqual(unzipped_hash, zipped_hash)
def test_writeHTMLForTracesToFile(self):
# Note: We can't use "with" when working with tempfile.NamedTemporaryFile as
# that does not work on Windows. We use the longer, more clunky version
# instead. See https://bugs.python.org/issue14243 for detials.
raw_tmpfile = tempfile.NamedTemporaryFile(
mode='w', suffix='.html', delete=False)
raw_tmpfile.close()
try:
with codecs.open(raw_tmpfile.name, 'w', encoding='utf-8') as tmpfile:
simple_trace_path = os.path.join(
os.path.dirname(__file__),
'..', 'test_data', 'simple_trace.json')
big_trace_path = os.path.join(
os.path.dirname(__file__),
'..', 'test_data', 'big_trace.json')
non_json_trace_path = os.path.join(
os.path.dirname(__file__),
'..', 'test_data', 'android_systrace.txt')
trace2html.WriteHTMLForTracesToFile(
[big_trace_path, simple_trace_path, non_json_trace_path], tmpfile)
finally:
os.remove(raw_tmpfile.name)
def _PackageTracesAsHtml(trace_files, html_file):
with codecs.open(html_file, mode='w', encoding='utf-8') as f:
trace2html.WriteHTMLForTracesToFile(trace_files, f)
for trace_file in trace_files:
os.unlink(trace_file)