|
| 1 | +import unittest |
| 2 | +import re |
| 3 | +from unittest import TestCase |
| 4 | +from spdx.parsers.rdf import Parser |
| 5 | +from spdx.parsers.rdfbuilders import Builder |
| 6 | +from spdx.parsers.loggers import StandardLogger |
| 7 | +from tests import utils_test |
| 8 | + |
| 9 | +class TestParser(TestCase): |
| 10 | + |
| 11 | + def test_extracted_licenses(self): |
| 12 | + parser = Parser(Builder(), StandardLogger()) |
| 13 | + test_file = utils_test.get_test_loc('../../data/SPDXRdfExample.rdf', test_data_dir=utils_test.test_data_dir) |
| 14 | + with open(test_file, 'r') as file: |
| 15 | + document, _ = parser.parse(file) |
| 16 | + assert len(document.extracted_licenses) == 4 |
| 17 | + # It is needed to sort the list because the order changes when parsing |
| 18 | + licenses = sorted(document.extracted_licenses) |
| 19 | + assert licenses[0].identifier.toPython() == 'LicenseRef-1' |
| 20 | + assert licenses[1].identifier.toPython() == 'LicenseRef-2' |
| 21 | + assert licenses[2].identifier.toPython() == 'LicenseRef-3' |
| 22 | + assert licenses[3].identifier.toPython() == 'LicenseRef-4' |
| 23 | + |
| 24 | + def test_pkg_lic_decl(self): |
| 25 | + parser = Parser(Builder(), StandardLogger()) |
| 26 | + test_file = utils_test.get_test_loc('../../data/SPDXRdfExample.rdf', test_data_dir=utils_test.test_data_dir) |
| 27 | + with open(test_file, 'r') as file: |
| 28 | + document, _ = parser.parse(file) |
| 29 | + # It is needed to ckeck it as sorted lists because the order changes when parsing |
| 30 | + lic_expected = ['Apache-2.0', 'LicenseRef-1', 'LicenseRef-2', 'LicenseRef-3', 'LicenseRef-4', 'MPL-1.1'] |
| 31 | + lic_result = sorted(document.package.license_declared.identifier.split(' AND ')) |
| 32 | + assert lic_result == lic_expected |
| 33 | + |
| 34 | + def test_pkg_lic_conc(self): |
| 35 | + parser = Parser(Builder(), StandardLogger()) |
| 36 | + test_file = utils_test.get_test_loc('../../data/SPDXRdfExample.rdf', test_data_dir=utils_test.test_data_dir) |
| 37 | + with open(test_file, 'r') as file: |
| 38 | + document, _ = parser.parse(file) |
| 39 | + # It is needed to ckeck it as sorted lists because the order changes when parsing |
| 40 | + lic_expected = ['Apache-1.0', 'Apache-2.0', 'LicenseRef-1', 'LicenseRef-2', 'LicenseRef-3', 'LicenseRef-4', 'MPL-1.1'] |
| 41 | + lic_result = sorted(document.package.conc_lics.identifier.split(' AND ')) |
| 42 | + assert lic_result == lic_expected |
| 43 | + |
| 44 | + def test_file_lic_conc(self): |
| 45 | + parser = Parser(Builder(), StandardLogger()) |
| 46 | + test_file = utils_test.get_test_loc('../../data/SPDXRdfExample.rdf', test_data_dir=utils_test.test_data_dir) |
| 47 | + with open(test_file, 'r') as file: |
| 48 | + document, _ = parser.parse(file) |
| 49 | + files = sorted(document.package.files) |
| 50 | + assert files[0].conc_lics.identifier.toPython() == 'LicenseRef-1' |
| 51 | + assert files[1].conc_lics.identifier == 'Apache-2.0' |
| 52 | + |
| 53 | +if __name__ == '__main__': |
| 54 | + unittest.main() |
0 commit comments