add3prf.py: Add support for zlib license detection
Bug: 321066417
Test: Run add3prf.py against tinyvec-1.6.0 and added unit tests
Change-Id: I824f93a0335c8d8de13b11df9efe6775a02870f2
diff --git a/scripts/add3prf_test.py b/scripts/add3prf_test.py
index a75b31f..8389b6e 100755
--- a/scripts/add3prf_test.py
+++ b/scripts/add3prf_test.py
@@ -118,6 +118,24 @@
self.assertEqual(preferred_license.group, add3prf.LicenseGroup.PERMISSIVE)
self.assertEqual(preferred_license.filename, "UNLICENSE.txt")
+ def test_zlib_license(self):
+ self.fs.create_file("LICENSE")
+ licenses = add3prf.decide_license_type("Zlib")
+ self.assertEqual(len(licenses), 1)
+ preferred_license = licenses[0]
+ self.assertEqual(preferred_license.type, add3prf.LicenseType.ZLIB)
+ self.assertEqual(preferred_license.group, add3prf.LicenseGroup.NOTICE)
+ self.assertEqual(preferred_license.filename, "LICENSE")
+
+ def test_zlib_licensefile(self):
+ self.fs.create_file("LICENSE-ZLIB")
+ licenses = add3prf.decide_license_type("")
+ self.assertEqual(len(licenses), 1)
+ preferred_license = licenses[0]
+ self.assertEqual(preferred_license.type, add3prf.LicenseType.ZLIB)
+ self.assertEqual(preferred_license.group, add3prf.LicenseGroup.NOTICE)
+ self.assertEqual(preferred_license.filename, "LICENSE-ZLIB")
+
class AddModuleLicenseTestCase(fake_filesystem_unittest.TestCase):
def setUp(self):