diff options
author | android-build-team Robot <android-build-team-robot@google.com> | 2020-02-15 03:23:48 +0000 |
---|---|---|
committer | android-build-team Robot <android-build-team-robot@google.com> | 2020-02-15 03:23:48 +0000 |
commit | 3b46527da85b8139d5fcf887387bcd4de20dec41 (patch) | |
tree | 35fa0fbaeaaddd9cc2a126a05eee3527b51e83a8 /afdo_redaction/redact_profile_test.py | |
parent | 62564f20dc6f3a88caed664d54fc00dbf3bf7373 (diff) | |
parent | b75f321fc8978b92ce3db6886ccb966768f0c7a8 (diff) | |
download | toolchain-utils-android11-d1-s6-release.tar.gz |
Snap for 6210127 from b75f321fc8978b92ce3db6886ccb966768f0c7a8 to rvc-d1-releaseandroid-11.0.0_r9android-11.0.0_r8android-11.0.0_r7android-11.0.0_r15android-11.0.0_r14android-11.0.0_r13android-11.0.0_r12android-11.0.0_r11android-11.0.0_r10android11-d1-s7-releaseandroid11-d1-s6-releaseandroid11-d1-s5-releaseandroid11-d1-s1-releaseandroid11-d1-release
Change-Id: I508eae962f523cfa857b4da7c77fdb8a70dfab1e
Diffstat (limited to 'afdo_redaction/redact_profile_test.py')
-rwxr-xr-x | afdo_redaction/redact_profile_test.py | 30 |
1 files changed, 16 insertions, 14 deletions
diff --git a/afdo_redaction/redact_profile_test.py b/afdo_redaction/redact_profile_test.py index 27fb534e..e2438972 100755 --- a/afdo_redaction/redact_profile_test.py +++ b/afdo_redaction/redact_profile_test.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python2 +#!/usr/bin/env python3 # -*- coding: utf-8 -*- # Copyright 2018 The Chromium OS Authors. All rights reserved. # Use of this source code is governed by a BSD-style license that can be @@ -8,12 +8,12 @@ from __future__ import division, print_function -import StringIO +import io import unittest -import redact_profile +from afdo_redaction import redact_profile -_redact_limit = redact_profile.dedup_records.func_defaults[0] +_redact_limit = redact_profile.dedup_records.__defaults__[0] def _redact(input_lines, summary_to=None): @@ -21,17 +21,18 @@ def _redact(input_lines, summary_to=None): input_lines = input_lines.splitlines() if summary_to is None: - summary_to = StringIO.StringIO() + summary_to = io.StringIO() - output_to = StringIO.StringIO() - redact_profile.run(profile_input_file=input_lines, - summary_output_file=summary_to, - profile_output_file=output_to) + output_to = io.StringIO() + redact_profile.run( + profile_input_file=input_lines, + summary_output_file=summary_to, + profile_output_file=output_to) return output_to.getvalue() def _redact_with_summary(input_lines): - summary = StringIO.StringIO() + summary = io.StringIO() result = _redact(input_lines, summary_to=summary) return result, summary.getvalue() @@ -64,6 +65,7 @@ def _generate_repeated_function_body(repeats, fn_name='_some_name'): class Tests(unittest.TestCase): """All of our tests for redact_profile.""" + def test_no_input_works(self): self.assertEqual(_redact(''), '') @@ -93,13 +95,13 @@ class Tests(unittest.TestCase): result_file = '\n'.join(kept_lines) + '\n' - lines = _generate_repeated_function_body(_redact_limit, - fn_name='_discard_me') + lines = _generate_repeated_function_body( + _redact_limit, fn_name='_discard_me') self.assertEqual(_redact(kept_lines + lines), result_file) self.assertEqual(_redact(lines + kept_lines), result_file) - more_lines = _generate_repeated_function_body(_redact_limit, - fn_name='_and_discard_me') + more_lines = _generate_repeated_function_body( + _redact_limit, fn_name='_and_discard_me') self.assertEqual(_redact(lines + kept_lines + more_lines), result_file) self.assertEqual(_redact(lines + more_lines), '') |