trace-viewer/tracing/build/vulcanize_trace_viewer_unittest.py

# Copyright (c) 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import codecs
import os
import unittest
import tempfile

from tracing.build import vulcanize_trace_viewer


class Trace2HTMLTests(unittest.TestCase):
  def test_writeHTMLForTracesToFile(self):
    try:
      # 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()
      with codecs.open(raw_tmpfile.name, 'w', encoding='utf-8') as tmpfile:
        vulcanize_trace_viewer.WriteTraceViewer(tmpfile)
    finally:
      os.remove(raw_tmpfile.name)