| # Copyright 2024 The Pigweed Authors |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); you may not |
| # use this file except in compliance with the License. You may obtain a copy of |
| # the License at |
| # |
| # https://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| # License for the specific language governing permissions and limitations under |
| # the License. |
| """Tests for the clang-format formatter.""" |
| |
| import importlib.resources |
| from pathlib import Path |
| from tempfile import TemporaryDirectory |
| from typing import Final, Sequence |
| import unittest |
| |
| from format_testing_utils import CapturingToolRunner |
| from pw_presubmit.format.cpp import ClangFormatFormatter |
| |
| _TEST_DATA_FILES = importlib.resources.files('pw_presubmit.format.test_data') |
| |
| _TEST_SRC_FILE = _TEST_DATA_FILES / 'cpp_test_data.cc' |
| _TEST_GOLDEN = _TEST_DATA_FILES / 'cpp_test_data_golden.cc' |
| |
| _CLANG_FORMAT_CONFIG_PATH = _TEST_DATA_FILES / 'clang_format_config' |
| _CLANG_FORMAT_ARGS: Final[Sequence[str]] = ( |
| f'--style=file:{_CLANG_FORMAT_CONFIG_PATH}', |
| ) |
| |
| |
| class TestClangFormatFormatter(unittest.TestCase): |
| """Tests for the ClangFormatFormatter.""" |
| |
| def test_check_file(self): |
| tool_runner = CapturingToolRunner() |
| formatter = ClangFormatFormatter(_CLANG_FORMAT_ARGS) |
| formatter.run_tool = tool_runner |
| |
| result = formatter.format_file_in_memory( |
| _TEST_SRC_FILE, _TEST_SRC_FILE.read_bytes() |
| ) |
| self.assertTrue(result.ok) |
| self.assertEqual(result.error_message, None) |
| |
| self.assertEqual( |
| tool_runner.command_history.pop(0), |
| ' '.join( |
| ( |
| 'clang-format', |
| f'--style=file:{_CLANG_FORMAT_CONFIG_PATH}', |
| str(_TEST_SRC_FILE), |
| ) |
| ), |
| ) |
| |
| self.assertMultiLineEqual( |
| result.formatted_file_contents.decode(), _TEST_GOLDEN.read_text() |
| ) |
| |
| def test_fix_file(self): |
| tool_runner = CapturingToolRunner() |
| formatter = ClangFormatFormatter(_CLANG_FORMAT_ARGS) |
| formatter.run_tool = tool_runner |
| |
| with TemporaryDirectory() as temp_dir: |
| file_to_fix = Path(temp_dir) / _TEST_SRC_FILE.name |
| file_to_fix.write_bytes(_TEST_SRC_FILE.read_bytes()) |
| |
| errors = formatter.format_files([file_to_fix]) |
| |
| # Error dictionary should be empty. |
| self.assertFalse(list(errors)) |
| |
| self.assertEqual( |
| tool_runner.command_history.pop(0), |
| ' '.join( |
| ( |
| 'clang-format', |
| '-i', |
| f'--style=file:{_CLANG_FORMAT_CONFIG_PATH}', |
| str(file_to_fix), |
| ) |
| ), |
| ) |
| |
| self.assertMultiLineEqual( |
| file_to_fix.read_text(), _TEST_GOLDEN.read_text() |
| ) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |