[issue-487] add tests for the document_utils module

armintaenzertng · armintaenzertng · commit 6fbe644379ec · 2023-03-24T09:19:50.000+01:00
Signed-off-by: Armin Tänzer &lt;armin.taenzer@tngtech.com&gt;
diff --git a/tests/spdx/test_document_utils.py b/tests/spdx/test_document_utils.py
@@ -0,0 +1,27 @@
+#  SPDX-FileCopyrightText: Copyright (c) 2023 spdx contributors
+#
+#  SPDX-License-Identifier: Apache-2.0
+from unittest import TestCase
+
+import pytest
+
+from spdx.document_utils import get_element_from_spdx_id, get_contained_spdx_element_ids
+from tests.spdx.fixtures import document_fixture, snippet_fixture, package_fixture, file_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
