python-license-check/tests/test_get_packages_info.py at b5ab8aac845ad116efd3cbca7b563e1f12f9562f · dhatim/python-license-check · GitHub

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
78
79
80
81
82
83
84
85
86
87
import sys

import importlib.metadata as ilm
import pytest

from liccheck.command_line import get_packages_info


def test_license_strip(tmpfile):
    tmpfh, tmppath = tmpfile
    tmpfh.write("pip\n")
    tmpfh.close()
    assert get_packages_info(tmppath)[0]["licenses"] == ["MIT"]


def test_license_strip_with_return_carriage(tmp_path, mocker):
    resolve = mocker.patch("liccheck.command_line.resolve")
    req_path = tmp_path.joinpath("requirements.txt").as_posix()
    with open(req_path, "w") as tmpfh:
        tmpfh.write("pip\n")
    pkg_info_path = tmp_path.joinpath("PKG-INFO").as_posix()
    with open(pkg_info_path, "wb") as tmpfh:
        tmpfh.write(b"Metadata-Version: 2.1\r\n")
        tmpfh.write(b"Name: pip\r\n")
        tmpfh.write(b"Version: 23.3.1\r\n")
        tmpfh.write(b"Classifier: License :: OSI Approved :: MIT License\r\n")
    resolve.return_value = [ilm.Distribution.at(tmp_path)]
    assert get_packages_info(req_path)[0]["licenses"] == ["MIT"]


def test_requirements_markers(tmpfile):
    tmpfh, tmppath = tmpfile
    tmpfh.write(
        "python3-openid;python_version>=\"3.9\"\n"
    )
    tmpfh.close()
    if sys.version_info.minor >= 9:
        assert len(get_packages_info(tmppath)) == 2
    else:
        assert len(get_packages_info(tmppath)) == 0


def test_editable_requirements_get_ignored(tmpfile):
    tmpfh, tmppath = tmpfile
    tmpfh.write(
        "-e file:some_editable_req\n"
        "pip\n"
    )
    tmpfh.close()

    packages_info = get_packages_info(tmppath)
    assert len(packages_info) == 1
    assert packages_info[0]["name"] == "pip"


@pytest.mark.parametrize(
    ('no_deps', 'expected_packages'), (
        pytest.param(
            False,
            ('liccheck', 'semantic-version', 'toml'),
            id='with deps'
        ),
        pytest.param(True, ('liccheck',), id='without deps'),
    )
)
def test_deps(tmpfile, no_deps, expected_packages):
    tmpfh, tmppath = tmpfile
    tmpfh.write('liccheck\n')
    tmpfh.close()
    packages_info = get_packages_info(tmppath, no_deps)
    packages = tuple(package['name'] for package in packages_info)
    assert packages == expected_packages


def test_license_expression(tmp_path, mocker):
    resolve = mocker.patch("liccheck.command_line.resolve")
    req_path = tmp_path.joinpath("requirements.txt").as_posix()
    with open(req_path, "w") as tmpfh:
        tmpfh.write("Twisted\n")
    pkg_info_path = tmp_path.joinpath("PKG-INFO").as_posix()
    with open(pkg_info_path, "w") as tmpfh:
        tmpfh.write("Metadata-Version: 2.1\n")
        tmpfh.write("Name: Twisted\n")
        tmpfh.write("Version: 23.8.0\n")
        tmpfh.write("License-Expression: MIT\n")
    resolve.return_value = [ilm.Distribution.at(tmp_path)]
    assert get_packages_info(req_path)[0]["licenses"] == ["MIT"]