chromium/src/third_party/blink/tools/blinkpy/presubmit/audit_non_blink_usage_test.py

#!/usr/bin/env python
# Copyright 2018 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.

import re
import unittest

from blinkpy.presubmit import audit_non_blink_usage


class TestAuditNonBlinkUsageTest(unittest.TestCase):
    # This is not great but it allows us to check that something is a regexp.
    _REGEXP_CLASS = re.compile(r"foo").__class__

    def test_valid_compiled_config(self):
        # We need to test this protected data.
        # pylint: disable=W0212
        for entry in audit_non_blink_usage._COMPILED_CONFIG:
            for path in entry['paths']:
                self.assertIsInstance(path, str)
            if 'allowed' in entry:
                self.assertIsInstance(entry['allowed'], self._REGEXP_CLASS)
            if 'disallowed' in entry:
                self.assertIsInstance(entry['disallowed'], self._REGEXP_CLASS)
            for match, advice, warning in entry.get('advice', []):
                self.assertIsInstance(match, self._REGEXP_CLASS)
                self.assertIsInstance(advice, str)

    def test_for_special_cases(self):
        for entry in audit_non_blink_usage._COMPILED_CONFIG:
            if entry['paths'] == ['third_party/blink/renderer/']:
                check_list = [
                    {
                        'type': 'url::mojom::Origin',
                        'allowed': False
                    },
                    {
                        'type': '::media::mojom::InterfaceFactory',
                        'allowed': False
                    },
                    {
                        'type': 'Hogenetwork::mojom::URLLoaderFactory',
                        'allowed': False
                    },
                    {
                        'type': 'url::mojom::blink::Origin',
                        'allowed': True
                    },
                    {
                        'type': '::media::mojom::blink::InterfaceFactory',
                        'allowed': True
                    },
                    {
                        'type': 'network::mojom::URLLoaderFactory',
                        'allowed': True
                    },
                    {
                        'type': '::network::mojom::URLLoaderFactory',
                        'allowed': True
                    },
                ]
                for item in check_list:
                    if item['allowed']:
                        self.assertIsNone(
                            re.match(entry['disallowed'], item['type']))
                    elif not item['allowed']:
                        self.assertIsNotNone(
                            re.match(entry['disallowed'], item['type']))


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