diff options
Diffstat (limited to 'systrace/catapult/systrace/systrace/tracing_agents/atrace_agent_unittest.py')
-rwxr-xr-x | systrace/catapult/systrace/systrace/tracing_agents/atrace_agent_unittest.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/systrace/catapult/systrace/systrace/tracing_agents/atrace_agent_unittest.py b/systrace/catapult/systrace/systrace/tracing_agents/atrace_agent_unittest.py index 203e846..9ccc6e3 100755 --- a/systrace/catapult/systrace/systrace/tracing_agents/atrace_agent_unittest.py +++ b/systrace/catapult/systrace/systrace/tracing_agents/atrace_agent_unittest.py @@ -16,7 +16,6 @@ from systrace.tracing_agents import atrace_agent from devil.android import device_utils from devil.android.sdk import intent -from py_utils import tempfile_ext DEVICE_SERIAL = 'AG8404EC0444AGC' @@ -51,7 +50,9 @@ class AtraceAgentTest(unittest.TestCase): devices = device_utils.DeviceUtils.HealthyDevices() package_info = util.get_supported_browsers()['stable'] device = devices[0] - with tempfile_ext.TemporaryFileName() as output_file_name: + output_file_name = util.generate_random_filename_for_test() + + try: # Launch the browser before tracing. device.StartActivity( intent.Intent(activity=package_info.activity, @@ -75,7 +76,12 @@ class AtraceAgentTest(unittest.TestCase): # Verify results. with open(output_file_name, 'r') as f: full_trace = f.read() - self.assertTrue('CPU#' in full_trace) + self.assertTrue('CPU#'in full_trace) + except: + raise + finally: + if os.path.exists(output_file_name): + os.remove(output_file_name) @decorators.HostOnlyTest def test_construct_atrace_args(self): |