tools/profile_chrome/trace_packager_unittest.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import json
   6 import tempfile
   7 import unittest
   8
   9 from profile_chrome import trace_packager
  10
  11
  12 class TracePackagerTest(unittest.TestCase):
  13   def testJsonTraceMerging(self):
  14     t1 = {'traceEvents': [{'ts': 123, 'ph': 'b'}]}
  15     t2 = {'traceEvents': [], 'stackFrames': ['blah']}
  16
  17     # Both trace files will be merged to a third file and will get deleted in
  18     # the process, so there's no need for NamedTemporaryFile to do the
  19     # deletion.
  20     with tempfile.NamedTemporaryFile(delete=False) as f1, \
  21         tempfile.NamedTemporaryFile(delete=False) as f2:
  22       f1.write(json.dumps(t1))
  23       f2.write(json.dumps(t2))
  24       f1.flush()
  25       f2.flush()
  26
  27       with tempfile.NamedTemporaryFile() as output:
  28         trace_packager.PackageTraces([f1.name, f2.name],
  29                                      output.name,
  30                                      compress=False,
  31                                      write_json=True)
  32         with open(output.name) as output:
  33           output = json.load(output)
  34           self.assertEquals(output['traceEvents'], t1['traceEvents'])
  35           self.assertEquals(output['stackFrames'], t2['stackFrames'])