diff options
Diffstat (limited to 'catapult/common/py_trace_event/py_trace_event/trace_event_impl/trace_test.py')
-rw-r--r-- | catapult/common/py_trace_event/py_trace_event/trace_event_impl/trace_test.py | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/catapult/common/py_trace_event/py_trace_event/trace_event_impl/trace_test.py b/catapult/common/py_trace_event/py_trace_event/trace_event_impl/trace_test.py index 7047e0ea..1216037f 100644 --- a/catapult/common/py_trace_event/py_trace_event/trace_event_impl/trace_test.py +++ b/catapult/common/py_trace_event/py_trace_event/trace_event_impl/trace_test.py @@ -1,7 +1,6 @@ # Copyright 2016 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 tempfile import unittest #from .log import * @@ -9,6 +8,7 @@ import unittest from log import * from parsed_trace_events import * +from py_utils import tempfile_ext class TraceTest(unittest.TestCase): def __init__(self, *args): @@ -25,16 +25,16 @@ class TraceTest(unittest.TestCase): Enables tracing, runs the provided callback, and if successful, returns a TraceEvents object with the results. """ - self._file = tempfile.NamedTemporaryFile() - trace_enable(open(self._file.name, 'a+')) - - try: - cb() - finally: - trace_disable() - e = ParsedTraceEvents(trace_filename = self._file.name) - self._file.close() - self._file = None + with tempfile_ext.TemporaryFileName() as filename: + self._file = open(filename, 'a+') + trace_enable(self._file) + try: + cb() + finally: + trace_disable() + e = ParsedTraceEvents(trace_filename=self._file.name) + self._file.close() + self._file = None return e @property |