diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 16:04:21 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 16:04:21 +0000 |
commit | 8a754e0858d922e955e71b253c139e071ecec432 (patch) | |
tree | 527d16e74bfd1840c85efd675fdecad056c54107 /test/units/module_utils/urls/test_fetch_file.py | |
parent | Initial commit. (diff) | |
download | ansible-core-upstream.tar.xz ansible-core-upstream.zip |
Adding upstream version 2.14.3.upstream/2.14.3upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/units/module_utils/urls/test_fetch_file.py')
-rw-r--r-- | test/units/module_utils/urls/test_fetch_file.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/test/units/module_utils/urls/test_fetch_file.py b/test/units/module_utils/urls/test_fetch_file.py new file mode 100644 index 0000000..ed11227 --- /dev/null +++ b/test/units/module_utils/urls/test_fetch_file.py @@ -0,0 +1,45 @@ +# -*- coding: utf-8 -*- +# Copyright: Contributors to the Ansible project +# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) + +from __future__ import absolute_import, division, print_function +__metaclass__ = type + +import os + +from ansible.module_utils.urls import fetch_file + +import pytest +from units.compat.mock import MagicMock + + +class FakeTemporaryFile: + def __init__(self, name): + self.name = name + + +@pytest.mark.parametrize( + 'url, prefix, suffix, expected', ( + ('http://ansible.com/foo.tar.gz?foo=%s' % ('bar' * 100), 'foo', '.tar.gz', 'foo.tar.gz'), + ('https://www.gnu.org/licenses/gpl-3.0.txt', 'gpl-3.0', '.txt', 'gpl-3.0.txt'), + ('http://pyyaml.org/download/libyaml/yaml-0.2.5.tar.gz', 'yaml-0.2.5', '.tar.gz', 'yaml-0.2.5.tar.gz'), + ( + 'https://github.com/mozilla/geckodriver/releases/download/v0.26.0/geckodriver-v0.26.0-linux64.tar.gz', + 'geckodriver-v0.26.0-linux64', + '.tar.gz', + 'geckodriver-v0.26.0-linux64.tar.gz' + ), + ) +) +def test_file_multiple_extensions(mocker, url, prefix, suffix, expected): + module = mocker.Mock() + module.tmpdir = '/tmp' + module.add_cleanup_file = mocker.Mock(side_effect=AttributeError('raised intentionally')) + + mock_NamedTemporaryFile = mocker.patch('ansible.module_utils.urls.tempfile.NamedTemporaryFile', + return_value=FakeTemporaryFile(os.path.join(module.tmpdir, expected))) + + with pytest.raises(AttributeError, match='raised intentionally'): + fetch_file(module, url) + + mock_NamedTemporaryFile.assert_called_with(dir=module.tmpdir, prefix=prefix, suffix=suffix, delete=False) |