| #!/usr/bin/env vpython |
| # pylint: disable=relative-import,protected-access,unused-argument |
| |
| # Copyright 2017 The WebRTC project authors. All Rights Reserved. |
| # |
| # Use of this source code is governed by a BSD-style license |
| # that can be found in the LICENSE file in the root of the source |
| # tree. An additional intellectual property rights grant can be found |
| # in the file PATENTS. All contributing project authors may |
| # be found in the AUTHORS file in the root of the source tree. |
| |
| import unittest |
| import mock |
| |
| from generate_licenses import LicenseBuilder |
| |
| |
| class TestLicenseBuilder(unittest.TestCase): |
| |
| @staticmethod |
| def _FakeRunGN(buildfile_dir, target): |
| return """ |
| { |
| "target1": { |
| "deps": [ |
| "//a/b/third_party/libname1:c", |
| "//a/b/third_party/libname2:c(//d/e/f:g)", |
| "//a/b/third_party/libname3/c:d(//e/f/g:h)", |
| "//a/b/not_third_party/c" |
| ] |
| } |
| } |
| """ |
| |
| def testParseLibraryName(self): |
| self.assertEquals( |
| LicenseBuilder._ParseLibraryName('//a/b/third_party/libname1:c'), |
| 'libname1') |
| self.assertEquals( |
| LicenseBuilder._ParseLibraryName('//a/b/third_party/libname2:c(d)'), |
| 'libname2') |
| self.assertEquals( |
| LicenseBuilder._ParseLibraryName('//a/b/third_party/libname3/c:d(e)'), |
| 'libname3') |
| self.assertEquals( |
| LicenseBuilder._ParseLibraryName('//a/b/not_third_party/c'), None) |
| |
| def testParseLibrarySimpleMatch(self): |
| builder = LicenseBuilder([], [], {}, {}) |
| self.assertEquals( |
| builder._ParseLibrary('//a/b/third_party/libname:c'), 'libname') |
| |
| def testParseLibraryRegExNoMatchFallbacksToDefaultLibname(self): |
| lib_dict = { |
| 'libname:foo.*': ['path/to/LICENSE'], |
| } |
| builder = LicenseBuilder([], [], lib_dict, {}) |
| self.assertEquals( |
| builder._ParseLibrary('//a/b/third_party/libname:bar_java'), 'libname') |
| |
| def testParseLibraryRegExMatch(self): |
| lib_regex_dict = { |
| 'libname:foo.*': ['path/to/LICENSE'], |
| } |
| builder = LicenseBuilder([], [], {}, lib_regex_dict) |
| self.assertEquals( |
| builder._ParseLibrary('//a/b/third_party/libname:foo_bar_java'), |
| 'libname:foo.*') |
| |
| def testParseLibraryRegExMatchWithSubDirectory(self): |
| lib_regex_dict = { |
| 'libname/foo:bar.*': ['path/to/LICENSE'], |
| } |
| builder = LicenseBuilder([], [], {}, lib_regex_dict) |
| self.assertEquals( |
| builder._ParseLibrary('//a/b/third_party/libname/foo:bar_java'), |
| 'libname/foo:bar.*') |
| |
| def testParseLibraryRegExMatchWithStarInside(self): |
| lib_regex_dict = { |
| 'libname/foo.*bar.*': ['path/to/LICENSE'], |
| } |
| builder = LicenseBuilder([], [], {}, lib_regex_dict) |
| self.assertEquals( |
| builder._ParseLibrary('//a/b/third_party/libname/fooHAHA:bar_java'), |
| 'libname/foo.*bar.*') |
| |
| @mock.patch('generate_licenses.LicenseBuilder._RunGN', _FakeRunGN) |
| def testGetThirdPartyLibrariesWithoutRegex(self): |
| builder = LicenseBuilder([], [], {}, {}) |
| self.assertEquals( |
| builder._GetThirdPartyLibraries('out/arm', 'target1'), |
| set(['libname1', 'libname2', 'libname3'])) |
| |
| @mock.patch('generate_licenses.LicenseBuilder._RunGN', _FakeRunGN) |
| def testGetThirdPartyLibrariesWithRegex(self): |
| lib_regex_dict = { |
| 'libname2:c.*': ['path/to/LICENSE'], |
| } |
| builder = LicenseBuilder([], [], {}, lib_regex_dict) |
| self.assertEquals( |
| builder._GetThirdPartyLibraries('out/arm', 'target1'), |
| set(['libname1', 'libname2:c.*', 'libname3'])) |
| |
| @mock.patch('generate_licenses.LicenseBuilder._RunGN', _FakeRunGN) |
| def testGenerateLicenseTextFailIfUnknownLibrary(self): |
| lib_dict = { |
| 'simple_library': ['path/to/LICENSE'], |
| } |
| builder = LicenseBuilder(['dummy_dir'], ['dummy_target'], lib_dict, {}) |
| |
| with self.assertRaises(Exception) as context: |
| builder.GenerateLicenseText('dummy/dir') |
| |
| self.assertEquals( |
| context.exception.message, |
| 'Missing licenses for following third_party targets: ' |
| 'libname1, libname2, libname3') |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |