summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/tools_webrtc/libs/generate_licenses_test.py
blob: 6dfd8f3e229c713f8aeeb3fa7be701d44496bc37 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#!/usr/bin/env vpython3

# pylint: disable=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
from mock import patch

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.assertEqual(
        LicenseBuilder._ParseLibraryName('//a/b/third_party/libname1:c'),
        'libname1')
    self.assertEqual(
        LicenseBuilder._ParseLibraryName('//a/b/third_party/libname2:c(d)'),
        'libname2')
    self.assertEqual(
        LicenseBuilder._ParseLibraryName('//a/b/third_party/libname3/c:d(e)'),
        'libname3')
    self.assertEqual(
        LicenseBuilder._ParseLibraryName('//a/b/not_third_party/c'), None)

  def testParseLibrarySimpleMatch(self):
    builder = LicenseBuilder([], [], {}, {})
    self.assertEqual(builder._ParseLibrary('//a/b/third_party/libname:c'),
                     'libname')

  def testParseLibraryRegExNoMatchFallbacksToDefaultLibname(self):
    lib_dict = {
        'libname:foo.*': ['path/to/LICENSE'],
    }
    builder = LicenseBuilder([], [], lib_dict, {})
    self.assertEqual(
        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.assertEqual(
        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.assertEqual(
        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.assertEqual(
        builder._ParseLibrary('//a/b/third_party/libname/fooHAHA:bar_java'),
        'libname/foo.*bar.*')

  @patch('generate_licenses.LicenseBuilder._RunGN', _FakeRunGN)
  def testGetThirdPartyLibrariesWithoutRegex(self):
    builder = LicenseBuilder([], [], {}, {})
    self.assertEqual(builder._GetThirdPartyLibraries('out/arm', 'target1'),
                     set(['libname1', 'libname2', 'libname3']))

  @patch('generate_licenses.LicenseBuilder._RunGN', _FakeRunGN)
  def testGetThirdPartyLibrariesWithRegex(self):
    lib_regex_dict = {
        'libname2:c.*': ['path/to/LICENSE'],
    }
    builder = LicenseBuilder([], [], {}, lib_regex_dict)
    self.assertEqual(builder._GetThirdPartyLibraries('out/arm', 'target1'),
                     set(['libname1', 'libname2:c.*', 'libname3']))

  @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.assertEqual(
        context.exception.args[0],
        'Missing licenses for following third_party targets: '
        'libname1, libname2, libname3')


if __name__ == '__main__':
  unittest.main()