File: test_document_utils.py

package info (click to toggle)
python-spdx-tools 0.8.3-4
  • links: PTS, VCS
  • area: main
  • in suites: forky
  • size: 5,084 kB
  • sloc: python: 18,675; xml: 12,553; sh: 46; makefile: 6
file content (125 lines) | stat: -rw-r--r-- 4,798 bytes parent folder | download | duplicates (2)
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# SPDX-FileCopyrightText: 2023 spdx contributors
#
# SPDX-License-Identifier: Apache-2.0
from unittest import TestCase

import pytest

from spdx_tools.spdx.document_utils import (
    create_document_without_duplicates,
    create_list_without_duplicates,
    get_contained_spdx_element_ids,
    get_contained_spdx_elements,
    get_element_from_spdx_id,
)
from spdx_tools.spdx.model import FileType, SpdxNoAssertion, SpdxNone
from tests.spdx.fixtures import (
    actor_fixture,
    checksum_fixture,
    creation_info_fixture,
    document_fixture,
    external_document_ref_fixture,
    external_package_ref_fixture,
    extracted_licensing_info_fixture,
    file_fixture,
    package_fixture,
    snippet_fixture,
)


@pytest.fixture
def variables():
    return document_fixture(), package_fixture(), file_fixture(), snippet_fixture()


def test_contained_element_ids(variables):
    document, package, file, snippet = variables
    element_ids = get_contained_spdx_element_ids(document)
    TestCase().assertCountEqual(element_ids, [package.spdx_id, file.spdx_id, snippet.spdx_id])


def test_get_element_from_spdx_id(variables):
    document, package, file, snippet = variables
    assert get_element_from_spdx_id(document, package.spdx_id) == package
    assert get_element_from_spdx_id(document, file.spdx_id) == file
    assert get_element_from_spdx_id(document, snippet.spdx_id) == snippet
    assert get_element_from_spdx_id(document, "unknown_id") is None


def test_get_contained_spdx_elements(variables):
    document, package, file, snippet = variables
    contained_elements = get_contained_spdx_elements(document)
    assert contained_elements[package.spdx_id] == package
    assert contained_elements[file.spdx_id] == file
    assert contained_elements[snippet.spdx_id] == snippet


def test_create_list_without_duplicates():
    list_with_duplicates = [1, 2, 3, 5, 1, 67, 9, 67]

    list_without_duplicates = create_list_without_duplicates(list_with_duplicates)

    assert list_without_duplicates == [1, 2, 3, 5, 67, 9]


def test_create_document_without_duplicates():
    document = document_fixture(
        creation_info=creation_info_fixture(
            creators=[actor_fixture(name="creatorName"), actor_fixture(name="creatorName")],
            external_document_refs=[external_document_ref_fixture(), external_document_ref_fixture()],
        ),
        packages=[
            package_fixture(
                checksums=[checksum_fixture(), checksum_fixture()],
                license_info_from_files=[SpdxNoAssertion(), SpdxNoAssertion()],
                external_references=[external_package_ref_fixture(), external_package_ref_fixture()],
                attribution_texts=["duplicated text", "duplicated text"],
            )
        ],
        files=[
            file_fixture(
                checksums=[checksum_fixture(), checksum_fixture()],
                file_types=[FileType.TEXT, FileType.TEXT],
                license_info_in_file=[SpdxNoAssertion(), SpdxNoAssertion()],
                contributors=["duplicated contributor", "duplicated contributor"],
                attribution_texts=["duplicated text", "duplicated text"],
            )
        ],
        snippets=[
            snippet_fixture(
                license_info_in_snippet=[SpdxNone(), SpdxNone()],
                attribution_texts=["duplicated text", "duplicated text"],
            )
        ],
        extracted_licensing_info=[
            extracted_licensing_info_fixture(cross_references=["duplicated reference", "duplicated reference"])
        ],
    )
    expected_document = document_fixture(
        creation_info=creation_info_fixture(
            creators=[actor_fixture(name="creatorName")], external_document_refs=[external_document_ref_fixture()]
        ),
        packages=[
            package_fixture(
                checksums=[checksum_fixture()],
                license_info_from_files=[SpdxNoAssertion()],
                external_references=[external_package_ref_fixture()],
                attribution_texts=["duplicated text"],
            )
        ],
        files=[
            file_fixture(
                checksums=[checksum_fixture()],
                file_types=[FileType.TEXT],
                license_info_in_file=[SpdxNoAssertion()],
                contributors=["duplicated contributor"],
                attribution_texts=["duplicated text"],
            )
        ],
        snippets=[snippet_fixture(license_info_in_snippet=[SpdxNone()], attribution_texts=["duplicated text"])],
        extracted_licensing_info=[extracted_licensing_info_fixture(cross_references=["duplicated reference"])],
    )

    document_without_duplicates = create_document_without_duplicates(document)

    assert document_without_duplicates == expected_document