| #!/usr/bin/env vpython3 |
| # Copyright 2019 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import subprocess |
| import unittest |
| from unittest import mock |
| |
| import merge_lib as merger |
| |
| # Protected access is allowed for unittests. |
| # pylint: disable=protected-access |
| |
| class MergeLibTest(unittest.TestCase): |
| |
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.maxDiff = None |
| |
| @mock.patch.object(subprocess, 'check_output') |
| def test_validate_and_convert_profraw(self, mock_cmd): |
| test_cases = [ |
| ([''], [['mock.profdata'], [], []]), |
| (['Counter overflow'], [[], ['mock.profraw'], ['mock.profraw']]), |
| (subprocess.CalledProcessError( |
| 255, |
| 'llvm-cov merge -o mock.profdata -sparse=true mock.profraw', |
| output='Malformed profile'), [[], ['mock.profraw'], []]), |
| ] |
| for side_effect, expected_results in test_cases: |
| mock_cmd.side_effect = side_effect |
| output_profdata_files = [] |
| invalid_profraw_files = [] |
| counter_overflows = [] |
| merger._validate_and_convert_profraw('mock.profraw', |
| output_profdata_files, |
| invalid_profraw_files, |
| counter_overflows, |
| '/usr/bin/llvm-cov', |
| show_profdata=False) |
| self.assertEqual( |
| expected_results, |
| [output_profdata_files, invalid_profraw_files, counter_overflows]) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |