1 files changed, 62 insertions, 0 deletions
diff --git a/tools_webrtc/libs/generate_licenses_test.py b/tools_webrtc/libs/generate_licenses_test.py
new file mode 100755
index 0000000000..9ac541adf9
--- /dev/null
+++ b/tools_webrtc/libs/generate_licenses_test.py
@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+# 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 os
+import sys
+
+SRC = os.path.abspath(os.path.join(
+                      os.path.dirname((__file__)), os.pardir, os.pardir))
+sys.path.append(os.path.join(SRC, 'third_party', 'pymock'))
+
+import mock
+import unittest
+
+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 testParseLibrary(self):
+    self.assertEquals(LicenseBuilder._ParseLibrary(
+            '//a/b/third_party/libname1:c').group(1),
+        'libname1')
+    self.assertEquals(LicenseBuilder._ParseLibrary(
+            '//a/b/third_party/libname2:c(d)').group(1),
+        'libname2')
+    self.assertEquals(LicenseBuilder._ParseLibrary(
+            '//a/b/third_party/libname3/c:d(e)').group(1),
+        'libname3')
+    self.assertEquals(LicenseBuilder._ParseLibrary('//a/b/not_third_party/c'),
+        None)
+
+  @mock.patch('generate_licenses.LicenseBuilder._RunGN', _FakeRunGN)
+  def testGetThirdPartyLibraries(self):
+    self.assertEquals(LicenseBuilder._GetThirdPartyLibraries(
+            'out/arm', 'target1'),
+        set(['libname1', 'libname2', 'libname3']))
+
+
+if __name__ == '__main__':
+  unittest.main()