100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
|
from unittest.case import TestCase
|
||
|
|
import oe.license
|
||
|
|
|
||
|
|
class SeenVisitor(oe.license.LicenseVisitor):
|
||
|
|
def __init__(self):
|
||
|
|
self.seen = []
|
||
|
|
oe.license.LicenseVisitor.__init__(self)
|
||
|
|
|
||
|
|
def visit_Str(self, node):
|
||
|
|
self.seen.append(node.s)
|
||
|
|
|
||
|
|
class TestSingleLicense(TestCase):
|
||
|
|
licenses = [
|
||
|
|
"GPLv2",
|
||
|
|
"LGPL-2.0",
|
||
|
|
"Artistic",
|
||
|
|
"MIT",
|
||
|
|
"GPLv3+",
|
||
|
|
"FOO_BAR",
|
||
|
|
]
|
||
|
|
invalid_licenses = ["GPL/BSD"]
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def parse(licensestr):
|
||
|
|
visitor = SeenVisitor()
|
||
|
|
visitor.visit_string(licensestr)
|
||
|
|
return visitor.seen
|
||
|
|
|
||
|
|
def test_single_licenses(self):
|
||
|
|
for license in self.licenses:
|
||
|
|
licenses = self.parse(license)
|
||
|
|
self.assertListEqual(licenses, [license])
|
||
|
|
|
||
|
|
def test_invalid_licenses(self):
|
||
|
|
for license in self.invalid_licenses:
|
||
|
|
with self.assertRaises(oe.license.InvalidLicense) as cm:
|
||
|
|
self.parse(license)
|
||
|
|
self.assertEqual(cm.exception.license, license)
|
||
|
|
|
||
|
|
class TestSimpleCombinations(TestCase):
|
||
|
|
tests = {
|
||
|
|
"FOO&BAR": ["FOO", "BAR"],
|
||
|
|
"BAZ & MOO": ["BAZ", "MOO"],
|
||
|
|
"ALPHA|BETA": ["ALPHA"],
|
||
|
|
"BAZ&MOO|FOO": ["FOO"],
|
||
|
|
"FOO&BAR|BAZ": ["FOO", "BAR"],
|
||
|
|
}
|
||
|
|
preferred = ["ALPHA", "FOO", "BAR"]
|
||
|
|
|
||
|
|
def test_tests(self):
|
||
|
|
def choose(a, b):
|
||
|
|
if all(lic in self.preferred for lic in b):
|
||
|
|
return b
|
||
|
|
else:
|
||
|
|
return a
|
||
|
|
|
||
|
|
for license, expected in self.tests.items():
|
||
|
|
licenses = oe.license.flattened_licenses(license, choose)
|
||
|
|
self.assertListEqual(licenses, expected)
|
||
|
|
|
||
|
|
class TestComplexCombinations(TestSimpleCombinations):
|
||
|
|
tests = {
|
||
|
|
"FOO & (BAR | BAZ)&MOO": ["FOO", "BAR", "MOO"],
|
||
|
|
"(ALPHA|(BETA&THETA)|OMEGA)&DELTA": ["OMEGA", "DELTA"],
|
||
|
|
"((ALPHA|BETA)&FOO)|BAZ": ["BETA", "FOO"],
|
||
|
|
"(GPL-2.0|Proprietary)&BSD-4-clause&MIT": ["GPL-2.0", "BSD-4-clause", "MIT"],
|
||
|
|
}
|
||
|
|
preferred = ["BAR", "OMEGA", "BETA", "GPL-2.0"]
|
||
|
|
|
||
|
|
class TestIsIncluded(TestCase):
|
||
|
|
tests = {
|
||
|
|
("FOO | BAR", None, None):
|
||
|
|
[True, ["FOO"]],
|
||
|
|
("FOO | BAR", None, "FOO"):
|
||
|
|
[True, ["BAR"]],
|
||
|
|
("FOO | BAR", "BAR", None):
|
||
|
|
[True, ["BAR"]],
|
||
|
|
("FOO | BAR & FOOBAR", "*BAR", None):
|
||
|
|
[True, ["BAR", "FOOBAR"]],
|
||
|
|
("FOO | BAR & FOOBAR", None, "FOO*"):
|
||
|
|
[False, ["FOOBAR"]],
|
||
|
|
("(FOO | BAR) & FOOBAR | BARFOO", None, "FOO"):
|
||
|
|
[True, ["BAR", "FOOBAR"]],
|
||
|
|
("(FOO | BAR) & FOOBAR | BAZ & MOO & BARFOO", None, "FOO"):
|
||
|
|
[True, ["BAZ", "MOO", "BARFOO"]],
|
||
|
|
("GPL-3.0 & GPL-2.0 & LGPL-2.1 | Proprietary", None, None):
|
||
|
|
[True, ["GPL-3.0", "GPL-2.0", "LGPL-2.1"]],
|
||
|
|
("GPL-3.0 & GPL-2.0 & LGPL-2.1 | Proprietary", None, "GPL-3.0"):
|
||
|
|
[True, ["Proprietary"]],
|
||
|
|
("GPL-3.0 & GPL-2.0 & LGPL-2.1 | Proprietary", None, "GPL-3.0 Proprietary"):
|
||
|
|
[False, ["GPL-3.0"]]
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_tests(self):
|
||
|
|
for args, expected in self.tests.items():
|
||
|
|
is_included, licenses = oe.license.is_included(
|
||
|
|
args[0], (args[1] or '').split(), (args[2] or '').split())
|
||
|
|
self.assertEqual(is_included, expected[0])
|
||
|
|
self.assertListEqual(licenses, expected[1])
|