xref: /linux/scripts/sbom/tests/spdx_graph/test_kernel_file.py (revision 3d5e48944e824bddc20d7b874e784f7b279636fe)
1# SPDX-License-Identifier: GPL-2.0-only OR MIT
2# Copyright (C) 2025 TNG Technology Consulting GmbH
3
4import unittest
5from pathlib import Path
6import tempfile
7from sbom.spdx_graph.kernel_file import _parse_spdx_license_identifier  # type: ignore
8
9
10class TestKernelFile(unittest.TestCase):
11    def setUp(self):
12        self.tmpdir = tempfile.TemporaryDirectory()
13        self.src_tree = Path(self.tmpdir.name)
14
15    def tearDown(self):
16        self.tmpdir.cleanup()
17
18    def test_parse_spdx_license_identifier(self):
19        # REUSE-IgnoreStart
20        test_cases: list[tuple[str, str | None]] = [
21            ("/* SPDX-License-Identifier: MIT*/", "MIT"),
22            ("// SPDX-License-Identifier: GPL-2.0-only", "GPL-2.0-only"),
23            ("# SPDX-License-Identifier: GPL-2.0-only", "GPL-2.0-only"),
24            ("#!/bin/bash\n# SPDX-License-Identifier: GPL-2.0-only", "GPL-2.0-only"),
25            ("/* SPDX-License-Identifier: GPL-2.0-or-later OR MIT */", "GPL-2.0-or-later OR MIT"),
26            ("/* SPDX-License-Identifier: Apache-2.0 */\n extra text", "Apache-2.0"),
27            ("<!-- SPDX-License-Identifier: GPL-2.0 -->", "GPL-2.0"),
28            ("int main() { return 0; }", None),
29        ]
30        # REUSE-IgnoreEnd
31
32        for i, (file_content, expected_identifier) in enumerate(test_cases):
33            file_path = self.src_tree / f"file_{i}.c"
34            file_path.write_text(file_content)
35            self.assertEqual(_parse_spdx_license_identifier(str(file_path)), expected_identifier)
36