diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_setup.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/tests/test_setup.py b/tests/test_setup.py new file mode 100644 index 0000000..1850319 --- /dev/null +++ b/tests/test_setup.py @@ -0,0 +1,56 @@ +# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +"""Test for module setup.""" +from __future__ import annotations + +from unittest.mock import Mock +from unittest.mock import patch + +import pytest + +from setup import replace_markdown_relative_paths + + +@pytest.mark.parametrize( + "linked_file_found, file_content, expected_result", + [ + [ + True, + "[Test](README.md)", + "[Test](https://review.mlplatform.org/plugins/gitiles/" + "ml/mlia/+/refs/tags/0.1.0/README.md)", + ], + [ + True, + "![Test](image.png)", + "![Test](https://git.mlplatform.org/ml/mlia.git/plain/image.png?h=0.1.0)", + ], + [ + False, + "[Test](http://review.mlplatform.org)", + "[Test](http://review.mlplatform.org)", + ], + [False, "[Test](README.md)", "[Test](README.md)"], + [ + True, + "Lorem ipsum dolor sit amet, consectetur adipiscing elit", + "Lorem ipsum dolor sit amet, consectetur adipiscing elit", + ], + ], +) +@patch("pathlib.Path") +def test_replace_markdown_relative_paths( + path_mock: Mock, + linked_file_found: bool, + file_content: str, + expected_result: str, +) -> None: + """Test replacement of relative md paths with links to review.mlplatform.org.""" + # Set a mock setuptools scm version for testing + tag = "0.1.0" + path_mock.read_text.return_value = file_content + path_mock.exists.return_value = linked_file_found + path_mock.joinpath.return_value = path_mock + + result = replace_markdown_relative_paths(path_mock, "test.md", tag) + assert result == expected_result |