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)