tools/tests/licenses_test.py

#!/usr/bin/env python
# Copyright (c) 2017 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for //tools/licenses.py.
"""

import os
import sys
import unittest

REPOSITORY_ROOT = os.path.abspath(os.path.join(
    os.path.dirname(__file__), '..', '..'))
sys.path.append(os.path.join(REPOSITORY_ROOT, 'tools'))

import licenses


class LicensesTest(unittest.TestCase):

    def test_get_third_party_deps_from_gn_deps_output(self):

        def construct_absolute_path(path):
            return os.path.join(REPOSITORY_ROOT, *path.split('/')).replace(
                os.sep, '/')

        prune_path = next(iter(licenses.PRUNE_PATHS))
        gn_deps = [
            construct_absolute_path('net/BUILD.gn'),
            construct_absolute_path('third_party/zlib/BUILD.gn'),
            construct_absolute_path('third_party/cld_3/src/src/BUILD.gn'),
            construct_absolute_path(prune_path + '/BUILD.gn'),
        ]
        third_party_deps = licenses.GetThirdPartyDepsFromGNDepsOutput(
            '\n'.join(gn_deps), None)

        # 'net' is not in the output because it's not a third_party dependency.
        #
        # It must return the direct sub-directory of "third_party". So it should
        # return 'third_party/cld_3', not 'third_party/cld_3/src/src'.
        assert third_party_deps == set([
            os.path.join('third_party', 'zlib'),
            os.path.join('third_party', 'cld_3'),
        ])


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