summaryrefslogtreecommitdiffstats
path: root/test/units/module_utils/urls/test_split.py
blob: 7fd5fc13cf8fb216fbc0b184fd36d366eb2d54ae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# -*- 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 pytest

from ansible.module_utils.urls import _split_multiext


@pytest.mark.parametrize(
    'name, expected',
    (
        ('', ('', '')),
        ('a', ('a', '')),
        ('file.tar', ('file', '.tar')),
        ('file.tar.', ('file.tar.', '')),
        ('file.hidden', ('file.hidden', '')),
        ('file.tar.gz', ('file', '.tar.gz')),
        ('yaml-0.2.5.tar.gz', ('yaml-0.2.5', '.tar.gz')),
        ('yaml-0.2.5.zip', ('yaml-0.2.5', '.zip')),
        ('yaml-0.2.5.zip.hidden', ('yaml-0.2.5.zip.hidden', '')),
        ('geckodriver-v0.26.0-linux64.tar', ('geckodriver-v0.26.0-linux64', '.tar')),
        ('/var/lib/geckodriver-v0.26.0-linux64.tar', ('/var/lib/geckodriver-v0.26.0-linux64', '.tar')),
        ('https://acme.com/drivers/geckodriver-v0.26.0-linux64.tar', ('https://acme.com/drivers/geckodriver-v0.26.0-linux64', '.tar')),
        ('https://acme.com/drivers/geckodriver-v0.26.0-linux64.tar.bz', ('https://acme.com/drivers/geckodriver-v0.26.0-linux64', '.tar.bz')),
    )
)
def test__split_multiext(name, expected):
    assert expected == _split_multiext(name)


@pytest.mark.parametrize(
    'args, expected',
    (
        (('base-v0.26.0-linux64.tar.gz', 4, 4), ('base-v0.26.0-linux64.tar.gz', '')),
        (('base-v0.26.0.hidden', 1, 7), ('base-v0.26', '.0.hidden')),
        (('base-v0.26.0.hidden', 3, 4), ('base-v0.26.0.hidden', '')),
        (('base-v0.26.0.hidden.tar', 1, 7), ('base-v0.26.0', '.hidden.tar')),
        (('base-v0.26.0.hidden.tar.gz', 1, 7), ('base-v0.26.0.hidden', '.tar.gz')),
        (('base-v0.26.0.hidden.tar.gz', 4, 7), ('base-v0.26.0.hidden.tar.gz', '')),
    )
)
def test__split_multiext_min_max(args, expected):
    assert expected == _split_multiext(*args)


@pytest.mark.parametrize(
    'kwargs, expected', (
        (({'name': 'base-v0.25.0.tar.gz', 'count': 1}), ('base-v0.25.0.tar', '.gz')),
        (({'name': 'base-v0.25.0.tar.gz', 'count': 2}), ('base-v0.25.0', '.tar.gz')),
        (({'name': 'base-v0.25.0.tar.gz', 'count': 3}), ('base-v0.25.0', '.tar.gz')),
        (({'name': 'base-v0.25.0.tar.gz', 'count': 4}), ('base-v0.25.0', '.tar.gz')),
        (({'name': 'base-v0.25.foo.tar.gz', 'count': 3}), ('base-v0.25', '.foo.tar.gz')),
        (({'name': 'base-v0.25.foo.tar.gz', 'count': 4}), ('base-v0', '.25.foo.tar.gz')),
    )
)
def test__split_multiext_count(kwargs, expected):
    assert expected == _split_multiext(**kwargs)


@pytest.mark.parametrize(
    'name',
    (
        list(),
        tuple(),
        dict(),
        set(),
        1.729879,
        247,
    )
)
def test__split_multiext_invalid(name):
    with pytest.raises((TypeError, AttributeError)):
        _split_multiext(name)